cleanup

1 files changed, 84 insertions, 0 deletions

diff --git a/script/vm/global.lua b/script/vm/global.lua
new file mode 100644
index 00000000..d5974e63
--- /dev/null
+++ b/script/vm/global.lua
@@ -0,0 +1,84 @@
+local util = require 'utility'
+
+---@class vm.node.global.link
+---@field gets   parser.object[]
+---@field sets   parser.object[]
+
+---@class vm.node.global
+---@field links table<uri, vm.node.global.link>
+---@field setsCache parser.object[]
+---@field getsCache parser.object[]
+local mt = {}
+mt.__index = mt
+mt.type = 'global'
+mt.name = ''
+
+---@param uri    uri
+---@param source parser.object
+function mt:addSet(uri, source)
+    local link = self.links[uri]
+    link.sets[#link.sets+1] = source
+end
+
+---@param uri    uri
+---@param source parser.object
+function mt:addGet(uri, source)
+    local link = self.links[uri]
+    link.gets[#link.gets+1] = source
+end
+
+---@return parser.object[]
+function mt:getSets()
+    if not self.setsCache then
+        self.setsCache = {}
+        for _, link in pairs(self.links) do
+            for _, source in ipairs(link.sets) do
+                self.setsCache[#self.setsCache+1] = source
+            end
+        end
+    end
+    return self.setsCache
+end
+
+---@return parser.object[]
+function mt:getGets()
+    if not self.getsCache then
+        self.getsCache = {}
+        for _, link in pairs(self.links) do
+            for _, source in ipairs(link.gets) do
+                self.getsCache[#self.getsCache+1] = source
+            end
+        end
+    end
+    return self.getsCache
+end
+
+---@param uri uri
+function mt:dropUri(uri)
+    self.links[uri] = nil
+    self.setsCache = nil
+    self.getsCache = nil
+end
+
+---@return string
+function mt:getName()
+    return self.name
+end
+
+---@return boolean
+function mt:isAlive()
+    return next(self.links) ~= nil
+end
+
+---@return vm.node.global
+return function (name)
+    return setmetatable({
+        name  = name,
+        links = util.defaultTable(function ()
+            return {
+                sets   = {},
+                gets   = {},
+            }
+        end),
+    }, mt)
+end