1 files changed, 85 insertions, 16 deletions

diff --git a/src/daemon.rs b/src/daemon.rs
index 4122c37..b0da698 100644
--- a/src/daemon.rs
+++ b/src/daemon.rs
@@ -149,6 +149,16 @@ impl Queue {
         self.update_shuffle_indices();
     }
 
+    fn insert_after_current(&mut self, path: PathBuf) {
+        let insert_idx = self.position + 1;
+        self.tracks.insert(insert_idx, Track::from_path(path));
+        self.update_shuffle_indices();
+    }
+
+    fn track_at(&self, idx: usize) -> Option<&Track> {
+        self.tracks.get(idx)
+    }
+
     fn set_shuffle(&mut self, mode: ShuffleMode) {
         self.shuffle = mode.clone();
         self.update_shuffle_indices();
@@ -545,30 +555,89 @@ fn process_command(
             }
         }
         _ if cmd.starts_with("play ") => {
-            let path_str = &cmd[5..].trim();
-            let path = std::path::PathBuf::from(path_str);
+            let rest = &cmd[5..].trim();
+            let (position, path_str): (&str, &str) = if let Some((pos, p)) = rest.split_once(' ') {
+                if pos == "now" || pos == "next" || pos == "last" {
+                    (pos, p)
+                } else {
+                    ("last", rest)
+                }
+            } else {
+                ("last", rest)
+            };
+            let path_str = path_str.to_string();
+
+            if path_str.is_empty() {
+                return "Usage: play [now|next|last] <path>\n".to_string();
+            }
+
+            let path = std::path::PathBuf::from(path_str.clone());
             if path.is_file() {
-                // Add to queue and play
                 if let Ok(mut q) = queue.lock() {
-                    q.add_track(path.clone());
+                    match position {
+                        "now" => {
+                            q.insert_after_current(path.clone());
+                            q.position += 1;
+                            if let Some(track) = q.track_at(q.position) {
+                                audio.play(track.path.clone());
+                            }
+                            format!("Playing now: {}\n", path_str)
+                        }
+                        "next" => {
+                            q.insert_after_current(path.clone());
+                            format!("Queued next: {}\n", path_str)
+                        }
+                        _ => {
+                            q.add_track(path.clone());
+                            audio.play(path);
+                            format!("Playing: {}\n", path_str)
+                        }
+                    }
+                } else {
+                    "Error accessing queue\n".to_string()
                 }
-                audio.play(path);
-                format!("Playing: {}\n", path_str)
             } else if path.is_dir() {
-                // Find all audio files in directory
                 match find_audio_in_dir(&path) {
                     Some(files) if !files.is_empty() => {
                         if let Ok(mut q) = queue.lock() {
-                            for file in &files {
-                                q.add_track(file.clone());
-                            }
-                            // If shuffle is on, shuffled_indices is already set
-                            // Play the first track from the shuffled list
-                            if let Some(track) = q.current() {
-                                audio.play(track.path.clone());
+                            match position {
+                                "now" => {
+                                    for file in &files {
+                                        q.insert_after_current(file.clone());
+                                    }
+                                    q.position += 1;
+                                    if let Some(track) = q.track_at(q.position) {
+                                        audio.play(track.path.clone());
+                                    }
+                                    format!(
+                                        "Playing now: {} tracks from {}\n",
+                                        files.len(),
+                                        path_str
+                                    )
+                                }
+                                "next" => {
+                                    for file in &files {
+                                        q.insert_after_current(file.clone());
+                                    }
+                                    format!(
+                                        "Queued next: {} tracks from {}\n",
+                                        files.len(),
+                                        path_str
+                                    )
+                                }
+                                _ => {
+                                    for file in &files {
+                                        q.add_track(file.clone());
+                                    }
+                                    if let Some(track) = q.current() {
+                                        audio.play(track.path.clone());
+                                    }
+                                    format!("Queued {} tracks from: {}\n", files.len(), path_str)
+                                }
                             }
+                        } else {
+                            "Error accessing queue\n".to_string()
                         }
-                        format!("Queued {} tracks from: {}\n", files.len(), path_str)
                     }
                     _ => format!("Error: No audio files found in '{}'\n", path_str),
                 }
@@ -621,7 +690,7 @@ fn process_command(
                     for (display_idx, &track_idx) in indices.iter().enumerate() {
                         if let Some(track) = q.tracks.get(track_idx) {
                             let track_str = track.display_string(display_idx);
-                            if display_idx == current_position {
+                            if track_idx == current_position {
                                 output.push_str(&format!("> {} (playing)\n", track_str));
                             } else {
                                 output.push_str(&format!("  {}\n", track_str));