1 files changed, 157 insertions, 0 deletions
diff --git a/server-beta/src/searcher-old/getglobal.lua b/server-beta/src/searcher-old/getglobal.lua
new file mode 100644
index 00000000..1295d769
--- /dev/null
+++ b/server-beta/src/searcher-old/getglobal.lua
@@ -0,0 +1,157 @@
+local guide    = require 'parser.guide'
+local checkSMT = require 'searcher.setmetatable'
+
+local m = {}
+
+function m:eachDef(source, callback)
+    -- _ENV
+    local key = guide.getKeyName(source)
+    self:eachField(source.node, key, function (info)
+        if info.mode == 'set' then
+            callback(info)
+        end
+    end)
+    self:eachSpecial(function (name, src)
+        if name == '_G' then
+            local parent = src.parent
+            if guide.getKeyName(parent) == key then
+                callback {
+                    source = parent,
+                    uri    = self.uri,
+                    mode   = 'set',
+                }
+            end
+        elseif name == 'rawset' then
+            local t, k = self:callArgOf(src.parent)
+            if self:getSpecialName(t) == '_G'
+            and guide.getKeyName(k) == key then
+                callback {
+                    source = src.parent,
+                    uri    = self.uri,
+                    mode   = 'set',
+                }
+            end
+        end
+    end)
+end
+
+function m:eachRef(source, callback)
+    -- _ENV
+    local key = guide.getKeyName(source)
+    self:eachField(source.node, key, function (info)
+        if info.mode == 'set' or info.mode == 'get' then
+            callback(info)
+        end
+    end)
+    self:eachSpecial(function (name, src)
+        if name == '_G' then
+            local parent = src.parent
+            if guide.getKeyName(parent) == key then
+                if parent.type:sub(1, 3) == 'set' then
+                    callback {
+                        source = parent,
+                        uri    = self.uri,
+                        mode   = 'set',
+                    }
+                else
+                    callback {
+                        source = parent,
+                        uri    = self.uri,
+                        mode   = 'get',
+                    }
+                end
+            end
+        elseif name == 'rawset' then
+            local t, k = self:callArgOf(src.parent)
+            if self:getSpecialName(t) == '_G'
+            and guide.getKeyName(k) == key then
+                callback {
+                    source = src.parent,
+                    uri    = self.uri,
+                    mode   = 'set',
+                }
+            end
+        elseif name == 'rawget' then
+            local t, k = self:callArgOf(src.parent)
+            if self:getSpecialName(t) == '_G'
+            and guide.getKeyName(k) == key then
+                callback {
+                    source = src.parent,
+                    uri    = self.uri,
+                    mode   = 'set',
+                }
+            end
+        end
+    end)
+end
+
+function m:eachField(source, key, callback)
+    local used = {}
+    local found = false
+    used[source] = true
+
+    self:eachRef(source, function (info)
+        local src = info.source
+        used[src] = true
+        local child, mode, value = info.searcher:childMode(src)
+        if child then
+            if key == guide.getKeyName(child) then
+                callback {
+                    source = child,
+                    uri    = info.uri,
+                    mode   = mode,
+                }
+            end
+            if value then
+                info.searcher:eachField(value, key, callback)
+            end
+            return
+        end
+        if src.type == 'getglobal' then
+            local parent = src.parent
+            child, mode, value = info.searcher:childMode(parent)
+            if child then
+                if key == guide.getKeyName(child) then
+                    callback {
+                        source = child,
+                        uri    = info.uri,
+                        mode   = mode,
+                    }
+                end
+                if value then
+                    info.searcher:eachField(value, key, callback)
+                end
+            end
+        elseif src.type == 'setglobal' then
+            info.searcher:eachField(src.value, key, callback)
+        else
+            info.searcher:eachField(src, key, callback)
+        end
+    end)
+
+    checkSMT(self, key, used, found, callback)
+end
+
+function m:eachValue(source, callback)
+    callback {
+        source = source,
+        uri    = self.uri,
+    }
+    self:eachDef(source, function (info)
+        local src = info.source
+        if src.value then
+            callback {
+                source = src.value,
+                uri    = info.uri,
+            }
+        end
+    end)
+end
+
+function m:getValue(source)
+    if source.type == 'setglobal' then
+        return source.value and self:getValue(source.value)
+    end
+end
+
+return m