commands/utils/AddsnippetCommand.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, MessageAttachment, Permissions } 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';
import { download } from '../../utils/util';
import path from 'path';
import { fetchEmoji } from '../../utils/Emoji';

export default class AddsnippetCommand extends BaseCommand {
    permissions = [Permissions.FLAGS.MANAGE_MESSAGES];

    constructor() {
        super('addsnippet', 'utils', []);
    }

    async run(client: DiscordClient, msg: Message | CommandInteraction, options: CommandOptions | InteractionOptions) {
        if (!options.isInteraction && options.args[1] === undefined) {
            await msg.reply({
                embeds: [
                    new MessageEmbed()
                    .setColor('#f14a60')
                    .setDescription(`This command requires at least two arguments.`)
                ]
            });

            return;
        }

        let name: string;
        let content: string;
        let files: MessageAttachment[] = [];
        let filenames: string[] = [];

        if (options.isInteraction) {
            await (msg as CommandInteraction).deferReply();
            name = <string> await options.options.getString('name');
            content = <string> await options.options.getString('content');

            if (options.options.getAttachment('file')) 
                files.push(await options.options.getAttachment('file')!);
        }
        else {
            name = options.args[0];
            options.args.shift();
            content = options.args.join(' ');

            if ((msg as Message).attachments.first()) {
                files = (msg as Message).attachments.map(a => {
                    return {
                        name: a.name,
                        url: a.proxyURL
                    } as MessageAttachment;
                });
            }
        }

        if (client.snippetManager.get(msg.guild!.id, name)) {
            await this.deferReply(msg, {
                embeds: [
                    new MessageEmbed()
                    .setColor('#f14a60')
                    .setDescription("A snippet already exists with that name.")
                ]
            });

            return;
        }

        for await (const file of files) {
            try {
                let filename = Math.round(Math.random() * 1000000) + '_' + file.name!;
                filenames.push(filename);
                await download(file.url, path.resolve(process.env.SUDO_PREFIX ?? path.join(__dirname, '../../..'), 'storage', filename));
            }
            catch (e) {
                console.log(e);                
            }
        }

        await client.snippetManager.set(msg.guild!.id, name, content, filenames);
        await client.snippetManager.write();

        await this.deferReply(msg, {
            embeds: [
                new MessageEmbed()
                .setDescription((await fetchEmoji('check'))!.toString() + " Snippet created")
            ]
        });
    }
}
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, MessageAttachment, Permissions } 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			import { download } from '../../utils/util';
27			import path from 'path';
28			import { fetchEmoji } from '../../utils/Emoji';
29
30			export default class AddsnippetCommand extends BaseCommand {
31			permissions = [Permissions.FLAGS.MANAGE_MESSAGES];
32
33			constructor() {
34			super('addsnippet', 'utils', []);
35			}
36
37			async run(client: DiscordClient, msg: Message \| CommandInteraction, options: CommandOptions \| InteractionOptions) {
38			if (!options.isInteraction && options.args[1] === undefined) {
39			await msg.reply({
40			embeds: [
41			new MessageEmbed()
42			.setColor('#f14a60')
43			.setDescription(`This command requires at least two arguments.`)
44			]
45			});
46
47			return;
48			}
49
50			let name: string;
51			let content: string;
52			let files: MessageAttachment[] = [];
53			let filenames: string[] = [];
54
55			if (options.isInteraction) {
56			await (msg as CommandInteraction).deferReply();
57			name = <string> await options.options.getString('name');
58			content = <string> await options.options.getString('content');
59
60			if (options.options.getAttachment('file'))
61			files.push(await options.options.getAttachment('file')!);
62			}
63			else {
64			name = options.args[0];
65			options.args.shift();
66			content = options.args.join(' ');
67
68			if ((msg as Message).attachments.first()) {
69			files = (msg as Message).attachments.map(a => {
70			return {
71			name: a.name,
72			url: a.proxyURL
73			} as MessageAttachment;
74			});
75			}
76			}
77
78			if (client.snippetManager.get(msg.guild!.id, name)) {
79			await this.deferReply(msg, {
80			embeds: [
81			new MessageEmbed()
82			.setColor('#f14a60')
83			.setDescription("A snippet already exists with that name.")
84			]
85			});
86
87			return;
88			}
89
90			for await (const file of files) {
91			try {
92			let filename = Math.round(Math.random() * 1000000) + '_' + file.name!;
93			filenames.push(filename);
94			await download(file.url, path.resolve(process.env.SUDO_PREFIX ?? path.join(__dirname, '../../..'), 'storage', filename));
95			}
96			catch (e) {
97			console.log(e);
98			}
99			}
100
101			await client.snippetManager.set(msg.guild!.id, name, content, filenames);
102			await client.snippetManager.write();
103
104			await this.deferReply(msg, {
105			embeds: [
106			new MessageEmbed()
107			.setDescription((await fetchEmoji('check'))!.toString() + " Snippet created")
108			]
109			});
110			}
111			}