aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--queries/lua/highlights.scm10
1 files changed, 7 insertions, 3 deletions
diff --git a/queries/lua/highlights.scm b/queries/lua/highlights.scm
index 6ad5c64b2..507d21989 100644
--- a/queries/lua/highlights.scm
+++ b/queries/lua/highlights.scm
@@ -100,9 +100,13 @@
(spread) @constant ;; "..."
;; Functions
-("function" @keyword.function
- [(function_name) (identifier)] @function
- "end" @keyword.function)
+(function [(function_name) (identifier)] @function)
+(function ["function" "end"] @function)
+
+(local_function [(function_name) (identifier)] @function)
+(local_function ["function" "end"] @function)
+
+(function_definition ["function" "end"] @function)
(property_identifier) @property
(method) @method