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	/**
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	}