You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
51 lines
1.6 KiB
Rust
51 lines
1.6 KiB
Rust
use serenity::client::Context;
|
|
use serenity::framework::standard::macros::command;
|
|
use serenity::framework::standard::{Args, CommandResult};
|
|
use serenity::model::channel::Message;
|
|
|
|
use crate::providers::settings::ALL_SETTINGS;
|
|
use crate::utils::context_data::get_database_from_context;
|
|
|
|
#[command]
|
|
#[only_in(guilds)]
|
|
#[description(
|
|
"Set a guild setting. If no value is given the setting will be reset to the default value."
|
|
)]
|
|
#[usage("<setting> [<value>]")]
|
|
#[example("music.autoshuffle true")]
|
|
#[min_args(1)]
|
|
#[max_args(2)]
|
|
#[required_permissions("MANAGE_GUILD")]
|
|
#[bucket("general")]
|
|
async fn set(ctx: &Context, msg: &Message, mut args: Args) -> CommandResult {
|
|
let key = args.single::<String>().unwrap().to_lowercase();
|
|
let all_settings: Vec<String> = ALL_SETTINGS.iter().map(|s| s.to_string()).collect();
|
|
|
|
if !all_settings.contains(&key) {
|
|
msg.channel_id
|
|
.say(ctx, format!("Invalid setting `{}`", key))
|
|
.await?;
|
|
return Ok(());
|
|
}
|
|
let database = get_database_from_context(ctx).await;
|
|
let guild = msg.guild(&ctx.cache).unwrap();
|
|
|
|
if let Ok(value) = args.single::<String>() {
|
|
database
|
|
.set_guild_setting(guild.id.0, key.clone(), value.clone())
|
|
.await?;
|
|
msg.channel_id
|
|
.say(ctx, format!("Set `{}` to `{}`", key, value))
|
|
.await?;
|
|
} else {
|
|
database
|
|
.delete_guild_setting(guild.id.0, key.clone())
|
|
.await?;
|
|
msg.channel_id
|
|
.say(ctx, format!("Setting `{}` reset to default", key))
|
|
.await?;
|
|
}
|
|
|
|
Ok(())
|
|
}
|