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 { v4 as uuid } from "uuid";
import path from "path";

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

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 }: 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 = uuid();
        const model = await QueuedJob.create({ uuid: id, data, runOn: runAfter ? new Date(Date.now() + runAfter) : runAt!, createdAt: new Date(), className: queueClass.name });
        const queue = new queueClass(this.client, { model, id, runAt, runAfter });
        this.setQueue(queue);
    }

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

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