1 files changed, 92 insertions, 0 deletions
diff --git a/src/browser.rs b/src/browser.rs
new file mode 100755
index 0000000..0ed94ff
--- /dev/null
+++ b/src/browser.rs
@@ -0,0 +1,92 @@
+use mpd::Query;
+
+use crate::{app::AppResult, connection::Connection};
+
+#[derive(Debug)]
+pub struct FileBrowser {
+    pub filetree: Vec<(String, String)>,
+    pub selected: usize,
+    pub prev_selected: usize,
+    pub path: String,
+    pub prev_path: String,
+}
+
+impl FileBrowser {
+    pub fn new() -> FileBrowser {
+        FileBrowser {
+            filetree: Vec::new(),
+            selected: 0,
+            prev_selected: 0,
+            path: ".".to_string(),
+            prev_path: ".".to_string(),
+        }
+    }
+
+    pub fn update_directory(&mut self, conn: &mut Connection) -> AppResult<()> {
+        self.filetree.clear();
+        self.filetree = conn
+            .conn
+            .listfiles(self.path.as_str())?
+            .into_iter()
+            .filter(|(f, _)| f == "directory" || f == "file")
+            .collect::<Vec<(String, String)>>();
+
+        Ok(())
+    }
+
+    // Go to next item in filetree
+    pub fn next(&mut self) {
+        // if self.selected < self.filetree.len() - 1 {
+        //     self.selected += 1;
+        // }
+
+        if self.selected == self.filetree.len() - 1 {
+            self.selected = 0;
+        } else {
+            self.selected += 1;
+        }
+    }
+
+    /// Go to previous item in filetree
+    pub fn prev(&mut self) {
+        if self.selected == 0 {
+            self.selected = self.filetree.len() - 1;
+        } else {
+            self.selected -= 1;
+        }
+    }
+
+    pub fn handle_enter(&mut self, conn: &mut Connection) -> AppResult<()> {
+        let (t, path) = self.filetree.get(self.selected).unwrap();
+        if t == "directory" {
+            if path != "." {
+                self.prev_path = self.path.clone();
+                self.path = path.to_string();
+                self.update_directory(conn)?;
+                self.prev_selected = self.selected;
+                self.selected = 0;
+            }
+        } else {
+            let list = conn
+                .songs_filenames
+                .iter()
+                .map(|f| f.as_str())
+                .collect::<Vec<&str>>();
+            let (filename, _) = rust_fuzzy_search::fuzzy_search_sorted(&path, &list)
+                .get(0)
+                .unwrap()
+                .clone();
+
+            let song = conn.get_song_with_only_filename(filename);
+            conn.push(&song)?;
+        }
+        Ok(())
+    }
+
+    pub fn handle_go_back(&mut self, conn: &mut Connection) -> AppResult<()> {
+        self.path = self.prev_path.clone();
+        self.selected = self.prev_selected;
+        self.update_directory(conn)?;
+        Ok(())
+    }
+}