TravBot-v3/src/commands/admin.ts

import Command from "../core/command";
import {CommonLibrary, logs, botHasPermission, clean} from "../core/lib";
import {Config, Storage} from "../core/structures";
import {PermissionNames, getPermissionLevel} from "../core/permissions";
import {Permissions} from "discord.js";
import * as discord from "discord.js";

function getLogBuffer(type: string) {
    return {
        files: [
            {
                attachment: Buffer.alloc(logs[type].length, logs[type]),
                name: `${Date.now()}.${type}.log`
            }
        ]
    };
}

const activities = ["playing", "listening", "streaming", "watching"];
const statuses = ["online", "idle", "dnd", "invisible"];

export default new Command({
    description:
        "An all-in-one command to do admin stuff. You need to be either an admin of the server or one of the bot's mechanics to use this command.",
    async run($: CommonLibrary): Promise<any> {
        if (!$.member)
            return $.channel.send(
                "Couldn't find a member object for you! Did you make sure you used this in a server?"
            );
        const permLevel = getPermissionLevel($.member);
        $.channel.send(
            `${$.author.toString()}, your permission level is \`${
                PermissionNames[permLevel]
            }\` (${permLevel}).`
        );
    },
    subcommands: {
        set: new Command({
            description: "Set different per-guild settings for the bot.",
            run: "You have to specify the option you want to set.",
            permission: Command.PERMISSIONS.ADMIN,
            subcommands: {
                prefix: new Command({
                    description:
                        "Set a custom prefix for your guild. Removes your custom prefix if none is provided.",
                    usage: "(<prefix>)",
                    async run($: CommonLibrary): Promise<any> {
                        Storage.getGuild($.guild?.id || "N/A").prefix = null;
                        Storage.save();
                        $.channel.send(
                            `The custom prefix for this guild has been removed. My prefix is now back to \`${Config.prefix}\`.`
                        );
                    },
                    any: new Command({
                        async run($: CommonLibrary): Promise<any> {
                            Storage.getGuild($.guild?.id || "N/A").prefix =
                                $.args[0];
                            Storage.save();
                            $.channel.send(
                                `The custom prefix for this guild is now \`${$.args[0]}\`.`
                            );
                        }
                    })
                })
            }
        }),
        diag: new Command({
            description:
                'Requests a debug log with the "info" verbosity level.',
            permission: Command.PERMISSIONS.BOT_SUPPORT,
            async run($: CommonLibrary): Promise<any> {
                $.channel.send(getLogBuffer("info"));
            },
            any: new Command({
                description: `Select a verbosity to listen to. Available levels: \`[${Object.keys(
                    logs
                ).join(", ")}]\``,
                async run($: CommonLibrary): Promise<any> {
                    const type = $.args[0];

                    if (type in logs) $.channel.send(getLogBuffer(type));
                    else
                        $.channel.send(
                            `Couldn't find a verbosity level named \`${type}\`! The available types are \`[${Object.keys(
                                logs
                            ).join(", ")}]\`.`
                        );
                }
            })
        }),
        status: new Command({
            description: "Changes the bot's status.",
            permission: Command.PERMISSIONS.BOT_SUPPORT,
            async run($: CommonLibrary): Promise<any> {
                $.channel.send("Setting status to `online`...");
            },
            any: new Command({
                description: `Select a status to set to. Available statuses: \`[${statuses.join(
                    ", "
                )}]\`.`,
                async run($: CommonLibrary): Promise<any> {
                    if (!statuses.includes($.args[0]))
                        return $.channel.send("That status doesn't exist!");
                    else {
                        $.client.user?.setStatus($.args[0]);
                        $.channel.send(`Setting status to \`${$.args[0]}\`...`);
                    }
                }
            })
        }),
        purge: new Command({
            description: "Purges bot messages.",
            permission: Command.PERMISSIONS.BOT_SUPPORT,
            async run($: CommonLibrary): Promise<any> {
                if ($.message.channel instanceof discord.DMChannel) {
                    return;
                }
                $.message.delete();
                const msgs = await $.channel.messages.fetch({
                    limit: 100
                });
                const travMessages = msgs.filter(
                    (m) => m.author.id === $.client.user?.id
                );

                await $.message.channel
                    .send(`Found ${travMessages.size} messages to delete.`)
                    .then((m) =>
                        m.delete({
                            timeout: 5000
                        })
                    );
                await $.message.channel.bulkDelete(travMessages);
            }
        }),
        clear: new Command({
            description: "Clears a given amount of messages.",
            usage: "<amount>",
            run: "A number was not provided.",
            number: new Command({
                description: "Amount of messages to delete.",
                async run($: CommonLibrary): Promise<any> {
                    $.message.delete();
                    const fetched = await $.channel.messages.fetch({
                        limit: $.args[0]
                    });
                    $.channel
                        /// @ts-ignore
                        .bulkDelete(fetched)
                        .catch((error: any) =>
                            $.channel.send(`Error: ${error}`)
                        );
                }
            })
        }),
        eval: new Command({
            description: "Evaluate code.",
            usage: "<code>",
            permission: Command.PERMISSIONS.BOT_OWNER,
            async run($: CommonLibrary): Promise<any> {
                try {
                    const code = $.args.join(" ");
                    let evaled = eval(code);

                    if (typeof evaled !== "string")
                        evaled = require("util").inspect(evaled);
                    $.channel.send(clean(evaled), {code: "x1"});
                } catch (err) {
                    $.channel.send(`\`ERROR\` \`\`\`x1\n${clean(err)}\n\`\`\``);
                }
            }
        }),
        nick: new Command({
            description: "Change the bot's nickname.",
            permission: Command.PERMISSIONS.BOT_SUPPORT,
            async run($: CommonLibrary): Promise<any> {
                const nickName = $.args.join(" ");
                const trav = $.guild?.members.cache.find(
                    (member) => member.id === $.client.user?.id
                );
                await trav?.setNickname(nickName);
                if (
                    botHasPermission($.guild, Permissions.FLAGS.MANAGE_MESSAGES)
                )
                    $.message.delete({timeout: 5000}).catch($.handler.bind($));
                $.channel
                    .send(`Nickname set to \`${nickName}\``)
                    .then((m) => m.delete({timeout: 5000}));
            }
        }),
        guilds: new Command({
            description: "Shows a list of all guilds the bot is a member of.",
            permission: Command.PERMISSIONS.BOT_SUPPORT,
            async run($: CommonLibrary): Promise<any> {
                const guildList = $.client.guilds.cache
                    .array()
                    .map((e) => e.name);
                $.channel.send(guildList);
            }
        }),
        activity: new Command({
            description: "Set the activity of the bot.",
            permission: Command.PERMISSIONS.BOT_SUPPORT,
            usage: "<type> <string>",
            async run($: CommonLibrary): Promise<any> {
                $.client.user?.setActivity(".help", {
                    type: "LISTENING"
                });
                $.channel.send("Activity set to default.");
            },
            any: new Command({
                description: `Select an activity type to set. Available levels: \`[${activities.join(
                    ", "
                )}]\``,
                async run($: CommonLibrary): Promise<any> {
                    const type = $.args[0];

                    if (activities.includes(type)) {
                        $.client.user?.setActivity($.args.slice(1).join(" "), {
                            type: $.args[0].toUpperCase()
                        });
                        $.channel.send(
                            `Set activity to \`${$.args[0].toUpperCase()}\` \`${$.args
                                .slice(1)
                                .join(" ")}\`.`
                        );
                    } else
                        $.channel.send(
                            `Couldn't find an activity type named \`${type}\`! The available types are \`[${activities.join(
                                ", "
                            )}]\`.`
                        );
                }
            })
        })
    }
});