feat: Implement FileQL DSL files_count function

Author: AmrDeveloper

CHANGELOG.md

@@ -1,6 +1,12 @@
 Change Log
 ==========
 
+Version 0.5.0 *(2024-06-13)*
+-----------------------------
+
+* Upgrade to GitQL SDk 0.23.0
+* Implement `files_count` function.
+
 Version 0.4.0 *(2024-06-04)*
 -----------------------------
 

README.md

@@ -38,6 +38,7 @@ SELECT "File Query Language" LIKE "%Query%"
 SELECT * FROM files
 SELECT COUNT(path) from files
 SELECT DISTINCT parent AS folders FROM files
+SELECT CHILDREN_COUNT(parent) FROM files
 ```
 
 ---
@@ -55,6 +56,14 @@ SELECT DISTINCT parent AS folders FROM files
 
 ---
 
+### Files QL Functions
+
+| Name        | Parameters | Return  | Description                                                  |
+| ----------- | ---------- | ------- | ------------------------------------------------------------ |
+| FILES_COUNT | Text       | Integer | Number of children for directory or 0 if it's not valid path |
+
+---
+
 ### Download or Install
 
 - Install from Cargo.io

src/functions.rs

@@ -0,0 +1,43 @@
+use gitql_core::signature::Function;
+use gitql_core::signature::Signature;
+use gitql_core::types::DataType;
+use gitql_core::value::Value;
+use gitql_std::function::standard_function_signatures;
+use gitql_std::function::standard_functions;
+use std::collections::HashMap;
+use std::sync::OnceLock;
+
+pub fn fileql_std_functions() -> &'static HashMap<&'static str, Function> {
+    static HASHMAP: OnceLock<HashMap<&'static str, Function>> = OnceLock::new();
+    HASHMAP.get_or_init(|| {
+        let mut map: HashMap<&'static str, Function> =
+            HashMap::from(standard_functions().to_owned());
+        map.insert("files_count", files_count);
+        map
+    })
+}
+
+pub fn fileql_std_signatures() -> &'static HashMap<&'static str, Signature> {
+    static HASHMAP: OnceLock<HashMap<&'static str, Signature>> = OnceLock::new();
+    HASHMAP.get_or_init(|| {
+        let mut map: HashMap<&'static str, Signature> =
+            HashMap::from(standard_function_signatures().to_owned());
+        map.insert(
+            "files_count",
+            Signature {
+                parameters: vec![DataType::Text],
+                return_type: DataType::Integer,
+            },
+        );
+        map
+    })
+}
+
+fn files_count(values: &[Value]) -> Value {
+    let path = values[0].as_text();
+    if let Ok(entries) = std::fs::read_dir(path) {
+        let count = entries.flatten().count();
+        return Value::Integer(count as i64);
+    }
+    Value::Integer(0)
+}

src/main.rs

@@ -18,13 +18,12 @@ use gitql_parser::parser;
 use gitql_parser::tokenizer;
 use gitql_std::aggregation::aggregation_function_signatures;
 use gitql_std::aggregation::aggregation_functions;
-use gitql_std::function::standard_function_signatures;
-use gitql_std::function::standard_functions;
 use schema::tables_fields_names;
 use schema::tables_fields_types;
 
 mod arguments;
 mod data_provider;
+mod functions;
 mod schema;
 
 fn main() {
@@ -49,8 +48,8 @@ fn main() {
                 tables_fields_types: tables_fields_types().to_owned(),
             };
 
-            let std_signatures = standard_function_signatures();
-            let std_functions = standard_functions();
+            let std_signatures = functions::fileql_std_signatures();
+            let std_functions = functions::fileql_std_functions();
 
             let aggregation_signatures = aggregation_function_signatures();
             let aggregation_functions = aggregation_functions();
@@ -86,8 +85,8 @@ fn launch_fileql_repl(arguments: Arguments) {
         tables_fields_types: tables_fields_types().to_owned(),
     };
 
-    let std_signatures = standard_function_signatures();
-    let std_functions = standard_functions();
+    let std_signatures = functions::fileql_std_signatures();
+    let std_functions = functions::fileql_std_functions();
 
     let aggregation_signatures = aggregation_function_signatures();
     let aggregation_functions = aggregation_functions();