1 files changed, 16 insertions, 0 deletions

diff --git a/src/queue.rs b/src/queue.rs
index 2c20356..d42203f 100644
--- a/src/queue.rs
+++ b/src/queue.rs
@@ -117,6 +117,22 @@ impl Track {
             format!("{:0width$}. {} - {}", index + 1, album_artist, title)
         }
     }
+
+    pub fn format(&self, format_str: &str) -> String {
+        let artist = self.artist.clone().unwrap_or_default();
+        let album = self.album.clone().unwrap_or_default();
+        let title = self.title.clone().unwrap_or_else(|| {
+            self.path
+                .file_name()
+                .and_then(|n| n.to_str())
+                .unwrap_or("Unknown")
+                .to_string()
+        });
+        format_str
+            .replace("$ARTIST", &artist)
+            .replace("$ALBUM", &album)
+            .replace("$TRACK", &title)
+    }
 }
 
 pub struct Queue {