PluralKit/crates/command_definitions/src/system.rs

use command_parser::token::TokensIterator;

use crate::utils::get_list_flags;

use super::*;

pub fn cmds() -> impl Iterator<Item = Command> {
    edit()
}

pub fn system() -> (&'static str, [&'static str; 1]) {
    ("system", ["s"])
}

pub fn targeted() -> TokensIterator {
    tokens!(system(), SystemRef)
}

pub fn edit() -> impl Iterator<Item = Command> {
    let system = system();
    let system_target = targeted();

    let system_new = tokens!(system, ("new", ["n"]));
    let system_new_cmd = [
        command!(system_new => "system_new").help("Creates a new system"),
        command!(system_new, ("name", OpaqueStringRemainder) => "system_new_name")
            .help("Creates a new system (using the provided name)"),
    ]
    .into_iter();

    let system_webhook = tokens!(system, ("webhook", ["hook"]));
    let system_webhook_cmd = [
        command!(system_webhook => "system_webhook_show").help("Shows your system's webhook URL"),
        command!(system_webhook, CLEAR => "system_webhook_clear")
            .flag(YES)
            .help("Clears your system's webhook URL"),
        command!(system_webhook, ("url", OpaqueString) => "system_webhook_set")
            .help("Sets your system's webhook URL"),
    ]
    .into_iter();

    let add_info_flags = |cmd: Command| {
        cmd.flag(("public", ["pub"]))
            .flag(("private", ["priv"]))
            .flag(ALL)
    };
    let system_info_cmd_self = std::iter::once(add_info_flags(
        command!(system => "system_info_self").help("Shows information about your system"),
    ));
    let system_info_cmd = std::iter::once(add_info_flags(
        command!(system_target, ("info", ["show", "view"]) => "system_info")
            .help("Shows information about your system"),
    ));

    let system_name = tokens!(system_target, "name");
    let system_name_cmd =
        std::iter::once(command!(system_name => "system_show_name").help("Shows the systems name"));

    let system_name_self = tokens!(system, "name");
    let system_name_self_cmd = [
        command!(system_name_self => "system_show_name_self").help("Shows your system's name"),
        command!(system_name_self, CLEAR => "system_clear_name")
            .flag(YES)
            .help("Clears your system's name"),
        command!(system_name_self, ("name", OpaqueStringRemainder) => "system_rename")
            .help("Renames your system"),
    ]
    .into_iter();

    let system_server_name = tokens!(system_target, ("servername", ["sn", "guildname"]));
    let system_server_name_cmd = std::iter::once(
        command!(system_server_name => "system_show_server_name")
            .help("Shows the system's server name"),
    );

    let system_server_name_self = tokens!(system, ("servername", ["sn", "guildname"]));
    let system_server_name_self_cmd = [
        command!(system_server_name_self => "system_show_server_name_self")
            .help("Shows your system's server name"),
        command!(system_server_name_self, CLEAR => "system_clear_server_name")
            .flag(YES)
            .help("Clears your system's server name"),
        command!(system_server_name_self, ("name", OpaqueStringRemainder) => "system_rename_server_name")
            .help("Renames your system's server name"),
    ]
    .into_iter();

    let system_description = tokens!(system_target, ("description", ["desc", "d"]));
    let system_description_cmd = std::iter::once(
        command!(system_description => "system_show_description")
            .help("Shows the system's description"),
    );

    let system_description_self = tokens!(system, ("description", ["desc", "d"]));
    let system_description_self_cmd = [
        command!(system_description_self => "system_show_description_self").help("Shows your system's description"),
        command!(system_description_self, CLEAR => "system_clear_description")
        .flag(YES)
            .help("Clears your system's description"),
        command!(system_description_self, ("description", OpaqueStringRemainder) => "system_change_description")
            .help("Changes your system's description"),
    ]
    .into_iter();

    let system_color = tokens!(system_target, ("color", ["colour"]));
    let system_color_cmd = std::iter::once(
        command!(system_color => "system_show_color").help("Shows the system's color"),
    );

    let system_color_self = tokens!(system, ("color", ["colour"]));
    let system_color_self_cmd = [
        command!(system_color_self => "system_show_color_self").help("Shows your system's color"),
        command!(system_color_self, CLEAR => "system_clear_color")
            .flag(YES)
            .help("Clears your system's color"),
        command!(system_color_self, ("color", OpaqueString) => "system_change_color")
            .help("Changes your system's color"),
    ]
    .into_iter();

    let system_tag = tokens!(system_target, ("tag", ["suffix"]));
    let system_tag_cmd =
        std::iter::once(command!(system_tag => "system_show_tag").help("Shows the system's tag"));

    let system_tag_self = tokens!(system, ("tag", ["suffix"]));
    let system_tag_self_cmd = [
        command!(system_tag_self => "system_show_tag_self").help("Shows your system's tag"),
        command!(system_tag_self, CLEAR => "system_clear_tag")
            .flag(YES)
            .help("Clears your system's tag"),
        command!(system_tag_self, ("tag", OpaqueStringRemainder) => "system_change_tag")
            .help("Changes your system's tag"),
    ]
    .into_iter();

    let system_server_tag = tokens!(system_target, ("servertag", ["st", "guildtag"]));
    let system_server_tag_cmd = std::iter::once(
        command!(system_server_tag => "system_show_server_tag")
            .help("Shows the system's server tag"),
    );

    let system_server_tag_self = tokens!(system, ("servertag", ["st", "guildtag"]));
    let system_server_tag_self_cmd = [
        command!(system_server_tag_self => "system_show_server_tag_self")
            .help("Shows your system's server tag"),
        command!(system_server_tag_self, CLEAR => "system_clear_server_tag")
            .flag(YES)
            .help("Clears your system's server tag"),
        command!(system_server_tag_self, ("tag", OpaqueStringRemainder) => "system_change_server_tag")
            .help("Changes your system's server tag"),
    ]
    .into_iter();

    let system_pronouns = tokens!(system_target, ("pronouns", ["prns"]));
    let system_pronouns_cmd = std::iter::once(
        command!(system_pronouns => "system_show_pronouns").help("Shows the system's pronouns"),
    );

    let system_pronouns_self = tokens!(system, ("pronouns", ["prns"]));
    let system_pronouns_self_cmd = [
        command!(system_pronouns_self => "system_show_pronouns_self")
            .help("Shows your system's pronouns"),
        command!(system_pronouns_self, CLEAR => "system_clear_pronouns")
            .flag(YES)
            .help("Clears your system's pronouns"),
        command!(system_pronouns_self, ("pronouns", OpaqueStringRemainder) => "system_change_pronouns")
            .help("Changes your system's pronouns"),
    ]
    .into_iter();

    let system_avatar = tokens!(system_target, ("avatar", ["pfp"]));
    let system_avatar_cmd = std::iter::once(
        command!(system_avatar => "system_show_avatar").help("Shows the system's avatar"),
    );

    let system_avatar_self = tokens!(system, ("avatar", ["pfp"]));
    let system_avatar_self_cmd = [
        command!(system_avatar_self => "system_show_avatar_self")
            .help("Shows your system's avatar"),
        command!(system_avatar_self, CLEAR => "system_clear_avatar")
            .flag(YES)
            .help("Clears your system's avatar"),
        command!(system_avatar_self, ("avatar", Avatar) => "system_change_avatar")
            .help("Changes your system's avatar"),
    ]
    .into_iter();

    let system_server_avatar = tokens!(system_target, ("serveravatar", ["spfp"]));
    let system_server_avatar_cmd = std::iter::once(
        command!(system_server_avatar => "system_show_server_avatar")
            .help("Shows the system's server avatar"),
    );

    let system_server_avatar_self = tokens!(system, ("serveravatar", ["spfp"]));
    let system_server_avatar_self_cmd = [
        command!(system_server_avatar_self => "system_show_server_avatar_self")
            .help("Shows your system's server avatar"),
        command!(system_server_avatar_self, CLEAR => "system_clear_server_avatar")
            .flag(YES)
            .help("Clears your system's server avatar"),
        command!(system_server_avatar_self, ("avatar", Avatar) => "system_change_server_avatar")
            .help("Changes your system's server avatar"),
    ]
    .into_iter();

    let system_banner = tokens!(system_target, ("banner", ["cover"]));
    let system_banner_cmd = std::iter::once(
        command!(system_banner => "system_show_banner").help("Shows the system's banner"),
    );

    let system_banner_self = tokens!(system, ("banner", ["cover"]));
    let system_banner_self_cmd = [
        command!(system_banner_self => "system_show_banner_self")
            .help("Shows your system's banner"),
        command!(system_banner_self, CLEAR => "system_clear_banner")
            .flag(YES)
            .help("Clears your system's banner"),
        command!(system_banner_self, ("banner", Avatar) => "system_change_banner")
            .help("Changes your system's banner"),
    ]
    .into_iter();

    let system_delete = std::iter::once(
        command!(system, ("delete", ["erase", "remove", "yeet"]) => "system_delete")
            .flag(("no-export", ["ne"]))
            .help("Deletes the system"),
    );

    let system_proxy = tokens!(system, "proxy");
    let system_proxy_cmd = [
        command!(system_proxy => "system_show_proxy_current")
            .help("Shows your system's proxy setting for the guild you are in"),
        command!(system_proxy, Skip(Toggle) => "system_toggle_proxy_current")
            .help("Toggle your system's proxy for the guild you are in"),
        command!(system_proxy, GuildRef => "system_show_proxy")
            .help("Shows your system's proxy setting for a guild"),
        command!(system_proxy, GuildRef, Toggle => "system_toggle_proxy")
            .help("Toggle your system's proxy for a guild"),
    ]
    .into_iter();

    let system_privacy = tokens!(system, ("privacy", ["priv"]));
    let system_privacy_cmd = [
        command!(system_privacy => "system_show_privacy")
            .help("Shows your system's privacy settings"),
            command!(system_privacy, ALL, ("level", PrivacyLevel) => "system_change_privacy_all")
            .help("Changes all privacy settings for your system"),
        command!(system_privacy, ("privacy", SystemPrivacyTarget), ("level", PrivacyLevel) => "system_change_privacy")
            .help("Changes a specific privacy setting for your system"),
    ].into_iter();

    let system_front = tokens!(system_target, ("front", ["fronter", "fronters", "f"]));
    let system_front_cmd = [
        command!(system_front => "system_fronter"),
        command!(system_front, ("history", ["h"]) => "system_fronter_history").flag(CLEAR),
        command!(system_front, ("percent", ["p", "%"]) => "system_fronter_percent")
            .flag(("duration", OpaqueString))
            .flag(("fronters-only", ["fo"]))
            .flag("flat"),
    ]
    .into_iter();

    let system_link = [
        command!("link", ("account", UserRef) => "system_link"),
        command!("unlink", ("account", OpaqueString) => "system_unlink").flag(YES),
    ]
    .into_iter();

    let system_list = ("members", ["list", "l", "find", "f"]);
    let search = tokens!(
        ("search", ["query", "find"]),
        ("query", OpaqueStringRemainder),
    );
    let add_list_flags = |cmd: Command| cmd.flags(get_list_flags());
    let system_list_cmd = [
        command!(system_target, system_list => "system_members_list"),
        command!(system_target, search => "system_members_search"),
    ]
    .into_iter()
    .map(add_list_flags);
    let system_list_self_cmd = [
        command!(system_list => "system_members_list_self"),
        command!(system, system_list => "system_members_list_self"),
        command!(system, search => "system_members_search_self"),
    ]
    .into_iter()
    .map(add_list_flags);

    let system_groups = tokens!(system_target, ("groups", ["gs"]));
    let system_groups_cmd = [
        command!(system_groups => "system_list_groups"),
        command!(system_groups, ("list", ["ls"]) => "system_list_groups"),
        command!(system_groups, search => "system_search_groups"),
    ]
    .into_iter()
    .map(add_list_flags);

    let system_display_id_self_cmd =
        std::iter::once(command!(system, "id" => "system_display_id_self"));
    let system_display_id_cmd =
        std::iter::once(command!(system_target, "id" => "system_display_id"));

    system_info_cmd_self
        .chain(system_new_cmd)
        .chain(system_webhook_cmd)
        .chain(system_name_self_cmd)
        .chain(system_server_name_self_cmd)
        .chain(system_description_self_cmd)
        .chain(system_color_self_cmd)
        .chain(system_tag_self_cmd)
        .chain(system_server_tag_self_cmd)
        .chain(system_pronouns_self_cmd)
        .chain(system_avatar_self_cmd)
        .chain(system_server_avatar_self_cmd)
        .chain(system_banner_self_cmd)
        .chain(system_list_self_cmd)
        .chain(system_display_id_self_cmd)
        .chain(system_delete)
        .chain(system_privacy_cmd)
        .chain(system_proxy_cmd)
        .chain(system_name_cmd)
        .chain(system_server_name_cmd)
        .chain(system_description_cmd)
        .chain(system_color_cmd)
        .chain(system_tag_cmd)
        .chain(system_server_tag_cmd)
        .chain(system_pronouns_cmd)
        .chain(system_avatar_cmd)
        .chain(system_server_avatar_cmd)
        .chain(system_banner_cmd)
        .chain(system_info_cmd)
        .chain(system_front_cmd)
        .chain(system_link)
        .chain(system_list_cmd)
        .chain(system_groups_cmd)
        .chain(system_display_id_cmd)
}