special treat `local x; x = function() end`

1 files changed, 11 insertions, 0 deletions

diff --git a/script/vm/compiler.lua b/script/vm/compiler.lua
index c249b44d..b2cdb1be 100644
--- a/script/vm/compiler.lua
+++ b/script/vm/compiler.lua
@@ -1008,6 +1008,17 @@ local function compileLocal(source)
             vm.setNode(source, vm.compileNode(source.value))
         end
     end
+    if not hasMarkValue and not hasMarkValue then
+        if source.ref then
+            for _, ref in ipairs(source.ref) do
+                if  ref.type == 'setlocal'
+                and ref.value
+                and ref.value.type == 'function' then
+                    vm.setNode(source, vm.compileNode(ref.value))
+                end
+            end
+        end
+    end
     -- function x.y(self, ...) --> function x:y(...)
     if  source[1] == 'self'
     and not hasMarkDoc