1 files changed, 13 insertions, 2 deletions

diff --git a/src/cli.rs b/src/cli.rs
index 05cd61f..f5fdd2c 100644
--- a/src/cli.rs
+++ b/src/cli.rs
@@ -35,6 +35,9 @@ pub enum Commands {
         /// Add to queue after current track (don't skip)
         #[arg(short = 'n', long = "next")]
         next: bool,
+        /// Queue rest of album after this track (requires a track query)
+        #[arg(short = 'a', long = "album")]
+        album: bool,
         #[arg(value_name = "QUERY")]
         query: Option<String>,
     },
@@ -129,7 +132,11 @@ impl Cli {
                 crate::daemon::restart_daemon()?;
             }
             Commands::Play {
-                now, next, query, ..
+                now,
+                next,
+                album,
+                query,
+                ..
             } => {
                 if let Some(q) = query {
                     let pos = if *now {
@@ -139,7 +146,11 @@ impl Cli {
                     } else {
                         "last"
                     };
-                    crate::daemon::send_command(format!("play {} {}", pos, q))?;
+                    if *album {
+                        crate::daemon::send_command(format!("play-album {} {}", pos, q))?;
+                    } else {
+                        crate::daemon::send_command(format!("play {} {}", pos, q))?;
+                    }
                 } else {
                     crate::daemon::send_command("play last".to_string())?;
                 }