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, 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;    

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