commands/moderation/LockCommand.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 { CommandInteraction, Message, Permissions, Role, TextChannel, User } from 'discord.js';
import BaseCommand from '../../utils/structures/BaseCommand';
import DiscordClient from '../../client/Client';
import CommandOptions from '../../types/CommandOptions';
import InteractionOptions from '../../types/InteractionOptions';
import MessageEmbed from '../../client/MessageEmbed';

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

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

    async run(client: DiscordClient, msg: Message | CommandInteraction, options: CommandOptions | InteractionOptions) {
        let channel: TextChannel = <TextChannel> msg.channel;
        let role: Role = <Role> msg.guild!.roles.everyone;

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

        if (options.isInteraction) {
            if (options.options.getChannel('channel')) {
                channel = await <TextChannel> options.options.getChannel('channel');
            }

            if (options.options.getRole('role')) {
                role = await <Role> options.options.getRole('role');
            }
        }
        else {
            if ((msg as Message).mentions.roles.first()) {
                role = await <Role> (msg as Message).mentions.roles.first();
            }
            else if (options.normalArgs[0] && options.normalArgs[0] !== 'everyone') {
                role = <Role> await (msg as Message).guild?.roles.fetch(options.normalArgs[0]);
            }

            if ((msg as Message).mentions.channels.first()) {
                channel = await <TextChannel> (msg as Message).mentions.channels.first();
            }
            else if (options.normalArgs[1]) {
                channel = <TextChannel> await (msg as Message).guild?.channels.fetch(options.normalArgs[1]);
            }

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

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

        try {
            const result = await client.channelLock.lock(channel, msg.member!.user as User, { sendConfirmation: true });

            let error = null;

            if (!result) {
                error = 'This channel is already locked' + (role.id === msg.guild!.id ? '' : ' for the given role') + '.';
            }

            if (error) {
                await this.deferReply(msg, {
                    content: error,
                });

                return;
            }
            
            if (options.isInteraction) {
                await this.deferReply(msg, {
                    content: "Channel locked.",
                });
            }
            else {
                await (msg as Message).react('🔒');
            }
        }
        catch (e) {
            console.log(e);
            
            await msg.reply({
                embeds: [
                    new MessageEmbed()
                    .setColor('#f14a60')
                    .setDescription(`Failed to lock channel. Maybe missing permissions?`)
                ],
                ephemeral: true
            });

            return;
        }
    }
}
1	rakinar2	577	/**
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 { CommandInteraction, Message, Permissions, Role, TextChannel, User } from 'discord.js';
21			import BaseCommand from '../../utils/structures/BaseCommand';
22			import DiscordClient from '../../client/Client';
23			import CommandOptions from '../../types/CommandOptions';
24			import InteractionOptions from '../../types/InteractionOptions';
25			import MessageEmbed from '../../client/MessageEmbed';
26
27			export default class LockCommand extends BaseCommand {
28			supportsInteractions: boolean = true;
29			permissions = [Permissions.FLAGS.MANAGE_CHANNELS];
30
31			constructor() {
32			super('lock', 'moderation', []);
33			}
34
35			async run(client: DiscordClient, msg: Message \| CommandInteraction, options: CommandOptions \| InteractionOptions) {
36			let channel: TextChannel = <TextChannel> msg.channel;
37			let role: Role = <Role> msg.guild!.roles.everyone;
38
39			if (msg instanceof CommandInteraction)
40			await msg.deferReply({ ephemeral: true });
41
42			if (options.isInteraction) {
43			if (options.options.getChannel('channel')) {
44			channel = await <TextChannel> options.options.getChannel('channel');
45			}
46
47			if (options.options.getRole('role')) {
48			role = await <Role> options.options.getRole('role');
49			}
50			}
51			else {
52			if ((msg as Message).mentions.roles.first()) {
53			role = await <Role> (msg as Message).mentions.roles.first();
54			}
55			else if (options.normalArgs[0] && options.normalArgs[0] !== 'everyone') {
56			role = <Role> await (msg as Message).guild?.roles.fetch(options.normalArgs[0]);
57			}
58
59			if ((msg as Message).mentions.channels.first()) {
60			channel = await <TextChannel> (msg as Message).mentions.channels.first();
61			}
62			else if (options.normalArgs[1]) {
63			channel = <TextChannel> await (msg as Message).guild?.channels.fetch(options.normalArgs[1]);
64			}
65
66			if (!role) {
67			await msg.reply({
68			embeds: [
69			new MessageEmbed()
70			.setColor('#f14a60')
71			.setDescription(`Invalid role given.`)
72			],
73			ephemeral: true
74			});
75
76			return;
77			}
78
79			if (!channel \|\| channel.type !== 'GUILD_TEXT') {
80			await msg.reply({
81			embeds: [
82			new MessageEmbed()
83			.setColor('#f14a60')
84			.setDescription(`Invalid text channel given.`)
85			],
86			ephemeral: true
87			});
88
89			return;
90			}
91			}
92
93			try {
94			const result = await client.channelLock.lock(channel, msg.member!.user as User, { sendConfirmation: true });
95
96			let error = null;
97
98			if (!result) {
99			error = 'This channel is already locked' + (role.id === msg.guild!.id ? '' : ' for the given role') + '.';
100			}
101
102			if (error) {
103			await this.deferReply(msg, {
104			content: error,
105			});
106
107			return;
108			}
109
110			if (options.isInteraction) {
111			await this.deferReply(msg, {
112			content: "Channel locked.",
113			});
114			}
115			else {
116			await (msg as Message).react('🔒');
117			}
118			}
119			catch (e) {
120			console.log(e);
121
122			await msg.reply({
123			embeds: [
124			new MessageEmbed()
125			.setColor('#f14a60')
126			.setDescription(`Failed to lock channel. Maybe missing permissions?`)
127			],
128			ephemeral: true
129			});
130
131			return;
132			}
133			}
134			}