infer by break

Author: sumneko

script/parser/guide.lua

@@ -59,6 +59,7 @@ local type         = type
 ---@field filter                parser.object
 ---@field hasGoTo?              true
 ---@field hasReturn?            true
+---@field hasBreak?             true
 ---@field _root                 parser.object
 
 ---@class guide

script/parser/newparser.lua

@@ -3606,6 +3606,15 @@ local function parseBreak()
             break
         end
     end
+    for i = #Chunk, 1, -1 do
+        local chunk = Chunk[i]
+        if chunk.type == 'ifblock'
+        or chunk.type == 'elseifblock'
+        or chunk.type == 'elseblock' then
+            chunk.hasBreak = true
+            break
+        end
+    end
     if not ok and Mode == 'Lua' then
         pushError {
             type   = 'BREAK_OUTSIDE',

script/vm/runner.lua

@@ -193,7 +193,8 @@ function mt:_compileBlock(block)
             self.steps[#self.steps+1] = blockStep
             self:_compileNarrowByFilter(childBlock.filter, outStep, blockStep)
             if  not childBlock.hasReturn
-            and not childBlock.hasGoTo then
+            and not childBlock.hasGoTo
+            and not childBlock.hasBreak then
                 finals[#finals+1] = blockStep
             end
             self.steps[#self.steps+1] = {
@@ -212,7 +213,12 @@ function mt:_compileBlock(block)
         end
     end
 
-    if block.type == 'function' then
+    if block.type == 'function'
+    or block.type == 'while'
+    or block.type == 'loop'
+    or block.type == 'in'
+    or block.type == 'repeat'
+    or block.type == 'for' then
         self.steps[#self.steps+1] = {
             type  = 'load',
             pos   = block.start,

test/type_inference/init.lua

@@ -2023,3 +2023,28 @@ end
 
 print(<?x?>)
 ]]
+
+TEST 'integer' [[
+---@type integer?
+local x
+
+while true do
+    if not x then
+        break
+    end
+    print(<?x?>)
+end
+]]
+
+TEST 'integer?' [[
+---@type integer?
+local x
+
+while true do
+    if not x then
+        break
+    end
+end
+
+print(<?x?>)
+]]