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