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

import {
    Command,
    NamedCommand,
    botHasPermission,
    getPermissionLevel,
    getPermissionName,
    CHANNEL_TYPE,
    RestCommand
} from "../../core";
import {clean} from "../../lib";
import {Config, Storage} from "../../structures";
import {Permissions, TextChannel, User, Role} from "discord.js";
import {logs} from "../../modules/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 NamedCommand({
    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({send, author, member}) {
        const permLevel = getPermissionLevel(author, member);
        return send(`${author}, your permission level is \`${getPermissionName(permLevel)}\` (${permLevel}).`);
    },
    subcommands: {
        set: new NamedCommand({
            description: "Set different per-guild settings for the bot.",
            run: "You have to specify the option you want to set.",
            permission: PERMISSIONS.ADMIN,
            channelType: CHANNEL_TYPE.GUILD,
            subcommands: {
                prefix: new NamedCommand({
                    description: "Set a custom prefix for your guild. Removes your custom prefix if none is provided.",
                    usage: "(<prefix>) (<@bot>)",
                    async run({send, guild}) {
                        Storage.getGuild(guild!.id).prefix = null;
                        Storage.save();
                        send(
                            `The custom prefix for this guild has been removed. My prefix is now back to \`${Config.prefix}\`.`
                        );
                    },
                    any: new Command({
                        async run({send, guild, args}) {
                            Storage.getGuild(guild!.id).prefix = args[0];
                            Storage.save();
                            send(`The custom prefix for this guild is now \`${args[0]}\`.`);
                        },
                        user: new Command({
                            description: "Specifies the bot in case of conflicting prefixes.",
                            async run({send, guild, client, args}) {
                                if ((args[1] as User).id === client.user!.id) {
                                    Storage.getGuild(guild!.id).prefix = args[0];
                                    Storage.save();
                                    send(`The custom prefix for this guild is now \`${args[0]}\`.`);
                                }
                            }
                        })
                    })
                }),
                welcome: new NamedCommand({
                    description: "Configure your server's welcome settings for the bot.",
                    usage: "type/channel <...>",
                    run: "You need to specify which part to modify, `type`/`channel`.",
                    subcommands: {
                        type: new NamedCommand({
                            description:
                                "Sets how welcome messages are displayed for your server. Removes welcome messages if unspecified.",
                            usage: "`none`/`text`/`graphical`",
                            async run({send, guild}) {
                                Storage.getGuild(guild!.id).welcomeType = "none";
                                Storage.save();
                                send("Set this server's welcome type to `none`.");
                            },
                            // I should probably make this a bit more dynamic... Oh well.
                            subcommands: {
                                text: new NamedCommand({
                                    async run({send, guild}) {
                                        Storage.getGuild(guild!.id).welcomeType = "text";
                                        Storage.save();
                                        send("Set this server's welcome type to `text`.");
                                    }
                                }),
                                graphical: new NamedCommand({
                                    async run({send, guild}) {
                                        Storage.getGuild(guild!.id).welcomeType = "graphical";
                                        Storage.save();
                                        send("Set this server's welcome type to `graphical`.");
                                    }
                                })
                            }
                        }),
                        channel: new NamedCommand({
                            description: "Sets the welcome channel for your server. Type `#` to reference the channel.",
                            usage: "(<channel mention>)",
                            async run({send, channel, guild}) {
                                Storage.getGuild(guild!.id).welcomeChannel = channel.id;
                                Storage.save();
                                send(`Successfully set ${channel} as the welcome channel for this server.`);
                            },
                            id: "channel",
                            channel: new Command({
                                async run({send, guild, args}) {
                                    const result = args[0] as TextChannel;
                                    Storage.getGuild(guild!.id).welcomeChannel = result.id;
                                    Storage.save();
                                    send(`Successfully set this server's welcome channel to ${result}.`);
                                }
                            })
                        }),
                        message: new NamedCommand({
                            description:
                                "Sets a custom welcome message for your server. Use `%user%` as the placeholder for the user.",
                            usage: "(<message>)",
                            async run({send, guild}) {
                                Storage.getGuild(guild!.id).welcomeMessage = null;
                                Storage.save();
                                send("Reset your server's welcome message to the default.");
                            },
                            any: new RestCommand({
                                async run({send, guild, combined}) {
                                    Storage.getGuild(guild!.id).welcomeMessage = combined;
                                    Storage.save();
                                    send(`Set your server's welcome message to \`${combined}\`.`);
                                }
                            })
                        })
                    }
                }),
                stream: new NamedCommand({
                    description: "Set a channel to send stream notifications. Type `#` to reference the channel.",
                    usage: "(<channel mention>)",
                    async run({send, channel, guild}) {
                        const targetGuild = Storage.getGuild(guild!.id);

                        if (targetGuild.streamingChannel) {
                            targetGuild.streamingChannel = null;
                            send("Removed your server's stream notifications channel.");
                        } else {
                            targetGuild.streamingChannel = channel.id;
                            send(`Set your server's stream notifications channel to ${channel}.`);
                        }

                        Storage.save();
                    },
                    id: "channel",
                    channel: new Command({
                        async run({send, guild, args}) {
                            const result = args[0] as TextChannel;
                            Storage.getGuild(guild!.id).streamingChannel = result.id;
                            Storage.save();
                            send(`Successfully set this server's stream notifications channel to ${result}.`);
                        }
                    })
                }),
                streamrole: new NamedCommand({
                    description: "Sets/removes a stream notification role (and the corresponding category name)",
                    usage: "set/remove <...>",
                    run: "You need to enter in a role.",
                    subcommands: {
                        set: new NamedCommand({
                            usage: "<role> <category>",
                            id: "role",
                            role: new Command({
                                run: "You need to enter a category name.",
                                any: new RestCommand({
                                    async run({send, guild, args, combined}) {
                                        const role = args[0] as Role;
                                        Storage.getGuild(guild!.id).streamingRoles[role.id] = combined;
                                        Storage.save();
                                        send(
                                            `Successfully set the category \`${combined}\` to notify \`${role.name}\`.`
                                        );
                                    }
                                })
                            })
                        }),
                        remove: new NamedCommand({
                            usage: "<role>",
                            id: "role",
                            role: new Command({
                                async run({send, guild, args}) {
                                    const role = args[0] as Role;
                                    const guildStorage = Storage.getGuild(guild!.id);
                                    const category = guildStorage.streamingRoles[role.id];
                                    delete guildStorage.streamingRoles[role.id];
                                    Storage.save();
                                    send(
                                        `Successfully removed the category \`${category}\` to notify \`${role.name}\`.`
                                    );
                                }
                            })
                        })
                    }
                })
            }
        }),
        diag: new NamedCommand({
            description: 'Requests a debug log with the "info" verbosity level.',
            permission: PERMISSIONS.BOT_SUPPORT,
            async run({send}) {
                send(getLogBuffer("info"));
            },
            any: new Command({
                description: `Select a verbosity to listen to. Available levels: \`[${Object.keys(logs).join(", ")}]\``,
                async run({send, args}) {
                    const type = args[0];

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

                    await send(`Found ${travMessages.size} messages to delete.`).then((m) =>
                        m.delete({
                            timeout: 5000
                        })
                    );
                    await (channel as TextChannel).bulkDelete(travMessages);
                } else {
                    send("This command must be executed in a guild where I have the `MANAGE_MESSAGES` permission.");
                }
            }
        }),
        clear: new NamedCommand({
            description: "Clears a given amount of messages.",
            usage: "<amount>",
            channelType: CHANNEL_TYPE.GUILD,
            run: "A number was not provided.",
            number: new Command({
                description: "Amount of messages to delete.",
                async run({message, channel, args}) {
                    message.delete();
                    const fetched = await channel.messages.fetch({
                        limit: args[0]
                    });
                    return await (channel as TextChannel).bulkDelete(fetched);
                }
            })
        }),
        eval: new NamedCommand({
            description: "Evaluate code.",
            usage: "<code>",
            permission: PERMISSIONS.BOT_OWNER,
            run: "You have to enter some code to execute first.",
            any: new RestCommand({
                // 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({send, message, channel, guild, author, member, client, args, combined}) {
                    try {
                        let evaled = eval(combined);
                        if (typeof evaled !== "string") evaled = require("util").inspect(evaled);
                        send(clean(evaled), {code: "js", split: true});
                    } catch (err) {
                        send(clean(err), {code: "js", split: true});
                    }
                }
            })
        }),
        nick: new NamedCommand({
            description: "Change the bot's nickname.",
            permission: PERMISSIONS.BOT_SUPPORT,
            channelType: CHANNEL_TYPE.GUILD,
            run: "You have to specify a nickname to set for the bot",
            any: new RestCommand({
                async run({send, message, guild, combined}) {
                    await guild!.me?.setNickname(combined);
                    if (botHasPermission(guild, Permissions.FLAGS.MANAGE_MESSAGES)) message.delete({timeout: 5000});
                    send(`Nickname set to \`${combined}\``).then((m) => m.delete({timeout: 5000}));
                }
            })
        }),
        guilds: new NamedCommand({
            description: "Shows a list of all guilds the bot is a member of.",
            permission: PERMISSIONS.BOT_SUPPORT,
            async run({send, client}) {
                const guildList = client.guilds.cache.array().map((e) => e.name);
                send(guildList, {split: true});
            }
        }),
        activity: new NamedCommand({
            description: "Set the activity of the bot.",
            permission: PERMISSIONS.BOT_SUPPORT,
            usage: "<type> <string>",
            async run({send, client}) {
                client.user?.setActivity(".help", {
                    type: "LISTENING"
                });
                send("Activity set to default.");
            },
            any: new RestCommand({
                description: `Select an activity type to set. Available levels: \`[${activities.join(", ")}]\``,
                async run({send, client, args}) {
                    const type = args[0];

                    if (activities.includes(type)) {
                        client.user?.setActivity(args.slice(1).join(" "), {
                            type: args[0].toUpperCase()
                        });
                        send(`Set activity to \`${args[0].toUpperCase()}\` \`${args.slice(1).join(" ")}\`.`);
                    } else
                        send(
                            `Couldn't find an activity type named \`${type}\`! The available types are \`[${activities.join(
                                ", "
                            )}]\`.`
                        );
                }
            })
        }),
        syslog: new NamedCommand({
            description: "Sets up the current channel to receive system logs.",
            permission: PERMISSIONS.BOT_ADMIN,
            channelType: CHANNEL_TYPE.GUILD,
            async run({send, channel}) {
                Config.systemLogsChannel = channel.id;
                Config.save();
                send(`Successfully set ${channel} as the system logs channel.`);
            },
            channel: new Command({
                async run({send, args}) {
                    const targetChannel = args[0] as TextChannel;
                    Config.systemLogsChannel = targetChannel.id;
                    Config.save();
                    send(`Successfully set ${targetChannel} as the system logs channel.`);
                }
            })
        })
    }
});