Add Zig grammar

1 files changed, 8 insertions, 2 deletions

diff --git a/main.c b/main.c
index b923cdc..5b4a80a 100644
--- a/main.c
+++ b/main.c
@@ -20,10 +20,11 @@
 #include "queries/cpp.h"
 #include "queries/go.h"
 #include "queries/javascript.h"
+#include "queries/lua.h"
 #include "queries/php.h"
 #include "queries/python.h"
 #include "queries/rust.h"
-#include "queries/lua.h"
+#include "queries/zig.h"
 
 int debug_enabled = 0;
 
@@ -35,6 +36,7 @@ TSLanguage *tree_sitter_php(void);
 TSLanguage *tree_sitter_rust(void);
 TSLanguage *tree_sitter_javascript(void);
 TSLanguage *tree_sitter_lua(void);
+TSLanguage *tree_sitter_zig(void);
 
 #define MIN(a, b) ((a) < (b) ? (a) : (b))
 
@@ -203,7 +205,7 @@ void parse_source_file(void *arg) {
 					// We treat it as a match, but result pointer logic is different
 					// For printing purposes effectively a match.
 					// We'll just set result to non-null to trigger the print.
-					result = (char *)fn.fname; 
+					result = (char *)fn.fname;
 				}
 			} else {
 				if (case_sensitive) {
@@ -345,6 +347,10 @@ int main(int argc, char *argv[]) {
 				lang = tree_sitter_lua();
 				query_string = (const char *)query_lua;
 				query_len = query_lua_len;
+			} else if (strcmp(extension, "zig") == 0) {
+				lang = tree_sitter_zig();
+				query_string = (const char *)query_zig;
+				query_len = query_zig_len;
 			}
 		}