jude/reminder-bot
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 3 Wiki Activity

added count to del. changed inner joins to left outers. moved all consts and lazies into consts.rs. more formatting fixes

Browse Source
This commit is contained in:
jude
2020-10-11 17:41:26 +01:00
parent 1f41343e2e
commit 47ad3f3822
4 changed files with 104 additions and 50 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
src/commands/moderation_cmds.rs
View File

@ -14,8 +14,6 @@ use serenity::{
},
};
use regex::Regex;
use chrono_tz::Tz;
use chrono::offset::Utc;
@ -31,19 +29,16 @@ use crate::{
SQLPool,
FrameworkCtx,
framework::SendIterator,
consts::{
REGEX_ALIAS,
REGEX_CHANNEL,
REGEX_COMMANDS,
REGEX_ROLE,
},
};
use std::iter;
lazy_static! {
static ref REGEX_CHANNEL: Regex = Regex::new(r#"^\s*<#(\d+)>\s*$"#).unwrap();
static ref REGEX_ROLE: Regex = Regex::new(r#"<@&([0-9]+)>"#).unwrap();
static ref REGEX_COMMANDS: Regex = Regex::new(r#"([a-z]+)"#).unwrap();
static ref REGEX_ALIAS: Regex = Regex::new(r#"(?P<name>[\S]{1,12})(?:(?: (?P<cmd>.*)$)|$)"#).unwrap();
}
#[command]
#[supports_dm(false)]
@ -82,7 +77,6 @@ async fn timezone(ctx: &Context, msg: &Message, args: String) -> CommandResult {
.get::<SQLPool>().cloned().expect("Could not get SQLPool from data");
let mut user_data = UserData::from_user(&msg.author, &ctx, &pool).await.unwrap();
let guild_data = GuildData::from_guild(msg.guild(&ctx).await.unwrap(), &pool).await.unwrap();
if !args.is_empty() {
match args.parse::<Tz>() {
@ -106,7 +100,7 @@ async fn timezone(ctx: &Context, msg: &Message, args: String) -> CommandResult {
}
else {
let content = user_data.response(&pool, "timezone/no_argument").await
.replace("{prefix}", &guild_data.prefix)
.replace("{prefix}", &GuildData::prefix_from_id(msg.guild_id, &pool).await)
.replacen("{timezone}", &user_data.timezone, 1);
let _ = msg.channel_id.say(&ctx, content).await;

58
src/commands/reminder_cmds.rs
View File

@ -23,6 +23,13 @@ use serenity::{
use tokio::process::Command;
use crate::{
consts::{
REGEX_CHANNEL,
REGEX_CHANNEL_USER,
MIN_INTERVAL,
MAX_TIME,
CHARACTERS,
},
models::{
ChannelData,
GuildData,
@ -71,18 +78,6 @@ use regex::Regex;
use serde_json::json;
lazy_static! {
static ref REGEX_CHANNEL: Regex = Regex::new(r#"^\s*<#(\d+)>\s*$"#).unwrap();
static ref REGEX_CHANNEL_USER: Regex = Regex::new(r#"^\s*<(#|@)(?:!)?(\d+)>\s*$"#).unwrap();
static ref MIN_INTERVAL: i64 = env::var("MIN_INTERVAL").ok().map(|inner| inner.parse::<i64>().ok()).flatten().unwrap_or(600);
static ref MAX_TIME: i64 = env::var("MAX_TIME").ok().map(|inner| inner.parse::<i64>().ok()).flatten().unwrap_or(60*60*24*365*50);
}
static CHARACTERS: &str = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789_";
#[command]
#[supports_dm(false)]
@ -330,7 +325,7 @@ SELECT
reminders.id, reminders.time, messages.content, channels.channel
FROM
reminders
INNER JOIN
LEFT OUTER JOIN
channels
ON
channels.id = reminders.channel_id
@ -392,14 +387,14 @@ SELECT
reminders.id, reminders.time, messages.content, channels.channel
FROM
reminders
INNER JOIN
LEFT OUTER JOIN
channels
ON
reminders.channel_id = channels.id
channels.id = reminders.channel_id
INNER JOIN
messages
ON
reminders.message_id = messages.id
messages.id = reminders.message_id
WHERE
channels.guild_id = (SELECT id FROM guilds WHERE guild = ?)
", guild_id)
@ -467,17 +462,30 @@ WHERE
.collect::<Vec<String>>();
if parts.len() == valid_parts.len() {
let joined = valid_parts.join(",");
let count_row = sqlx::query!(
"
SELECT COUNT(1) AS count FROM reminders WHERE FIND_IN_SET(id, ?)
", joined)
.fetch_one(&pool)
.await
.unwrap();
sqlx::query!(
"
DELETE FROM reminders WHERE FIND_IN_SET(id, ?)
", valid_parts.join(","))
", joined)
.execute(&pool)
.await
.unwrap();
// TODO add deletion events to event list
let _ = msg.channel_id.say(&ctx, user_data.response(&pool, "del/count").await).await;
let content = user_data.response(&pool, "del/count").await
.replacen("{}", &count_row.count.to_string(), 1);
let _ = msg.channel_id.say(&ctx, content).await;
}
}
@ -599,6 +607,7 @@ custom_error!{ReminderError
InvalidTag = "Invalid reminder scope",
NotEnoughArgs = "Not enough args",
InvalidTime = "Invalid time provided",
NeedSubscription = "Subscription required and not found",
DiscordError = "Bad response received from Discord"
}
@ -618,6 +627,7 @@ impl ToResponse for ReminderError {
Self::InvalidTag => "remind/invalid_tag",
Self::NotEnoughArgs => "remind/no_argument",
Self::InvalidTime => "remind/invalid_time",
Self::NeedSubscription => "interval/donor",
Self::DiscordError => "remind/no_webhook",
}.to_string()
}
@ -625,6 +635,7 @@ impl ToResponse for ReminderError {
fn to_response_natural(&self) -> String {
match self {
Self::LongTime => "natural/long_time".to_string(),
Self::InvalidTime => "natural/invalid_time".to_string(),
_ => self.to_response(),
}
}
@ -709,6 +720,9 @@ async fn remind_command(ctx: &Context, msg: &Message, args: String, command: Rem
Err(ReminderError::NotEnoughArgs)
}
}
else if command == RemindCommand::Interval {
Err(ReminderError::NeedSubscription)
}
else {
let content = args_iter.collect::<Vec<&str>>().join(" ");
@ -769,6 +783,7 @@ async fn remind_command(ctx: &Context, msg: &Message, args: String, command: Rem
let offset = time_parser.map(|tp| tp.displacement().ok()).flatten().unwrap_or(0) as u64;
let str_response = user_data.response(&pool, &response.to_response()).await
.replace("{prefix}", &GuildData::prefix_from_id(msg.guild_id, &pool).await)
.replacen("{location}", &scope_id.mention(), 1)
.replacen("{offset}", &shorthand_displacement(offset), 1)
.replacen("{min_interval}", &MIN_INTERVAL.to_string(), 1)
@ -887,6 +902,7 @@ async fn natural(ctx: &Context, msg: &Message, args: String) -> CommandResult {
let offset = timestamp as u64 - since_epoch.as_secs();
let str_response = user_data.response(&pool, &res.to_response_natural()).await
.replace("{prefix}", &GuildData::prefix_from_id(msg.guild_id, &pool).await)
.replacen("{location}", &location_id.mention(), 1)
.replacen("{offset}", &shorthand_displacement(offset), 1)
.replacen("{min_interval}", &MIN_INTERVAL.to_string(), 1)
@ -1004,7 +1020,7 @@ async fn create_reminder<T: TryInto<i64>, S: ToString + Type<MySql> + Encode<MyS
let unix_time = SystemTime::now().duration_since(UNIX_EPOCH).unwrap().as_secs() as i64;
if time > unix_time {
if time >= unix_time {
if time > unix_time + *MAX_TIME {
Err(ReminderError::LongTime)
}
@ -1032,6 +1048,10 @@ INSERT INTO reminders (uid, message_id, channel_id, time, `interval`, method, se
Ok(())
}
}
else if time < 0 {
// case required for if python returns -1
Err(ReminderError::InvalidTime)
}
else {
Err(ReminderError::PastTime)
}