commands/moderation/LockallCommand.ts

/**
* This file is part of SudoBot.
* 
* Copyright (C) 2021-2022 OSN Inc.
*
* 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, CommandInteraction, Message, Permissions, Role, TextChannel, User } from 'discord.js';
import BaseCommand from '../../utils/structures/BaseCommand';
import CommandOptions from '../../types/CommandOptions';
import InteractionOptions from '../../types/InteractionOptions';
import DiscordClient from '../../client/Client';
import MessageEmbed from '../../client/MessageEmbed';

export async function lockAll(client: DiscordClient, role: Role, channels: Collection <string, TextChannel> | TextChannel[], user: User, send: boolean = true, reason?: string) {
    if (role) {
        // const gen = await channels.first()!.guild.roles.fetch(client.config.props[channels.first()!.guild.id].gen_role);

        try {
            return await client.channelLock.lockAll(channels instanceof Collection ? [...channels.values()] : channels, user, {
                role,
                sendConfirmation: send,
                reason
            });
        }
        catch (e) {
            console.log(e);
        }
    }
}

export default class LockallCommand extends BaseCommand {
    supportsInteractions: boolean = true;
    permissions = [Permissions.FLAGS.MANAGE_CHANNELS];

    constructor() {
        super('lockall', 'moderation', []);
    }

    async run(client: DiscordClient, msg: Message | CommandInteraction, options: CommandOptions | InteractionOptions) {
        if (!options.isInteraction && typeof options.args[0] === 'undefined') {
            await msg.reply({
                embeds: [
                    new MessageEmbed()
                    .setColor('#f14a60')
                    .setDescription(`This command requires at least one argument.`)
                ]
            });

            return;
        }

        if (msg instanceof CommandInteraction) {
            await msg.deferReply({ ephemeral: true });
        }

        const raid = options.isInteraction ? options.options.getBoolean('raid') === true : (options.options.indexOf('--raid') !== -1);

        let role: Role = <Role> msg.guild!.roles.everyone;
        let lockall: string[] = [], lockallChannels: TextChannel[] = [];
        let reason: string | undefined;
        // const force = options.isInteraction ? options.options.getBoolean('force') === true : (options.options.indexOf('--force') !== -1);

        if (options.isInteraction) {
            if (options.options.getString('channels'))
                lockall = options.options.getString('channels')!.split(' ').filter(a => /^\d+$/gi.test(a) || a.startsWith('<#'));

            if (options.options.getRole('role')) {
                role = await <Role> options.options.getRole('role');
            }

            if (options.options.getString('reason')) {
                reason = await <string> options.options.getString('reason');
            }
        }
        else {
            if ((msg as Message).mentions.roles.first()) {
                role = await <Role> (msg as Message).mentions.roles.first();
            }

            if (!role) {
                await msg.reply({
                    embeds: [
                        new MessageEmbed()
                        .setColor('#f14a60')
                        .setDescription(`Invalid role given.`)
                    ],
                    ephemeral: true
                });
    
                return;
            }

            if (!raid) {
                lockall = options.normalArgs.filter(a => /^\d+$/gi.test(a) || a.startsWith('<#'));
            }
            else {
                if (msg.guild!.channels.cache.size < 2) {
                    await msg.guild?.channels.fetch();
                }

                const raidChannels = client.config.get('raid').channels;

                if (client.config.get('raid').exclude) {
                    lockall = [];

                    for await (const [id, { parent, type }] of msg.guild!.channels.cache) {
                        if (type === 'GUILD_CATEGORY')
                            continue;

                        if ((raidChannels.includes(id) || raidChannels.includes(parent?.id))) {
                            continue;
                        }

                        lockall.push(id);
                    }
                }
                else {
                    // for await (const [id, { parent, type }] of msg.guild!.channels.cache) {
                    //     if (type === 'GUILD_CATEGORY')
                    //         continue;

                    //     if ((!raidChannels.includes(id) && !raidChannels.includes(parent?.id))) {
                    //         continue;
                    //     }

                    //     lockall.push(id);
                    // }

                    lockall = raidChannels;
                }

                console.log("Raid", lockall);
                console.log("Raid 1", client.config.get('raid').channels);
            }
        }

        if (lockall.length === 0 && !raid) {
            await this.deferReply(msg, {
                content: "No channel specified!"
            });

            return;
        }

        if (msg.guild!.channels.cache.size < 2) {
            await msg.guild?.channels.fetch();
        }

        for await (const c of lockall) {
            console.log(c);
            const id = c.startsWith('<#') ? c.substring(2, c.length - 1) : c;

            if (lockallChannels.find(c => c.id === id))
                continue;
            
            const channel = msg.guild?.channels.cache.get(id);

            if (!channel) {
                continue;
            }
                
            if (!channel.isText()) {
                lockallChannels = [...lockallChannels, ...(msg.guild?.channels.cache.filter(c => c.parent?.id === id) as Collection<string, TextChannel>).values()!];
                continue;
            }

            lockallChannels.push(channel as TextChannel);
        }

        console.log("Array: ", lockallChannels, lockall);

        const [success, failure] = (await lockAll(client, role, lockallChannels, msg.member!.user as User, true, reason))!;

        if (options.isInteraction) {
            await this.deferReply(msg, {
                content: "Locked " + lockallChannels.length + " channel(s)." + (failure > 0 ? ` ${success} successful locks and ${failure} failed locks.` : '')
            });
        }
        else {
            await (msg as Message).react('🔒');
        }
    }
}
1	/**
2	* This file is part of SudoBot.
3	*
4	* Copyright (C) 2021-2022 OSN Inc.
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, CommandInteraction, Message, Permissions, Role, TextChannel, User } from 'discord.js';
21	import BaseCommand from '../../utils/structures/BaseCommand';
22	import CommandOptions from '../../types/CommandOptions';
23	import InteractionOptions from '../../types/InteractionOptions';
24	import DiscordClient from '../../client/Client';
25	import MessageEmbed from '../../client/MessageEmbed';
26
27	export async function lockAll(client: DiscordClient, role: Role, channels: Collection <string, TextChannel> \| TextChannel[], user: User, send: boolean = true, reason?: string) {
28	if (role) {
29	// const gen = await channels.first()!.guild.roles.fetch(client.config.props[channels.first()!.guild.id].gen_role);
30
31	try {
32	return await client.channelLock.lockAll(channels instanceof Collection ? [...channels.values()] : channels, user, {
33	role,
34	sendConfirmation: send,
35	reason
36	});
37	}
38	catch (e) {
39	console.log(e);
40	}
41	}
42	}
43
44	export default class LockallCommand extends BaseCommand {
45	supportsInteractions: boolean = true;
46	permissions = [Permissions.FLAGS.MANAGE_CHANNELS];
47
48	constructor() {
49	super('lockall', 'moderation', []);
50	}
51
52	async run(client: DiscordClient, msg: Message \| CommandInteraction, options: CommandOptions \| InteractionOptions) {
53	if (!options.isInteraction && typeof options.args[0] === 'undefined') {
54	await msg.reply({
55	embeds: [
56	new MessageEmbed()
57	.setColor('#f14a60')
58	.setDescription(`This command requires at least one argument.`)
59	]
60	});
61
62	return;
63	}
64
65	if (msg instanceof CommandInteraction) {
66	await msg.deferReply({ ephemeral: true });
67	}
68
69	const raid = options.isInteraction ? options.options.getBoolean('raid') === true : (options.options.indexOf('--raid') !== -1);
70
71	let role: Role = <Role> msg.guild!.roles.everyone;
72	let lockall: string[] = [], lockallChannels: TextChannel[] = [];
73	let reason: string \| undefined;
74	// const force = options.isInteraction ? options.options.getBoolean('force') === true : (options.options.indexOf('--force') !== -1);
75
76	if (options.isInteraction) {
77	if (options.options.getString('channels'))
78	lockall = options.options.getString('channels')!.split(' ').filter(a => /^\d+$/gi.test(a) \|\| a.startsWith('<#'));
79
80	if (options.options.getRole('role')) {
81	role = await <Role> options.options.getRole('role');
82	}
83
84	if (options.options.getString('reason')) {
85	reason = await <string> options.options.getString('reason');
86	}
87	}
88	else {
89	if ((msg as Message).mentions.roles.first()) {
90	role = await <Role> (msg as Message).mentions.roles.first();
91	}
92
93	if (!role) {
94	await msg.reply({
95	embeds: [
96	new MessageEmbed()
97	.setColor('#f14a60')
98	.setDescription(`Invalid role given.`)
99	],
100	ephemeral: true
101	});
102
103	return;
104	}
105
106	if (!raid) {
107	lockall = options.normalArgs.filter(a => /^\d+$/gi.test(a) \|\| a.startsWith('<#'));
108	}
109	else {
110	if (msg.guild!.channels.cache.size < 2) {
111	await msg.guild?.channels.fetch();
112	}
113
114	const raidChannels = client.config.get('raid').channels;
115
116	if (client.config.get('raid').exclude) {
117	lockall = [];
118
119	for await (const [id, { parent, type }] of msg.guild!.channels.cache) {
120	if (type === 'GUILD_CATEGORY')
121	continue;
122
123	if ((raidChannels.includes(id) \|\| raidChannels.includes(parent?.id))) {
124	continue;
125	}
126
127	lockall.push(id);
128	}
129	}
130	else {
131	// for await (const [id, { parent, type }] of msg.guild!.channels.cache) {
132	// if (type === 'GUILD_CATEGORY')
133	// continue;
134
135	// if ((!raidChannels.includes(id) && !raidChannels.includes(parent?.id))) {
136	// continue;
137	// }
138
139	// lockall.push(id);
140	// }
141
142	lockall = raidChannels;
143	}
144
145	console.log("Raid", lockall);
146	console.log("Raid 1", client.config.get('raid').channels);
147	}
148	}
149
150	if (lockall.length === 0 && !raid) {
151	await this.deferReply(msg, {
152	content: "No channel specified!"
153	});
154
155	return;
156	}
157
158	if (msg.guild!.channels.cache.size < 2) {
159	await msg.guild?.channels.fetch();
160	}
161
162	for await (const c of lockall) {
163	console.log(c);
164	const id = c.startsWith('<#') ? c.substring(2, c.length - 1) : c;
165
166	if (lockallChannels.find(c => c.id === id))
167	continue;
168
169	const channel = msg.guild?.channels.cache.get(id);
170
171	if (!channel) {
172	continue;
173	}
174
175	if (!channel.isText()) {
176	lockallChannels = [...lockallChannels, ...(msg.guild?.channels.cache.filter(c => c.parent?.id === id) as Collection<string, TextChannel>).values()!];
177	continue;
178	}
179
180	lockallChannels.push(channel as TextChannel);
181	}
182
183	console.log("Array: ", lockallChannels, lockall);
184
185	const [success, failure] = (await lockAll(client, role, lockallChannels, msg.member!.user as User, true, reason))!;
186
187	if (options.isInteraction) {
188	await this.deferReply(msg, {
189	content: "Locked " + lockallChannels.length + " channel(s)." + (failure > 0 ? ` ${success} successful locks and ${failure} failed locks.` : '')
190	});
191	}
192	else {
193	await (msg as Message).react('🔒');
194	}
195	}
196	}