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