fix(mpd): stops working if connection lost

The client will now attempt to reconnect when a connection loss is detected.

Fixes #21.

src/clients/music/mod.rs

@@ -7,7 +7,11 @@ use tokio::sync::broadcast;
 pub mod mpd;
 pub mod mpris;
 
-pub type PlayerUpdate = (Option<Track>, Status);
+#[derive(Clone, Debug)]
+pub enum PlayerUpdate {
+    Update(Box<Option<Track>>, Status),
+    Disconnect,
+}
 
 #[derive(Clone, Debug)]
 pub struct Track {

src/clients/music/mpd.rs

@@ -19,7 +19,7 @@ use tokio::spawn;
 use tokio::sync::broadcast::{channel, error::SendError, Receiver, Sender};
 use tokio::sync::Mutex;
 use tokio::time::sleep;
-use tracing::{debug, error};
+use tracing::{debug, error, info};
 
 lazy_static! {
     static ref CONNECTIONS: Arc<Mutex<HashMap<String, Arc<MpdClient>>>> =
@@ -72,7 +72,9 @@ impl MpdClient {
                         Subsystem::Player | Subsystem::Queue | Subsystem::Mixer,
                     ) = change
                     {
-                        Self::send_update(&client, &tx, &music_dir).await?;
+                        Self::send_update(&client, &tx, &music_dir)
+                            .await
+                            .expect("Failed to send update");
                     }
                 }
 
@@ -92,7 +94,7 @@ impl MpdClient {
         client: &Client,
         tx: &Sender<PlayerUpdate>,
         music_dir: &Path,
-    ) -> Result<(), SendError<(Option<Track>, Status)>> {
+    ) -> Result<(), SendError<PlayerUpdate>> {
         let current_song = client.command(commands::CurrentSong).await;
         let status = client.command(commands::Status).await;
 
@@ -100,12 +102,22 @@ impl MpdClient {
             let track = current_song.map(|s| Self::convert_song(&s.song, music_dir));
             let status = Status::from(status);
 
-            tx.send((track, status))?;
+            tx.send(PlayerUpdate::Update(Box::new(track), status))?;
         }
 
         Ok(())
     }
 
+    fn is_connected(&self) -> bool {
+        !self.client.is_connection_closed()
+    }
+
+    fn send_disconnect_update(&self) -> Result<(), SendError<PlayerUpdate>> {
+        info!("Connection to MPD server lost");
+        self.tx.send(PlayerUpdate::Disconnect)?;
+        Ok(())
+    }
+
     fn convert_song(song: &Song, music_dir: &Path) -> Track {
         let (track, disc) = song.number();
 
@@ -189,7 +201,20 @@ pub async fn get_client(
             connections.insert(host.to_string(), Arc::clone(&client));
             Ok(client)
         }
-        Some(client) => Ok(Arc::clone(client)),
+        Some(client) => {
+            if client.is_connected() {
+                Ok(Arc::clone(client))
+            } else {
+                client
+                    .send_disconnect_update()
+                    .expect("Failed to send disconnect update");
+
+                let client = MpdClient::new(host, music_dir).await?;
+                let client = Arc::new(client);
+                connections.insert(host.to_string(), Arc::clone(&client));
+                Ok(client)
+            }
+        }
     }
 }
 

src/clients/music/mpris.rs

@@ -170,7 +170,7 @@ impl Client {
 
         let track = Track::from(metadata);
 
-        let player_update: PlayerUpdate = (Some(track), status);
+        let player_update = PlayerUpdate::Update(Box::new(Some(track)), status);
 
         tx.send(player_update)
             .expect("Failed to send player update");