Compare commits

...

3 Commits

Author SHA1 Message Date
bf34721e55 stub of new remind command 2021-05-13 18:19:14 +01:00
2c91a72640 slash commands 2021-04-30 00:13:14 +01:00
4a64238ee4 database migration 2021-04-17 16:57:46 +01:00
8 changed files with 840 additions and 148 deletions

2
Cargo.lock generated
View File

@ -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",

View File

@ -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]

File diff suppressed because one or more lines are too long

View File

@ -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}", &current_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();
}

View File

@ -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)]

View File

@ -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

View File

@ -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)

View File

@ -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,