diff options
| -rw-r--r-- | src/commands/mod.rs | 5 | ||||
| -rw-r--r-- | src/commands/today.rs | 109 |
2 files changed, 113 insertions, 1 deletions
diff --git a/src/commands/mod.rs b/src/commands/mod.rs index 74cdaf5..705d697 100644 --- a/src/commands/mod.rs +++ b/src/commands/mod.rs @@ -16,6 +16,7 @@ pub use self::{ playback::*, sound_levels::*, roll::{roll, ROLL_COMMAND}, + today::{today, TODAY_COMMAND}, }; #[cfg(feature = "diesel")] pub use self::meme::*; @@ -23,8 +24,9 @@ pub use self::meme::*; pub(crate) mod playback; pub(crate) mod sound_levels; pub(crate) mod roll; -mod help; +pub(crate) mod today; +mod help; group!({ name: "general", @@ -33,6 +35,7 @@ group!({ }, commands: [ roll, + today, ], }); diff --git a/src/commands/today.rs b/src/commands/today.rs new file mode 100644 index 0000000..eea323c --- /dev/null +++ b/src/commands/today.rs @@ -0,0 +1,109 @@ +use serenity::{ + prelude::*, + model::{ + channel::Message, + }, + framework::standard::{ + Args, + macros::command, + }, +}; +use chrono::{Datelike, Duration}; +use either::Left; +use lazy_static::lazy_static; +use rand::{ + thread_rng, + seq::SliceRandom, +}; + +use crate::{ + Result, + CtxExt, + audio::{ + PlayArgs, + PlayQueue, + }, +}; + +#[derive(Clone, Debug, Hash, Default)] +struct TodayArgs { + url: &'static str, + start: Option<Duration>, + end: Option<Duration>, +} + +impl TodayArgs { + #[inline] + fn as_play_args(&self, msg: &Message) -> PlayArgs { + PlayArgs { + initiator: "you have done this to yourself :^)".to_string(), + data: Left(self.url.to_owned()), + sender_channel: msg.channel_id, + start: self.start, + end: self.end, + } + } +} + +lazy_static! { + static ref SEPT_21_CHOICES: Vec<TodayArgs> = vec![ + TodayArgs { + url: "https://www.youtube.com/watch?v=kPwG6L73-VU", + ..Default::default() + }, + TodayArgs { + url: "https://www.youtube.com/watch?v=fPpUYXZb2AA", + ..Default::default() + }, + TodayArgs { + url: "https://www.youtube.com/watch?v=CG7YHFT4hjw", + end: Some(Duration::seconds(69)), + ..Default::default() + }, + TodayArgs { + url: "https://www.youtube.com/watch?v=_hpU6UEq8hA", + end: Some(Duration::seconds(67)), + ..Default::default() + }, + TodayArgs { + url: "https://www.youtube.com/watch?v=_zzEDrYTkkg", + end: Some(Duration::seconds(68)), + ..Default::default() + }, + TodayArgs { + url: "https://www.youtube.com/watch?v=Gs069dndIYk", + ..Default::default() + }, + ]; +} + +#[command] +pub fn today(ctx: &mut Context, msg: &Message, _: Args) -> Result<()> { + let today = chrono::Local::today().naive_local(); + + let args: Option<PlayArgs> = match (today.month(), today.day()) { + (9, 21) => SEPT_21_CHOICES.choose(&mut thread_rng()) + .map(|choice| choice.as_play_args(msg)), + _ => { + let result = TodayArgs { + url: "https://www.youtube.com/watch?v=W78AGkm_AtE", + start: None, + end: Some(Duration::seconds(6)) + }.as_play_args(msg); + + Some(result) + }, + }; + + if let Some(args) = args { + let queue_lock = ctx.data.write().get::<PlayQueue>().cloned().unwrap(); + let mut play_queue = queue_lock.write().unwrap(); + + play_queue.general_queue.push_front(args); + } else { + ctx.send(msg.channel_id, "no", false)?; + ctx.send(msg.channel_id, ":angry:", false)?; + } + + Ok(()) +} |