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

patreon gated repeat argument

Browse Source
This commit is contained in:
jellywx 2021-11-06 23:30:38 +00:00
parent a21f518b21
commit 23c6b3869e
1 changed files with 37 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

51
src/commands/reminder_cmds.rs
View File

@ -11,6 +11,7 @@ use regex_command_attr::command;
use serenity::{builder::CreateEmbed, client::Context, model::channel::Channel}; use serenity::{builder::CreateEmbed, client::Context, model::channel::Channel};
use crate::{ use crate::{
check_subscription,
component_models::{ component_models::{
pager::{DelPager, LookPager, Pager}, pager::{DelPager, LookPager, Pager},
ComponentDataModel, DelSelector, ComponentDataModel, DelSelector,
@ -665,6 +666,17 @@ DELETE FROM timers WHERE owner = ? AND name = ?
)] )]
#[hook(CHECK_GUILD_PERMISSIONS_HOOK)] #[hook(CHECK_GUILD_PERMISSIONS_HOOK)]
async fn remind(ctx: &Context, invoke: &mut CommandInvoke, args: CommandOptions) { async fn remind(ctx: &Context, invoke: &mut CommandInvoke, args: CommandOptions) {
if args.get("repeat").is_none() && args.get("expires").is_some() {
let _ = invoke
.respond(
&ctx,
CreateGenericResponse::new().content("`expires` can only be used with `repeat`"),
)
.await;
return;
}
invoke.defer(&ctx).await; invoke.defer(&ctx).await;
let user_data = ctx.user_data(invoke.author_id()).await.unwrap(); let user_data = ctx.user_data(invoke.author_id()).await.unwrap();
@ -698,22 +710,33 @@ async fn remind(ctx: &Context, invoke: &mut CommandInvoke, args: CommandOptions)
} }
}; };
// todo gate this on patreon subscription let (interval, expires) = if let Some(repeat) = args.get("repeat") {
let interval = args if check_subscription(&ctx, invoke.author_id()).await {
.get("repeat") (
.map(|arg| { humantime::parse_duration(&repeat.to_string())
humantime::parse_duration(&arg.to_string()) .or_else(|_| {
.or_else(|_| humantime::parse_duration(&format!("1 {}", arg.to_string()))) humantime::parse_duration(&format!("1 {}", repeat.to_string()))
.map(|duration| duration.as_secs() as i64) })
.ok() .map(|duration| duration.as_secs() as i64)
}) .ok(),
.flatten(); {
let expires = { if let Some(arg) = args.get("expires") {
if let Some(arg) = args.get("expires") { natural_parser(&arg.to_string(), &timezone.to_string()).await
natural_parser(&arg.to_string(), &timezone.to_string()).await } else {
None
}
},
)
} else { } else {
None let _ = invoke
.respond(&ctx, CreateGenericResponse::new()
.content("`repeat` is only available to Patreon subscribers or self-hosted users")
).await;
return;
} }
} else {
(None, None)
}; };
let mut builder = MultiReminderBuilder::new(ctx, invoke.guild_id()) let mut builder = MultiReminderBuilder::new(ctx, invoke.guild_id())