src/queues/CommandQueue.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, GuildMember, Snowflake, TextChannel } from "discord.js";
import Queue from "../utils/Queue";
import { safeChannelFetch, safeMemberFetch, safeMessageFetch } from "../utils/fetch";
import { log, logError } from "../utils/logger";

export default class CommandQueue extends Queue {
    cloneObject(obj: object) {
        const clonedObj = Object.create(Object.getPrototypeOf(obj));

        for (const key of Reflect.ownKeys(obj)) {
            const descriptor = Object.getOwnPropertyDescriptor(obj, key);

            if (descriptor) {
                Object.defineProperty(clonedObj, key, descriptor);
            }
        }

        return clonedObj;
    }

    copyCollection<K, V>(destination: Collection<K, V>, source: Collection<K, V>) {
        for (const [key, value] of source) {
            destination.set(key, value);
        }
    }

    async run(channelId: Snowflake, messageId: Snowflake, contentWithoutPrefix: string) {
        try {
            const channel = await safeChannelFetch(this.guild, channelId);

            if (!channel || !channel.isTextBased()) {
                return;
            }

            let message = await safeMessageFetch(channel, messageId);
            let member = message?.member as GuildMember | null;

            if (!message) {
                message =
                    (this.guild.channels.cache.find(c => c.isTextBased() && c.lastMessage) as TextChannel | null)?.lastMessage ??
                    null;

                member = await safeMemberFetch(this.guild, this.userId);

                if (message) {
                    message = this.cloneObject(message);

                    message!.reply = (...args: [any]) => channel.send(...args);
                    message!.delete = (...args: any[]) => Promise.resolve(message!);
                    message!.react = (...args: any[]) => Promise.resolve(null as any);
                }
            } else {
                message = this.cloneObject(message);
            }

            if (!member) {
                log("Aborting command queue as the member who ran the command was not found.");
                return;
            }

            if (!message) {
                log(
                    "Aborting command queue as no message alternative strategy can be used to run this command queue in a safe sandbox."
                );
                return;
            }

            const userMentions = message.mentions.users.clone();
            const roleMentions = message.mentions.roles.clone();
            const memberMentions = message.mentions.members?.clone();
            const channelMentions = message.mentions.channels.clone();

            message.mentions.users.clear();
            message.mentions.roles.clear();
            message.mentions.members?.clear();
            message.mentions.channels.clear();

            message.mentions.users = userMentions;
            message.mentions.roles = roleMentions;

            if (message.mentions.members && memberMentions) {
                this.copyCollection(message.mentions.members, memberMentions);
            }

            this.copyCollection(message.mentions.channels, channelMentions);

            message.content = `${this.client.configManager.config[this.guild.id]?.prefix ?? "-"}${contentWithoutPrefix}`;
            message.channelId = channel.id;

            if (member) {
                Object.defineProperty(message, "member", {
                    get: () => member
                });

                Object.defineProperty(message, "author", {
                    get: () => member?.user
                });
            }

            Object.defineProperty(message, "channel", {
                get: () => channel
            });

            Object.defineProperty(message, "url", {
                get: () => `https://discord.com/channels/${this.guild.id}/${channel.id}/${messageId}`
            });

            await this.client.commandManager.runCommandFromMessage(message);
        } catch (e) {
            logError(e);
        }
    }
}
1	/*
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, GuildMember, Snowflake, TextChannel } from "discord.js";
21	import Queue from "../utils/Queue";
22	import { safeChannelFetch, safeMemberFetch, safeMessageFetch } from "../utils/fetch";
23	import { log, logError } from "../utils/logger";
24
25	export default class CommandQueue extends Queue {
26	cloneObject(obj: object) {
27	const clonedObj = Object.create(Object.getPrototypeOf(obj));
28
29	for (const key of Reflect.ownKeys(obj)) {
30	const descriptor = Object.getOwnPropertyDescriptor(obj, key);
31
32	if (descriptor) {
33	Object.defineProperty(clonedObj, key, descriptor);
34	}
35	}
36
37	return clonedObj;
38	}
39
40	copyCollection<K, V>(destination: Collection<K, V>, source: Collection<K, V>) {
41	for (const [key, value] of source) {
42	destination.set(key, value);
43	}
44	}
45
46	async run(channelId: Snowflake, messageId: Snowflake, contentWithoutPrefix: string) {
47	try {
48	const channel = await safeChannelFetch(this.guild, channelId);
49
50	if (!channel \|\| !channel.isTextBased()) {
51	return;
52	}
53
54	let message = await safeMessageFetch(channel, messageId);
55	let member = message?.member as GuildMember \| null;
56
57	if (!message) {
58	message =
59	(this.guild.channels.cache.find(c => c.isTextBased() && c.lastMessage) as TextChannel \| null)?.lastMessage ??
60	null;
61
62	member = await safeMemberFetch(this.guild, this.userId);
63
64	if (message) {
65	message = this.cloneObject(message);
66
67	message!.reply = (...args: [any]) => channel.send(...args);
68	message!.delete = (...args: any[]) => Promise.resolve(message!);
69	message!.react = (...args: any[]) => Promise.resolve(null as any);
70	}
71	} else {
72	message = this.cloneObject(message);
73	}
74
75	if (!member) {
76	log("Aborting command queue as the member who ran the command was not found.");
77	return;
78	}
79
80	if (!message) {
81	log(
82	"Aborting command queue as no message alternative strategy can be used to run this command queue in a safe sandbox."
83	);
84	return;
85	}
86
87	const userMentions = message.mentions.users.clone();
88	const roleMentions = message.mentions.roles.clone();
89	const memberMentions = message.mentions.members?.clone();
90	const channelMentions = message.mentions.channels.clone();
91
92	message.mentions.users.clear();
93	message.mentions.roles.clear();
94	message.mentions.members?.clear();
95	message.mentions.channels.clear();
96
97	message.mentions.users = userMentions;
98	message.mentions.roles = roleMentions;
99
100	if (message.mentions.members && memberMentions) {
101	this.copyCollection(message.mentions.members, memberMentions);
102	}
103
104	this.copyCollection(message.mentions.channels, channelMentions);
105
106	message.content = `${this.client.configManager.config[this.guild.id]?.prefix ?? "-"}${contentWithoutPrefix}`;
107	message.channelId = channel.id;
108
109	if (member) {
110	Object.defineProperty(message, "member", {
111	get: () => member
112	});
113
114	Object.defineProperty(message, "author", {
115	get: () => member?.user
116	});
117	}
118
119	Object.defineProperty(message, "channel", {
120	get: () => channel
121	});
122
123	Object.defineProperty(message, "url", {
124	get: () => `https://discord.com/channels/${this.guild.id}/${channel.id}/${messageId}`
125	});
126
127	await this.client.commandManager.runCommandFromMessage(message);
128	} catch (e) {
129	logError(e);
130	}
131	}
132	}