fix: allow reading from special files (device, fifo, etc) (#886)

sigoden · Sep 25, 2024 · 16c5952 · 16c5952
1 parent 4686c47
commit 16c5952
Showing 1 changed file with 11 additions and 14 deletions.
diff --git a/src/utils/path.rs b/src/utils/path.rs
@@ -84,21 +84,18 @@ async fn list_files(
     if !entry_path.exists() {
         bail!("Not found: {}", entry_path.display());
     }
-    if entry_path.is_file() {
-        add_file(files, suffixes, entry_path);
-        return Ok(());
-    }
-    if !entry_path.is_dir() {
-        bail!("Not a directory: {:?}", entry_path);
-    }
-    let mut reader = tokio::fs::read_dir(entry_path).await?;
-    while let Some(entry) = reader.next_entry().await? {
-        let path = entry.path();
-        if path.is_file() {
-            add_file(files, suffixes, &path);
-        } else if path.is_dir() {
-            list_files(files, &path, suffixes).await?;
+    if entry_path.is_dir() {
+        let mut reader = tokio::fs::read_dir(entry_path).await?;
+        while let Some(entry) = reader.next_entry().await? {
+            let path = entry.path();
+            if path.is_dir() {
+                list_files(files, &path, suffixes).await?;
+            } else {
+                add_file(files, suffixes, &path);
+            }
         }
+    } else {
+        add_file(files, suffixes, entry_path);
     }
     Ok(())
 }