update narrow assign

1 files changed, 12 insertions, 0 deletions

diff --git a/script/vm/compiler.lua b/script/vm/compiler.lua
index eefed5a8..e4fb633e 100644
--- a/script/vm/compiler.lua
+++ b/script/vm/compiler.lua
@@ -1180,8 +1180,20 @@ local compilerSwitch = util.switch()
         if bindDocs(source) then
             return
         end
+        local locNode = vm.compileNode(source.node)
+        if not source.value then
+            vm.setNode(source, locNode)
+            return
+        end
         local valueNode = vm.compileNode(source.value)
         vm.setNode(source, valueNode)
+        if  locNode:getData 'hasDefined'
+        and guide.isLiteral(source.value) then
+            vm.setNode(source, locNode)
+            vm.getNode(source):narrow(guide.getUri(source), source.value.type)
+        else
+            vm.setNode(source, valueNode)
+        end
     end)
     : case 'getlocal'
     ---@async