diff options
Diffstat (limited to 'server-beta/src/core/engineer.lua')
| -rw-r--r-- | server-beta/src/core/engineer.lua | 94 |
1 files changed, 61 insertions, 33 deletions
diff --git a/server-beta/src/core/engineer.lua b/server-beta/src/core/engineer.lua index be64a66c..192de506 100644 --- a/server-beta/src/core/engineer.lua +++ b/server-beta/src/core/engineer.lua @@ -153,32 +153,15 @@ mt['getglobal'] = function (self, source, mode, callback) end end mt['setglobal'] = mt['getglobal'] -mt['eachfield'] = function (self, node, key, mode, callback) - self:eachRef(node, 'field', function (src) - if key == guide.getKeyName(src) then - if mode == 'def' then - if src.type == 'setfield' - or src.type == 'tablefield' then - callback(src.field, 'set') - elseif src.type == 'setindex' - or src.type == 'tableindex' then - callback(src.index, 'set') - elseif src.type == 'setmethod' then - callback(src.method, 'set') - end - end - end - end) -end mt['field'] = function (self, source, mode, callback) local node = source.parent.node local key = guide.getKeyName(source) - mt['eachfield'](self, node, key, mode, callback) + self:eachField(node, key, mode, callback) end mt['method'] = function (self, source, mode, callback) local node = source.parent.node local key = guide.getKeyName(source) - mt['eachfield'](self, node, key, mode, callback) + self:eachField(node, key, mode, callback) end mt['special'] = function (self, source, mode, callback) local name = self:getSpecial(source) @@ -193,22 +176,14 @@ mt['special'] = function (self, source, mode, callback) end end end -mt['asindex'] = function (self, source, mode, callback) - local parent = source.parent - if not parent then - return - end - if parent.type ~= 'setindex' and parent.type ~= 'getindex' then - return - end - local node = parent.node - local key = guide.getKeyName(source) - mt['eachfield'](self, node, key, mode, callback) +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['number'] = mt['asindex'] -mt['boolean'] = mt['asindex'] mt['string'] = function (self, source, mode, callback) - mt['asindex'](self, source, mode, callback) + self:asindex(source, mode, callback) end mt['table'] = function (self, source, mode, callback) if mode == 'field' then @@ -221,6 +196,41 @@ mt['table'] = function (self, source, mode, callback) end end end +mt['select'] = function (self, source, mode, callback) + local vararg = source.vararg + if vararg.type == 'call' then + local func = vararg.node + self:specialreturn(func, source.index, mode, callback) + end +end + +function mt:specialreturn(source, index, mode, callback) + local name = self:getSpecial(source) + if not name then + return + end + local call = source.parent + if name == 's|setmetatable' then + if index ~= 1 then + return + end + self:eachField(call.args[2], 's|__index', 'def', callback) + return + end +end + +function mt:asindex(source, mode, callback) + local parent = source.parent + if not parent then + return + end + if parent.type ~= 'setindex' and parent.type ~= 'getindex' then + return + end + local node = parent.node + local key = guide.getKeyName(source) + self:eachField(node, key, mode, callback) +end function mt:getSpecial(source) local node = source.node @@ -239,6 +249,24 @@ function mt:search(source, method, mode, callback) f(self, source, mode, callback) end +function mt:eachField(node, key, mode, callback) + self:eachRef(node, 'field', function (src) + if key == guide.getKeyName(src) then + if mode == 'def' then + if src.type == 'setfield' + or src.type == 'tablefield' then + callback(src.field, 'set') + elseif src.type == 'setindex' + or src.type == 'tableindex' then + callback(src.index, 'set') + elseif src.type == 'setmethod' then + callback(src.method, 'set') + end + end + end + end) +end + function mt:eachRef(source, mode, callback) local tp = source.type self:search(source, tp, mode, callback) |