add basic file searching

Author: conaticus

src-tauri/Cargo.lock

@@ -17,6 +17,8 @@ tauri = { version = "1.3", features = ["shell-open"] }
 serde = { version = "1.0", features = ["derive"] }
 serde_json = "1.0"
 sysinfo = "0.29.2"
+walkdir = "2.3.3"
+fuzzy-matcher = "*"
 
 [features]
 # this feature is used for production builds or when `devPath` points to the filesystem

src-tauri/Cargo.toml

@@ -14,7 +14,7 @@ pub struct Drive {
     letter: String,
 }
 
-#[derive(Serialize)]
+#[derive(Serialize, Clone)]
 pub enum DirectoryChild {
     File(String), // String is the path to the file
     Directory(String)

src-tauri/src/file_explorer/mod.rs

@@ -1,10 +1,47 @@
+use std::ops::Deref;
+use fuzzy_matcher::clangd::fuzzy_match;
+use fuzzy_matcher::FuzzyMatcher;
 use crate::file_explorer::DirectoryChild;
+use walkdir::WalkDir;
+use crate::util::conversions::ostr_to_string;
+use fuzzy_matcher::skim::SkimMatcherV2;
 
 #[tauri::command]
 pub fn search_directory(query: String, search_directory: String, extension: String, accept_files: bool, accept_directories: bool) -> Vec<DirectoryChild> {
-    return walk(query, search_directory, extension, accept_files, accept_directories);
-}
+    let mut results = Vec::new();
+    let mut fuzzy_scores = Vec::new();
+    let matcher = SkimMatcherV2::default();
 
-fn walk(query: String, directory: String, extension: String, accept_files: bool, accept_directories: bool) -> Vec<DirectoryChild> {
-    Vec::new()
+    for entry in WalkDir::new(search_directory).into_iter().flatten() {
+        let file_name = entry.file_name().to_string_lossy();
+
+        if entry.file_type().is_file() {
+            if !accept_files { continue }
+
+            if extension.len() > 0 && !file_name.ends_with(extension.as_str()) { continue }
+
+
+            // Remove extension from file name.
+            let cleaned_filename = entry.path().file_stem().and_then(|stem| stem.to_str()).unwrap_or("");
+            let score = matcher.fuzzy_match(cleaned_filename, &query).unwrap_or(0);
+            if score < 20 { continue }
+
+            fuzzy_scores.push(score);
+            results.push(DirectoryChild::File(file_name.to_string()));
+
+            continue
+        } else if !accept_directories { continue }
+
+        let score = matcher.fuzzy_match(&file_name, &query).unwrap_or(0);
+        if score < 40 { continue }
+
+        results.push(DirectoryChild::Directory(file_name.to_string()));
+        fuzzy_scores.push(score);
+    }
+
+    let mut tuples: Vec<(usize, _)> = fuzzy_scores.iter().enumerate().collect();
+    tuples.sort_by(|a, b| b.1.cmp(a.1));
+    results = tuples.into_iter().map(|(index, _)| results[index].clone()).collect();
+
+    results
 }

src-tauri/src/file_explorer/search.rs

@@ -12,4 +12,4 @@ fn main() {
         .invoke_handler(tauri::generate_handler![get_disks, open_directory, search_directory])
         .run(tauri::generate_context!())
         .expect("error while running tauri application");
-}
+}

src-tauri/src/main.rs

@@ -40,7 +40,7 @@ export default function SearchBar({ currentDirectoryPath, setSearchResults }: Pr
 
     return (
         <div>
-            <Input value={searchValue} setValue={setSearchValue} placeholder={`Search ${currentPlace || "PC."}`} className="rounded-bl-none rounded-br-none" onSubmit={onSearch} size={InputSize.Large} />
+            <Input value={searchValue} setValue={setSearchValue} placeholder={`Search ${currentPlace || "PC"}`} className="rounded-bl-none rounded-br-none" onSubmit={onSearch} size={InputSize.Large} />
             <SearchFilter filters={searchFilter} setFilters={setSearchFilter} />
         </div>
     )