feat: Support selecting is_dir and is_file

Author: AmrDeveloper

README.md

@@ -44,12 +44,14 @@ SELECT DISTINCT parent AS folders FROM files
 
 ### Files table structure
 
-| Name      | Type    | Description           |
-| --------- | ------- | --------------------- |
-| path      | Text    | File path             |
-| parent    | Text    | File parent path      |
-| extension | Text    | Extension of the file |
-| size      | Integer | Size of the file      |
+| Name      | Type    | Description              |
+| --------- | ------- | ------------------------ |
+| path      | Text    | File path                |
+| parent    | Text    | File parent path         |
+| extension | Text    | Extension of the file    |
+| is_dir    | Boolean | True if it's a directory |
+| is_file   | Boolean | True if it's a file      |
+| size      | Integer | Size of the file         |
 
 ---
 

src/data_provider.rs

@@ -62,6 +62,7 @@ impl DataProvider for FileDataProvider {
 
         for file in files {
             let mut values: Vec<Value> = vec![];
+            let path = Path::new(&file);
 
             for index in 0..names_len {
                 let field_name = &fields_names[index as usize];
@@ -76,14 +77,12 @@ impl DataProvider for FileDataProvider {
                 }
 
                 if field_name == "path" {
-                    let path = Path::new(&file);
                     let file_path_string = path.to_str().unwrap_or("");
                     values.push(Value::Text(file_path_string.to_string()));
                     continue;
                 }
 
                 if field_name == "parent" {
-                    let path = Path::new(&file);
                     let parent_path = if let Some(parent) = path.parent() {
                         parent.to_str().unwrap_or("")
                     } else {
@@ -94,12 +93,21 @@ impl DataProvider for FileDataProvider {
                 }
 
                 if field_name == "extension" {
-                    let path = Path::new(&file);
                     let extension = path.extension().and_then(|ext| ext.to_str()).unwrap_or("");
                     values.push(Value::Text(extension.to_string()));
                     continue;
                 }
 
+                if field_name == "is_dir" {
+                    values.push(Value::Boolean(path.is_dir()));
+                    continue;
+                }
+
+                if field_name == "is_file" {
+                    values.push(Value::Boolean(path.is_file()));
+                    continue;
+                }
+
                 if field_name == "size" {
                     let file_size = if let Ok(meta_data) = std::fs::metadata(&file) {
                         meta_data.len() as i64

src/schema.rs

@@ -8,6 +8,8 @@ lazy_static! {
         map.insert("path", DataType::Text);
         map.insert("parent", DataType::Text);
         map.insert("extension", DataType::Text);
+        map.insert("is_dir", DataType::Boolean);
+        map.insert("is_file", DataType::Boolean);
         map.insert("size", DataType::Integer);
         map
     };
@@ -16,7 +18,10 @@ lazy_static! {
 lazy_static! {
     pub static ref TABLES_FIELDS_NAMES: HashMap<&'static str, Vec<&'static str>> = {
         let mut map = HashMap::new();
-        map.insert("files", vec!["path", "parent", "extension", "size"]);
+        map.insert(
+            "files",
+            vec!["path", "parent", "extension", "is_dir", "is_file", "size"],
+        );
         map
     };
 }