update

2 files changed, 77 insertions, 0 deletions

diff --git a/script/core/diagnostics/cast-type-mismatch.lua b/script/core/diagnostics/cast-type-mismatch.lua
new file mode 100644
index 00000000..ecee6a84
--- /dev/null
+++ b/script/core/diagnostics/cast-type-mismatch.lua
@@ -0,0 +1,45 @@
+local files   = require 'files'
+local guide   = require 'parser.guide'
+local lang    = require 'language'
+local vm      = require 'vm'
+local await   = require 'await'
+
+---@async
+return function (uri, callback)
+    local state = files.getState(uri)
+    if not state then
+        return
+    end
+
+    if not state.ast.docs then
+        return
+    end
+
+    for _, doc in ipairs(state.ast.docs) do
+        if doc.type == 'doc.cast' then
+            await.delay()
+            local defs = vm.getDefs(doc.loc)
+            local loc = defs[1]
+            if loc then
+                local defNode = vm.compileNode(loc)
+                if defNode:getData 'hasDefined' then
+                    for _, cast in ipairs(doc.casts) do
+                        if not cast.mode and cast.extends then
+                            local refNode = vm.compileNode(cast.extends)
+                            if not vm.canCastType(uri, defNode, refNode) then
+                                callback {
+                                    start   = cast.extends.start,
+                                    finish  = cast.extends.finish,
+                                    message = lang.script('DIAG_UNKNOWN_CAST_VARIABLE', {
+                                        def = vm.getInfer(defNode):view(uri),
+                                        ref = vm.getInfer(refNode):view(uri),
+                                    })
+                                }
+                            end
+                        end
+                    end
+                end
+            end
+        end
+    end
+end
diff --git a/script/core/diagnostics/unknown-cast-variable.lua b/script/core/diagnostics/unknown-cast-variable.lua
new file mode 100644
index 00000000..cfa25ed8
--- /dev/null
+++ b/script/core/diagnostics/unknown-cast-variable.lua
@@ -0,0 +1,32 @@
+local files   = require 'files'
+local guide   = require 'parser.guide'
+local lang    = require 'language'
+local vm      = require 'vm'
+local await   = require 'await'
+
+---@async
+return function (uri, callback)
+    local state = files.getState(uri)
+    if not state then
+        return
+    end
+
+    if not state.ast.docs then
+        return
+    end
+
+    for _, doc in ipairs(state.ast.docs) do
+        if doc.type == 'doc.cast' then
+            await.delay()
+            local defs = vm.getDefs(doc.loc)
+            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[1])
+                }
+            end
+        end
+    end
+end