Allow for favoriting and unfavoriting songs

Author: Ross

src/app/mpris.rs

@@ -1,6 +1,6 @@
+use anyhow::Result;
 use mpris_server::{PlaybackStatus, Property};
 use notify_rust::{Hint, Notification};
-use anyhow::Result;
 
 use crate::app::Track;
 
@@ -15,8 +15,8 @@ impl App {
             PlaybackStatus::Stopped
         };
 
-        let can_next = self.queue_tab.index < self.queue_tab.len().saturating_sub(1);
-        let can_prev = self.queue_tab.index > 0;
+        let can_next = self.playing_index < self.queue_tab.len().saturating_sub(1);
+        let can_prev = self.playing_index > 0;
         let current_pos = self.player.lock().await.get_position();
 
         if let Ok(mut state) = self.shared_state.write() {

src/app/queue.rs

@@ -1,8 +1,7 @@
+use super::App;
 use crate::app::{ActiveSection, ActiveTab, Track};
 use anyhow::Result;
 use futures::future;
-use super::App;
-
 
 impl App {
     pub fn add_search_result_to_queue(&mut self) {
@@ -69,4 +68,28 @@ impl App {
         }
         Ok(())
     }
+    pub async fn make_favorite(&mut self, remove: bool) -> Result<()> {
+        match (&self.active_section, &self.active_tab) {
+            (ActiveSection::Queue, _) => {
+                if !self.queue_tab.data.is_empty() {
+                    let song = self.queue_tab.get().unwrap();
+                    self.subsonic_client.favorite_a_song(song, remove).await?;
+                }
+            }
+            (ActiveSection::Others, ActiveTab::Songs) => {
+                if !self.tracks_tab.data.is_empty() {
+                    let song = self.tracks_tab.get().unwrap();
+                    self.subsonic_client.favorite_a_song(song, remove).await?;
+                }
+            }
+            (ActiveSection::Others, ActiveTab::Search) => {
+                if !self.search_tab.data.is_empty() {
+                    let song = self.search_tab.get().unwrap();
+                    self.subsonic_client.favorite_a_song(song, remove).await?;
+                }
+            }
+            _ => (),
+        }
+        Ok(())
+    }
 }

src/main.rs

@@ -10,7 +10,7 @@ mod ui;
 use anyhow::Result;
 use app::App;
 use crossterm::{
-    event::{self, DisableMouseCapture, EnableMouseCapture, Event, KeyCode},
+    event::{self, DisableMouseCapture, EnableMouseCapture, Event, KeyCode, KeyModifiers},
     execute,
     terminal::{EnterAlternateScreen, LeaveAlternateScreen, disable_raw_mode, enable_raw_mode},
 };
@@ -95,14 +95,18 @@ async fn run_app<B: Backend>(terminal: &mut Terminal<B>, mut app: App) -> Result
                             app.start_inline_search();
                         }
                     }
+                    KeyCode::Char('f') if key.modifiers.contains(KeyModifiers::SHIFT) => {
+                        app.make_favorite(true).await?
+                    }
+                    KeyCode::Char('f') => app.make_favorite(false).await?,
                     KeyCode::Char('n') => app.play_next().await?,
                     KeyCode::Char('p') => app.play_previous().await?,
                     _ => {}
                 }
             }
         }
         app.update().await?;
-        interval(Duration::from_millis(100)).tick().await;
-        // tokio::time::sleep(tokio::time::Duration::from_millis(10)).await;
+        // interval(Duration::from_millis(100)).tick().await;
+        tokio::time::sleep(tokio::time::Duration::from_millis(100)).await;
     }
 }

src/subsonic.rs

@@ -2,6 +2,7 @@ use crate::app;
 use crate::app::{Album, Artist, Playlists, Track};
 use crate::config::Config;
 use anyhow::{Context, Ok, Result};
+use ratatui::widgets::LegendPosition;
 use serde::Deserialize;
 use url::Url;
 
@@ -346,6 +347,17 @@ impl SubsonicClient {
             .await?;
         Ok(())
     }
+    pub async fn favorite_a_song(&self, track: &Track, remove: bool) -> Result<()> {
+        #[derive(Deserialize)]
+        struct Empty {}
+
+        let mut endpoint = "star";
+        if remove {
+            endpoint = "unstar"
+        }
+        let _: Empty = self.get(endpoint, vec![("id", track.id.clone())]).await?;
+        Ok(())
+    }
     pub async fn get_songs_in_album(&self, album: &Album) -> Result<Vec<Track>> {
         let data: GetAlbumResponse = self.get("getAlbum", vec![("id", album.id.clone())]).await?;
         self.songs_to_tracks(data.album.song)