commands/moderation/LockallCommand.ts

import { Collection, CommandInteraction, GuildBasedChannel, GuildChannel, 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;

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