1 files changed, 65 insertions, 0 deletions
diff --git a/script-beta/vm/eachDef.lua b/script-beta/vm/eachDef.lua
new file mode 100644
index 00000000..0274cbee
--- /dev/null
+++ b/script-beta/vm/eachDef.lua
@@ -0,0 +1,65 @@
+local vm    = require 'vm.vm'
+local guide = require 'parser.guide'
+local files = require 'files'
+
+local function checkPath(source, info)
+    if source.type == 'goto' then
+        return true
+    end
+    local src = info.source
+    local mode = guide.getPath(source, src)
+    if not mode then
+        return true
+    end
+    if mode == 'before' then
+        return false
+    end
+    return true
+end
+
+function vm.eachDef(source, callback)
+    local results = {}
+    local valueUris = {}
+    local sourceUri = guide.getRoot(source).uri
+    vm.eachRef(source, function (info)
+        if info.mode == 'declare'
+        or info.mode == 'set'
+        or info.mode == 'return'
+        or info.mode == 'value' then
+            results[#results+1] = info
+            local src = info.source
+            if info.mode == 'return' then
+                local uri = guide.getRoot(src).uri
+                valueUris[uri] = info.source
+            end
+        end
+    end)
+
+    for _, info in ipairs(results) do
+        local src = info.source
+        local destUri = guide.getRoot(src).uri
+        -- 如果是同一个文件，则检查位置关系后放行
+        if sourceUri == destUri then
+            if checkPath(source, info) then
+                callback(info)
+            end
+            goto CONTINUE
+        end
+        -- 如果是global或field，则直接放行（因为无法确定顺序）
+        if src.type == 'setindex'
+        or src.type == 'setfield'
+        or src.type == 'setmethod'
+        or src.type == 'tablefield'
+        or src.type == 'tableindex'
+        or src.type == 'setglobal' then
+            callback(info)
+            goto CONTINUE
+        end
+        -- 如果不是同一个文件，则必须在该文件 return 后才放行
+        if valueUris[destUri] then
+            callback(info)
+            goto CONTINUE
+        end
+        ::CONTINUE::
+    end
+end