use poise::serenity_prelude::Activity; use crate::{ManifoldContext, ManifoldData, ManifoldResult}; use crate::error::ManifoldError; #[poise::command(slash_command, prefix_command, owners_only)] async fn dump_config(ctx: ManifoldContext<'_>) -> ManifoldResult<()> { ctx.say(t!("commands.admin.config_dump", dump = format!("{:?}", &ctx.data().bot_config))).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(t!("commands.admin.start_watching", target = target)).await?; Ok(()) } #[poise::command(slash_command, prefix_command, owners_only)] async fn save_world(ctx: ManifoldContext<'_>) -> ManifoldResult<()> { let my_message = ctx.send(|f| f.content(t!("commands.admin.saving_the_world")).reply(true)).await?; let mut user_count = 0; let db = &ctx.data().database; let userinfo = ctx.data().user_info.lock().await; for user in userinfo.iter() { debug!("Saving user {:?}", user.0); match user.1.insert(db) { Ok(_) => debug!("User {:?} inserted successfully", user.0), Err(e) => { debug!("User {:?} was not inserted: {:?}, attempting to save instead", user.1, e); match user.1.insert(db) { Ok(_) => debug!("User {:?} saved successfully, user is saved in database.", user.0), Err(e) => debug!("User {:?} could not be saved: {:?}", user.1, e), }; } } user_count = userinfo.len(); } my_message.edit(ctx, |f| f.content(t!("commands.admin.save_complete", user_count = user_count))).await?; Ok(()) } pub fn commands() -> [poise::Command; 4] { [dump_config(), register_commands(), set_activity(), save_world()] }