add metadata in directory tree

1 files changed, 57 insertions, 1 deletions

diff --git a/src/browser.rs b/src/browser.rs
index 6af4f59..73f9cdd 100755
--- a/src/browser.rs
+++ b/src/browser.rs
@@ -1,3 +1,8 @@
+use std::ffi::OsStr;
+use std::path::Path;
+
+use mpd::Song;
+
 use crate::{app::AppResult, connection::Connection};
 
 #[derive(Debug)]
@@ -7,6 +12,24 @@ pub struct FileBrowser {
     pub prev_selected: usize,
     pub path: String,
     pub prev_path: String,
+    pub songs: Vec<Song>,
+}
+
+// https://stackoverflow.com/questions/72392835/check-if-a-file-is-of-a-given-type
+pub trait FileExtension {
+    fn has_extension<S: AsRef<str>>(&self, extensions: &[S]) -> bool;
+}
+
+impl<P: AsRef<Path>> FileExtension for P {
+    fn has_extension<S: AsRef<str>>(&self, extensions: &[S]) -> bool {
+        if let Some(ref extension) = self.as_ref().extension().and_then(OsStr::to_str) {
+            return extensions
+                .iter()
+                .any(|x| x.as_ref().eq_ignore_ascii_case(extension));
+        }
+
+        false
+    }
 }
 
 impl FileBrowser {
@@ -17,6 +40,7 @@ impl FileBrowser {
             prev_selected: 0,
             path: ".".to_string(),
             prev_path: ".".to_string(),
+            songs: vec![],
         }
     }
 
@@ -26,9 +50,41 @@ impl FileBrowser {
             .conn
             .listfiles(self.path.as_str())?
             .into_iter()
-            .filter(|(f, _)| f == "directory" || f == "file")
+            .filter(|(f, l)| {
+                f == "directory"
+                    || f == "file" && Path::new(l).has_extension(&["mp3", "ogg", "flac", "m4a", "wav", "aac" ,"opus", "ape", "wma", "mpc", "aiff", "dff", "mp2", "mka"])
+            })
             .collect::<Vec<(String, String)>>();
 
+        self.songs.clear();
+        for (t, song) in self.filetree.iter() {
+            if t == "file" {
+                let v = conn
+                    .conn
+                    .lsinfo(Song {
+                        file: conn
+                            .get_full_path(song)
+                            .unwrap_or_else(|| "Not a song".to_string()),
+                        ..Default::default()
+                    })
+                    .unwrap_or_else(|_| {
+                        vec![Song {
+                            file: "Not a song".to_string(),
+                            ..Default::default()
+                        }]
+                    });
+
+                self.songs.push(v.get(0).unwrap().clone());
+            } else {
+                let v = Song {
+                    file: "".to_string(),
+                    ..Default::default()
+                };
+
+                self.songs.push(v);
+            }
+        }
+
         Ok(())
     }