diff options
| -rw-r--r-- | queries/lua/highlights.scm | 1 | ||||
| -rw-r--r-- | queries/lua/locals.scm | 19 |
2 files changed, 12 insertions, 8 deletions
diff --git a/queries/lua/highlights.scm b/queries/lua/highlights.scm index 4ce331fa6..3077be108 100644 --- a/queries/lua/highlights.scm +++ b/queries/lua/highlights.scm @@ -50,6 +50,7 @@ ;; Nodes (function "function" @function "end" @function) +(local_function "function" @function "end" @function) (table "{" @operator "}" @operator) (comment) @comment (string) @string diff --git a/queries/lua/locals.scm b/queries/lua/locals.scm index 4764c021c..5f21e0aaf 100644 --- a/queries/lua/locals.scm +++ b/queries/lua/locals.scm @@ -17,13 +17,19 @@ (parameters (identifier) @definition)) (set! definition.kind "v")) +;; Loops +((loop_expression + (identifier) @definition) + (set! definition.kind "v")) + ;; Function definitions ;; Functions definitions creates both a definition and a new scope ((function - (function_name_field - object: (identifier) @definition.associated - (property_identifier) @definition)) @scope - (set! definition.kind "m")) + (function_name + (function_name_field + (identifier) @definition.associated + (property_identifier) @definition))) @scope + (set! definition.kind "m")) ((function (function_name (identifier) @definition)) @scope @@ -36,10 +42,7 @@ ((if_statement) @scope) ((for_in_statement) @scope) ((repeat_statement) @scope) -;; Loops -((loop_expression - (identifier) @definition) - (set! definition.kind "v")) +((while_statement) @scope) ;;; REFERENCES ((identifier) @reference) |