can now use reactions on lang command
This commit is contained in:
parent
06d13ec810
commit
3c1fe1f091
@ -3,7 +3,7 @@ use regex_command_attr::command;
|
||||
use serenity::{
|
||||
client::Context,
|
||||
framework::Framework,
|
||||
model::{channel::Message, id::ChannelId, id::RoleId},
|
||||
model::{channel::Message, channel::ReactionType, id::ChannelId, id::RoleId},
|
||||
};
|
||||
|
||||
use chrono_tz::{Tz, TZ_VARIANTS};
|
||||
@ -22,7 +22,7 @@ use crate::{
|
||||
FrameworkCtx, SQLPool,
|
||||
};
|
||||
|
||||
use std::iter;
|
||||
use std::{iter, time::Duration};
|
||||
|
||||
#[command]
|
||||
#[supports_dm(false)]
|
||||
@ -205,7 +205,7 @@ async fn timezone(ctx: &Context, msg: &Message, args: String) {
|
||||
.format("%H:%M")
|
||||
.to_string()
|
||||
),
|
||||
false,
|
||||
true,
|
||||
)
|
||||
});
|
||||
|
||||
@ -243,30 +243,103 @@ async fn language(ctx: &Context, msg: &Message, args: String) {
|
||||
|
||||
let mut user_data = UserData::from_user(&msg.author, &ctx, &pool).await.unwrap();
|
||||
|
||||
if !args.is_empty() {
|
||||
match lm.get_language(&args) {
|
||||
Some(row) => {
|
||||
user_data.language = row.to_string();
|
||||
Some(lang) => {
|
||||
user_data.language = lang.to_string();
|
||||
|
||||
user_data.commit_changes(&pool).await;
|
||||
|
||||
let _ = msg
|
||||
.channel_id
|
||||
.say(&ctx, lm.get(&user_data.language, "lang/set_p"))
|
||||
.send_message(&ctx, |m| {
|
||||
m.embed(|e| {
|
||||
e.title(lm.get(&user_data.language, "lang/set_p_title"))
|
||||
.color(*THEME_COLOR)
|
||||
.description(lm.get(&user_data.language, "lang/set_p"))
|
||||
})
|
||||
})
|
||||
.await;
|
||||
}
|
||||
|
||||
None => {
|
||||
let language_codes = lm
|
||||
let language_codes = lm.all_languages().map(|(k, v)| {
|
||||
(
|
||||
format!("{} {}", lm.get(k, "flag"), v.to_title_case()),
|
||||
format!("`$lang {}`", k.to_uppercase()),
|
||||
true,
|
||||
)
|
||||
});
|
||||
|
||||
let _ = msg
|
||||
.channel_id
|
||||
.send_message(&ctx, |m| {
|
||||
m.embed(|e| {
|
||||
e.title(lm.get(&user_data.language, "lang/invalid_title"))
|
||||
.color(*THEME_COLOR)
|
||||
.description(lm.get(&user_data.language, "lang/invalid"))
|
||||
.fields(language_codes)
|
||||
})
|
||||
})
|
||||
.await;
|
||||
}
|
||||
}
|
||||
} else {
|
||||
let language_codes = lm.all_languages().map(|(k, v)| {
|
||||
(
|
||||
format!("{} {}", lm.get(k, "flag"), v.to_title_case()),
|
||||
format!("`$lang {}`", k.to_uppercase()),
|
||||
true,
|
||||
)
|
||||
});
|
||||
|
||||
let flags = lm
|
||||
.all_languages()
|
||||
.map(|(k, v)| format!("{} ({})", v.to_title_case(), k.to_uppercase()))
|
||||
.collect::<Vec<String>>()
|
||||
.join("\n");
|
||||
.map(|(k, _)| ReactionType::Unicode(lm.get(k, "flag").to_string()));
|
||||
|
||||
let content =
|
||||
lm.get(&user_data.language, "lang/invalid")
|
||||
.replacen("{}", &language_codes, 1);
|
||||
let reactor = msg
|
||||
.channel_id
|
||||
.send_message(&ctx, |m| {
|
||||
m.embed(|e| {
|
||||
e.title(lm.get(&user_data.language, "lang/select_title"))
|
||||
.color(*THEME_COLOR)
|
||||
.description(lm.get(&user_data.language, "lang/select"))
|
||||
.fields(language_codes)
|
||||
})
|
||||
.reactions(flags)
|
||||
})
|
||||
.await;
|
||||
|
||||
let _ = msg.channel_id.say(&ctx, content).await;
|
||||
if let Ok(sent_msg) = reactor {
|
||||
let reaction_reply = sent_msg
|
||||
.await_reaction(&ctx)
|
||||
.timeout(Duration::from_secs(45))
|
||||
.await;
|
||||
|
||||
if let Some(reaction_action) = reaction_reply {
|
||||
if reaction_action.is_added() {
|
||||
if let ReactionType::Unicode(emoji) = &reaction_action.as_inner_ref().emoji {
|
||||
if let Some(lang) = lm.get_language_by_flag(emoji) {
|
||||
user_data.language = lang.to_string();
|
||||
|
||||
user_data.commit_changes(&pool).await;
|
||||
|
||||
let _ = msg
|
||||
.channel_id
|
||||
.send_message(&ctx, |m| {
|
||||
m.embed(|e| {
|
||||
e.title(lm.get(&user_data.language, "lang/set_p_title"))
|
||||
.color(*THEME_COLOR)
|
||||
.description(lm.get(&user_data.language, "lang/set_p"))
|
||||
})
|
||||
})
|
||||
.await;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
let _ = sent_msg.delete_reactions(&ctx).await;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
@ -47,6 +47,14 @@ impl LanguageManager {
|
||||
.next()
|
||||
}
|
||||
|
||||
pub fn get_language_by_flag(&self, flag: &str) -> Option<&str> {
|
||||
self.languages
|
||||
.iter()
|
||||
.filter(|(k, _)| self.get(k, "flag") == flag)
|
||||
.map(|(k, _)| k.as_str())
|
||||
.next()
|
||||
}
|
||||
|
||||
pub fn all_languages(&self) -> impl Iterator<Item = (&str, &str)> {
|
||||
self.languages.iter().map(|(k, v)| (k.as_str(), v.as_str()))
|
||||
}
|
||||
|
@ -217,7 +217,9 @@ async fn main() -> Result<(), Box<dyn std::error::Error + Send + Sync>> {
|
||||
.intents(
|
||||
GatewayIntents::GUILD_MESSAGES
|
||||
| GatewayIntents::GUILDS
|
||||
| GatewayIntents::DIRECT_MESSAGES,
|
||||
| GatewayIntents::GUILD_MESSAGE_REACTIONS
|
||||
| GatewayIntents::DIRECT_MESSAGES
|
||||
| GatewayIntents::DIRECT_MESSAGE_REACTIONS,
|
||||
)
|
||||
.event_handler(Handler)
|
||||
.framework_arc(framework_arc.clone())
|
||||
|
Loading…
Reference in New Issue
Block a user