Something went wrong. Try again.
(MIRROR) Grabs zig bindings for tree-sitter git.gae.moe/grace/eazy-zig-tree-sitter
zig tree-sitter
Something went wrong. Try again.
11 kB · 279 lines
Zig
at main
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280const std = @import("std");const testing = std.testing;const ts = @import("tree-sitter");const ezts = @import("eazy_zig_tree_sitter");const grammars = @import("grammars");const queries_mod = @import("queries");
const Entry = struct { key: []const u8, source: []const u8 };
fn runParseTest(entries: []const Entry) !void { for (entries) |e| { const lang = ezts.getLang(e.key) orelse @panic("unknown lang"); defer lang.destroy(); var parser = ts.Parser.create(); defer parser.destroy(); try parser.setLanguage(lang); const tree = parser.parseString(e.source, null) orelse { std.debug.print("PARSE FAIL: {s}\n", .{e.key}); return error.ParseFailed; }; defer tree.destroy(); }}
fn runQueryTest(entries: []const Entry) !void { for (entries) |e| { const lang = ezts.getLang(e.key) orelse @panic("unknown lang"); defer lang.destroy(); var parser = ts.Parser.create(); defer parser.destroy(); try parser.setLanguage(lang); const tree = parser.parseString(e.source, null) orelse continue; defer tree.destroy(); const query_src = ezts.getQuery(e.key, "highlights") orelse { std.debug.print("NO QUERY: {s}\n", .{e.key}); continue; }; var error_offset: u32 = 0; const query = ts.Query.create(lang, query_src, &error_offset) catch |err| { std.debug.print("QUERY ERROR {s}: {s} at {d}\n", .{ e.key, @errorName(err), error_offset }); return error.QueryFailed; }; defer query.destroy(); const cursor = ts.QueryCursor.create(); defer cursor.destroy(); cursor.exec(query, tree.rootNode()); _ = cursor.nextCapture(); }}
const parse_entries: []const Entry = &.{ .{ .key = "c", .source = "int a;" }, .{ .key = "md", .source = "# hi" }, .{ .key = "md_inline", .source = "**hi**" }, .{ .key = "lua", .source = "x=1" }, .{ .key = "cuda", .source = "int a;" }, .{ .key = "glsl", .source = "int a;" }, .{ .key = "diff", .source = "+a" }, .{ .key = "ungrammar", .source = "A = 'a'" }, .{ .key = "thrift", .source = "struct A{}" }, .{ .key = "meson", .source = "x=1" }, .{ .key = "gitattributes", .source = "* text" }, .{ .key = "pymanifest", .source = "" }, .{ .key = "poe_filter", .source = "" }, .{ .key = "chatito", .source = "" }, .{ .key = "ino", .source = "void setup(){}" }, .{ .key = "capnp", .source = "struct A{}" }, .{ .key = "kdl", .source = "node1" }, .{ .key = "func", .source = "" }, .{ .key = "gosum", .source = "" }, .{ .key = "ron", .source = "()" }, .{ .key = "po", .source = "" }, .{ .key = "starlark", .source = "x=1" }, .{ .key = "yuck", .source = "" }, .{ .key = "smali", .source = ".class public A" }, .{ .key = "bicep", .source = "" }, .{ .key = "cpon", .source = "" }, .{ .key = "firrtl", .source = "" }, .{ .key = "luap", .source = "" }, .{ .key = "squirrel", .source = "x=1" }, .{ .key = "uxntal", .source = "" }, .{ .key = "cairo", .source = "func main(){}" }, .{ .key = "odin", .source = "main :: proc(){}" }, .{ .key = "hare", .source = "export fn main() void;" }, .{ .key = "pony", .source = "actor Main" }, .{ .key = "tablegen", .source = "" }, .{ .key = "luadoc", .source = "" }, .{ .key = "puppet", .source = "$a=1" }, .{ .key = "luau", .source = "x=1" }, .{ .key = "objc", .source = "int a;" }, .{ .key = "ispc", .source = "int a;" }, .{ .key = "pem", .source = "" }, .{ .key = "requirements", .source = "a" }, .{ .key = "gpg", .source = "" }, .{ .key = "bb", .source = "A = \"1\"" }, .{ .key = "csv", .source = "a,b" }, .{ .key = "psv", .source = "a|b" }, .{ .key = "tsv", .source = "a\tb" }, .{ .key = "printf", .source = "" }, .{ .key = "re2c", .source = "" }, .{ .key = "doxygen", .source = "" }, .{ .key = "ssh_config", .source = "Host *" }, .{ .key = "nqc", .source = "" }, .{ .key = "kconfig", .source = "mainmenu \"x\"" }, .{ .key = "gn", .source = "executable(\"a\") {}" }, .{ .key = "udev", .source = "" }, .{ .key = "xcompose", .source = "" }, .{ .key = "linkerscript", .source = "" }, .{ .key = "properties", .source = "key=value" }, .{ .key = "readline", .source = "" }, .{ .key = "Makefile", .source = "x=a" }, .{ .key = "yaml", .source = "a: 1" }, .{ .key = "toml", .source = "a=1" }, .{ .key = "svelte", .source = "" }, .{ .key = "scss", .source = "$a:1" }, .{ .key = "cst", .source = "" }, .{ .key = "jl", .source = "x=1" }, .{ .key = "haskell", .source = "x=1" }, .{ .key = "cyberchef", .source = "" }, .{ .key = "js", .source = "1" }, .{ .key = "json", .source = "1" }, .{ .key = "cpp", .source = "int a;" }, .{ .key = "rb", .source = "x=1" }, .{ .key = "go", .source = "package p" }, .{ .key = "cs", .source = "class A{}" }, .{ .key = "py", .source = "x=1" }, .{ .key = "rs", .source = "fn f(){}" }, .{ .key = "sh", .source = "true" }, .{ .key = "java", .source = "class A{}" }, .{ .key = "scala", .source = "class A" }, .{ .key = "agda", .source = "" }, .{ .key = "html", .source = "<p>" }, .{ .key = "embedded_template", .source = "" }, .{ .key = "regex", .source = "a" }, .{ .key = "css", .source = "a{}" }, .{ .key = "jsdoc", .source = "/** */" }, .{ .key = "ql", .source = "" }, .{ .key = "query", .source = "" }, .{ .key = "lisp", .source = "()" }, .{ .key = "tf", .source = "" }, .{ .key = "hcl", .source = "" }, .{ .key = "hlsl", .source = "int a;" }, .{ .key = "tcl", .source = "" }, .{ .key = "chatl", .source = "" }, .{ .key = "wgsl_bevy", .source = "" }, .{ .key = "move", .source = "" }, .{ .key = "kt", .source = "fun f(){}" }, .{ .key = "vim", .source = "" }, .{ .key = "hyprlang", .source = "" }, .{ .key = "gstlaunch", .source = "" }, .{ .key = "slang", .source = "" }, .{ .key = "dtd", .source = "" }, .{ .key = "xml", .source = "<a/>" }, .{ .key = "core_schema", .source = "" }, .{ .key = "json_schema", .source = "" }, .{ .key = "legacy_schema", .source = "" }, .{ .key = "vue", .source = "" }, .{ .key = "qmldir", .source = "" }, .{ .key = "zig", .source = "const a=1;" }, .{ .key = "test", .source = "" }, .{ .key = "v", .source = "" }, .{ .key = "tsx", .source = "<a/>" }, .{ .key = "ts", .source = "let a=1" }, .{ .key = "php", .source = "<?php" }, .{ .key = "php_only", .source = "<?php" }, .{ .key = "mli", .source = "" }, .{ .key = "ml", .source = "let a=1" }, .{ .key = "mly", .source = "" }, .{ .key = "fluent", .source = "" }, .{ .key = "tsq", .source = "" }, .{ .key = "ql_dbscheme", .source = "" },};
const query_entries: []const Entry = &.{ .{ .key = "c", .source = "int a;" }, .{ .key = "md", .source = "# hi" }, .{ .key = "md_inline", .source = "**hi**" }, .{ .key = "lua", .source = "x=1" }, .{ .key = "cuda", .source = "int a;" }, .{ .key = "glsl", .source = "int a;" }, .{ .key = "diff", .source = "+a" }, .{ .key = "ungrammar", .source = "A = 'a'" }, .{ .key = "thrift", .source = "struct A{}" }, .{ .key = "meson", .source = "x=1" }, .{ .key = "gitattributes", .source = "* text" }, .{ .key = "pymanifest", .source = "" }, .{ .key = "poe_filter", .source = "" }, .{ .key = "chatito", .source = "" }, .{ .key = "ino", .source = "void setup(){}" }, .{ .key = "capnp", .source = "struct A{}" }, .{ .key = "kdl", .source = "node1" }, .{ .key = "func", .source = "" }, .{ .key = "gosum", .source = "" }, .{ .key = "ron", .source = "()" }, .{ .key = "po", .source = "" }, .{ .key = "starlark", .source = "x=1" }, .{ .key = "yuck", .source = "" }, .{ .key = "smali", .source = ".class public A" }, .{ .key = "bicep", .source = "" }, .{ .key = "cpon", .source = "" }, .{ .key = "firrtl", .source = "" }, .{ .key = "luap", .source = "" }, .{ .key = "squirrel", .source = "x=1" }, .{ .key = "uxntal", .source = "" }, .{ .key = "cairo", .source = "func main(){}" }, .{ .key = "odin", .source = "main :: proc(){}" }, .{ .key = "hare", .source = "export fn main() void;" }, .{ .key = "pony", .source = "actor Main" }, .{ .key = "tablegen", .source = "" }, .{ .key = "luadoc", .source = "" }, .{ .key = "puppet", .source = "$a=1" }, .{ .key = "luau", .source = "x=1" }, .{ .key = "objc", .source = "int a;" }, .{ .key = "ispc", .source = "int a;" }, .{ .key = "pem", .source = "" }, .{ .key = "requirements", .source = "a" }, .{ .key = "gpg", .source = "" }, .{ .key = "bb", .source = "A = \"1\"" }, .{ .key = "csv", .source = "a,b" }, .{ .key = "psv", .source = "a|b" }, .{ .key = "tsv", .source = "a\tb" }, .{ .key = "printf", .source = "" }, .{ .key = "re2c", .source = "" }, .{ .key = "doxygen", .source = "" }, .{ .key = "ssh_config", .source = "Host *" }, .{ .key = "nqc", .source = "" }, .{ .key = "kconfig", .source = "mainmenu \"x\"" }, .{ .key = "gn", .source = "executable(\"a\") {}" }, .{ .key = "udev", .source = "" }, .{ .key = "xcompose", .source = "" }, .{ .key = "linkerscript", .source = "" }, .{ .key = "properties", .source = "key=value" }, .{ .key = "readline", .source = "" }, .{ .key = "Makefile", .source = "x=a" }, .{ .key = "yaml", .source = "a: 1" }, .{ .key = "toml", .source = "a=1" }, .{ .key = "svelte", .source = "" }, .{ .key = "scss", .source = "$a:1" }, .{ .key = "cst", .source = "" }, .{ .key = "jl", .source = "x=1" }, .{ .key = "haskell", .source = "x=1" }, .{ .key = "cyberchef", .source = "" }, .{ .key = "js", .source = "1" }, .{ .key = "json", .source = "1" }, .{ .key = "cpp", .source = "int a;" }, .{ .key = "rb", .source = "x=1" }, .{ .key = "go", .source = "package p" }, .{ .key = "cs", .source = "class A{}" }, .{ .key = "py", .source = "x=1" }, .{ .key = "rs", .source = "fn f(){}" }, .{ .key = "sh", .source = "true" }, .{ .key = "java", .source = "class A{}" }, .{ .key = "scala", .source = "class A" }, .{ .key = "agda", .source = "" }, .{ .key = "html", .source = "<p>" }, .{ .key = "embedded_template", .source = "" }, .{ .key = "regex", .source = "a" }, .{ .key = "css", .source = "a{}" }, .{ .key = "jsdoc", .source = "/** */" }, .{ .key = "ql", .source = "" },};
comptime { @setEvalBranchQuota(30000); for (parse_entries) |e| { if (!grammars.languages.has(e.key)) @compileError("unknown lang: " ++ e.key); } for (query_entries) |e| { if (!queries_mod.queries.has(e.key)) @compileError("unknown query: " ++ e.key); }}
test "parse all languages" { try runParseTest(parse_entries);}
test "execute highlight queries" { try runQueryTest(query_entries);}