automatic derivation function param type

Author: fesily

script/vm/compiler.lua

@@ -1090,6 +1090,37 @@ local function compileLocal(source)
                 end
             end
         end
+        if not hasDocArg
+        and func.parent.type == 'local' then
+            local refs = vm.getRefs(func, function (_)
+                return false
+            end)
+            local findCall
+            if refs then
+                for i, ref in ipairs(refs) do
+                    if ref.parent.type == 'call' then
+                        findCall = ref.parent
+                        break
+                    end
+                end
+            end
+            if findCall and findCall.args then
+                local index
+                for i, arg in ipairs(source.parent) do
+                    if arg == source then
+                        index = i
+                        break
+                    end
+                end
+                if index then
+                    local callerArg = findCall.args[index]
+                    if callerArg then
+                        hasDocArg = true
+                        vm.setNode(source, vm.compileNode(callerArg))
+                    end
+                end
+            end
+        end
         if not hasDocArg then
             vm.setNode(source, vm.declareGlobal('type', 'any'))
         end

test/diagnostics/await-in-sync.lua

@@ -119,7 +119,7 @@ end
 
 TEST [[
 local function f(cb)
-    cb()
+    <!cb!>()
 end
 
 local function af()

test/diagnostics/redundant-parameter.lua

@@ -94,7 +94,7 @@ print(1, 2, 3, 4, 5)
 
 TEST [[
 local function f(callback)
-    callback(1, 2, 3)
+    callback(<!1!>, <!2!>, <!3!>)
 end
 f(function () end)
 ]]

test/hover/init.lua

@@ -276,15 +276,15 @@ function string.lower(s: string|number)
   -> string
 ]]
 
--- 不根据传入值推测参数类型
+-- 根据传入值推测参数类型
 TEST [[
 local function x(a, ...)
 end
 
 <?x?>(1, 2, 3, 4, 5, 6, 7)
 ]]
 [[
-function x(a: any, ...any)
+function x(a: integer, ...any)
 ]]
 
 TEST [[

test/signature/init.lua

@@ -88,7 +88,7 @@ end
 
 x(1, 2, 3, <??>
 ]]
-{'function x(a: any, <!...any!>)'}
+{'function x(a: integer, <!...any!>)'}
 
 TEST [[
 (''):sub(<??>
@@ -106,7 +106,7 @@ end
 
 f(1, 'string<??>')
 ]]
-{'function f(a: any, <!b: any!>, c: any)'}
+{'function f(a: integer, <!b: string!>, c: any)'}
 
 TEST [[
 pcall(function () <??> end)
@@ -156,7 +156,7 @@ end
 
 f({},<??>)
 ]]
-{'function f(a: any, <!b: any!>, c: any)'}
+{'function f(a: table, <!b: any!>, c: any)'}
 
 TEST [[
 for _ in pairs(<??>) do
@@ -188,31 +188,31 @@ end
 
 x(  aaaa  <??>, 2)
 ]]
-{"function x(<!a: any!>, b: any)"}
+{"function x(<!a: any!>, b: integer)"}
 
 TEST [[
 local function x(a, b)
 end
 
 x(<??>   aaaa  , 2)
 ]]
-{'function x(<!a: any!>, b: any)'}
+{'function x(<!a: any!>, b: integer)'}
 
 TEST [[
 local function x(a, b)
 end
 
 x(aaaa  ,<??>    2)
 ]]
-{'function x(a: any, <!b: any!>)'}
+{'function x(a: any, <!b: integer!>)'}
 
 TEST [[
 local function x(a, b)
 end
 
 x(aaaa  ,    2     <??>)
 ]]
-{'function x(a: any, <!b: any!>)'}
+{'function x(a: any, <!b: integer!>)'}
 
 TEST [[
 local fooC