2 files changed, 83 insertions, 0 deletions
diff --git a/script/plugins/nodeHelper.lua b/script/plugins/nodeHelper.lua
new file mode 100644
index 00000000..3f90b152
--- /dev/null
+++ b/script/plugins/nodeHelper.lua
@@ -0,0 +1,75 @@
+local vm = require 'vm'
+local guide = require 'parser.guide'
+
+local _M = {}
+
+---@class node.match.pattern
+---@field next node.match.pattern?
+
+local function deepCompare(source, pattern)
+    local type1, type2 = type(source), type(pattern)
+    if type1 ~= type2 then
+        return false
+    end
+
+    if type1 ~= "table" then
+        return source == pattern
+    end
+
+    for key2, value2 in pairs(pattern) do
+        local value1 = source[key2]
+        if value1 == nil or not deepCompare(value1, value2) then
+            return false
+        end
+    end
+
+    return true
+end
+
+---@param source parser.object
+---@param pattern node.match.pattern
+---@return boolean
+function _M.matchPattern(source, pattern)
+    if source.type == 'local' then
+        if source.parent.type == 'funcargs' and source.parent.parent.type == 'function' then
+            for i, ref in ipairs(source.ref) do
+                if deepCompare(ref, pattern) then
+                    return true
+                end
+            end
+        end
+    end
+    return false
+end
+
+local vaildVarRegex = "()([a-zA-Z][a-zA-Z0-9_]*)()"
+---创建类型 *.field.field形式的 pattern
+---@param pattern string
+---@return node.match.pattern?, string?
+function _M.createFieldPattern(pattern)
+    local ret = { next = nil }
+    local next = ret
+    local init = 1
+    while true do
+        local startpos, matched, endpos
+        if pattern:sub(1, 1) == "*" then
+            startpos, matched, endpos = init, "*", init + 1
+        else
+            startpos, matched, endpos = vaildVarRegex:match(pattern, init)
+        end
+        if not startpos then
+            break
+        end
+        if startpos ~= init then
+            return nil, "invalid pattern"
+        end
+        local field = matched == "*" and { next = nil }
+            or { field = { type = 'field', matched }, type = 'getfield', next = nil }
+        next.next = field
+        next = field
+        pattern = pattern:sub(endpos)
+    end
+    return ret
+end
+
+return _M
diff --git a/script/vm/compiler.lua b/script/vm/compiler.lua
index 8a1fa96a..6b8b76cd 100644
--- a/script/vm/compiler.lua
+++ b/script/vm/compiler.lua
@@ -5,6 +5,7 @@ local rpath      = require 'workspace.require-path'
 local files      = require 'files'
 ---@class vm
 local vm         = require 'vm.vm'
+local plugin     = require 'plugin'
 
 ---@class parser.object
 ---@field _compiledNodes        boolean
@@ -1091,6 +1092,13 @@ local function compileLocal(source)
             end
         end
         if not hasDocArg then
+            local suc, node = plugin.dispatch("OnNodeCompileFunctionParam", guide.getUri(source), source)
+            if suc and node then
+                hasDocArg = true
+                vm.setNode(source, node)
+            end
+        end
+        if not hasDocArg then
             vm.setNode(source, vm.declareGlobal('type', 'any'))
         end
     end