commands/settings/BlockedFileAddCommand.ts

/**
 * This file is part of SudoBot.
 *
 * Copyright (C) 2021-2023 OSN Developers.
 *
 * 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 { Message, escapeMarkdown } from "discord.js";
import Command, { AnyCommandContext, CommandMessage, CommandReturn, ValidationRule } from "../../core/Command";

export default class BlockedFileAddCommand extends Command {
    public readonly name = "blockedfile__add";
    public readonly validationRules: ValidationRule[] = [];
    public readonly permissions = [];
    public readonly supportsInteractions = false;
    public readonly supportsLegacy = false;
    public readonly description = "Add a file hash to the blocklist";

    async execute(message: CommandMessage, context: AnyCommandContext): Promise<CommandReturn> {
        await this.deferIfInteraction(message);
        const attachments =
            message instanceof Message ? [...message.attachments.values()] : [message.options.getAttachment("file", true)];

        if (!this.client.configManager.config[message.guildId!]) {
            return;
        }

        if (!this.client.configManager.config[message.guildId!]!.file_filter?.enabled) {
            await this.error(message, "The blocked file filter is not enabled for this server.");
            return;
        }

        if (attachments.length === 0) {
            await this.error(message, "Please specify at least one file to block!");
            return;
        }

        this.client.configManager.config[message.guildId!]!.file_filter ??= {
            blocked_hashes: {},
            enabled: false,
            disabled_channels: []
        };

        let description = "";

        for (const attachment of attachments) {
            const hash = await this.client.fileFilter.getFileHashFromURL(attachment.proxyURL);

            if (!hash) {
                continue;
            }

            this.client.configManager.config[message.guildId!]!.file_filter!.blocked_hashes[hash] = attachment.contentType;
            description +=
                `* [${escapeMarkdown(attachment.name)}](${attachment.proxyURL}): \`${hash}\`` +
                (attachment.contentType ? ` (${attachment.contentType})` : "") +
                "\n";
        }

        await this.client.configManager.write();
        await this.deferredReply(message, {
            embeds: [
                {
                    description: `${this.emoji("check")} **Successfully blocked the file hashes.**\n\n${description}`,
                    color: 0x007bff
                }
            ]
        });
    }
}
1	/**
2	* This file is part of SudoBot.
3	*
4	* Copyright (C) 2021-2023 OSN Developers.
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 { Message, escapeMarkdown } from "discord.js";
21	import Command, { AnyCommandContext, CommandMessage, CommandReturn, ValidationRule } from "../../core/Command";
22
23	export default class BlockedFileAddCommand extends Command {
24	public readonly name = "blockedfile__add";
25	public readonly validationRules: ValidationRule[] = [];
26	public readonly permissions = [];
27	public readonly supportsInteractions = false;
28	public readonly supportsLegacy = false;
29	public readonly description = "Add a file hash to the blocklist";
30
31	async execute(message: CommandMessage, context: AnyCommandContext): Promise<CommandReturn> {
32	await this.deferIfInteraction(message);
33	const attachments =
34	message instanceof Message ? [...message.attachments.values()] : [message.options.getAttachment("file", true)];
35
36	if (!this.client.configManager.config[message.guildId!]) {
37	return;
38	}
39
40	if (!this.client.configManager.config[message.guildId!]!.file_filter?.enabled) {
41	await this.error(message, "The blocked file filter is not enabled for this server.");
42	return;
43	}
44
45	if (attachments.length === 0) {
46	await this.error(message, "Please specify at least one file to block!");
47	return;
48	}
49
50	this.client.configManager.config[message.guildId!]!.file_filter ??= {
51	blocked_hashes: {},
52	enabled: false,
53	disabled_channels: []
54	};
55
56	let description = "";
57
58	for (const attachment of attachments) {
59	const hash = await this.client.fileFilter.getFileHashFromURL(attachment.proxyURL);
60
61	if (!hash) {
62	continue;
63	}
64
65	this.client.configManager.config[message.guildId!]!.file_filter!.blocked_hashes[hash] = attachment.contentType;
66	description +=
67	`* [${escapeMarkdown(attachment.name)}](${attachment.proxyURL}): \`${hash}\`` +
68	(attachment.contentType ? ` (${attachment.contentType})` : "") +
69	"\n";
70	}
71
72	await this.client.configManager.write();
73	await this.deferredReply(message, {
74	embeds: [
75	{
76	description: `${this.emoji("check")} Successfully blocked the file hashes.\n\n${description}`,
77	color: 0x007bff
78	}
79	]
80	});
81	}
82	}