1 files changed, 54 insertions, 2 deletions

diff --git a/src/queue.rs b/src/queue.rs
index 17f2143..c985755 100644
--- a/src/queue.rs
+++ b/src/queue.rs
@@ -8,9 +8,11 @@
 use rand::SeedableRng;
 use rand::rngs::StdRng;
 use rand::seq::SliceRandom;
+use serde::{Deserialize, Serialize};
 use std::path::{Path, PathBuf};
 
-#[derive(Debug, Clone, PartialEq)]
+#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash, Serialize, Deserialize)]
+#[serde(rename_all = "lowercase")]
 pub enum ShuffleMode {
     None,
     All,
@@ -94,7 +96,7 @@ pub fn extract_metadata(path: &Path) -> Option<AudioMetadata> {
     }
 }
 
-#[derive(Debug, Clone)]
+#[derive(Debug, Clone, Serialize, Deserialize)]
 pub struct Track {
     pub path: PathBuf,
     pub title: Option<String>,
@@ -199,6 +201,18 @@ impl Track {
     }
 }
 
+/// A queue captured for persistence.
+///
+/// `shuffled_indices` carries the current play order, so restoring a shuffled
+/// queue keeps the order it was played in instead of reshuffling the tail.
+#[derive(Debug, Clone, Serialize, Deserialize)]
+pub struct QueueSnapshot {
+    pub tracks: Vec<Track>,
+    pub position: usize,
+    pub shuffle: ShuffleMode,
+    pub shuffled_indices: Vec<usize>,
+}
+
 pub struct Queue {
     tracks: Vec<Track>,
     position: usize,
@@ -591,4 +605,42 @@ impl Queue {
         self.position = 0;
         self.shuffled_indices.clear();
     }
+
+    /// Everything needed to rebuild this queue exactly.
+    pub fn snapshot(&self) -> QueueSnapshot {
+        QueueSnapshot {
+            tracks: self.tracks.clone(),
+            position: self.position,
+            shuffle: self.shuffle,
+            shuffled_indices: self.shuffled_indices.clone(),
+        }
+    }
+
+    pub fn from_snapshot(snapshot: QueueSnapshot) -> Self {
+        let mut queue = Queue::new();
+        queue.tracks = snapshot.tracks;
+        queue.shuffle = snapshot.shuffle;
+        queue.position = if queue.tracks.is_empty() {
+            0
+        } else {
+            snapshot.position.min(queue.tracks.len() - 1)
+        };
+        // Trust the saved play order only when it still describes this queue;
+        // otherwise derive a new one instead of playing stale indices.
+        if queue.shuffle != ShuffleMode::None
+            && snapshot.shuffled_indices.len() == queue.tracks.len()
+            && Self::is_permutation(&snapshot.shuffled_indices)
+        {
+            queue.shuffled_indices = snapshot.shuffled_indices;
+        } else {
+            queue.update_shuffle_indices();
+        }
+        queue
+    }
+
+    fn is_permutation(indices: &[usize]) -> bool {
+        let mut sorted = indices.to_vec();
+        sorted.sort_unstable();
+        sorted.iter().enumerate().all(|(i, &idx)| i == idx)
+    }
 }