Compare commits
3 Commits
Author | SHA1 | Date | |
---|---|---|---|
bf34721e55 | |||
2c91a72640 | |||
4a64238ee4 |
2
Cargo.lock
generated
2
Cargo.lock
generated
@ -1270,7 +1270,7 @@ dependencies = [
|
|||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "reminder_rs"
|
name = "reminder_rs"
|
||||||
version = "1.4.13"
|
version = "1.5.0"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"Inflector",
|
"Inflector",
|
||||||
"chrono",
|
"chrono",
|
||||||
|
@ -1,6 +1,6 @@
|
|||||||
[package]
|
[package]
|
||||||
name = "reminder_rs"
|
name = "reminder_rs"
|
||||||
version = "1.4.13"
|
version = "1.5.0"
|
||||||
authors = ["jellywx <judesouthworth@pm.me>"]
|
authors = ["jellywx <judesouthworth@pm.me>"]
|
||||||
edition = "2018"
|
edition = "2018"
|
||||||
|
|
||||||
@ -23,7 +23,7 @@ rand = "0.7"
|
|||||||
Inflector = "0.11"
|
Inflector = "0.11"
|
||||||
levenshtein = "1.0"
|
levenshtein = "1.0"
|
||||||
# serenity = { version = "0.10", features = ["collector"] }
|
# serenity = { version = "0.10", features = ["collector"] }
|
||||||
serenity = { git = "https://github.com/serenity-rs/serenity", branch = "next", features = ["collector"] }
|
serenity = { git = "https://github.com/serenity-rs/serenity", branch = "next", features = ["collector", "unstable_discord_api"] }
|
||||||
sqlx = { version = "0.5", features = ["runtime-tokio-rustls", "macros", "mysql", "bigdecimal", "chrono"]}
|
sqlx = { version = "0.5", features = ["runtime-tokio-rustls", "macros", "mysql", "bigdecimal", "chrono"]}
|
||||||
|
|
||||||
[dependencies.regex_command_attr]
|
[dependencies.regex_command_attr]
|
||||||
|
160
migration/reminder_message_embed.sql
Normal file
160
migration/reminder_message_embed.sql
Normal file
File diff suppressed because one or more lines are too long
@ -1,16 +1,22 @@
|
|||||||
use regex_command_attr::command;
|
use regex_command_attr::command;
|
||||||
|
|
||||||
use serenity::{client::Context, model::channel::Message};
|
use serenity::{
|
||||||
|
builder::CreateEmbedFooter,
|
||||||
|
client::Context,
|
||||||
|
model::{
|
||||||
|
channel::Message,
|
||||||
|
interactions::{Interaction, InteractionResponseType},
|
||||||
|
},
|
||||||
|
};
|
||||||
|
|
||||||
use chrono::offset::Utc;
|
use chrono::offset::Utc;
|
||||||
|
|
||||||
use crate::{
|
use crate::{
|
||||||
command_help, consts::DEFAULT_PREFIX, get_ctx_data, language_manager::LanguageManager,
|
command_help, consts::DEFAULT_PREFIX, get_ctx_data, language_manager::LanguageManager,
|
||||||
models::UserData, FrameworkCtx, THEME_COLOR,
|
models::CtxGuildData, models::UserData, FrameworkCtx, THEME_COLOR,
|
||||||
};
|
};
|
||||||
|
|
||||||
use crate::models::CtxGuildData;
|
use inflector::Inflector;
|
||||||
use serenity::builder::CreateEmbedFooter;
|
|
||||||
use std::sync::Arc;
|
use std::sync::Arc;
|
||||||
use std::time::{SystemTime, UNIX_EPOCH};
|
use std::time::{SystemTime, UNIX_EPOCH};
|
||||||
|
|
||||||
@ -130,6 +136,137 @@ async fn help(ctx: &Context, msg: &Message, args: String) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pub async fn help_interaction(ctx: &Context, interaction: Interaction) {
|
||||||
|
async fn default_help(
|
||||||
|
ctx: &Context,
|
||||||
|
interaction: Interaction,
|
||||||
|
lm: Arc<LanguageManager>,
|
||||||
|
language: &str,
|
||||||
|
) {
|
||||||
|
let desc = lm.get(language, "help/desc").replace("{prefix}", "/");
|
||||||
|
let footer = footer(ctx).await;
|
||||||
|
|
||||||
|
interaction
|
||||||
|
.create_interaction_response(ctx, |response| {
|
||||||
|
response
|
||||||
|
.kind(InteractionResponseType::ChannelMessageWithSource)
|
||||||
|
.interaction_response_data(|data| {
|
||||||
|
data.embed(move |e| {
|
||||||
|
e.title("Help Menu")
|
||||||
|
.description(desc)
|
||||||
|
.field(
|
||||||
|
lm.get(language, "help/setup_title"),
|
||||||
|
"`lang` `timezone` `meridian`",
|
||||||
|
true,
|
||||||
|
)
|
||||||
|
.field(
|
||||||
|
lm.get(language, "help/mod_title"),
|
||||||
|
"`prefix` `blacklist` `restrict` `alias`",
|
||||||
|
true,
|
||||||
|
)
|
||||||
|
.field(
|
||||||
|
lm.get(language, "help/reminder_title"),
|
||||||
|
"`remind` `interval` `natural` `look` `countdown`",
|
||||||
|
true,
|
||||||
|
)
|
||||||
|
.field(
|
||||||
|
lm.get(language, "help/reminder_mod_title"),
|
||||||
|
"`del` `offset` `pause` `nudge`",
|
||||||
|
true,
|
||||||
|
)
|
||||||
|
.field(
|
||||||
|
lm.get(language, "help/info_title"),
|
||||||
|
"`help` `info` `donate` `clock`",
|
||||||
|
true,
|
||||||
|
)
|
||||||
|
.field(
|
||||||
|
lm.get(language, "help/todo_title"),
|
||||||
|
"`todo` `todos` `todoc`",
|
||||||
|
true,
|
||||||
|
)
|
||||||
|
.field(lm.get(language, "help/other_title"), "`timer`", true)
|
||||||
|
.footer(footer)
|
||||||
|
.color(*THEME_COLOR)
|
||||||
|
})
|
||||||
|
})
|
||||||
|
})
|
||||||
|
.await
|
||||||
|
.unwrap();
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn command_help(
|
||||||
|
ctx: &Context,
|
||||||
|
interaction: Interaction,
|
||||||
|
lm: Arc<LanguageManager>,
|
||||||
|
language: &str,
|
||||||
|
command_name: &str,
|
||||||
|
) {
|
||||||
|
interaction
|
||||||
|
.create_interaction_response(ctx, |r| {
|
||||||
|
r.kind(InteractionResponseType::ChannelMessageWithSource)
|
||||||
|
.interaction_response_data(|data| {
|
||||||
|
data.embed(move |e| {
|
||||||
|
e.title(format!("{} Help", command_name.to_title_case()))
|
||||||
|
.description(
|
||||||
|
lm.get(&language, &format!("help/{}", command_name))
|
||||||
|
.replace("{prefix}", "/"),
|
||||||
|
)
|
||||||
|
.footer(|f| {
|
||||||
|
f.text(concat!(
|
||||||
|
env!("CARGO_PKG_NAME"),
|
||||||
|
" ver ",
|
||||||
|
env!("CARGO_PKG_VERSION")
|
||||||
|
))
|
||||||
|
})
|
||||||
|
.color(*THEME_COLOR)
|
||||||
|
})
|
||||||
|
})
|
||||||
|
})
|
||||||
|
.await
|
||||||
|
.unwrap();
|
||||||
|
}
|
||||||
|
|
||||||
|
let (pool, lm) = get_ctx_data(&ctx).await;
|
||||||
|
|
||||||
|
let language = UserData::language_of(interaction.member.user.id, &pool);
|
||||||
|
|
||||||
|
if let Some(data) = &interaction.data {
|
||||||
|
if let Some(command_name) = data
|
||||||
|
.options
|
||||||
|
.first()
|
||||||
|
.map(|opt| {
|
||||||
|
opt.value
|
||||||
|
.clone()
|
||||||
|
.map(|inner| inner.as_str().unwrap().to_string())
|
||||||
|
})
|
||||||
|
.flatten()
|
||||||
|
{
|
||||||
|
let framework = ctx
|
||||||
|
.data
|
||||||
|
.read()
|
||||||
|
.await
|
||||||
|
.get::<FrameworkCtx>()
|
||||||
|
.cloned()
|
||||||
|
.expect("Could not get FrameworkCtx from data");
|
||||||
|
|
||||||
|
let matched = framework
|
||||||
|
.commands
|
||||||
|
.get(&command_name)
|
||||||
|
.map(|inner| inner.name);
|
||||||
|
|
||||||
|
if let Some(command_name) = matched {
|
||||||
|
command_help(ctx, interaction, lm, &language.await, command_name).await
|
||||||
|
} else {
|
||||||
|
default_help(ctx, interaction, lm, &language.await).await;
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
default_help(ctx, interaction, lm, &language.await).await;
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
default_help(ctx, interaction, lm, &language.await).await;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
#[command]
|
#[command]
|
||||||
async fn info(ctx: &Context, msg: &Message, _args: String) {
|
async fn info(ctx: &Context, msg: &Message, _args: String) {
|
||||||
let (pool, lm) = get_ctx_data(&ctx).await;
|
let (pool, lm) = get_ctx_data(&ctx).await;
|
||||||
@ -158,6 +295,36 @@ async fn info(ctx: &Context, msg: &Message, _args: String) {
|
|||||||
.await;
|
.await;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pub async fn info_interaction(ctx: &Context, interaction: Interaction) {
|
||||||
|
let (pool, lm) = get_ctx_data(&ctx).await;
|
||||||
|
|
||||||
|
let language = UserData::language_of(&interaction.member, &pool);
|
||||||
|
let current_user = ctx.cache.current_user();
|
||||||
|
let footer = footer(ctx).await;
|
||||||
|
|
||||||
|
let desc = lm
|
||||||
|
.get(&language.await, "info")
|
||||||
|
.replacen("{user}", ¤t_user.await.name, 1)
|
||||||
|
.replace("{default_prefix}", &*DEFAULT_PREFIX)
|
||||||
|
.replace("{prefix}", "/");
|
||||||
|
|
||||||
|
interaction
|
||||||
|
.create_interaction_response(ctx, |response| {
|
||||||
|
response
|
||||||
|
.kind(InteractionResponseType::ChannelMessageWithSource)
|
||||||
|
.interaction_response_data(|data| {
|
||||||
|
data.embed(move |e| {
|
||||||
|
e.title("Info")
|
||||||
|
.description(desc)
|
||||||
|
.footer(footer)
|
||||||
|
.color(*THEME_COLOR)
|
||||||
|
})
|
||||||
|
})
|
||||||
|
})
|
||||||
|
.await
|
||||||
|
.unwrap();
|
||||||
|
}
|
||||||
|
|
||||||
#[command]
|
#[command]
|
||||||
async fn donate(ctx: &Context, msg: &Message, _args: String) {
|
async fn donate(ctx: &Context, msg: &Message, _args: String) {
|
||||||
let (pool, lm) = get_ctx_data(&ctx).await;
|
let (pool, lm) = get_ctx_data(&ctx).await;
|
||||||
@ -179,6 +346,30 @@ async fn donate(ctx: &Context, msg: &Message, _args: String) {
|
|||||||
.await;
|
.await;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pub async fn donate_interaction(ctx: &Context, interaction: Interaction) {
|
||||||
|
let (pool, lm) = get_ctx_data(&ctx).await;
|
||||||
|
|
||||||
|
let language = UserData::language_of(&interaction.member, &pool).await;
|
||||||
|
let desc = lm.get(&language, "donate");
|
||||||
|
let footer = footer(ctx).await;
|
||||||
|
|
||||||
|
interaction
|
||||||
|
.create_interaction_response(ctx, |response| {
|
||||||
|
response
|
||||||
|
.kind(InteractionResponseType::ChannelMessageWithSource)
|
||||||
|
.interaction_response_data(|data| {
|
||||||
|
data.embed(move |e| {
|
||||||
|
e.title("Donate")
|
||||||
|
.description(desc)
|
||||||
|
.footer(footer)
|
||||||
|
.color(*THEME_COLOR)
|
||||||
|
})
|
||||||
|
})
|
||||||
|
})
|
||||||
|
.await
|
||||||
|
.unwrap();
|
||||||
|
}
|
||||||
|
|
||||||
#[command]
|
#[command]
|
||||||
async fn dashboard(ctx: &Context, msg: &Message, _args: String) {
|
async fn dashboard(ctx: &Context, msg: &Message, _args: String) {
|
||||||
let footer = footer(ctx).await;
|
let footer = footer(ctx).await;
|
||||||
@ -216,3 +407,30 @@ async fn clock(ctx: &Context, msg: &Message, _args: String) {
|
|||||||
)
|
)
|
||||||
.await;
|
.await;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pub async fn clock_interaction(ctx: &Context, interaction: Interaction) {
|
||||||
|
let (pool, lm) = get_ctx_data(&ctx).await;
|
||||||
|
|
||||||
|
let language = UserData::language_of(&interaction.member, &pool).await;
|
||||||
|
let timezone = UserData::timezone_of(&interaction.member, &pool).await;
|
||||||
|
let meridian = UserData::meridian_of(&interaction.member, &pool).await;
|
||||||
|
|
||||||
|
let now = Utc::now().with_timezone(&timezone);
|
||||||
|
|
||||||
|
let clock_display = lm.get(&language, "clock/time");
|
||||||
|
|
||||||
|
interaction
|
||||||
|
.create_interaction_response(ctx, |response| {
|
||||||
|
response
|
||||||
|
.kind(InteractionResponseType::ChannelMessageWithSource)
|
||||||
|
.interaction_response_data(|data| {
|
||||||
|
data.content(clock_display.replacen(
|
||||||
|
"{}",
|
||||||
|
&now.format(meridian.fmt_str()).to_string(),
|
||||||
|
1,
|
||||||
|
))
|
||||||
|
})
|
||||||
|
})
|
||||||
|
.await
|
||||||
|
.unwrap();
|
||||||
|
}
|
||||||
|
@ -6,8 +6,8 @@ use serenity::{
|
|||||||
model::{
|
model::{
|
||||||
channel::ReactionType,
|
channel::ReactionType,
|
||||||
channel::{Channel, Message},
|
channel::{Channel, Message},
|
||||||
id::ChannelId,
|
id::{ChannelId, RoleId},
|
||||||
id::RoleId,
|
interactions::{Interaction, InteractionResponseType},
|
||||||
},
|
},
|
||||||
};
|
};
|
||||||
|
|
||||||
@ -28,9 +28,6 @@ use crate::{
|
|||||||
FrameworkCtx, PopularTimezones,
|
FrameworkCtx, PopularTimezones,
|
||||||
};
|
};
|
||||||
|
|
||||||
#[cfg(feature = "prefix-cache")]
|
|
||||||
use crate::PrefixCache;
|
|
||||||
|
|
||||||
use crate::models::CtxGuildData;
|
use crate::models::CtxGuildData;
|
||||||
use std::{collections::HashMap, iter, time::Duration};
|
use std::{collections::HashMap, iter, time::Duration};
|
||||||
|
|
||||||
@ -218,6 +215,116 @@ async fn timezone(ctx: &Context, msg: &Message, args: String) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pub async fn timezone_interaction(ctx: &Context, interaction: Interaction) {
|
||||||
|
let (pool, lm) = get_ctx_data(&&ctx).await;
|
||||||
|
|
||||||
|
let mut user_data = UserData::from_user(&interaction.member.user, &ctx, &pool)
|
||||||
|
.await
|
||||||
|
.unwrap();
|
||||||
|
|
||||||
|
let footer_text = lm.get(&user_data.language, "timezone/footer").replacen(
|
||||||
|
"{timezone}",
|
||||||
|
&user_data.timezone,
|
||||||
|
1,
|
||||||
|
);
|
||||||
|
|
||||||
|
if let Some(data) = &interaction.data {
|
||||||
|
if let Some(timezone) = data
|
||||||
|
.options
|
||||||
|
.first()
|
||||||
|
.map(|inner| {
|
||||||
|
inner
|
||||||
|
.value
|
||||||
|
.clone()
|
||||||
|
.map(|v| v.as_str().map(|s| s.to_string()))
|
||||||
|
.flatten()
|
||||||
|
})
|
||||||
|
.flatten()
|
||||||
|
.map(|tz| tz.parse::<Tz>().ok())
|
||||||
|
.flatten()
|
||||||
|
{
|
||||||
|
user_data.timezone = timezone.to_string();
|
||||||
|
user_data.commit_changes(&pool).await;
|
||||||
|
|
||||||
|
let now = Utc::now().with_timezone(&user_data.timezone());
|
||||||
|
|
||||||
|
let content = lm
|
||||||
|
.get(&user_data.language, "timezone/set_p")
|
||||||
|
.replacen("{timezone}", &user_data.timezone, 1)
|
||||||
|
.replacen(
|
||||||
|
"{time}",
|
||||||
|
&now.format(user_data.meridian().fmt_str_short()).to_string(),
|
||||||
|
1,
|
||||||
|
);
|
||||||
|
|
||||||
|
interaction
|
||||||
|
.create_interaction_response(&ctx, |response| {
|
||||||
|
response
|
||||||
|
.kind(InteractionResponseType::ChannelMessageWithSource)
|
||||||
|
.interaction_response_data(|data| {
|
||||||
|
data.embed(|e| {
|
||||||
|
e.title(lm.get(&user_data.language, "timezone/set_p_title"))
|
||||||
|
.description(content)
|
||||||
|
.color(*THEME_COLOR)
|
||||||
|
.footer(|f| {
|
||||||
|
f.text(
|
||||||
|
lm.get(&user_data.language, "timezone/footer")
|
||||||
|
.replacen("{timezone}", &user_data.timezone, 1),
|
||||||
|
)
|
||||||
|
})
|
||||||
|
})
|
||||||
|
})
|
||||||
|
})
|
||||||
|
.await
|
||||||
|
.unwrap();
|
||||||
|
} else {
|
||||||
|
let content = lm
|
||||||
|
.get(&user_data.language, "timezone/no_argument")
|
||||||
|
.replace("{prefix}", "/");
|
||||||
|
|
||||||
|
let popular_timezones = ctx
|
||||||
|
.data
|
||||||
|
.read()
|
||||||
|
.await
|
||||||
|
.get::<PopularTimezones>()
|
||||||
|
.cloned()
|
||||||
|
.unwrap();
|
||||||
|
|
||||||
|
let popular_timezones_iter = popular_timezones.iter().map(|t| {
|
||||||
|
(
|
||||||
|
t.to_string(),
|
||||||
|
format!(
|
||||||
|
"🕗 `{}`",
|
||||||
|
Utc::now()
|
||||||
|
.with_timezone(t)
|
||||||
|
.format(user_data.meridian().fmt_str_short())
|
||||||
|
.to_string()
|
||||||
|
),
|
||||||
|
true,
|
||||||
|
)
|
||||||
|
});
|
||||||
|
|
||||||
|
interaction
|
||||||
|
.create_interaction_response(&ctx, |response| {
|
||||||
|
response
|
||||||
|
.kind(InteractionResponseType::ChannelMessageWithSource)
|
||||||
|
.interaction_response_data(|data| {
|
||||||
|
data.embed(|e| {
|
||||||
|
e.title(lm.get(&user_data.language, "timezone/no_argument_title"))
|
||||||
|
.description(content)
|
||||||
|
.color(*THEME_COLOR)
|
||||||
|
.fields(popular_timezones_iter)
|
||||||
|
.footer(|f| f.text(footer_text))
|
||||||
|
.url("https://gist.github.com/JellyWX/913dfc8b63d45192ad6cb54c829324ee")
|
||||||
|
})
|
||||||
|
})
|
||||||
|
})
|
||||||
|
.await
|
||||||
|
.unwrap();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
#[command("meridian")]
|
#[command("meridian")]
|
||||||
async fn change_meridian(ctx: &Context, msg: &Message, args: String) {
|
async fn change_meridian(ctx: &Context, msg: &Message, args: String) {
|
||||||
let (pool, lm) = get_ctx_data(&ctx).await;
|
let (pool, lm) = get_ctx_data(&ctx).await;
|
||||||
@ -393,15 +500,44 @@ async fn language(ctx: &Context, msg: &Message, args: String) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pub async fn language_interaction(ctx: &Context, interaction: Interaction) {
|
||||||
|
let (pool, lm) = get_ctx_data(&ctx).await;
|
||||||
|
|
||||||
|
let mut user_data = UserData::from_user(&interaction.member.user, &ctx, &pool)
|
||||||
|
.await
|
||||||
|
.unwrap();
|
||||||
|
|
||||||
|
if let Some(data) = &interaction.data {
|
||||||
|
let option = &data.options[0];
|
||||||
|
|
||||||
|
user_data.language = option.value.clone().unwrap().as_str().unwrap().to_string();
|
||||||
|
user_data.commit_changes(&pool).await;
|
||||||
|
|
||||||
|
interaction
|
||||||
|
.create_interaction_response(ctx, |response| {
|
||||||
|
response
|
||||||
|
.kind(InteractionResponseType::ChannelMessageWithSource)
|
||||||
|
.interaction_response_data(|data| {
|
||||||
|
data.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
|
||||||
|
.unwrap();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
#[command]
|
#[command]
|
||||||
#[supports_dm(false)]
|
#[supports_dm(false)]
|
||||||
#[permission_level(Restricted)]
|
#[permission_level(Restricted)]
|
||||||
async fn prefix(ctx: &Context, msg: &Message, args: String) {
|
async fn prefix(ctx: &Context, msg: &Message, args: String) {
|
||||||
let (pool, lm) = get_ctx_data(&ctx).await;
|
let (pool, lm) = get_ctx_data(&ctx).await;
|
||||||
|
|
||||||
let mut guild_data = GuildData::from_guild(msg.guild(&ctx).await.unwrap(), &pool)
|
let guild_data = ctx.guild_data(msg.guild_id.unwrap()).await.unwrap();
|
||||||
.await
|
|
||||||
.unwrap();
|
|
||||||
let language = UserData::language_of(&msg.author, &pool).await;
|
let language = UserData::language_of(&msg.author, &pool).await;
|
||||||
|
|
||||||
if args.len() > 5 {
|
if args.len() > 5 {
|
||||||
@ -415,23 +551,62 @@ async fn prefix(ctx: &Context, msg: &Message, args: String) {
|
|||||||
.say(&ctx, lm.get(&language, "prefix/no_argument"))
|
.say(&ctx, lm.get(&language, "prefix/no_argument"))
|
||||||
.await;
|
.await;
|
||||||
} else {
|
} else {
|
||||||
guild_data.prefix = args;
|
guild_data.write().await.prefix = args;
|
||||||
|
guild_data.read().await.commit_changes(&pool).await;
|
||||||
|
|
||||||
#[cfg(feature = "prefix-cache")]
|
let content = lm.get(&language, "prefix/success").replacen(
|
||||||
let prefix_cache = ctx.data.read().await.get::<PrefixCache>().cloned().unwrap();
|
"{prefix}",
|
||||||
#[cfg(feature = "prefix-cache")]
|
&guild_data.read().await.prefix,
|
||||||
prefix_cache.insert(msg.guild_id.unwrap(), guild_data.prefix.clone());
|
1,
|
||||||
|
);
|
||||||
guild_data.commit_changes(&pool).await;
|
|
||||||
|
|
||||||
let content =
|
|
||||||
lm.get(&language, "prefix/success")
|
|
||||||
.replacen("{prefix}", &guild_data.prefix, 1);
|
|
||||||
|
|
||||||
let _ = msg.channel_id.say(&ctx, content).await;
|
let _ = msg.channel_id.say(&ctx, content).await;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pub async fn prefix_interaction(ctx: &Context, interaction: Interaction) {
|
||||||
|
let (pool, lm) = get_ctx_data(&ctx).await;
|
||||||
|
|
||||||
|
let guild_data = ctx.guild_data(interaction.guild_id).await.unwrap();
|
||||||
|
|
||||||
|
let language = UserData::language_of(&interaction.member, &pool).await;
|
||||||
|
|
||||||
|
if let Some(data) = &interaction.data {
|
||||||
|
let option = &data.options[0];
|
||||||
|
|
||||||
|
let new_prefix = option.value.clone().unwrap().as_str().unwrap().to_string();
|
||||||
|
|
||||||
|
if new_prefix.len() > 5 {
|
||||||
|
interaction
|
||||||
|
.create_interaction_response(ctx, |response| {
|
||||||
|
response
|
||||||
|
.kind(InteractionResponseType::ChannelMessageWithSource)
|
||||||
|
.interaction_response_data(|data| {
|
||||||
|
data.content(lm.get(&language, "prefix/too_long"))
|
||||||
|
})
|
||||||
|
})
|
||||||
|
.await
|
||||||
|
.unwrap();
|
||||||
|
} else {
|
||||||
|
guild_data.write().await.prefix = new_prefix.clone();
|
||||||
|
guild_data.read().await.commit_changes(&pool).await;
|
||||||
|
|
||||||
|
let content = lm
|
||||||
|
.get(&language, "prefix/success")
|
||||||
|
.replacen("{prefix}", &new_prefix, 1);
|
||||||
|
|
||||||
|
interaction
|
||||||
|
.create_interaction_response(ctx, |response| {
|
||||||
|
response
|
||||||
|
.kind(InteractionResponseType::ChannelMessageWithSource)
|
||||||
|
.interaction_response_data(|data| data.content(content))
|
||||||
|
})
|
||||||
|
.await
|
||||||
|
.unwrap();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
#[command]
|
#[command]
|
||||||
#[supports_dm(false)]
|
#[supports_dm(false)]
|
||||||
#[permission_level(Restricted)]
|
#[permission_level(Restricted)]
|
||||||
|
@ -50,6 +50,7 @@ use std::{
|
|||||||
use crate::models::{CtxGuildData, MeridianType};
|
use crate::models::{CtxGuildData, MeridianType};
|
||||||
use regex::Captures;
|
use regex::Captures;
|
||||||
use serenity::model::channel::Channel;
|
use serenity::model::channel::Channel;
|
||||||
|
use serenity::model::interactions::Interaction;
|
||||||
|
|
||||||
fn shorthand_displacement(seconds: u64) -> String {
|
fn shorthand_displacement(seconds: u64) -> String {
|
||||||
let (days, seconds) = seconds.div_rem(&DAY);
|
let (days, seconds) = seconds.div_rem(&DAY);
|
||||||
@ -193,7 +194,7 @@ UPDATE reminders
|
|||||||
INNER JOIN `channels`
|
INNER JOIN `channels`
|
||||||
ON `channels`.id = reminders.channel_id
|
ON `channels`.id = reminders.channel_id
|
||||||
SET
|
SET
|
||||||
reminders.`time` = reminders.`time` + ?
|
reminders.`utc_time` = reminders.`utc_time` + ?
|
||||||
WHERE channels.guild_id = ?
|
WHERE channels.guild_id = ?
|
||||||
",
|
",
|
||||||
displacement,
|
displacement,
|
||||||
@ -205,7 +206,7 @@ UPDATE reminders
|
|||||||
} else {
|
} else {
|
||||||
sqlx::query!(
|
sqlx::query!(
|
||||||
"
|
"
|
||||||
UPDATE reminders SET `time` = `time` + ? WHERE reminders.channel_id = ?
|
UPDATE reminders SET `utc_time` = `utc_time` + ? WHERE reminders.channel_id = ?
|
||||||
",
|
",
|
||||||
displacement,
|
displacement,
|
||||||
user_data.dm_channel
|
user_data.dm_channel
|
||||||
@ -345,11 +346,11 @@ impl LookFlags {
|
|||||||
|
|
||||||
struct LookReminder {
|
struct LookReminder {
|
||||||
id: u32,
|
id: u32,
|
||||||
time: u32,
|
time: NaiveDateTime,
|
||||||
interval: Option<u32>,
|
interval: Option<u32>,
|
||||||
channel: u64,
|
channel: u64,
|
||||||
content: String,
|
content: String,
|
||||||
description: Option<String>,
|
description: String,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl LookReminder {
|
impl LookReminder {
|
||||||
@ -357,7 +358,7 @@ impl LookReminder {
|
|||||||
if self.content.len() > 0 {
|
if self.content.len() > 0 {
|
||||||
self.content.clone()
|
self.content.clone()
|
||||||
} else {
|
} else {
|
||||||
self.description.clone().unwrap_or(String::from(""))
|
self.description.clone()
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -370,7 +371,7 @@ impl LookReminder {
|
|||||||
) -> String {
|
) -> String {
|
||||||
let time_display = match flags.time_display {
|
let time_display = match flags.time_display {
|
||||||
TimeDisplayType::Absolute => timezone
|
TimeDisplayType::Absolute => timezone
|
||||||
.timestamp(self.time as i64, 0)
|
.from_utc_datetime(&self.time)
|
||||||
.format(meridian.fmt_str())
|
.format(meridian.fmt_str())
|
||||||
.to_string(),
|
.to_string(),
|
||||||
|
|
||||||
@ -380,7 +381,7 @@ impl LookReminder {
|
|||||||
.unwrap()
|
.unwrap()
|
||||||
.as_secs();
|
.as_secs();
|
||||||
|
|
||||||
longhand_displacement((self.time as u64).checked_sub(now).unwrap_or(1))
|
longhand_displacement((self.time.timestamp() as u64).checked_sub(now).unwrap_or(1))
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
@ -434,26 +435,18 @@ async fn look(ctx: &Context, msg: &Message, args: String) {
|
|||||||
LookReminder,
|
LookReminder,
|
||||||
"
|
"
|
||||||
SELECT
|
SELECT
|
||||||
reminders.id, reminders.time, reminders.interval, channels.channel, messages.content, embeds.description
|
reminders.id, reminders.utc_time AS time, reminders.interval, channels.channel, reminders.content, reminders.embed_description AS description
|
||||||
FROM
|
FROM
|
||||||
reminders
|
reminders
|
||||||
INNER JOIN
|
INNER JOIN
|
||||||
channels
|
channels
|
||||||
ON
|
ON
|
||||||
reminders.channel_id = channels.id
|
reminders.channel_id = channels.id
|
||||||
INNER JOIN
|
|
||||||
messages
|
|
||||||
ON
|
|
||||||
messages.id = reminders.message_id
|
|
||||||
LEFT JOIN
|
|
||||||
embeds
|
|
||||||
ON
|
|
||||||
embeds.id = messages.embed_id
|
|
||||||
WHERE
|
WHERE
|
||||||
channels.channel = ? AND
|
channels.channel = ? AND
|
||||||
FIND_IN_SET(reminders.enabled, ?)
|
FIND_IN_SET(reminders.enabled, ?)
|
||||||
ORDER BY
|
ORDER BY
|
||||||
reminders.time
|
reminders.utc_time
|
||||||
LIMIT
|
LIMIT
|
||||||
?
|
?
|
||||||
",
|
",
|
||||||
@ -509,21 +502,13 @@ async fn delete(ctx: &Context, msg: &Message, _args: String) {
|
|||||||
LookReminder,
|
LookReminder,
|
||||||
"
|
"
|
||||||
SELECT
|
SELECT
|
||||||
reminders.id, reminders.time, reminders.interval, channels.channel, messages.content, embeds.description
|
reminders.id, reminders.utc_time AS time, reminders.interval, channels.channel, reminders.content, reminders.embed_description AS description
|
||||||
FROM
|
FROM
|
||||||
reminders
|
reminders
|
||||||
LEFT OUTER JOIN
|
INNER JOIN
|
||||||
channels
|
channels
|
||||||
ON
|
ON
|
||||||
channels.id = reminders.channel_id
|
channels.id = reminders.channel_id
|
||||||
INNER JOIN
|
|
||||||
messages
|
|
||||||
ON
|
|
||||||
messages.id = reminders.message_id
|
|
||||||
LEFT JOIN
|
|
||||||
embeds
|
|
||||||
ON
|
|
||||||
embeds.id = messages.embed_id
|
|
||||||
WHERE
|
WHERE
|
||||||
FIND_IN_SET(channels.channel, ?)
|
FIND_IN_SET(channels.channel, ?)
|
||||||
",
|
",
|
||||||
@ -536,21 +521,13 @@ WHERE
|
|||||||
LookReminder,
|
LookReminder,
|
||||||
"
|
"
|
||||||
SELECT
|
SELECT
|
||||||
reminders.id, reminders.time, reminders.interval, channels.channel, messages.content, embeds.description
|
reminders.id, reminders.utc_time AS time, reminders.interval, channels.channel, reminders.content, reminders.embed_description AS description
|
||||||
FROM
|
FROM
|
||||||
reminders
|
reminders
|
||||||
LEFT OUTER JOIN
|
INNER JOIN
|
||||||
channels
|
channels
|
||||||
ON
|
ON
|
||||||
channels.id = reminders.channel_id
|
channels.id = reminders.channel_id
|
||||||
INNER JOIN
|
|
||||||
messages
|
|
||||||
ON
|
|
||||||
messages.id = reminders.message_id
|
|
||||||
LEFT JOIN
|
|
||||||
embeds
|
|
||||||
ON
|
|
||||||
embeds.id = messages.embed_id
|
|
||||||
WHERE
|
WHERE
|
||||||
channels.guild_id = (SELECT id FROM guilds WHERE guild = ?)
|
channels.guild_id = (SELECT id FROM guilds WHERE guild = ?)
|
||||||
",
|
",
|
||||||
@ -564,17 +541,9 @@ WHERE
|
|||||||
LookReminder,
|
LookReminder,
|
||||||
"
|
"
|
||||||
SELECT
|
SELECT
|
||||||
reminders.id, reminders.time, reminders.interval, channels.channel, messages.content, embeds.description
|
reminders.id, reminders.utc_time AS time, reminders.interval, channels.channel, reminders.content, reminders.embed_description AS description
|
||||||
FROM
|
FROM
|
||||||
reminders
|
reminders
|
||||||
INNER JOIN
|
|
||||||
messages
|
|
||||||
ON
|
|
||||||
reminders.message_id = messages.id
|
|
||||||
LEFT JOIN
|
|
||||||
embeds
|
|
||||||
ON
|
|
||||||
embeds.id = messages.embed_id
|
|
||||||
INNER JOIN
|
INNER JOIN
|
||||||
channels
|
channels
|
||||||
ON
|
ON
|
||||||
@ -593,7 +562,7 @@ WHERE
|
|||||||
|
|
||||||
let enumerated_reminders = reminders.iter().enumerate().map(|(count, reminder)| {
|
let enumerated_reminders = reminders.iter().enumerate().map(|(count, reminder)| {
|
||||||
reminder_ids.push(reminder.id);
|
reminder_ids.push(reminder.id);
|
||||||
let time = user_data.timezone().timestamp(reminder.time as i64, 0);
|
let time = user_data.timezone().timestamp(reminder.time.timestamp(), 0);
|
||||||
|
|
||||||
format!(
|
format!(
|
||||||
"**{}**: '{}' *<#{}>* at {}",
|
"**{}**: '{}' *<#{}>* at {}",
|
||||||
@ -1047,70 +1016,41 @@ async fn countdown(ctx: &Context, msg: &Message, args: String) {
|
|||||||
event_name, target_ts
|
event_name, target_ts
|
||||||
);
|
);
|
||||||
|
|
||||||
sqlx::query!(
|
|
||||||
"
|
|
||||||
INSERT INTO embeds (title, description, color) VALUES (?, ?, ?)
|
|
||||||
",
|
|
||||||
event_name,
|
|
||||||
description,
|
|
||||||
*THEME_COLOR
|
|
||||||
)
|
|
||||||
.execute(&pool)
|
|
||||||
.await
|
|
||||||
.unwrap();
|
|
||||||
|
|
||||||
let embed_id = sqlx::query!(
|
|
||||||
"
|
|
||||||
SELECT id FROM embeds WHERE title = ? AND description = ?
|
|
||||||
",
|
|
||||||
event_name,
|
|
||||||
description
|
|
||||||
)
|
|
||||||
.fetch_one(&pool)
|
|
||||||
.await
|
|
||||||
.unwrap();
|
|
||||||
|
|
||||||
sqlx::query!(
|
|
||||||
"
|
|
||||||
INSERT INTO messages (embed_id) VALUES (?)
|
|
||||||
",
|
|
||||||
embed_id.id
|
|
||||||
)
|
|
||||||
.execute(&pool)
|
|
||||||
.await
|
|
||||||
.unwrap();
|
|
||||||
|
|
||||||
sqlx::query!(
|
sqlx::query!(
|
||||||
"
|
"
|
||||||
INSERT INTO reminders (
|
INSERT INTO reminders (
|
||||||
`uid`,
|
`uid`,
|
||||||
`name`,
|
`name`,
|
||||||
`message_id`,
|
|
||||||
`channel_id`,
|
`channel_id`,
|
||||||
`time`,
|
`utc_time`,
|
||||||
`interval`,
|
`interval`,
|
||||||
`method`,
|
`expires`,
|
||||||
`set_by`,
|
`embed_title`,
|
||||||
`expires`
|
`embed_description`,
|
||||||
|
`embed_color`,
|
||||||
|
`set_by`
|
||||||
) VALUES (
|
) VALUES (
|
||||||
?,
|
?,
|
||||||
'Countdown',
|
'Countdown',
|
||||||
(SELECT id FROM messages WHERE embed_id = ?),
|
|
||||||
(SELECT id FROM channels WHERE channel = ?),
|
(SELECT id FROM channels WHERE channel = ?),
|
||||||
?,
|
?,
|
||||||
?,
|
?,
|
||||||
'countdown',
|
FROM_UNIXTIME(?),
|
||||||
(SELECT id FROM users WHERE user = ?),
|
?,
|
||||||
FROM_UNIXTIME(?)
|
?,
|
||||||
|
?,
|
||||||
|
(SELECT id FROM users WHERE user = ?)
|
||||||
)
|
)
|
||||||
",
|
",
|
||||||
generate_uid(),
|
generate_uid(),
|
||||||
embed_id.id,
|
|
||||||
msg.channel_id.as_u64(),
|
msg.channel_id.as_u64(),
|
||||||
first_time,
|
first_time,
|
||||||
interval,
|
interval,
|
||||||
|
target_ts,
|
||||||
|
event_name,
|
||||||
|
description,
|
||||||
|
*THEME_COLOR,
|
||||||
msg.author.id.as_u64(),
|
msg.author.id.as_u64(),
|
||||||
target_ts
|
|
||||||
)
|
)
|
||||||
.execute(&pool)
|
.execute(&pool)
|
||||||
.await
|
.await
|
||||||
@ -1250,6 +1190,7 @@ async fn remind_command(ctx: &Context, msg: &Message, args: String, command: Rem
|
|||||||
msg.guild_id,
|
msg.guild_id,
|
||||||
&scope,
|
&scope,
|
||||||
&time_parser,
|
&time_parser,
|
||||||
|
timezone.to_string(),
|
||||||
expires_parser.as_ref().clone(),
|
expires_parser.as_ref().clone(),
|
||||||
interval,
|
interval,
|
||||||
&mut content,
|
&mut content,
|
||||||
@ -1479,6 +1420,7 @@ async fn natural(ctx: &Context, msg: &Message, args: String) {
|
|||||||
msg.guild_id,
|
msg.guild_id,
|
||||||
&scope,
|
&scope,
|
||||||
timestamp,
|
timestamp,
|
||||||
|
user_data.timezone.clone(),
|
||||||
expires,
|
expires,
|
||||||
interval.clone(),
|
interval.clone(),
|
||||||
&mut content,
|
&mut content,
|
||||||
@ -1611,6 +1553,19 @@ async fn natural(ctx: &Context, msg: &Message, args: String) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pub async fn set_reminder(ctx: &Context, interaction: Interaction) {
|
||||||
|
let (pool, lm) = get_ctx_data(&ctx).await;
|
||||||
|
|
||||||
|
let now = SystemTime::now();
|
||||||
|
let since_epoch = now
|
||||||
|
.duration_since(UNIX_EPOCH)
|
||||||
|
.expect("Time calculated as going backwards. Very bad");
|
||||||
|
|
||||||
|
let user_data = UserData::from_user(&interaction.member.user, &ctx, &pool)
|
||||||
|
.await
|
||||||
|
.unwrap();
|
||||||
|
}
|
||||||
|
|
||||||
async fn create_reminder<'a, U: Into<u64>, T: TryInto<i64>>(
|
async fn create_reminder<'a, U: Into<u64>, T: TryInto<i64>>(
|
||||||
ctx: impl CacheHttp + AsRef<Cache>,
|
ctx: impl CacheHttp + AsRef<Cache>,
|
||||||
pool: &MySqlPool,
|
pool: &MySqlPool,
|
||||||
@ -1618,6 +1573,7 @@ async fn create_reminder<'a, U: Into<u64>, T: TryInto<i64>>(
|
|||||||
guild_id: Option<GuildId>,
|
guild_id: Option<GuildId>,
|
||||||
scope_id: &ReminderScope,
|
scope_id: &ReminderScope,
|
||||||
time_parser: T,
|
time_parser: T,
|
||||||
|
timezone: String,
|
||||||
expires_parser: Option<T>,
|
expires_parser: Option<T>,
|
||||||
interval: Option<i64>,
|
interval: Option<i64>,
|
||||||
content: &mut Content,
|
content: &mut Content,
|
||||||
@ -1698,29 +1654,17 @@ async fn create_reminder<'a, U: Into<u64>, T: TryInto<i64>>(
|
|||||||
} else {
|
} else {
|
||||||
sqlx::query!(
|
sqlx::query!(
|
||||||
"
|
"
|
||||||
INSERT INTO messages (content, tts, attachment, attachment_name) VALUES (?, ?, ?, ?)
|
INSERT INTO reminders (uid, content, tts, attachment, attachment_name, channel_id, `utc_time`, timezone, expires, `interval`, set_by) VALUES
|
||||||
|
(?, ?, ?, ?, ?, ?, FROM_UNIXTIME(?), ?, FROM_UNIXTIME(?), ?, (SELECT id FROM users WHERE user = ? LIMIT 1))
|
||||||
",
|
",
|
||||||
|
generate_uid(),
|
||||||
content.content,
|
content.content,
|
||||||
content.tts,
|
content.tts,
|
||||||
content.attachment,
|
content.attachment,
|
||||||
content.attachment_name,
|
content.attachment_name,
|
||||||
)
|
|
||||||
.execute(&pool.clone())
|
|
||||||
.await
|
|
||||||
.unwrap();
|
|
||||||
|
|
||||||
sqlx::query!(
|
|
||||||
"
|
|
||||||
INSERT INTO reminders (uid, message_id, channel_id, time, expires, `interval`, method, set_by) VALUES
|
|
||||||
(?,
|
|
||||||
(SELECT id FROM messages WHERE content = ? ORDER BY id DESC LIMIT 1),
|
|
||||||
?, ?, FROM_UNIXTIME(?), ?, 'remind',
|
|
||||||
(SELECT id FROM users WHERE user = ? LIMIT 1))
|
|
||||||
",
|
|
||||||
generate_uid(),
|
|
||||||
content.content,
|
|
||||||
db_channel_id,
|
db_channel_id,
|
||||||
time as u32,
|
time,
|
||||||
|
timezone,
|
||||||
expires,
|
expires,
|
||||||
interval,
|
interval,
|
||||||
user_id
|
user_id
|
||||||
|
211
src/main.rs
211
src/main.rs
@ -47,6 +47,9 @@ use dashmap::DashMap;
|
|||||||
use tokio::sync::RwLock;
|
use tokio::sync::RwLock;
|
||||||
|
|
||||||
use chrono_tz::Tz;
|
use chrono_tz::Tz;
|
||||||
|
use serenity::model::interactions::{Interaction, InteractionType};
|
||||||
|
use serenity::model::prelude::ApplicationCommandOptionType;
|
||||||
|
use std::collections::HashSet;
|
||||||
|
|
||||||
struct GuildDataCache;
|
struct GuildDataCache;
|
||||||
|
|
||||||
@ -194,6 +197,30 @@ DELETE FROM guilds WHERE guild = ?
|
|||||||
.await
|
.await
|
||||||
.unwrap();
|
.unwrap();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
async fn interaction_create(&self, ctx: Context, interaction: Interaction) {
|
||||||
|
match interaction.kind {
|
||||||
|
InteractionType::ApplicationCommand => {
|
||||||
|
if let Some(data) = &interaction.data {
|
||||||
|
match data.name.as_str() {
|
||||||
|
"timezone" => {
|
||||||
|
moderation_cmds::timezone_interaction(&ctx, interaction).await
|
||||||
|
}
|
||||||
|
"lang" => moderation_cmds::language_interaction(&ctx, interaction).await,
|
||||||
|
"prefix" => moderation_cmds::prefix_interaction(&ctx, interaction).await,
|
||||||
|
"help" => info_cmds::help_interaction(&ctx, interaction).await,
|
||||||
|
"info" => info_cmds::info_interaction(&ctx, interaction).await,
|
||||||
|
"donate" => info_cmds::donate_interaction(&ctx, interaction).await,
|
||||||
|
"clock" => info_cmds::clock_interaction(&ctx, interaction).await,
|
||||||
|
"remind" => reminder_cmds::set_reminder(&ctx, interaction).await,
|
||||||
|
_ => {}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
_ => {}
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
#[tokio::main]
|
#[tokio::main]
|
||||||
@ -280,6 +307,15 @@ async fn main() -> Result<(), Box<dyn std::error::Error + Send + Sync>> {
|
|||||||
.await
|
.await
|
||||||
.expect("Error occurred creating client");
|
.expect("Error occurred creating client");
|
||||||
|
|
||||||
|
let language_manager = Arc::new(
|
||||||
|
LanguageManager::from_compiled(include_str!(concat!(
|
||||||
|
env!("CARGO_MANIFEST_DIR"),
|
||||||
|
"/assets/",
|
||||||
|
env!("STRINGS_FILE")
|
||||||
|
)))
|
||||||
|
.unwrap(),
|
||||||
|
);
|
||||||
|
|
||||||
{
|
{
|
||||||
let guild_data_cache = dashmap::DashMap::new();
|
let guild_data_cache = dashmap::DashMap::new();
|
||||||
|
|
||||||
@ -289,13 +325,6 @@ async fn main() -> Result<(), Box<dyn std::error::Error + Send + Sync>> {
|
|||||||
.await
|
.await
|
||||||
.unwrap();
|
.unwrap();
|
||||||
|
|
||||||
let language_manager = LanguageManager::from_compiled(include_str!(concat!(
|
|
||||||
env!("CARGO_MANIFEST_DIR"),
|
|
||||||
"/assets/",
|
|
||||||
env!("STRINGS_FILE")
|
|
||||||
)))
|
|
||||||
.unwrap();
|
|
||||||
|
|
||||||
let popular_timezones = sqlx::query!(
|
let popular_timezones = sqlx::query!(
|
||||||
"SELECT timezone FROM users GROUP BY timezone ORDER BY COUNT(timezone) DESC LIMIT 21"
|
"SELECT timezone FROM users GROUP BY timezone ORDER BY COUNT(timezone) DESC LIMIT 21"
|
||||||
)
|
)
|
||||||
@ -314,9 +343,16 @@ async fn main() -> Result<(), Box<dyn std::error::Error + Send + Sync>> {
|
|||||||
data.insert::<PopularTimezones>(Arc::new(popular_timezones));
|
data.insert::<PopularTimezones>(Arc::new(popular_timezones));
|
||||||
data.insert::<ReqwestClient>(Arc::new(reqwest::Client::new()));
|
data.insert::<ReqwestClient>(Arc::new(reqwest::Client::new()));
|
||||||
data.insert::<FrameworkCtx>(framework_arc.clone());
|
data.insert::<FrameworkCtx>(framework_arc.clone());
|
||||||
data.insert::<LanguageManager>(Arc::new(language_manager))
|
data.insert::<LanguageManager>(language_manager.clone())
|
||||||
}
|
}
|
||||||
|
|
||||||
|
create_interactions(
|
||||||
|
&client.cache_and_http,
|
||||||
|
framework_arc.clone(),
|
||||||
|
language_manager.clone(),
|
||||||
|
)
|
||||||
|
.await;
|
||||||
|
|
||||||
if let Ok((Some(lower), Some(upper))) = env::var("SHARD_RANGE").map(|sr| {
|
if let Ok((Some(lower), Some(upper))) = env::var("SHARD_RANGE").map(|sr| {
|
||||||
let mut split = sr
|
let mut split = sr
|
||||||
.split(',')
|
.split(',')
|
||||||
@ -360,6 +396,165 @@ async fn main() -> Result<(), Box<dyn std::error::Error + Send + Sync>> {
|
|||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
|
async fn create_interactions(
|
||||||
|
cache_http: impl CacheHttp,
|
||||||
|
framework: Arc<RegexFramework>,
|
||||||
|
lm: Arc<LanguageManager>,
|
||||||
|
) {
|
||||||
|
let http = cache_http.http();
|
||||||
|
let app_id = {
|
||||||
|
let app_info = http.get_current_application_info().await.unwrap();
|
||||||
|
|
||||||
|
app_info.id.as_u64().to_owned()
|
||||||
|
};
|
||||||
|
|
||||||
|
if let Some(guild_id) = env::var("TEST_GUILD")
|
||||||
|
.map(|i| i.parse::<u64>().ok().map(|u| GuildId(u)))
|
||||||
|
.ok()
|
||||||
|
.flatten()
|
||||||
|
{
|
||||||
|
guild_id
|
||||||
|
.create_application_command(&http, app_id, |command| {
|
||||||
|
command
|
||||||
|
.name("timezone")
|
||||||
|
.description("Select your local timezone. Do `/timezone` for more information")
|
||||||
|
.create_interaction_option(|option| {
|
||||||
|
option
|
||||||
|
.name("region")
|
||||||
|
.description("Name of your time region")
|
||||||
|
.kind(ApplicationCommandOptionType::String)
|
||||||
|
})
|
||||||
|
})
|
||||||
|
.await
|
||||||
|
.unwrap();
|
||||||
|
|
||||||
|
guild_id
|
||||||
|
.create_application_command(&http, app_id, |command| {
|
||||||
|
command
|
||||||
|
.name("lang")
|
||||||
|
.description("Select your language")
|
||||||
|
.create_interaction_option(|option| {
|
||||||
|
option
|
||||||
|
.name("language")
|
||||||
|
.description("Name of supported language you wish to use")
|
||||||
|
.kind(ApplicationCommandOptionType::String)
|
||||||
|
.required(true);
|
||||||
|
|
||||||
|
for (code, language) in lm.all_languages() {
|
||||||
|
option.add_string_choice(language, code);
|
||||||
|
}
|
||||||
|
|
||||||
|
option
|
||||||
|
})
|
||||||
|
})
|
||||||
|
.await
|
||||||
|
.unwrap();
|
||||||
|
|
||||||
|
guild_id
|
||||||
|
.create_application_command(&http, app_id, |command| {
|
||||||
|
command
|
||||||
|
.name("prefix")
|
||||||
|
.description("Select the prefix for normal commands")
|
||||||
|
.create_interaction_option(|option| {
|
||||||
|
option
|
||||||
|
.name("prefix")
|
||||||
|
.description("New prefix to use")
|
||||||
|
.kind(ApplicationCommandOptionType::String)
|
||||||
|
.required(true)
|
||||||
|
})
|
||||||
|
})
|
||||||
|
.await
|
||||||
|
.unwrap();
|
||||||
|
|
||||||
|
guild_id
|
||||||
|
.create_application_command(&http, app_id, |command| {
|
||||||
|
command
|
||||||
|
.name("info")
|
||||||
|
.description("Get information about the bot")
|
||||||
|
})
|
||||||
|
.await
|
||||||
|
.unwrap();
|
||||||
|
|
||||||
|
guild_id
|
||||||
|
.create_application_command(&http, app_id, |command| {
|
||||||
|
command
|
||||||
|
.name("donate")
|
||||||
|
.description("View information about the Patreon")
|
||||||
|
})
|
||||||
|
.await
|
||||||
|
.unwrap();
|
||||||
|
|
||||||
|
guild_id
|
||||||
|
.create_application_command(&http, app_id, |command| {
|
||||||
|
command
|
||||||
|
.name("clock")
|
||||||
|
.description("View the current time in your timezone")
|
||||||
|
})
|
||||||
|
.await
|
||||||
|
.unwrap();
|
||||||
|
|
||||||
|
guild_id
|
||||||
|
.create_application_command(&http, app_id, |command| {
|
||||||
|
command
|
||||||
|
.name("help")
|
||||||
|
.description("Get details about commands. Do `/help` to view all commands")
|
||||||
|
.create_interaction_option(|option| {
|
||||||
|
option
|
||||||
|
.name("command")
|
||||||
|
.description("Name of the command to view help for")
|
||||||
|
.kind(ApplicationCommandOptionType::String);
|
||||||
|
|
||||||
|
let mut command_set = HashSet::new();
|
||||||
|
command_set.insert("help");
|
||||||
|
command_set.insert("info");
|
||||||
|
command_set.insert("donate");
|
||||||
|
|
||||||
|
for (_, command) in &framework.commands {
|
||||||
|
if !command_set.contains(command.name) {
|
||||||
|
option.add_string_choice(&command.name, &command.name);
|
||||||
|
|
||||||
|
command_set.insert(command.name);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
option
|
||||||
|
})
|
||||||
|
})
|
||||||
|
.await
|
||||||
|
.unwrap();
|
||||||
|
|
||||||
|
guild_id
|
||||||
|
.create_application_command(&http, app_id, |command| {
|
||||||
|
command
|
||||||
|
.name("remind")
|
||||||
|
.description("Set a reminder")
|
||||||
|
.create_interaction_option(|option| {
|
||||||
|
option
|
||||||
|
.name("message")
|
||||||
|
.description("Message to send with the reminder")
|
||||||
|
.kind(ApplicationCommandOptionType::String)
|
||||||
|
.required(true)
|
||||||
|
})
|
||||||
|
.create_interaction_option(|option| {
|
||||||
|
option
|
||||||
|
.name("time")
|
||||||
|
.description("Time to send the reminder")
|
||||||
|
.kind(ApplicationCommandOptionType::String)
|
||||||
|
.required(true)
|
||||||
|
})
|
||||||
|
.create_interaction_option(|option| {
|
||||||
|
option
|
||||||
|
.name("channel")
|
||||||
|
.description("Channel to send reminder to (default: this channel)")
|
||||||
|
.kind(ApplicationCommandOptionType::Channel)
|
||||||
|
.required(false)
|
||||||
|
})
|
||||||
|
})
|
||||||
|
.await
|
||||||
|
.unwrap();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
pub async fn check_subscription(cache_http: impl CacheHttp, user_id: impl Into<UserId>) -> bool {
|
pub async fn check_subscription(cache_http: impl CacheHttp, user_id: impl Into<UserId>) -> bool {
|
||||||
if let Some(subscription_guild) = *CNC_GUILD {
|
if let Some(subscription_guild) = *CNC_GUILD {
|
||||||
let guild_member = GuildId(subscription_guild)
|
let guild_member = GuildId(subscription_guild)
|
||||||
|
@ -32,7 +32,7 @@ enum ParseType {
|
|||||||
}
|
}
|
||||||
|
|
||||||
pub struct TimeParser {
|
pub struct TimeParser {
|
||||||
timezone: Tz,
|
pub timezone: Tz,
|
||||||
inverted: bool,
|
inverted: bool,
|
||||||
time_string: String,
|
time_string: String,
|
||||||
parse_type: ParseType,
|
parse_type: ParseType,
|
||||||
|
Loading…
Reference in New Issue
Block a user