chore: refactor nodetype labels

Author: fayekelmith

ast/src/lang/asg.rs

@@ -297,6 +297,34 @@ impl FromStr for NodeType {
         }
     }
 }
+impl NodeType {
+    pub fn all_labels() -> &'static [&'static str] {
+        &[
+            "Repository",
+            "Package",
+            "Language",
+            "Directory",
+            "File",
+            "Import",
+            "Library",
+            "Class",
+            "Trait",
+            "Instance",
+            "Function",
+            "Endpoint",
+            "Request",
+            "Datamodel",
+            "Feature",
+            "Page",
+            "Var",
+            "UnitTest",
+            "IntegrationTest",
+            "E2etest",
+            "Mock",
+        ]
+    }
+}
+
 impl std::fmt::Display for NodeType {
     fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
         let s = match self {

ast/src/lang/graphs/neo4j/queries/edges.rs

@@ -292,30 +292,7 @@ pub fn find_dynamic_edges_for_file_query(file: &str) -> (String, BoltMap) {
     let mut params = BoltMap::new();
     boltmap_insert_str(&mut params, "file", file);
 
-    let static_types = vec![
-        "Repository",
-        "Package",
-        "Language",
-        "Directory",
-        "File",
-        "Import",
-        "Library",
-        "Class",
-        "Trait",
-        "Instance",
-        "Function",
-        "Endpoint",
-        "Request",
-        "Datamodel",
-        "Feature",
-        "Page",
-        "Var",
-        "UnitTest",
-        "IntegrationTest",
-        "E2etest",
-    ];
-
-    let static_labels = static_types
+    let static_labels = NodeType::all_labels()
         .iter()
         .map(|t| format!("source:{}", t))
         .collect::<Vec<_>>()
@@ -336,30 +313,7 @@ pub fn find_dynamic_edges_for_file_query(file: &str) -> (String, BoltMap) {
 pub fn find_all_dynamic_edges_query() -> (String, BoltMap) {
     let params = BoltMap::new();
 
-    let static_types = vec![
-        "Repository",
-        "Package",
-        "Language",
-        "Directory",
-        "File",
-        "Import",
-        "Library",
-        "Class",
-        "Trait",
-        "Instance",
-        "Function",
-        "Endpoint",
-        "Request",
-        "Datamodel",
-        "Feature",
-        "Page",
-        "Var",
-        "UnitTest",
-        "IntegrationTest",
-        "E2etest",
-    ];
-
-    let static_labels = static_types
+    let static_labels = NodeType::all_labels()
         .iter()
         .map(|t| format!("source:{}", t))
         .collect::<Vec<_>>()

ast/src/lang/graphs/neo4j/queries/nodes.rs

@@ -500,12 +500,21 @@ pub fn get_muted_nodes_for_files_query(files: &[String]) -> (String, BoltMap) {
         .collect::<Vec<_>>();
     boltmap_insert_list(&mut params, "files", files_list);
 
-    let query = "MATCH (n:Data_Bank) 
+    let labels_list = NodeType::all_labels()
+        .iter()
+        .map(|t| format!("'{}'", t))
+        .collect::<Vec<_>>()
+        .join(", ");
+
+    let query = format!(
+        "MATCH (n:Data_Bank) 
                  WHERE n.file IN $files
                  AND (n.is_muted = true OR n.is_muted = 'true')
-                 WITH n, [label IN labels(n) WHERE label IN ['Function', 'Class', 'DataModel', 'Endpoint', 'Request', 'File', 'Directory', 'Repository', 'Language', 'Library', 'Import', 'Instance', 'Page', 'Var', 'UnitTest', 'IntegrationTest', 'E2eTest', 'Trait']][0] as node_type
+                 WITH n, [label IN labels(n) WHERE label IN [{}]][0] as node_type
                  WHERE node_type IS NOT NULL
-                 RETURN node_type, n.name as name, n.file as file".to_string();
+                 RETURN node_type, n.name as name, n.file as file",
+        labels_list
+    );
 
     (query, params)
 }