From 4753990edeeeec6d4ba32563b2d956f51bf59686 Mon Sep 17 00:00:00 2001
From: Julian Berman <Julian@GrayVines.com>
Date: Sat, 28 Dec 2024 12:22:39 -0500
Subject: [PATCH] Tweak the import tree sitter hlgroups.

(This still isn't really used yet until/unless tree-sitter-lean is
revived. But the hlgroup is wrong as-is so may as well fix it.)
---
 queries/lean/highlights.scm | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/queries/lean/highlights.scm b/queries/lean/highlights.scm
index 5ede0575..6c3d8f52 100644
--- a/queries/lean/highlights.scm
+++ b/queries/lean/highlights.scm
@@ -68,7 +68,8 @@
 
 ["for" "in" "do"] @repeat
 
-(import) @include
+(import
+  module: (identifier) @module)
 
 ; Tokens
 
@@ -173,6 +174,10 @@
   (trivial)
 ] @keyword
 
+[
+ "import"
+] @keyword.import
+
 [
   "catch"
   "finally"