From 0b50cf159283855b72f4c6b22f5ba3c9cf813b37 Mon Sep 17 00:00:00 2001
From: r001 <rob@nyar.eu>
Date: Mon, 23 Oct 2023 12:15:49 +0200
Subject: [PATCH] fix(leo): highlights fixes (#5563)

* make sure affine_group has priority, so parens and comma will not be miscolored; added `annotation` as @attribute

* `then` is not used as conditional, so moved to keywords

* finalize function highlight added

* differentiate between constant and variable variable identifiers
---
 queries/leo/highlights.scm | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/queries/leo/highlights.scm b/queries/leo/highlights.scm
index 4114f41f9..86939e168 100644
--- a/queries/leo/highlights.scm
+++ b/queries/leo/highlights.scm
@@ -1,4 +1,5 @@
-(variable) @variable
+(variable_identifier) @variable
+(constant_identifier) @constant
 
 [
  "assert"
@@ -13,6 +14,7 @@
  "record"
  "self"
  "struct"
+ "then"
 ] @keyword
 
  "in" @keyword.operator
@@ -43,7 +45,6 @@
 [ 
   "else"
   "if"
-  "then"
 ] @conditional
 
 [
@@ -143,6 +144,9 @@
 (transition_declaration
   name: (identifier) @function.builtin)
 
+(finalizer
+  name: (identifier) @function.builtin)
+
 (free_function_call
   (identifier) @function.call)
 
@@ -168,10 +172,12 @@
 
 [ 
   (address_literal)
-  (affine_group_literal) 
+  ((affine_group_literal) (#set! "priority" 101))
   (field_literal) 
   (product_group_literal) 
   (scalar_literal) 
   (signed_literal) 
   (unsigned_literal) 
 ] @number
+
+(annotation) @attribute