TravBot-v3/src/commands/system/admin.ts

import Command, {handler} from "../../core/command";
import {clean} from "../../core/lib";
import {botHasPermission} from "../../core/libd";
import {Config, Storage} from "../../core/structures";
import {getPermissionLevel, getPermissionName} from "../../core/permissions";
import {Permissions} from "discord.js";
import {logs} from "../../globals";

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($) {
        if (!$.member) {
            $.channel.send("Couldn't find a member object for you! Did you make sure you used this in a server?");
            return;
        }
        const permLevel = getPermissionLevel($.member);
        $.channel.send(
            `${$.author.toString()}, your permission level is \`${getPermissionName(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: 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($) {
                        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($) {
                            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: PERMISSIONS.BOT_SUPPORT,
            async run($) {
                $.channel.send(getLogBuffer("info"));
            },
            any: new Command({
                description: `Select a verbosity to listen to. Available levels: \`[${Object.keys(logs).join(", ")}]\``,
                async run($) {
                    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: PERMISSIONS.BOT_SUPPORT,
            async run($) {
                $.channel.send("Setting status to `online`...");
            },
            any: new Command({
                description: `Select a status to set to. Available statuses: \`[${statuses.join(", ")}]\`.`,
                async run($) {
                    if (!statuses.includes($.args[0])) {
                        $.channel.send("That status doesn't exist!");
                        return;
                    } else {
                        $.client.user?.setStatus($.args[0]);
                        $.channel.send(`Setting status to \`${$.args[0]}\`...`);
                    }
                }
            })
        }),
        purge: new Command({
            description: "Purges the bot's own messages.",
            permission: PERMISSIONS.BOT_SUPPORT,
            async run($) {
                // It's probably better to go through the bot's own messages instead of calling bulkDelete which requires MANAGE_MESSAGES.
                if (botHasPermission($.guild, Permissions.FLAGS.MANAGE_MESSAGES) && $.channel.type !== "dm") {
                    $.message.delete();
                    const msgs = await $.channel.messages.fetch({
                        limit: 100
                    });
                    const travMessages = msgs.filter((m) => m.author.id === $.client.user?.id);

                    await $.channel.send(`Found ${travMessages.size} messages to delete.`).then((m) =>
                        m.delete({
                            timeout: 5000
                        })
                    );
                    await $.channel.bulkDelete(travMessages);
                } else {
                    $.channel.send(
                        "This command must be executed in a guild where I have the `MANAGE_MESSAGES` permission."
                    );
                }
            }
        }),
        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($) {
                    if ($.channel.type === "dm") {
                        await $.channel.send("Can't clear messages in the DMs!");
                        return;
                    }
                    $.message.delete();
                    const fetched = await $.channel.messages.fetch({
                        limit: $.args[0]
                    });
                    await $.channel.bulkDelete(fetched);
                }
            })
        }),
        eval: new Command({
            description: "Evaluate code.",
            usage: "<code>",
            permission: PERMISSIONS.BOT_OWNER,
            // You have to bring everything into scope to use them. AFAIK, there isn't a more maintainable way to do this, but at least TS will let you know if anything gets removed.
            async run({args, author, channel, client, guild, member, message}) {
                try {
                    const code = args.join(" ");
                    let evaled = eval(code);

                    if (typeof evaled !== "string") evaled = require("util").inspect(evaled);
                    channel.send(clean(evaled), {code: "js", split: true});
                } catch (err) {
                    channel.send(clean(err), {code: "js", split: true});
                }
            }
        }),
        nick: new Command({
            description: "Change the bot's nickname.",
            permission: PERMISSIONS.BOT_SUPPORT,
            async run($) {
                const nickName = $.args.join(" ");
                await $.guild?.me?.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: PERMISSIONS.BOT_SUPPORT,
            async run($) {
                const guildList = $.client.guilds.cache.array().map((e) => e.name);
                $.channel.send(guildList, {split: true});
            }
        }),
        activity: new Command({
            description: "Set the activity of the bot.",
            permission: PERMISSIONS.BOT_SUPPORT,
            usage: "<type> <string>",
            async run($) {
                $.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($) {
                    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(
                                ", "
                            )}]\`.`
                        );
                }
            })
        })
    }
});