Make use of callback serenity

src/player/commands/play.rs

@@ -5,7 +5,8 @@ use serenity::voice::LockedAudio;
 
 use std::cell::RefCell;
 use std::fs;
-use std::ops::DerefMut;
+use std::ops::{Deref, DerefMut};
+use std::rc::Rc;
 use std::sync::{Arc, MutexGuard};
 
 use rand::{seq::SliceRandom, thread_rng};
@@ -84,7 +85,27 @@ impl Play {
 
         Self::check_join_channel(&manager_lock, msg);
 
-        MediaData::start_thread(&self.media, msg.channel_id, &manager_lock);
+        let check_finished = {
+            let media_clone = self.media.clone();
+            let channel_id = msg.channel_id;
+            let manager_lock_clone = manager_lock.clone();
+
+            Rc::new(move || {
+                let callback = match media_clone.next_callback.borrow().deref() {
+                    Some(callback) => callback.clone(),
+                    None => {
+                        println!("next_callback not set!");
+                        return;
+                    }
+                };
+
+                MediaData::next_song(&media_clone, channel_id, &manager_lock_clone, callback);
+            })
+        };
+
+        MediaData::start_playing(&self.media, check_finished, msg.channel_id, &manager_lock);
+
+        //MediaData::start_thread(&self.media, msg.channel_id, &manager_lock);
     }
 
     fn handle_http_request(

src/player/mediadata.rs

@@ -5,10 +5,8 @@ use serenity::voice::LockedAudio;
 
 use std::cell::RefCell;
 use std::ops::DerefMut;
+use std::rc::Rc;
 use std::sync::{Arc, Mutex, MutexGuard};
-use std::thread;
-use std::thread::JoinHandle;
-use std::time;
 
 use super::prelude::*;
 
@@ -20,7 +18,7 @@ pub struct MediaData {
     playlist: Mutex<RefCell<Vec<Song>>>,
     current_song: Mutex<RefCell<Option<LockedAudio>>>,
     song_name: Mutex<RefCell<String>>,
-    watcher_thread: Mutex<RefCell<Option<JoinHandle<()>>>>,
+    pub next_callback: RefCell<Option<Rc<Fn() -> ()>>>,
 }
 
 impl MediaData {
@@ -45,14 +43,19 @@ impl MediaData {
         }
 
         {
-            let manager_lock = ctx.data.lock().get::<VoiceManager>().cloned().unwrap();
+            *self.next_callback.borrow_mut() = None;
+        }
 
+        {
+            let manager_lock = ctx.data.lock().get::<VoiceManager>().cloned().unwrap();
             let mut manager = manager_lock.lock();
+
             let guild_id = match guild_id(msg.channel_id) {
                 Some(guild_id) => guild_id,
                 None => return Ok(()),
             };
 
+            {
                 let handler = match handler(guild_id, &mut manager) {
                     Some(handler) => handler,
                     None => return Ok(()),
@@ -63,6 +66,9 @@ impl MediaData {
                 handler.stop();
             }
 
+            manager.remove(guild_id);
+        }
+
         Ok(())
     }
 
@@ -78,135 +84,75 @@ impl MediaData {
         self.song_name.lock().unwrap()
     }
 
-    pub fn reset_thread(&self) {
+    pub fn start_playing(
-        let thread_mutex = self.watcher_thread.lock().unwrap();
+        mediadata: &Arc<MediaData>,
-        *thread_mutex.borrow_mut() = None;
+        callback: Rc<Fn() -> ()>,
-    }
-
-    pub fn start_thread(
-        media: &Arc<MediaData>,
         channel_id: ChannelId,
         manager_lock: &Arc<
             serenity::prelude::Mutex<serenity::client::bridge::voice::ClientVoiceManager>,
         >,
     ) {
-        let media_clone = media.clone();
+        // check if there is already playing
-        let manager_clone = manager_lock.clone();
+        let already_started = {
-
+            let song_lock = mediadata.song_mut();
-        let thread_mutex = media.watcher_thread.lock().unwrap();
+            let song = song_lock.borrow();
-        let mut watcher_mut = thread_mutex.borrow_mut();
+            song.is_some()
-
-        if watcher_mut.is_none() {
-            *watcher_mut = Some(thread::spawn(move || loop {
-                {
-                    let song_lock = media_clone.song_mut();
-                    let mut borrow = song_lock.borrow_mut();
-
-                    // if there is currently something playing
-                    if let Some(ref mut song) = borrow.deref_mut() {
-                        let song_lock = song.clone();
-                        let audio = song_lock.lock();
-
-                        if audio.finished {
-                            let playlist = media_clone.playlist_mut();
-                            let song_name = media_clone.name_mut();
-
-                            if !Self::next_song(
-                                song,
-                                song_name.borrow_mut().deref_mut(),
-                                playlist.borrow_mut().deref_mut(),
-                                channel_id,
-                                &manager_clone,
-                            ) {
-                                manager_clone
-                                    .lock()
-                                    .remove(check_option!(guild_id(channel_id)));
-
-                                media_clone.reset_thread();
-
-                                println!("left channel");
-
-                                return;
-                            }
-                        }
-
-                        continue;
-                    }
-
-                    // nothing is playing
-                    {
-                        let playlist_mutex = media_clone.playlist_mut();
-                        let mut playlist = playlist_mutex.borrow_mut();
-
-                        if !playlist.is_empty() {
-                            let mut manager = manager_clone.lock();
-
-                            if let Some(handler) =
-                                handler(check_option!(guild_id(channel_id)), &mut manager)
-                            {
-                                let first = playlist.remove(0);
-
-                                let source = match ffmpeg(first.name.clone()) {
-                                    Ok(mpeg) => mpeg,
-                                    Err(_) => {
-                                        playlist.clear();
-                                        *borrow = None;
-                                        handler.stop();
-                                        media_clone.reset_thread();
-
-                                        return;
-                                    }
         };
 
-                                *borrow = Some(handler.play_returning(source));
+        // if there isnt already a song playing, start a new one
-                                let song_name = media_clone.name_mut();
+        if !already_started {
-                                *song_name.borrow_mut() = first.name.clone();
+            *mediadata.next_callback.borrow_mut() = Some(callback.clone());
 
-                                print_error!(
+            Self::next_song(mediadata, channel_id, manager_lock, callback);
-                                    channel_id.say(format!("Playing song: {}", first.name))
-                                );
-                            } else {
-                                return;
-                            }
-                        } else {
-                            manager_clone
-                                .lock()
-                                .remove(check_option!(guild_id(channel_id)));
-
-                            media_clone.reset_thread();
-
-                            println!("left channel");
-
-                            return;
-                        }
-                    }
-                }
-
-                let two_sec = time::Duration::from_secs(2);
-                thread::sleep(two_sec);
-            }));
         }
     }
 
-    fn next_song(
+    pub fn next_song(
-        song: &mut LockedAudio,
+        mediadata: &Arc<MediaData>,
-        name: &mut String,
-        playlist: &mut Vec<Song>,
         channel_id: ChannelId,
         manager_lock: &Arc<
             serenity::prelude::Mutex<serenity::client::bridge::voice::ClientVoiceManager>,
         >,
-    ) -> bool {
+        callback: Rc<Fn() -> ()>,
+    ) {
         let mut manager = manager_lock.lock();
 
         let guild_id = match guild_id(channel_id) {
             Some(id) => id,
-            None => return false,
+            None => {
+                println!("error getting guild_id");
+                return;
+            }
         };
 
-        if let Some(handler) = handler(guild_id, &mut manager) {
+        let mut need_to_leave = false;
+
+        {
+            let handler = {
+                match handler(guild_id, &mut manager) {
+                    Some(handler) => handler,
+                    None => {
+                        println!("error getting handler");
+                        return;
+                    }
+                }
+            };
+
+            let playlist_lock = mediadata.playlist_mut();
+            let mut playlist = playlist_lock.borrow_mut();
+
+            let song_name_lock = mediadata.name_mut();
+            let mut name = song_name_lock.borrow_mut();
+
+            let song_lock = mediadata.song_mut();
+            let mut song = song_lock.borrow_mut();
+
             if playlist.is_empty() {
-                return false;
+                need_to_leave = true;
+                handler.stop();
+                *song = None;
+                *name = String::new();
+                *mediadata.next_callback.borrow_mut() = None;
             } else {
                 handler.stop();
 
@@ -217,20 +163,20 @@ impl MediaData {
                     Err(_) => {
                         playlist.clear();
 
-                        return false;
+                        return;
                     }
                 };
 
-                *song = handler.play_returning(source);
+                *song = Some(handler.play_returning_and_callback(source, callback));
                 *name = first.name.clone();
 
                 print_error!(channel_id.say(format!("Playing song: {}", first.name)));
-
-                return true;
             }
         }
 
-        return false;
+        if need_to_leave {
+            manager.remove(guild_id);
+        }
     }
 }
 
@@ -240,7 +186,7 @@ impl Default for MediaData {
             playlist: Mutex::new(RefCell::new(Vec::new())),
             current_song: Mutex::new(RefCell::new(None)),
             song_name: Mutex::new(RefCell::new(String::new())),
-            watcher_thread: Mutex::new(RefCell::new(None)),
+            next_callback: RefCell::new(None),
         }
     }
 }

src/player/player.rs

@@ -57,7 +57,7 @@ pub fn channel_contains_author(
                 Some(id) => id,
                 None => {
                     println!("error getting channel_id for bot");
-                    return false;
+                    return true;
                 }
             },
         )