dressed/dist/utils/ratelimit.js

A sleek, serverless-ready Discord bot framework.

import ora from "ora";
const buckets = new Map();
const endpoints = new Map();
export async function checkLimit(endpoint, method = "") {
    if (endpoint !== "global")
        await checkLimit("global");
    const bucket = endpoints.get(method + endpoint);
    if (!bucket)
        return;
    const limit = buckets.get(bucket);
    if (limit) {
        if (Date.now() > limit.resetAt) {
            buckets.delete(bucket);
            return;
        }
        const displayed = endpoint === "global" ? "all endpoints" : endpoint;
        if (limit.remaining === 0) {
            const waiting = ora(`Rate limit for ${displayed} reached! - waiting to try again...`).start();
            await new Promise((r) => setTimeout(r, Math.max(0, limit.resetAt - Date.now())));
            buckets.delete(bucket);
            waiting.warn(`A request was delayed because you hit the rate limit for ${displayed}`);
        }
        else if (limit.remaining === 1) {
            ora(`You are about to hit the rate limit for ${displayed}`).warn();
        }
    }
}
export function headerUpdateLimit(endpoint, res, method = "") {
    const remaining = res.headers.get("x-ratelimit-remaining");
    const resetAt = res.headers.get("x-ratelimit-reset");
    const bucket = res.headers.get("x-ratelimit-bucket");
    if (remaining && bucket) {
        updateLimit(bucket, Number(remaining), Number(resetAt) * 1000);
        endpoints.set(method + endpoint, bucket);
    }
}
export function updateLimit(bucket, remaining, resetAt) {
    buckets.set(bucket, { remaining, resetAt });
}
//# sourceMappingURL=ratelimit.js.map