src/services/QueueManager.ts

import { Collection } from "discord.js";
import DiscordClient from "../client/Client";
import QueuedJob from "../models/QueuedJob";
import Queue, { QueueOptions } from "../utils/structures/Queue";
import Service from "../utils/structures/Service";
import path from "path";
import { generate as randomstring } from 'randomstring';

export type QueueCreateOptions = {
    data?: { [key: string | number]: any };
    runAt?: Date;
    runAfter?: number;
    guild?: string;
};

export default class QueueManager extends Service {
    public readonly queues: Collection<string, Queue> = new Collection();

    async loadQueues() {
        const models = await QueuedJob.find();

        for await (const model of models) {
            const { default: Queue }: { default: new (client: DiscordClient, queueOptions: QueueOptions) => Queue } = await import(path.resolve(__dirname, '../queues/', model.className));
            console.log(Queue);
            this.queues.set(model.uuid, new Queue(this.client, { model, id: model.uuid, runAt: model.runOn }));
            console.log("Found queue: ", model.className);
        }
    }

    setQueue(queue: Queue) {
        this.queues.set(queue.id, queue);
    }

    cancelQueue(id: string) {
        return this.queues.get(id)?.cancel();
    }

    async addQueue(queueClass: new (client: DiscordClient, queueOptions: QueueOptions) => Queue, { data, runAt, runAfter, guild }: QueueCreateOptions) {
        if (runAfter !== 0 && runAfter !== 0 && !runAfter && !runAt) {
            throw new Error("One of runAfter or runAt must be specified for creating a queue");
        }

        const id = randomstring(7);
        const model = await QueuedJob.create({ uuid: id, data, runOn: runAfter ? new Date(Date.now() + runAfter) : runAt!, createdAt: new Date(), className: queueClass.name, guild });
        const queue = new queueClass(this.client, { model, id, runAt, runAfter });
        this.setQueue(queue);
        return queue;
    }

    runQueue(id: string) {
        return this.queues.get(id)?.run();
    }

    removeQueue(queue: Queue) {
        const { id } = queue;
        this.queues.delete(id);
    }
}   
1	import { Collection } from "discord.js";
2	import DiscordClient from "../client/Client";
3	import QueuedJob from "../models/QueuedJob";
4	import Queue, { QueueOptions } from "../utils/structures/Queue";
5	import Service from "../utils/structures/Service";
6	import path from "path";
7	import { generate as randomstring } from 'randomstring';
8
9	export type QueueCreateOptions = {
10	data?: { [key: string \| number]: any };
11	runAt?: Date;
12	runAfter?: number;
13	guild?: string;
14	};
15
16	export default class QueueManager extends Service {
17	public readonly queues: Collection<string, Queue> = new Collection();
18
19	async loadQueues() {
20	const models = await QueuedJob.find();
21
22	for await (const model of models) {
23	const { default: Queue }: { default: new (client: DiscordClient, queueOptions: QueueOptions) => Queue } = await import(path.resolve(__dirname, '../queues/', model.className));
24	console.log(Queue);
25	this.queues.set(model.uuid, new Queue(this.client, { model, id: model.uuid, runAt: model.runOn }));
26	console.log("Found queue: ", model.className);
27	}
28	}
29
30	setQueue(queue: Queue) {
31	this.queues.set(queue.id, queue);
32	}
33
34	cancelQueue(id: string) {
35	return this.queues.get(id)?.cancel();
36	}
37
38	async addQueue(queueClass: new (client: DiscordClient, queueOptions: QueueOptions) => Queue, { data, runAt, runAfter, guild }: QueueCreateOptions) {
39	if (runAfter !== 0 && runAfter !== 0 && !runAfter && !runAt) {
40	throw new Error("One of runAfter or runAt must be specified for creating a queue");
41	}
42
43	const id = randomstring(7);
44	const model = await QueuedJob.create({ uuid: id, data, runOn: runAfter ? new Date(Date.now() + runAfter) : runAt!, createdAt: new Date(), className: queueClass.name, guild });
45	const queue = new queueClass(this.client, { model, id, runAt, runAfter });
46	this.setQueue(queue);
47	return queue;
48	}
49
50	runQueue(id: string) {
51	return this.queues.get(id)?.run();
52	}
53
54	removeQueue(queue: Queue) {
55	const { id } = queue;
56	this.queues.delete(id);
57	}
58	}