1 files changed, 8 insertions, 49 deletions

diff --git a/script/core/diagnostics/missing-parameter.lua b/script/core/diagnostics/missing-parameter.lua
index 698680ca..78b94a09 100644
--- a/script/core/diagnostics/missing-parameter.lua
+++ b/script/core/diagnostics/missing-parameter.lua
@@ -2,68 +2,27 @@ local files  = require 'files'
 local guide  = require 'parser.guide'
 local vm     = require 'vm'
 local lang   = require 'language'
+local await  = require 'await'
 
-local function countCallArgs(source)
-    local result = 0
-    if not source.args then
-        return 0
-    end
-    result = result + #source.args
-    return result
-end
-
----@return integer
-local function countFuncArgs(source)
-    if not source.args or #source.args == 0 then
-        return 0
-    end
-    local count = 0
-    for i = #source.args, 1, -1 do
-        local arg = source.args[i]
-        if  arg.type ~= '...'
-        and not (arg.name and arg.name[1] =='...')
-        and not vm.compileNode(arg):isNullable() then
-            return i
-        end
-    end
-    return count
-end
-
-local function getFuncArgs(func)
-    local funcArgs
-    local defs = vm.getDefs(func)
-    for _, def in ipairs(defs) do
-        if def.type == 'function'
-        or def.type == 'doc.type.function' then
-            local args = countFuncArgs(def)
-            if not funcArgs or args < funcArgs then
-                funcArgs = args
-            end
-        end
-    end
-    return funcArgs
-end
-
+---@async
 return function (uri, callback)
     local state = files.getState(uri)
     if not state then
         return
     end
 
+    ---@async
     guide.eachSourceType(state.ast, 'call', function (source)
-        local callArgs = countCallArgs(source)
+        await.delay()
+        local _, callArgs = vm.countList(source.args)
 
-        local func = source.node
-        local funcArgs = getFuncArgs(func)
+        local funcNode = vm.compileNode(source.node)
+        local funcArgs = vm.countParamsOfNode(funcNode)
 
-        if not funcArgs then
+        if callArgs >= funcArgs then
             return
         end
 
-        local delta = callArgs - funcArgs
-        if delta >= 0 then
-            return
-        end
         callback {
             start  = source.start,
             finish = source.finish,