From 6da2b175e20ed3c03b0dfcfc9046de1e0e5d4444 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=9C=80=E8=90=8C=E5=B0=8F=E6=B1=90?= <sumneko@hotmail.com>
Date: Sat, 23 Nov 2019 00:05:30 +0800
Subject: =?UTF-8?q?=E6=AD=A3=E8=B7=AF=E7=9B=AE=E5=BD=95?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 script-beta/vm/getGlobals.lua | 45 +++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 45 insertions(+)
 create mode 100644 script-beta/vm/getGlobals.lua

(limited to 'script-beta/vm/getGlobals.lua')

diff --git a/script-beta/vm/getGlobals.lua b/script-beta/vm/getGlobals.lua
new file mode 100644
index 00000000..699dd270
--- /dev/null
+++ b/script-beta/vm/getGlobals.lua
@@ -0,0 +1,45 @@
+local guide = require 'parser.guide'
+local vm    = require 'vm.vm'
+
+local function getGlobals(root)
+    local env  = guide.getENV(root)
+    local cache = {}
+    local mark = {}
+    vm.eachField(env, function (info)
+        local src = info.source
+        if mark[src] then
+            return
+        end
+        mark[src] = true
+        local name = info.key
+        if not name then
+            return
+        end
+        if not cache[name] then
+            cache[name] = {
+                key  = name,
+                mode = {},
+            }
+        end
+        cache[name][#cache[name]+1] = info
+        cache[name].mode[info.mode] = true
+        vm.cache.getGlobal[src] = name
+    end)
+    return cache
+end
+
+function vm.getGlobals(source)
+    source = guide.getRoot(source)
+    local cache = vm.cache.getGlobals[source]
+    if cache ~= nil then
+        return cache
+    end
+    local unlock = vm.lock('getGlobals', source)
+    if not unlock then
+        return nil
+    end
+    cache = getGlobals(source) or false
+    vm.cache.getGlobals[source] = cache
+    unlock()
+    return cache
+end
-- 
cgit v1.2.3