1 files changed, 117 insertions, 0 deletions
diff --git a/src/commands/today/mod.rs b/src/commands/today/mod.rs
new file mode 100644
index 0000000..bf9a439
--- /dev/null
+++ b/src/commands/today/mod.rs
@@ -0,0 +1,117 @@
+use serenity::{
+    prelude::*,
+    model::{
+        channel::Message,
+    },
+    framework::standard::{
+        Args,
+        macros::command,
+    },
+};
+use chrono::{Duration};
+use either::Left;
+use lazy_static::lazy_static;
+use rand::{
+    thread_rng,
+    seq::SliceRandom,
+};
+use log::debug;
+
+use crate::{
+    Result,
+    CtxExt,
+    audio::{
+        PlayArgs,
+        PlayQueue,
+    },
+};
+
+mod prelude;
+
+mod sept_21;
+mod nov_5;
+
+mod halloween;
+
+mod wednesday;
+
+mod tomorrow;
+
+pub type TodayIter = Box<dyn Iterator<Item=TodayArgs>>;
+
+#[derive(Clone, Debug, Hash, Default)]
+pub struct TodayArgs {
+    pub url: &'static str,
+    pub start: Option<Duration>,
+    pub end: Option<Duration>,
+}
+
+impl TodayArgs {
+    #[inline]
+    pub 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 ALL: Vec<fn(chrono::NaiveDate) -> TodayIter> = vec! [
+        sept_21::sept_21,
+        nov_5::nov_5,
+
+        halloween::halloween,
+
+        wednesday::wednesday,
+
+        tomorrow::tomorrow,
+    ];
+}
+
+
+#[command]
+pub fn today(ctx: &mut Context, msg: &Message, args: Args) -> Result<()> {
+    let today = {
+        let mut result = chrono::Local::today().naive_local();
+
+        #[cfg(debug_assertions)] {
+            match args.parse::<chrono::NaiveDate>() {
+                Ok(date) => {
+                    log::debug!("overriding with date: {}", date);
+                    result = date;
+                },
+                Err(e) => {
+                    log::debug!("parsing date: {:?}", e);
+                }
+            };
+        }
+
+        result
+    };
+
+    let options: Vec<TodayArgs> = ALL.iter()
+        .flat_map(|f| f(today))
+        .collect();
+
+    debug!("selected {} options for {}", options.len(), today);
+
+    let play_args = options.choose(&mut thread_rng())
+        .map(|x| x.as_play_args(msg));
+
+    if let Some(play_args) = play_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(play_args);
+    } else {
+        ctx.send(msg.channel_id, "no", false)?;
+        ctx.send(msg.channel_id, ":angry:", false)?;
+    }
+
+    Ok(())
+}