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;
                         }
                 }