1 files changed, 13 insertions, 10 deletions

diff --git a/src/notification.rs b/src/notification.rs
index 2f7baa2..4947e6f 100644
--- a/src/notification.rs
+++ b/src/notification.rs
@@ -1,15 +1,18 @@
 use crate::queue::Track;
 
-pub fn notify_track(track: &Track) {
-    let body = match (track.artist.as_deref(), track.title.as_deref()) {
-        (Some(a), Some(t)) => format!("{} - {}", a, t),
-        (_, Some(t)) => t.to_string(),
-        _ => track
-            .path
-            .file_name()
-            .and_then(|n| n.to_str())
-            .unwrap_or("Unknown")
-            .to_string(),
+pub fn notify_track(track: &Track, format: Option<&str>) {
+    let body = match format {
+        Some(fmt) => track.format(fmt),
+        None => match (track.artist.as_deref(), track.title.as_deref()) {
+            (Some(a), Some(t)) => format!("{} - {}", a, t),
+            (_, Some(t)) => t.to_string(),
+            _ => track
+                .path
+                .file_name()
+                .and_then(|n| n.to_str())
+                .unwrap_or("Unknown")
+                .to_string(),
+        },
     };
 
     let _ = notify_rust::Notification::new()