refactor(watch/playlist_handler): Init

This facilitates outsourcing the mpv playlist operations and overlaying
them with an cache that provides the facility to convert for `playlist_entry_id`s
to `ExtractorHash`es even after their corresponding video has been
removed from the playlist.

1 files changed, 24 insertions, 70 deletions

diff --git a/src/watch/events.rs b/src/watch/events/mod.rs
index c1a2d13..9ca12fd 100644
--- a/src/watch/events.rs
+++ b/src/watch/events/mod.rs
@@ -8,12 +8,11 @@
 // You should have received a copy of the License along with this program.
 // If not, see <https://www.gnu.org/licenses/gpl-3.0.txt>.
 
-use std::{collections::HashMap, env::current_exe, time::Duration, usize};
+use std::{collections::HashMap, env::current_exe, mem, time::Duration, usize};
 
 use anyhow::{bail, Result};
 use libmpv2::{
     events::{Event, PlaylistEntryId},
-    mpv_node::MpvNode,
     EndFileReason, Mpv,
 };
 use log::{debug, info, warn};
@@ -30,72 +29,25 @@ use crate::{
     },
 };
 
+use playlist_handler::PlaylistHandler;
+
+mod playlist_handler;
+
 #[derive(Debug)]
 pub struct MpvEventHandler {
     watch_later_block_list: HashMap<ExtractorHash, ()>,
-    // current_playlist: HashMap<PlaylistEntryId, ExtractorHash>,
-    playlist_cache: HashMap<String, ExtractorHash>,
+    playlist_handler: PlaylistHandler,
 }
 
 impl MpvEventHandler {
     pub fn from_playlist(playlist_cache: HashMap<String, ExtractorHash>) -> Self {
+        let playlist_handler = PlaylistHandler::from_cache(playlist_cache);
         Self {
-            // current_playlist,
-            playlist_cache,
+            playlist_handler,
             watch_later_block_list: HashMap::new(),
         }
     }
 
-    fn get_current_mpv_playlist(
-        &self,
-        mpv: &Mpv,
-    ) -> Result<HashMap<PlaylistEntryId, ExtractorHash>> {
-        let mpv_playlist: Vec<(String, PlaylistEntryId)> = match mpv.get_property("playlist")? {
-            MpvNode::ArrayIter(array) => array
-                .map(|val| match val {
-                    MpvNode::MapIter(map) => {
-                        struct BuildPlaylistEntry {
-                            filename: Option<String>,
-                            id: Option<PlaylistEntryId>,
-                        }
-                        let mut entry = BuildPlaylistEntry {
-                            filename: None,
-                            id: None,
-                        };
-
-                        map.for_each(|(key, value)| match key.as_str() {
-                            "filename" => {
-                                entry.filename = Some(value.str().expect("work").to_owned())
-                            }
-                            "id" => {
-                                entry.id = Some(PlaylistEntryId::new(value.i64().expect("Works")))
-                            }
-                            _ => (),
-                        });
-                        (entry.filename.expect("is some"), entry.id.expect("is some"))
-                    }
-                    _ => unreachable!(),
-                })
-                .collect(),
-            _ => unreachable!(),
-        };
-
-        let mut playlist: HashMap<PlaylistEntryId, ExtractorHash> =
-            HashMap::with_capacity(mpv_playlist.len());
-        for (path, key) in mpv_playlist {
-            let hash = self
-                .playlist_cache
-                .get(&path)
-                .expect("All path should also be stored in the cache")
-                .to_owned();
-            playlist.insert(key, hash);
-        }
-
-        // debug!("Requested the current playlist: '{:#?}'", &playlist);
-
-        Ok(playlist)
-    }
-
     /// Checks, whether new videos are ready to be played
     pub async fn possibly_add_new_videos(
         &mut self,
@@ -114,7 +66,7 @@ impl MpvEventHandler {
         }
 
         let mut blocked_videos = 0;
-        let current_playlist = self.get_current_mpv_playlist(mpv)?;
+        let current_playlist = self.playlist_handler.playlist_ids(mpv)?;
         let play_things = play_things
             .into_iter()
             .filter(|val| {
@@ -142,9 +94,8 @@ impl MpvEventHandler {
             blocked_videos
         );
 
-        self.playlist_cache.reserve(play_things.len());
-
         let num = play_things.len();
+        self.playlist_handler.reserve(play_things.len());
         for play_thing in play_things {
             debug!("Adding '{}' to playlist.", play_thing.title);
 
@@ -155,8 +106,8 @@ impl MpvEventHandler {
             let args = &[&fmt_cache_path, "append-play"];
 
             mpv.execute("loadfile", args)?;
-            self.playlist_cache
-                .insert(cache_path.to_owned(), play_thing.extractor_hash);
+            self.playlist_handler
+                .add(cache_path.to_owned(), play_thing.extractor_hash);
         }
 
         if force_message || num > 0 {
@@ -182,7 +133,7 @@ impl MpvEventHandler {
     /// You can specify an offset, which is added to the playlist_position to get, for example, the
     /// previous video (-1) or the next video (+1).
     /// Beware that setting an offset can cause an property error if it's out of bound.
-    fn get_cvideo_hash(&self, mpv: &Mpv, offset: i64) -> Result<ExtractorHash> {
+    fn get_cvideo_hash(&mut self, mpv: &Mpv, offset: i64) -> Result<ExtractorHash> {
         let playlist_entry_id = {
             let playlist_position = {
                 let raw = mpv.get_property::<i64>("playlist-pos")?;
@@ -201,9 +152,11 @@ impl MpvEventHandler {
         // debug!("Trying to get playlist entry: '{}'", playlist_entry_id);
 
         let video_hash = self
-            .get_current_mpv_playlist(mpv)?
-            .remove(&playlist_entry_id)
-            .expect("The stored playling index should always be in the playlist");
+            .playlist_handler
+            .playlist_ids(mpv)?
+            .get(&playlist_entry_id)
+            .expect("The stored playling index should always be in the playlist")
+            .to_owned();
 
         Ok(video_hash)
     }
@@ -220,7 +173,7 @@ impl MpvEventHandler {
         mpv: &Mpv,
         playlist_index: PlaylistEntryId,
     ) -> Result<()> {
-        let current_playlist = self.get_current_mpv_playlist(mpv)?;
+        let current_playlist = self.playlist_handler.playlist_ids(mpv)?;
         let video_hash = current_playlist
             .get(&playlist_index)
             .expect("The video index should always be correctly tracked");
@@ -234,7 +187,7 @@ impl MpvEventHandler {
         mpv: &Mpv,
         playlist_index: PlaylistEntryId,
     ) -> Result<()> {
-        let current_playlist = self.get_current_mpv_playlist(mpv)?;
+        let current_playlist = self.playlist_handler.playlist_ids(mpv)?;
         let video_hash = current_playlist
             .get(&playlist_index)
             .expect("The video index should always be correctly tracked");
@@ -304,7 +257,8 @@ impl MpvEventHandler {
                     info!("Mpv quit. Exiting playback");
 
                     // draining the playlist is okay, as mpv is done playing
-                    let videos = self.get_current_mpv_playlist(mpv)?;
+                    let mut handler = mem::take(&mut self.playlist_handler);
+                    let videos = handler.playlist_ids(mpv)?;
                     for (_, hash) in videos {
                         self.mark_video_watched(app, &hash).await?;
                         set_state_change(&app, &hash, false).await?;
@@ -324,8 +278,8 @@ impl MpvEventHandler {
                 // We don't need to check, whether other videos are still active, as they should
                 // have been marked inactive in the `Stop` handler.
                 self.mark_video_active(app, mpv, entry_id).await?;
-                self.apply_options(app, mpv, &self.get_cvideo_hash(mpv, 0)?)
-                    .await?;
+                let hash = self.get_cvideo_hash(mpv, 0)?;
+                self.apply_options(app, mpv, &hash).await?;
             }
             Event::ClientMessage(a) => {
                 debug!("Got Client Message event: '{}'", a.join(" "));