seyfert/lib/websocket/structures/index.js

The most advanced framework for discord bots

"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
exports.DynamicBucket = void 0;
const common_1 = require("../../common");
class DynamicBucket {
    options;
    queue = [];
    used = 0;
    processing;
    refillsAt;
    timeoutId;
    constructor(options) {
        this.options = options;
    }
    get remaining() {
        if (this.options.limit < this.used) {
            return 0;
        }
        return this.options.limit - this.used;
    }
    refill() {
        this.refillsAt = undefined;
        if (this.timeoutId) {
            clearTimeout(this.timeoutId);
            this.timeoutId = undefined;
        }
        if (this.used > 0) {
            this.used = 0;
            this.timeoutId = setTimeout(() => {
                this.refill();
            }, this.options.refillInterval);
            this.refillsAt = Date.now() + this.options.refillInterval;
        }
    }
    async processQueue() {
        if (this.processing)
            return;
        this.processing = true;
        while (this.queue.length) {
            if (this.remaining) {
                this.options.debugger?.debug(`Processing queue. Remaining: ${this.remaining} Length: ${this.queue.length}`);
                this.queue.shift()();
                this.used++;
                if (!this.timeoutId) {
                    this.timeoutId = setTimeout(() => {
                        this.refill();
                    }, this.options.refillInterval);
                    this.refillsAt = Date.now() + this.options.refillInterval;
                }
            }
            else if (this.refillsAt) {
                const now = Date.now();
                if (this.refillsAt > now) {
                    this.options.debugger?.info(`Waiting ${this.refillsAt - now}ms to process queue`);
                    await (0, common_1.delay)(this.refillsAt - now);
                    this.used = 0;
                }
            }
        }
        this.processing = false;
    }
    acquire(force = false) {
        return new Promise(res => {
            this.queue[force ? 'unshift' : 'push'](res);
            this.processQueue();
        });
    }
    static chunk(array, chunks) {
        let index = 0;
        let resIndex = 0;
        const result = new Array(Math.ceil(array.length / chunks));
        while (index < array.length) {
            result[resIndex] = array.slice(index, (index += chunks));
            resIndex++;
        }
        return result;
    }
}
exports.DynamicBucket = DynamicBucket;