diff options
Diffstat (limited to 'server-beta/src/core/engineer.lua')
| -rw-r--r-- | server-beta/src/core/engineer.lua | 256 |
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) |