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.
45 lines
1.1 KiB
Rust
45 lines
1.1 KiB
Rust
use std::str::FromStr;
|
|
|
|
use serenity::client::Context;
|
|
use serenity::model::prelude::GuildId;
|
|
|
|
use crate::utils::context_data::DatabaseContainer;
|
|
use crate::utils::error::{BotError, BotResult};
|
|
|
|
pub static ALL_SETTINGS: &[Setting] = &[
|
|
Setting::MusicAutoShuffle,
|
|
Setting::BotAutoDelete,
|
|
Setting::MusicDjRole,
|
|
];
|
|
|
|
#[derive(Clone, Debug)]
|
|
pub enum Setting {
|
|
MusicAutoShuffle,
|
|
MusicDjRole,
|
|
BotAutoDelete,
|
|
}
|
|
|
|
impl ToString for Setting {
|
|
fn to_string(&self) -> String {
|
|
match self {
|
|
Self::MusicAutoShuffle => "music.autoshuffle".to_string(),
|
|
Self::BotAutoDelete => "bot.autodelete".to_string(),
|
|
Self::MusicDjRole => "music.dj-role".to_string(),
|
|
}
|
|
}
|
|
}
|
|
|
|
/// Returns a specific guild setting
|
|
pub async fn get_setting<T: 'static + FromStr>(
|
|
ctx: &Context,
|
|
guild_id: GuildId,
|
|
setting: Setting,
|
|
) -> BotResult<Option<T>> {
|
|
let data = ctx.data.read().await;
|
|
let database = data.get::<DatabaseContainer>().unwrap();
|
|
database
|
|
.get_guild_setting::<T, _>(guild_id.0, setting.to_string())
|
|
.await
|
|
.map_err(BotError::from)
|
|
}
|