feat(repo-map): zig support (yetone#663)

* feature: zig support for repo map * Update crates/avante-repo-map/Cargo.toml Co-authored-by: yetone <yetoneful@gmail.com> * fix: update lint error Signed-off-by: Aaron Pham <contact@aarnphm.xyz> --------- Signed-off-by: Aaron Pham <contact@aarnphm.xyz> Co-authored-by: Aaron Pham <Aaronpham0103@gmail.com> Co-authored-by: yetone <yetoneful@gmail.com> Co-authored-by: Aaron Pham <contact@aarnphm.xyz>
Hein00 · Sep 29, 2024 · bac46ce · bac46ce
1 parent d28fece
commit bac46ce
Show file tree

Hide file tree

Showing 4 changed files with 288 additions and 7 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/crates/avante-repo-map/Cargo.toml b/crates/avante-repo-map/Cargo.toml
@@ -26,6 +26,7 @@ tree-sitter-c = "0.23"
 tree-sitter-cpp = "0.23"
 tree-sitter-lua = "0.2"
 tree-sitter-ruby = "0.23"
+tree-sitter-zig = "1.0.2"
 
 [lints]
 workspace = true

diff --git a/crates/avante-repo-map/queries/tree-sitter-zig-defs.scm b/crates/avante-repo-map/queries/tree-sitter-zig-defs.scm
@@ -0,0 +1,23 @@
+ ;; Capture functions, structs, methods, variable definitions, and unions in Zig
+(variable_declaration (identifier)
+  (struct_declaration
+        (container_field) @class_variable))
+
+(variable_declaration (identifier)
+  (struct_declaration
+        (function_declaration
+            name: (identifier) @method)))
+
+(variable_declaration (identifier)
+  (enum_declaration
+    (container_field
+      type: (identifier) @enum_item)))
+
+(variable_declaration (identifier)
+  (union_declaration
+    (container_field
+      name: (identifier) @union_item)))
+
+(source_file (function_declaration) @function)
+
+(source_file (variable_declaration (identifier) @variable))