| 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 { |
| 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 |
} |
Parent Directory
| 
Revision Log