诊断设置常量

3 files changed, 45 insertions, 0 deletions

diff --git a/server/src/constant/DiagnosticDefaultSeverity.lua b/server/src/constant/DiagnosticDefaultSeverity.lua
index 2088a922..eef44107 100644
--- a/server/src/constant/DiagnosticDefaultSeverity.lua
+++ b/server/src/constant/DiagnosticDefaultSeverity.lua
@@ -14,4 +14,5 @@ return {
     ['empty-block']         = 'Information',
     ['redundant-value']     = 'Information',
     ['emmy-lua']            = 'Warning',
+    ['set-const']           = 'Error',
 }
diff --git a/server/src/core/diagnostics.lua b/server/src/core/diagnostics.lua
index 93cf34d9..458a4686 100644
--- a/server/src/core/diagnostics.lua
+++ b/server/src/core/diagnostics.lua
@@ -631,6 +631,38 @@ function mt:searchEmmyLua(callback)
     end)
 end
 
+function mt:searchSetConstLocal(callback)
+    local mark = {}
+    self.vm:eachSource(function (source)
+        local loc = source:bindLocal()
+        if not loc then
+            return
+        end
+        if mark[loc] then
+            return
+        end
+        mark[loc] = true
+        if not loc.tags then
+            return
+        end
+        local const
+        for _, tag in ipairs(loc.tags) do
+            if tag[1] == 'const' then
+                const = true
+                break
+            end
+        end
+        if not const then
+            return
+        end
+        loc:eachInfo(function (info, src)
+            if info.type == 'set' then
+                callback(src.start, src.finish)
+            end
+        end)
+    end)
+end
+
 function mt:doDiagnostics(func, code, callback)
     if config.config.diagnostics.disable[code] then
         return
@@ -772,5 +804,11 @@ return function (vm, lines, uri)
             related = related,
         }
     end)
+    -- 检查给const变量赋值
+    session:doDiagnostics(session.searchSetConstLocal, 'set-const', function ()
+        return {
+            message = lang.script.DIAG_SET_CONST
+        }
+    end)
     return session.datas
 end
diff --git a/server/test/diagnostics/init.lua b/server/test/diagnostics/init.lua
index eec4bf36..7884df06 100644
--- a/server/test/diagnostics/init.lua
+++ b/server/test/diagnostics/init.lua
@@ -442,3 +442,9 @@ TEST [[
 TEST [[
 ---@class Class : any
 ]]
+
+TEST [[
+local <const> x = 1
+<!x!> = 2
+return x
+]]