1 files changed, 137 insertions, 0 deletions
diff --git a/src/audio/play_queue.rs b/src/audio/play_queue.rs
new file mode 100644
index 0000000..6f3a130
--- /dev/null
+++ b/src/audio/play_queue.rs
@@ -0,0 +1,137 @@
+use std::{
+    sync::{Arc, RwLock},
+    thread,
+    collections::VecDeque,
+    time::Duration,
+};
+
+use typemap::Key;
+use either::{Left, Right};
+use serenity::prelude::*;
+
+use crate::{
+    audio::{
+        CurrentItem,
+        PlayArgs,
+        ytdl,
+    },
+    commands::{
+        sound_levels::DEFAULT_VOLUME,
+        send,
+    },
+    must_env_lookup,
+    TARGET_GUILD_ID,
+};
+
+#[derive(Clone)]
+pub struct PlayQueue {
+    pub queue: VecDeque<PlayArgs>,
+    pub playing: Option<CurrentItem>,
+    pub volume: f32,
+}
+
+impl Key for PlayQueue {
+    type Value = Arc<RwLock<PlayQueue>>;
+}
+
+impl PlayQueue {
+    pub fn new() -> Self {
+        PlayQueue {
+            queue: VecDeque::new(),
+            playing: None,
+            volume: DEFAULT_VOLUME,
+        }
+    }
+
+    pub fn register(c: &mut Client) {
+        let voice_manager = Arc::clone(&c.voice_manager);
+
+        let mut data = c.data.lock();
+        let queue = Arc::new(RwLock::new(PlayQueue::new()));
+
+        data.insert::<PlayQueue>(Arc::clone(&queue));
+
+        thread::spawn(move || {
+            let queue_lck = Arc::clone(&queue);
+            let voice_manager = voice_manager;
+
+            loop {
+                thread::sleep(Duration::from_millis(250));
+                let (queue_is_empty, queue_has_playing) = {
+                    let queue = queue_lck.read().unwrap();
+
+                    let allow_continue = queue.playing.clone().map_or(false, |x| !x.audio.lock().finished);
+
+                    if allow_continue {
+                        continue;
+                    }
+
+                    (queue.queue.is_empty(), queue.playing.is_some())
+                };
+
+                if queue_is_empty {
+                    if queue_has_playing {
+                        let mut queue = queue_lck.write().unwrap();
+
+                        assert!({
+                            let audio_lck = queue.playing.clone().unwrap().audio;
+                            let audio = audio_lck.lock();
+                            audio.finished
+                        });
+
+                        queue.playing = None;
+
+                        let mut manager = voice_manager.lock();
+                        manager.leave(*TARGET_GUILD_ID);
+                        debug!("disconnected because playback finished");
+                    }
+                    continue;
+                }
+
+                let mut queue = queue_lck.write().unwrap();
+                let item = queue.queue.pop_front().unwrap();
+
+                let src = match item.data {
+                    Left(ref url) => {
+                        match ytdl(url, item.start, item.end) {
+                            Ok(src) => src,
+                            Err(e) => {
+                                error!("bad link: {}; {:?}", url, e);
+                                let _ = send(item.sender_channel, "what the fuck", false);
+                                continue;
+                            }
+                        }
+                    },
+                    Right(ref vec) => {
+                        ::serenity::voice::opus(true, ::std::io::Cursor::new(vec.clone()))
+                    }
+                };
+
+
+                let mut manager = voice_manager.lock();
+                let handler = manager.join(*TARGET_GUILD_ID, must_env_lookup::<u64>("VOICE_CHANNEL"));
+
+                match handler {
+                    Some(handler) => {
+                        let audio = handler.play_only(src);
+                        {
+                            audio.lock().volume(queue.volume);
+                        }
+
+                        queue.playing = Some(CurrentItem {
+                            init_args: item,
+                            audio,
+                        });
+
+                        debug!("playing new song");
+                    },
+                    None => {
+                        error!("couldn't join channel");
+                        let _ = send(item.sender_channel, "something happened somewhere somehow.", false);
+                    }
+                }
+            }
+        });
+    }
+}
+