From dbf893e48be5cbb15bf2b87f84e5907eb3311bb0 Mon Sep 17 00:00:00 2001 From: John Doty Date: Thu, 5 Sep 2024 15:03:44 -0700 Subject: [PATCH] Generate queries a little better --- parser/tree_sitter.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/parser/tree_sitter.py b/parser/tree_sitter.py index cd170ae..3f27d31 100644 --- a/parser/tree_sitter.py +++ b/parser/tree_sitter.py @@ -319,8 +319,12 @@ def emit_tree_sitter_queries(grammar: parser.Grammar, path: pathlib.Path | str): for rule in grammar.terminals(): highlight = rule.meta.get("highlight") if isinstance(highlight, parser.HighlightMeta): - queries.append(f"({terminal_name(rule)} @{highlight.scope})") + queries.append(f"({terminal_name(rule)}) @{highlight.scope}") - path = pathlib.Path(path) / "highlight.scm" + path = pathlib.Path(path) / "queries" + if not path.exists(): + path.mkdir(parents=True) + + path = path / "highlights.scm" with open(path, "w", encoding="utf-8") as f: f.write("\n\n".join(queries))