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,
    MessageCreateOptions,
    MessagePayload,
    MessageReplyOptions,
    Snowflake,
    TextChannel
} from "discord.js";
import { log, logError } from "../utils/Logger";
import Queue from "../utils/Queue";
import { safeChannelFetch, safeMemberFetch, safeMessageFetch } from "../utils/fetch";

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