check await cross linked callback

1 files changed, 55 insertions, 3 deletions

diff --git a/script/vm/getDocs.lua b/script/vm/getDocs.lua
index ed2299ec..1af85409 100644
--- a/script/vm/getDocs.lua
+++ b/script/vm/getDocs.lua
@@ -201,6 +201,9 @@ function vm.isDeprecated(value, deep)
 end
 
 function vm.isAsync(value, deep)
+    if isAsync(value) then
+        return true
+    end
     if deep then
         local defs = vm.getDefs(value)
         if #defs == 0 then
@@ -211,10 +214,59 @@ function vm.isAsync(value, deep)
                 return true
             end
         end
-        return false
-    else
-        return isAsync(value)
     end
+    return false
+end
+
+local function isCalledInFunction(param)
+    local func = guide.getParentFunction(param)
+    for _, ref in ipairs(param.ref) do
+        if ref.type == 'getlocal' then
+            if  ref.parent.type == 'call'
+            and guide.getParentFunction(ref) == func then
+                return true
+            end
+            if  ref.parent.type == 'callargs'
+            and ref.parent[1] == ref
+            and guide.getParentFunction(ref) == func then
+                if ref.parent.parent.node.special == 'pcall'
+                or ref.parent.parent.node.special == 'xpcall' then
+                    return true
+                end
+            end
+        end
+    end
+    return false
+end
+
+local function isLinkedCall(node, index)
+    for _, def in ipairs(vm.getDefs(node)) do
+        if def.type == 'function' then
+            local param = def.args and def.args[index]
+            if param then
+                if isCalledInFunction(param) then
+                    return true
+                end
+            end
+        end
+    end
+    return false
+end
+
+function vm.isAsyncCall(call)
+    if vm.isAsync(call.node, true) then
+        return true
+    end
+    if not call.args then
+        return
+    end
+    for i, arg in ipairs(call.args) do
+        if  vm.isAsync(arg, true)
+        and isLinkedCall(call.node, i) then
+            return true
+        end
+    end
+    return false
 end
 
 local function makeDiagRange(uri, doc, results)