manifold/src/commands/core.rs

use poise::serenity_prelude::*;

use crate::{ManifoldContext, ManifoldData};
use crate::built_info;
use crate::error::{ManifoldError, ManifoldResult};

#[poise::command(prefix_command, track_edits, slash_command)]
async fn help(
    ctx: ManifoldContext<'_>,
    #[description = "Help about a specific command"] command: Option<String>
) -> ManifoldResult<()> {

    let responses = &ctx.data().responses;

    let config = poise::builtins::HelpConfiguration {
        extra_text_at_bottom: &*format!("{}", responses.get_response(&"help footer".to_string()).unwrap_or(&"".to_string())),
        ..Default::default()
    };

    poise::builtins::help(ctx, command.as_deref(), config).await?;

    Ok(())
}

#[poise::command(slash_command, prefix_command)]
async fn ping(ctx: ManifoldContext<'_>,) -> ManifoldResult<()> {
    let requestor = {
        let calling_user = ctx.author();
        calling_user.mention().to_string()
    };

    ctx.say(format!("{}: Ping? Pong!", requestor)).await?;

    Ok(())
}

#[poise::command(slash_command, prefix_command, owners_only)]
async fn register_commands(ctx: ManifoldContext<'_>) -> ManifoldResult<()> {
    poise::builtins::register_application_commands_buttons(ctx).await?;

    Ok(())
}

#[poise::command(slash_command, prefix_command, track_edits, aliases("sa"), required_permissions = "MODERATE_MEMBERS")]
async fn set_activity(ctx: ManifoldContext<'_>, #[rest] #[description="Who to watch"] target: String) -> ManifoldResult<()> {
    ctx.serenity_context().set_activity(Activity::watching(&target)).await;

    ctx.say(format!("Okay, I'll start watching {}", target)).await?;
    Ok(())
}

#[poise::command(slash_command, prefix_command)]
async fn version(ctx: ManifoldContext<'_>) -> ManifoldResult<()> {
    let version_string = &ctx.data().version_string;

    ctx.send(|f| f
        .reply(true)
        .content(version_string)).await?;

    Ok(())
}

#[poise::command(slash_command, prefix_command, owners_only)]
async fn get_config(ctx: ManifoldContext<'_>, #[description="Config key"] key: String) -> ManifoldResult<()> {
    let config = &ctx.data().bot_config;

    let value = match config.get_value(&key) {
        Ok(v) => v.clone(),
        Err(_) => "not found, sorry!".to_string()
    };

    ctx.say(format!("Value for key {} was {}", &key, &value)).await?;

    Ok(())
}

#[poise::command(slash_command, prefix_command, owners_only)]
async fn dump_config(ctx: ManifoldContext<'_>) -> ManifoldResult<()> {
    let config = &ctx.data().bot_config;

    ctx.say(format!("Config dump; {:?}", config.config)).await?;

    Ok(())
}

#[poise::command(slash_command, prefix_command, owners_only)]
async fn get_environment(ctx: ManifoldContext<'_>) -> ManifoldResult<()> {
    let environment = ctx.data().bot_config.get_environment();

    ctx.say(format!("Currently running under the {} environment", environment)).await?;

    Ok(())
}

pub fn commands() -> [poise::Command<ManifoldData, ManifoldError>; 8] {
    [help(), ping(), register_commands(), set_activity(), version(), get_config(), dump_config(), get_environment()]
}