From b10a09ebc3f9786c57604942ce5d7201a10c0337 Mon Sep 17 00:00:00 2001
From: kyazdani42 <yazdani.kiyan@protonmail.com>
Date: Tue, 8 Sep 2020 17:59:16 +0200
Subject: fix function/end keywords not applied for every scope

---
 queries/lua/highlights.scm | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

(limited to 'queries/lua/highlights.scm')

diff --git a/queries/lua/highlights.scm b/queries/lua/highlights.scm
index 6ad5c64b..507d2198 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
-- 
cgit v1.2.3