支持重定义局部变量

2 files changed, 8 insertions, 6 deletions

diff --git a/server-beta/src/parser/ast.lua b/server-beta/src/parser/ast.lua
index ae8f9a7a..39e7d871 100644
--- a/server-beta/src/parser/ast.lua
+++ b/server-beta/src/parser/ast.lua
@@ -862,7 +862,7 @@ local Defs = {
             return
         end
 
-        local loc = createLocal(name, start, actions)
+        local loc = createLocal(name, name.start, actions)
         loc.localfunction = true
 
         return loc
diff --git a/server-beta/src/parser/guide.lua b/server-beta/src/parser/guide.lua
index d8dd873e..02284b1e 100644
--- a/server-beta/src/parser/guide.lua
+++ b/server-beta/src/parser/guide.lua
@@ -316,12 +316,14 @@ function m.eachSourceType(ast, type, callback)
     if not cache then
         cache = {}
         ast.typeCache[type] = cache
+        local mark = {}
+        m.eachSource(ast, function (source)
+            if source.type == type and not mark[source] then
+                mark[source] = true
+                cache[#cache+1] = source
+            end
+        end)
     end
-    m.eachSource(ast, function (source)
-        if source.type == type then
-            cache[#cache+1] = source
-        end
-    end)
     for i = 1, #cache do
         callback(cache[i])
     end