检查修改了循环变量

1 files changed, 27 insertions, 0 deletions

diff --git a/script/core/diagnostics.lua b/script/core/diagnostics.lua
index 0a9d339d..f488dba5 100644
--- a/script/core/diagnostics.lua
+++ b/script/core/diagnostics.lua
@@ -855,6 +855,27 @@ function mt:searchSetConstLocal(callback)
     end)
 end
 
+function mt:searchSetForState(callback)
+    local locs = {}
+    self.vm:eachSource(function (source)
+        if source.type == 'loop' then
+            locs[#locs+1] = source.arg:bindLocal()
+        elseif source.type == 'in' then
+            self.vm:forList(source.arg, function (arg)
+                locs[#locs+1] = arg:bindLocal()
+            end)
+        end
+    end)
+    for i = 1, #locs do
+        local loc = locs[i]
+        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
@@ -1051,5 +1072,11 @@ return function (vm, lines, uri)
             message = lang.script.DIAG_SET_CONST
         }
     end)
+    -- 检查修改for的内置变量
+    session:doDiagnostics(session.searchSetForState, 'set-for-state', function ()
+        return {
+            message = lang.script.DIAG_SET_FOR_STATE,
+        }
+    end)
     return session.datas
 end