fix(leo): indent } now properly dedents

1 files changed, 32 insertions, 18 deletions

diff --git a/queries/leo/indents.scm b/queries/leo/indents.scm
index 473091fb1..97a456c87 100644
--- a/queries/leo/indents.scm
+++ b/queries/leo/indents.scm
@@ -1,33 +1,47 @@
 [
-  (record_declaration)
-  (struct_declaration)
-  (mapping_declaration)
+  (array_expression)
+  (assert_statement)
+  (assignment_statement)
+  (block)
   (constant_declaration)
-  (return_statement)
   (expression_statement)
-  (variable_declaration)
+  (items_block)
   (loop_statement)
-  (assignment_statement)
-  (assert_statement)
+  (mapping_declaration)
+  (parenthesized_expression)
+  (record_declaration)
+  (return_statement)
+  (struct_declaration)
   (struct_expression)
-  (array_expression)
   (tuple_expression)
-  (parenthesized_expression)
-  (items_block)
-  (block)
+  (variable_declaration)
 ] @indent.begin
 
-; if "if" statement and conditional statement are in separate lines
-; conditional should be indented and when the conditional block
-; starts dedented
-(branch
-  (block
-    "{" @indent.end)) @indent.begin
-
 ((function_parameters) @indent.align
   (#set! indent.open_delimiter "(")
   (#set! indent.close_delimiter ")"))
 
+(record_declaration
+  "}" @indent.branch)
+
+(struct_declaration
+  "}" @indent.branch)
+
+(struct_expression
+  "}" @indent.branch)
+
+(array_expression
+  "}" @indent.branch)
+
+(tuple_expression
+  "}" @indent.branch)
+
+(items_block
+  "}" @indent.branch)
+
+(block
+  "}" @indent.branch)
+
 [
   (comment)
   (ERROR)