plugin interface add OnNodeCompileFunctionParam

1 files changed, 8 insertions, 0 deletions

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