nudge command

This commit is contained in:
jude 2020-09-01 18:37:43 +01:00
parent 4bd24ff94c
commit e395886f6c
3 changed files with 48 additions and 6 deletions

View File

@ -38,10 +38,10 @@ async fn pause(ctx: &Context, msg: &Message, args: String) -> CommandResult {
channel.commit_changes(&pool).await; channel.commit_changes(&pool).await;
if channel.paused { if channel.paused {
let _ = msg.channel_id.say(&ctx, user_data.response(&pool, "paused/paused_indefinite").await).await; let _ = msg.channel_id.say(&ctx, user_data.response(&pool, "pause/paused_indefinite").await).await;
} }
else { else {
let _ = msg.channel_id.say(&ctx, user_data.response(&pool, "paused/unpaused").await).await; let _ = msg.channel_id.say(&ctx, user_data.response(&pool, "pause/unpaused").await).await;
} }
} }
else { else {
@ -55,11 +55,11 @@ async fn pause(ctx: &Context, msg: &Message, args: String) -> CommandResult {
channel.commit_changes(&pool).await; channel.commit_changes(&pool).await;
let _ = msg.channel_id.say(&ctx, user_data.response(&pool, "paused/paused_until").await).await; let _ = msg.channel_id.say(&ctx, user_data.response(&pool, "pause/paused_until").await).await;
}, },
Err(_) => { Err(_) => {
let _ = msg.channel_id.say(&ctx, user_data.response(&pool, "paused/invalid_time").await).await; let _ = msg.channel_id.say(&ctx, user_data.response(&pool, "pause/invalid_time").await).await;
}, },
} }
} }
@ -116,3 +116,43 @@ UPDATE reminders SET `time` = `time` + ? WHERE reminders.channel_id = ?
Ok(()) Ok(())
} }
#[command]
async fn nudge(ctx: &Context, msg: &Message, args: String) -> CommandResult {
let pool = ctx.data.read().await
.get::<SQLPool>().cloned().expect("Could not get SQLPool from data");
let user_data = UserData::from_id(&msg.author, &ctx, &pool).await.unwrap();
let mut channel = ChannelData::from_channel(msg.channel(&ctx).await.unwrap(), &pool).await.unwrap();
if args.len() == 0 {
let _ = msg.channel_id.say(&ctx, user_data.response(&pool, "nudge/invalid_time").await).await;
}
else {
let parser = TimeParser::new(args, user_data.timezone.parse().unwrap());
let nudge_time = parser.displacement();
match nudge_time {
Ok(displacement) => {
if displacement < i16::MIN as i64 || displacement > i16::MAX as i64 {
let _ = msg.channel_id.say(&ctx, user_data.response(&pool, "nudge/invalid_time").await).await;
}
else {
channel.nudge = displacement as i16;
channel.commit_changes(&pool).await;
let response = user_data.response(&pool, "nudge/success").await.replacen("{}", &displacement.to_string(), 1);
let _ = msg.channel_id.say(&ctx, response).await;
}
},
Err(_) => {
let _ = msg.channel_id.say(&ctx, user_data.response(&pool, "nudge/invalid_time").await).await;
},
}
}
Ok(())
}

View File

@ -73,6 +73,8 @@ async fn main() -> Result<(), Box<dyn std::error::Error + Send + Sync>> {
.add_command("pause", &reminder_cmds::PAUSE_COMMAND) .add_command("pause", &reminder_cmds::PAUSE_COMMAND)
.add_command("offset", &reminder_cmds::OFFSET_COMMAND) .add_command("offset", &reminder_cmds::OFFSET_COMMAND)
.add_command("nudge", &reminder_cmds::NUDGE_COMMAND)
.build(); .build();
let mut client = Client::new(&env::var("DISCORD_TOKEN").expect("Missing DISCORD_TOKEN from environment")) let mut client = Client::new(&env::var("DISCORD_TOKEN").expect("Missing DISCORD_TOKEN from environment"))

View File

@ -196,9 +196,9 @@ SELECT value FROM strings WHERE (language = ? OR language = 'EN') AND name = ? O
", self.language, name) ", self.language, name)
.fetch_one(pool) .fetch_one(pool)
.await .await
.expect("No string with that name"); .expect(&format!("No string with that name: {}", name));
row.value.expect("No string with that name") row.value.expect(&format!("Null string with that name: {}", name))
} }
pub fn timezone(&self) -> Tz { pub fn timezone(&self) -> Tz {