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	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, 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			}