整理代码

1 files changed, 52 insertions, 204 deletions

diff --git a/server-beta/src/core/engineer.lua b/server-beta/src/core/engineer.lua
index 9fc771b2..76a360cc 100644
--- a/server-beta/src/core/engineer.lua
+++ b/server-beta/src/core/engineer.lua
@@ -1,217 +1,57 @@
-local guide  = require 'parser.guide'
-local config = require 'config'
+local guide   = require 'parser.guide'
+local require = require
 
-local type         = type
 local setmetatable = setmetatable
 local ipairs       = ipairs
 
 _ENV = nil
-
 ---@class engineer
 local mt = {}
 mt.__index = mt
 mt.type = 'engineer'
 
-mt['local'] = function (self, source, mode, callback)
-    if mode == 'def' then
-        if source.tag ~= 'self' then
-            callback(source, 'local')
-        end
-        if source.ref then
-            for _, ref in ipairs(source.ref) do
-                if ref.type == 'setlocal' then
-                    callback(ref, 'set')
-                end
-            end
-        end
-        if source.tag == 'self' then
-            local method = source.method
-            local node = method.node
-            self:eachRef(node, mode, callback)
-        end
-    elseif mode == 'ref' then
-        if source.tag ~= 'self' then
-            callback(source, 'local')
-        end
-        if source.ref then
-            for _, ref in ipairs(source.ref) do
-                if ref.type == 'setlocal' then
-                    callback(ref, 'set')
-                elseif ref.type == 'getlocal' then
-                    callback(ref, 'get')
-                end
-            end
-        end
-        if source.tag == 'self' then
-            local method = source.method
-            local node = method.node
-            self:eachRef(node, mode, callback)
-        end
-    elseif mode == 'field' then
-        if source.ref then
-            for _, ref in ipairs(source.ref) do
-                if ref.type == 'getlocal' then
-                    local parent = ref.parent
-                    local tp     = parent.type
-                    if tp == 'setfield'
-                    or tp == 'getfield'
-                    or tp == 'setmethod'
-                    or tp == 'getmethod'
-                    or tp == 'setindex'
-                    or tp == 'getindex' then
-                        callback(parent)
-                    end
-                elseif ref.type == 'setlocal' then
-                    self:eachRef(ref.value, 'field', callback)
-                end
-            end
-        end
-        if source.tag == 'self' then
-            local method = source.method
-            local node = method.node
-            self:eachRef(node, 'field', callback)
-        end
-        if source.value then
-            self:eachField(source.value, nil, 'ref', callback)
-        end
-    end
-end
-mt['getlocal'] = function (self, source, mode, callback)
-    self:search(source.loc, 'local', mode, callback)
-end
-mt['setlocal'] = mt['getlocal']
-mt['_G'] = function (self, source, mode, callback)
-    if mode == 'def' then
-        local parent = source.parent
-        if parent.type == 'setfield'
-        or parent.type == 'setindex'
-        or parent.type == 'setmethod' then
-            callback(parent, 'set')
-        elseif parent.type == 'getfield'
-        or     parent.type == 'getindex' then
-            self:search(parent, 'special', mode, callback)
-        elseif parent.type == 'callargs' then
-            self:search(parent.parent, 'special', mode, callback)
-        end
-    elseif mode == 'ref' then
-        local parent = source.parent
-        if parent.type == 'setfield'
-        or parent.type == 'setindex'
-        or parent.type == 'setmethod' then
-            callback(parent, 'set')
-        elseif parent.type == 'getfield'
-        or     parent.type == 'getindex'
-        or     parent.type == 'getmethod' then
-            callback(parent, 'get')
-        elseif parent.type == 'getfield' then
-            self:search(parent, 'special', mode, callback)
-        elseif parent.type == 'callargs' then
-            self:search(parent.parent, 'special', mode, callback)
-        end
-    end
-end
-mt['getglobal'] = function (self, source, mode, callback)
-    local env = source.node
-    if mode == 'def' then
-        if env.ref then
-            for _, ref in ipairs(env.ref) do
-                if ref.type == 'setglobal' then
-                    callback(ref, 'set')
-                elseif ref.type == 'getglobal' then
-                    self:search(ref, 'special', mode, callback)
-                elseif ref.type == 'getlocal' then
-                    self:search(ref, '_G', mode, callback)
-                end
-            end
-        end
-    elseif mode == 'ref' then
-        if env.ref then
-            for _, ref in ipairs(env.ref) do
-                if ref.type == 'setglobal' then
-                    callback(ref, 'set')
-                elseif ref.type == 'getglobal' then
-                    callback(ref, 'get')
-                    self:search(ref, 'special', mode, callback)
-                elseif ref.type == 'getlocal' then
-                    self:search(ref, '_G', mode, callback)
-                end
-            end
-        end
-    elseif mode == 'field' then
-        self:search(source, 'getglobal', 'ref', function (src)
-            local parent = src.parent
-            local tp     = parent.type
-            if tp == 'setfield'
-            or tp == 'getfield'
-            or tp == 'setmethod'
-            or tp == 'getmethod'
-            or tp == 'setindex'
-            or tp == 'getindex' then
-                callback(parent)
-            end
-        end)
-    end
-end
+mt['local']     = require 'core.local'
+mt['getlocal']  = require 'core.getlocal'
+mt['setlocal']  = mt['getlocal']
+mt['_G']        = require 'core._G'
+mt['getglobal'] = require 'core.getglobal'
 mt['setglobal'] = mt['getglobal']
-mt['field'] = function (self, source, mode, callback)
-    local node = source.parent.node
-    local key = guide.getKeyName(source)
-    self:eachField(node, key, mode, callback)
-end
-mt['method'] = function (self, source, mode, callback)
-    local node = source.parent.node
-    local key = guide.getKeyName(source)
-    self:eachField(node, key, mode, callback)
-end
-mt['special'] = function (self, source, mode, callback)
-    local name = self:getSpecial(source)
-    if not name then
-        return
-    end
-    if mode == 'def' then
-        if name == 's|_G' then
-            self:search(source, '_G', mode, callback)
-        elseif name == 's|rawset' then
-            callback(source.parent, 'set')
-        end
-    end
-end
-mt['number']  = function (self, source, mode, callback)
-    self:asindex(source, mode, callback)
-end
-mt['boolean'] = function (self, source, mode, callback)
-    self:asindex(source, mode, callback)
-end
-mt['string'] = function (self, source, mode, callback)
-    self:asindex(source, mode, callback)
-end
-mt['table'] = function (self, source, mode, callback)
-    if mode == 'field' then
-        for i = 1, #source do
-            local src = source[i]
-            if src.type == 'tablefield'
-            or src.type == 'tableindex' then
-                callback(src)
-            end
-        end
-    end
-end
-mt['call'] = function (self, source, mode, callback)
-    local name = self:getSpecial(source)
-    if not name then
-        return
-    end
-    local parent = source.parent
-    if name == 's|setmetatable' then
-        if parent.index ~= 1 then
-            return
-        end
-        self:eachField(source.args[2], 's|__index', 'def', function (src)
-            self:eachField(src, nil, 'ref', callback)
-        end)
-        return
-    end
-end
+mt['field']     = require 'core.field'
+mt['method']    = require 'core.method'
+mt['number']    = require 'core.number'
+mt['boolean']   = require 'core.boolean'
+mt['string']    = require 'core.string'
+mt['table']     = require 'core.table'
+
+--mt['special'] = function (self, source, mode, callback)
+--    local name = self:getSpecial(source)
+--    if not name then
+--        return
+--    end
+--    if mode == 'def' then
+--        if name == 's|_G' then
+--            self:search(source, '_G', mode, callback)
+--        elseif name == 's|rawset' then
+--            callback(source.parent, 'set')
+--        end
+--    end
+--end
+--mt['call'] = function (self, source, mode, callback)
+--    local name = self:getSpecial(source)
+--    if not name then
+--        return
+--    end
+--    local parent = source.parent
+--    if name == 's|setmetatable' then
+--        if parent.index ~= 1 then
+--            return
+--        end
+--        self:eachField(source.args[2], 's|__index', 'def', function (src)
+--            self:eachField(src, nil, 'ref', callback)
+--        end)
+--        return
+--    end
+--end
 
 function mt:asindex(source, mode, callback)
     local parent = source.parent
@@ -279,7 +119,15 @@ end
 
 function mt:eachRef(source, mode, callback)
     local tp = source.type
-    self:search(source, tp, mode, callback)
+    local d = mt[tp]
+    if not d then
+        return
+    end
+    local f = d[mode]
+    if not f then
+        return
+    end
+    f(self, source, callback)
 end
 
 return function (ast)