Merge pull request #41 from desdic/20240227fixlua

Add missing lua function type

queries/lua/agrolens.callings.scm

@@ -1,3 +1,8 @@
 (function_call name:(identifier) @agrolens.name) @agrolens.scope
 
 (function_call name:(dot_index_expression table:(identifier) @agrolens.object field:(identifier) @agrolens.name)) @agrolens.scope
+
+(function_call
+ name:(method_index_expression
+  table:(identifier)
+  method:(identifier) @agrolens.name)) @agrolens.scope

queries/lua/agrolens.functions.scm

@@ -15,3 +15,9 @@
     (variable_list name:(dot_index_expression field:(identifier) @agrolens.name))
     (expression_list value:(function_definition))
 ) @agrolens.scope
+
+(function_declaration
+ name:(method_index_expression
+  table:(identifier)
+  method:(identifier) @agrolens.name)) @agrolens.scope
+

tests/lua/test.lua

@@ -24,8 +24,13 @@ function M.hello5()
     print("hello5")
 end
 
+function M:hello6()
+    print("hello6")
+end
+
 hello1()
 hello2()
 hello3()
 M.hello4()
 M.hello5()
+M:hello6()

tests/lua_spec.lua

@@ -20,25 +20,27 @@ describe("lua", function()
         local entries =
             core.get_captures({ queries = { "functions" }, bufids = buffers })
 
-        eq(#entries, 5)
+        eq(#entries, 6)
         eq(entries[1].lnum, 1)
         eq(entries[1].col, 0)
         eq(entries[1].line, "local hello1 = function()")
         eq(entries[2].line, "hello2 = function()")
         eq(entries[3].line, "function hello3()")
         eq(entries[4].line, "M.hello4 = function()")
         eq(entries[5].line, "function M.hello5()")
+        eq(entries[6].line, "function M:hello6()")
     end)
 
     it("callings", function()
         local entries =
             core.get_captures({ queries = { "callings" }, bufids = buffers })
 
-        eq(#entries, 10)
+        eq(#entries, 12)
         eq(entries[1].lnum, 2)
         eq(entries[1].col, 4)
         eq(entries[1].line, '    print("hello1")')
         eq(entries[2].line, '    print("hello2")')
+        eq(entries[12].line, 'M:hello6()')
     end)
 
     it("comments", function()