src/services/QueueManager.ts

/**
 * This file is part of SudoBot.
 *
 * Copyright (C) 2021-2023 OSN Developers.
 *
 * SudoBot is free software; you can redistribute it and/or modify it
 * under the terms of the GNU Affero General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * SudoBot is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with SudoBot. If not, see <https://www.gnu.org/licenses/>.
 */

import { Collection } from "discord.js";
import path from "path";
import Service from "../core/Service";
import QueueEntry from "../utils/QueueEntry";
import { log, logError } from "../utils/logger";

export const name = "queueManager";

export default class QueueManager extends Service {
    queues = new Collection<string, QueueEntry>();

    async onReady() {
        const queuesInDatabase = await this.client.prisma.queue.findMany();

        log(`Found ${queuesInDatabase.length} queues since the last startup.`);

        for (const queue of queuesInDatabase) {
            try {
                log(queue.guild_id, path.resolve(__dirname, "../queues/", queue.file_name));

                const entry = new QueueEntry({
                    channelId: queue.channel_id ?? undefined,
                    client: this.client,
                    createdAt: queue.createdAt,
                    guild: this.client.guilds.cache.get(queue.guild_id) || (await this.client.guilds.fetch(queue.guild_id)!),
                    messageId: queue.message_id ?? undefined,
                    willRunAt: queue.willRunAt,
                    filePath: path.resolve(__dirname, "../queues/", queue.file_name),
                    args: queue.args,
                    id: queue.id,
                    userId: queue.user_id,
                    name: queue.name
                });

                if (queue.willRunAt.getTime() <= Date.now()) {
                    await entry.run().catch(logError);
                    await entry.deleteDatabaseRecord();
                    continue;
                }

                this.queues.set(`${queue.id}`, entry);
                entry.setTimeout();
            } catch (e) {
                log(e);
                continue;
            }
        }
    }

    async add(queue: QueueEntry) {
        if (queue.creatingRecord) await queue.creatingRecord;

        this.queues.set(queue.id.toString(), queue);
        queue.setTimeout();
        return queue.id;
    }

    async remove(queue: QueueEntry) {
        if (queue.creatingRecord) await queue.creatingRecord;

        this.queues.delete(queue.id.toString());
        await queue.clearTimeout(true);
        return queue.id;
    }

    async removeById(id: number) {
        const queue = this.queues.get(id.toString());

        if (!queue) return null;

        return await this.remove(queue);
    }
}
1	rakinar2	577	/**
2			* This file is part of SudoBot.
3			*
4			* Copyright (C) 2021-2023 OSN Developers.
5			*
6			* SudoBot is free software; you can redistribute it and/or modify it
7			* under the terms of the GNU Affero General Public License as published by
8			* the Free Software Foundation, either version 3 of the License, or
9			* (at your option) any later version.
10			*
11			* SudoBot is distributed in the hope that it will be useful, but
12			* WITHOUT ANY WARRANTY; without even the implied warranty of
13			* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14			* GNU Affero General Public License for more details.
15			*
16			* You should have received a copy of the GNU Affero General Public License
17			* along with SudoBot. If not, see <https://www.gnu.org/licenses/>.
18			*/
19
20			import { Collection } from "discord.js";
21			import path from "path";
22			import Service from "../core/Service";
23			import QueueEntry from "../utils/QueueEntry";
24			import { log, logError } from "../utils/logger";
25
26			export const name = "queueManager";
27
28			export default class QueueManager extends Service {
29			queues = new Collection<string, QueueEntry>();
30
31			async onReady() {
32			const queuesInDatabase = await this.client.prisma.queue.findMany();
33
34			log(`Found ${queuesInDatabase.length} queues since the last startup.`);
35
36			for (const queue of queuesInDatabase) {
37			try {
38			log(queue.guild_id, path.resolve(__dirname, "../queues/", queue.file_name));
39
40			const entry = new QueueEntry({
41			channelId: queue.channel_id ?? undefined,
42			client: this.client,
43			createdAt: queue.createdAt,
44			guild: this.client.guilds.cache.get(queue.guild_id) \|\| (await this.client.guilds.fetch(queue.guild_id)!),
45			messageId: queue.message_id ?? undefined,
46			willRunAt: queue.willRunAt,
47			filePath: path.resolve(__dirname, "../queues/", queue.file_name),
48			args: queue.args,
49			id: queue.id,
50			userId: queue.user_id,
51			name: queue.name
52			});
53
54			if (queue.willRunAt.getTime() <= Date.now()) {
55			await entry.run().catch(logError);
56			await entry.deleteDatabaseRecord();
57			continue;
58			}
59
60			this.queues.set(`${queue.id}`, entry);
61			entry.setTimeout();
62			} catch (e) {
63			log(e);
64			continue;
65			}
66			}
67			}
68
69			async add(queue: QueueEntry) {
70			if (queue.creatingRecord) await queue.creatingRecord;
71
72			this.queues.set(queue.id.toString(), queue);
73			queue.setTimeout();
74			return queue.id;
75			}
76
77			async remove(queue: QueueEntry) {
78			if (queue.creatingRecord) await queue.creatingRecord;
79
80			this.queues.delete(queue.id.toString());
81			await queue.clearTimeout(true);
82			return queue.id;
83			}
84
85			async removeById(id: number) {
86			const queue = this.queues.get(id.toString());
87
88			if (!queue) return null;
89
90			return await this.remove(queue);
91			}
92			}