feat: Support query is_struct attribute in classes table

AmrDeveloper · May 1, 2024 · 83dd0cf · 83dd0cf
1 parent ed94c19
commit 83dd0cf
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 1 deletion.
diff --git a/src/data_provider.rs b/src/data_provider.rs
@@ -109,6 +109,11 @@ fn select_classes(
                 continue;
             }
 
+            if field_name == "is_struct" {
+                values.push(Value::Boolean(class.is_struct));
+                continue;
+            }
+
             if field_name == "file" {
                 values.push(Value::Text(class.location.file.to_string()));
                 continue;

diff --git a/src/schema.rs b/src/schema.rs
@@ -20,6 +20,7 @@ lazy_static! {
         map.insert("access_modifier", DataType::Integer);
         map.insert("is_variadic", DataType::Boolean);
         map.insert("is_volatile", DataType::Boolean);
+        map.insert("is_struct", DataType::Boolean);
 
         // Source code location columns
         map.insert("file", DataType::Text);
@@ -33,7 +34,10 @@ lazy_static! {
 lazy_static! {
     pub static ref TABLES_FIELDS_NAMES: HashMap<&'static str, Vec<&'static str>> = {
         let mut map = HashMap::new();
-        map.insert("classes", vec!["name", "line", "column", "offset"]);
+        map.insert(
+            "classes",
+            vec!["name", "is_struct", "line", "column", "offset"],
+        );
         map.insert(
             "functions",
             vec![

diff --git a/src/visitor/class.rs b/src/visitor/class.rs
@@ -10,6 +10,7 @@ use crate::visitor::location;
 
 pub struct ClassNode {
     pub name: String,
+    pub is_struct: bool,
     pub location: location::SourceLocation,
 }
 
@@ -60,8 +61,11 @@ extern "C" fn visit_children(
             let location = location::visit_source_location(cursor);
 
             let classes = &mut *(data as *mut Vec<ClassNode>);
+            let is_struct = cursor_kind == CXCursor_StructDecl;
+
             classes.push(ClassNode {
                 name: class_name.to_string(),
+                is_struct,
                 location,
             });