cast global

Author: sumneko

script/core/diagnostics/cast-type-mismatch.lua

@@ -16,9 +16,9 @@ return function (uri, callback)
     end
 
     for _, doc in ipairs(state.ast.docs) do
-        if doc.type == 'doc.cast' and doc.loc then
+        if doc.type == 'doc.cast' and doc.name then
             await.delay()
-            local defs = vm.getDefs(doc.loc)
+            local defs = vm.getDefs(doc.name)
             local loc = defs[1]
             if loc then
                 local defNode = vm.compileNode(loc)

script/core/diagnostics/unknown-cast-variable.lua

@@ -16,15 +16,15 @@ return function (uri, callback)
     end
 
     for _, doc in ipairs(state.ast.docs) do
-        if doc.type == 'doc.cast' and doc.loc then
+        if doc.type == 'doc.cast' and doc.name then
             await.delay()
-            local defs = vm.getDefs(doc.loc)
+            local defs = vm.getDefs(doc.name)
             local loc = defs[1]
             if not loc then
                 callback {
-                    start   = doc.loc.start,
-                    finish  = doc.loc.finish,
-                    message = lang.script('DIAG_UNKNOWN_CAST_VARIABLE', doc.loc[1])
+                    start   = doc.name.start,
+                    finish  = doc.name.finish,
+                    message = lang.script('DIAG_UNKNOWN_CAST_VARIABLE', doc.name[1])
                 }
             end
         end

script/parser/guide.lua

@@ -165,7 +165,7 @@ local childMap = {
     ['doc.version']        = {'#versions'},
     ['doc.diagnostic']     = {'#names'},
     ['doc.as']             = {'as'},
-    ['doc.cast']           = {'loc', '#casts'},
+    ['doc.cast']           = {'name', '#casts'},
     ['doc.cast.block']     = {'extends'},
     ['doc.operator']       = {'op', 'exp', 'extends'},
 }

script/parser/luadoc.lua

@@ -1316,7 +1316,7 @@ local docSwitch = util.switch()
             return result
         end
 
-        result.loc    = loc
+        result.name   = loc
         result.finish = loc.finish
 
         while true do

script/vm/tracer.lua

@@ -8,6 +8,7 @@ local util      = require 'utility'
 ---@field package _casts?  parser.object[]
 
 ---@class vm.tracer
+---@field name      string
 ---@field source    parser.object
 ---@field assigns   parser.object[]
 ---@field assignMap table<parser.object, true>
@@ -30,7 +31,7 @@ function mt:getCasts()
         root._casts = {}
         local docs = root.docs
         for _, doc in ipairs(docs) do
-            if doc.type == 'doc.cast' and doc.loc then
+            if doc.type == 'doc.cast' and doc.name then
                 root._casts[#root._casts+1] = doc
             end
         end
@@ -113,10 +114,10 @@ function mt:collectLocal()
 
     local casts = self:getCasts()
     for _, cast in ipairs(casts) do
-        if  cast.loc[1] == self.source[1]
+        if  cast.name[1] == self.name
         and cast.start  > startPos
         and cast.finish < finishPos
-        and guide.getLocal(self.source, self.source[1], cast.start) == self.source then
+        and guide.getLocal(self.source, self.name, cast.start) == self.source then
             self.casts[#self.casts+1] = cast
         end
     end
@@ -153,6 +154,17 @@ function mt:collectGlobal()
             finishPos = get.finish
         end
     end
+
+    local casts = self:getCasts()
+    for _, cast in ipairs(casts) do
+        if cast.name[1] == self.name then
+            self.casts[#self.casts+1] = cast
+        end
+    end
+
+    if #self.casts > 0 then
+        self.fastCalc = false
+    end
 end
 
 ---@param start  integer
@@ -794,8 +806,10 @@ local function createTracer(source)
 
     if source.type == 'local'
     or source.type == 'self' then
+        tracer.name = source[1]
         tracer:collectLocal()
     else
+        tracer.name = source.global:getName()
         tracer:collectGlobal()
     end
 

test/type_inference/init.lua

@@ -4157,3 +4157,12 @@ if type(X) ~= 'number' then
     print(<?X?>)
 end
 ]]
+
+TEST 'boolean' [[
+---@type number
+X = Y
+
+---@cast X boolean
+
+print(<?X?>)
+]]