search full refs for global

fix #1914

Author: sumneko

changelog.md

@@ -2,8 +2,10 @@
 
 ## 3.6.14
 * `FIX` [#1753]
+* `FIX` [#1914]
 
 [#1753]: https://github.com/LuaLS/lua-language-server/issues/1753
+[#1914]: https://github.com/LuaLS/lua-language-server/issues/1914
 
 ## 3.6.13
 `2023-3-2`

script/vm/ref.lua

@@ -107,29 +107,28 @@ local function searchWord(source, pushResult, defMap, fileNotify)
                     await.delay()
                 end
             end)
-        else
-            ---@async
-            guide.eachSourceTypes(state.ast, {'getfield', 'setfield'}, function (src)
-                if src.field and src.field[1] == key then
-                    checkDef(src)
-                    await.delay()
-                end
-            end)
-            ---@async
-            guide.eachSourceTypes(state.ast, {'getmethod', 'setmethod'}, function (src)
-                if src.method and src.method[1] == key then
-                    checkDef(src)
-                    await.delay()
-                end
-            end)
-            ---@async
-            guide.eachSourceTypes(state.ast, {'getindex', 'setindex'}, function (src)
-                if src.index and src.index.type == 'string' and src.index[1] == key then
-                    checkDef(src)
-                    await.delay()
-                end
-            end)
         end
+        ---@async
+        guide.eachSourceTypes(state.ast, {'getfield', 'setfield'}, function (src)
+            if src.field and src.field[1] == key then
+                checkDef(src)
+                await.delay()
+            end
+        end)
+        ---@async
+        guide.eachSourceTypes(state.ast, {'getmethod', 'setmethod'}, function (src)
+            if src.method and src.method[1] == key then
+                checkDef(src)
+                await.delay()
+            end
+        end)
+        ---@async
+        guide.eachSourceTypes(state.ast, {'getindex', 'setindex'}, function (src)
+            if src.index and src.index.type == 'string' and src.index[1] == key then
+                checkDef(src)
+                await.delay()
+            end
+        end)
     end
 
     searchInAllFiles(guide.getUri(source), findWord, fileNotify)

test/references/all.lua

@@ -126,3 +126,16 @@ end
 local v1 = Master:foobar("", Dog)
 v1.<!eat!>()
 ]]
+
+TEST [[
+---@class A
+A = {}
+
+function A:<~TestA~>()
+end
+
+---@param param A
+function TestB(param)
+    param:<!TestA!>()
+end
+]]