diff options
Diffstat (limited to 'server/src/matcher/definition.lua')
| -rw-r--r-- | server/src/matcher/definition.lua | 548 |
1 files changed, 20 insertions, 528 deletions
diff --git a/server/src/matcher/definition.lua b/server/src/matcher/definition.lua index 26a51a4f..41556745 100644 --- a/server/src/matcher/definition.lua +++ b/server/src/matcher/definition.lua @@ -1,503 +1,12 @@ -local env = require 'matcher.env' -local mt = {} -mt.__index = mt -function mt:isContainPos(obj) - return obj.start <= self.pos and obj.finish + 1 >= self.pos +local function isContainPos(obj, pos) + return obj.start <= pos and obj.finish + 1 >= pos end -function mt:getVar(key, source) - if key == nil then - return nil - end - local var = self.env.var[key] - or self:getField(self.env.var._ENV, key, source) -- 这里不需要用getVar来递归获取_ENV - if not var then - var = self:addField(self:getVar '_ENV', key, source) - end - if var and var.meta then - var = var.meta - end - return var -end - -function mt:addInfo(obj, type, source) - if not obj then - return nil - end - obj[#obj+1] = { - type = type, - source = source, - } - return obj -end - -function mt:createVar(type, key, source) - local var = { - type = type, - key = key, - source = source, - childs = {}, - } - self.results.vars[#self.results.vars+1] = var - return var -end - -function mt:createLabel(key) - local lbl = { - key = key, - type = 'label', - } - self.results.labels[#self.results.labels+1] = lbl - return lbl -end - -function mt:createDots() - local dots = { - type = 'dots', - } - self.results.dots[#self.results.dots+1] = dots - return dots -end - -function mt:createLocal(key, source, var) - if key == nil then - return nil - end - if not var then - var = self:createVar('local', key, source) - end - self.env.var[key] = var - return var -end - -function mt:addField(parent, key, source) - if parent == nil or key == nil then - return nil - end - assert(source) - local var = parent.childs[key] - if not var then - var = self:createVar('field', key, source) - parent.childs[key] = var - end - return var -end - -function mt:getField(parent, key, source) - if parent == nil or key == nil then - return nil - end - local var - if parent.childs then - var = parent.childs[key] - end - if not var and source then - var = self:addField(parent, key, source) - end - return var -end - -function mt:checkName(name) - local var = self:getVar(name[1], name) - self:addInfo(var, 'get', name) - return var -end - -function mt:checkDots(source) - local dots = self.env.dots - if not dots then - return - end - self:addInfo(dots, 'get', source) -end - -function mt:searchCall(call, simple, i) - local results = {} - for i, exp in ipairs(call) do - results[i] = self:searchExp(exp) - end - - -- 特殊处理 setmetatable - if i == 2 and simple[1][1] == 'setmetatable' then - local obj = results[1] - local metatable = results[2] - if metatable then - local index = self:getField(metatable, '__index') - if obj then - self:setTable(obj, index, 'copy') - return obj - else - return index - end - else - return obj - end - end - return nil -end - -function mt:searchSimple(simple) - local name = simple[1] - local var - if name.type == 'name' then - var = self:getVar(name[1], name) - end - self:searchExp(simple[1]) - for i = 2, #simple do - local obj = simple[i] - local tp = obj.type - if tp == 'call' then - var = self:searchCall(obj, simple, i) - elseif tp == ':' then - elseif tp == 'name' then - if obj.index then - self:checkName(obj) - var = nil - else - var = self:getField(var, obj[1], obj) - if i ~= #simple then - self:addInfo(var, 'get', obj) - end - end - else - if obj.index then - if obj.type == 'string' or obj.type == 'number' or obj.type == 'boolean' then - var = self:getField(var, obj[1], obj) - if i ~= #simple then - self:addInfo(var, 'get', obj) - end - end - else - self:searchExp(obj) - var = nil - end - end - end - return var -end - -function mt:searchBinary(exp) - self:searchExp(exp[1]) - self:searchExp(exp[2]) -end - -function mt:searchUnary(exp) - return self:searchExp(exp[1]) -end - -function mt:searchTable(exp) - local tbl = { - type = 'table', - childs = {}, - } - for _, obj in ipairs(exp) do - if obj.type == 'pair' then - local key, value = obj[1], obj[2] - local res = self:searchExp(value) - local var = self:addField(tbl, key[1], key) - self:setTable(var, res) - self:addInfo(var, 'set', key) - else - self:searchExp(obj) - end - end - return tbl -end - -function mt:searchExp(exp) - local tp = exp.type - if tp == 'nil' then - elseif tp == 'string' then - elseif tp == 'boolean' then - elseif tp == 'number' then - elseif tp == 'name' then - return self:checkName(exp) - elseif tp == 'simple' then - return self:searchSimple(exp) - elseif tp == 'binary' then - self:searchBinary(exp) - elseif tp == 'unary' then - self:searchUnary(exp) - elseif tp == '...' then - self:checkDots(exp) - elseif tp == 'function' then - self:searchFunction(exp) - elseif tp == 'table' then - return self:searchTable(exp) - end - return nil -end - -function mt:searchReturn(action) - for _, exp in ipairs(action) do - self:searchExp(exp) - end -end - -function mt:setTable(var, tbl, mode) - if not var or not tbl then - return - end - if mode == 'copy' then - for k, v in pairs(var.childs) do - if tbl.childs[k] then - for i, info in ipairs(v) do - table.insert(tbl.childs[k], 1, info) - end - end - tbl.childs[k] = v - end - end - var.childs = tbl.childs -end - -function mt:markSimple(simple) - local name = simple[1] - local var = self:getVar(name[1], name) - for i = 2, #simple do - local obj = simple[i] - local tp = obj.type - if tp == ':' then - var = self:createLocal('self', simple[i-1], self:getVar(simple[i-1][1])) - elseif tp == 'name' then - if not obj.index then - var = self:addField(var, obj[1], obj) - if i == #simple then - self:addInfo(var, 'set', obj) - end - else - var = nil - end - else - if obj.index then - var = self:addField(var, obj[1], obj) - if i == #simple then - self:addInfo(var, 'set', obj) - end - else - var = nil - end - end - end - return var -end - -function mt:markSet(simple, tbl) - if simple.type == 'name' then - local var = self:getVar(simple[1], simple) - self:addInfo(var, 'set', simple) - self:setTable(var, tbl) - else - self:searchSimple(simple) - local var = self:markSimple(simple) - self:setTable(var, tbl) - end -end - -function mt:markLocal(name, tbl) - if name.type == 'name' then - local str = name[1] - -- 创建一个局部变量 - local var = self:createLocal(str, name) - self:setTable(var, tbl) - elseif name.type == '...' then - local dots = self:createDots() - self:addInfo(dots, 'local', name) - self.env.dots = dots - elseif name.type == ':' then - -- 创建一个局部变量 - self:createLocal('self', name) - end -end - -function mt:forList(list, callback) - if not list then - return - end - if list.type == 'list' then - for i = 1, #list do - callback(list[i]) - end - else - callback(list) - end -end - -function mt:markSets(action) - local keys = action[1] - local values = action[2] - local results = {} - -- 要先计算赋值 - local i = 0 - self:forList(values, function (value) - i = i + 1 - results[i] = self:searchExp(value) - end) - local i = 0 - self:forList(keys, function (key) - i = i + 1 - self:markSet(key, results[i]) - end) -end - -function mt:markLocals(action) - local keys = action[1] - local values = action[2] - local results = {} - -- 要先计算赋值 - local i = 0 - self:forList(values, function (value) - i = i + 1 - results[i] = self:searchExp(value) - end) - local i = 0 - self:forList(keys, function (key) - i = i + 1 - self:markLocal(key, results[i]) - end) -end - -function mt:searchIfs(action) - for _, block in ipairs(action) do - self.env:push() - if block.filter then - self:searchExp(block.filter) - end - self:searchActions(block) - self.env:pop() - end -end - -function mt:searchLoop(action) - self.env:push() - self:markLocal(action.arg) - self:searchExp(action.min) - self:searchExp(action.max) - if action.step then - self:searchExp(action.step) - end - self:searchActions(action) - self.env:pop() -end - -function mt:searchIn(action) - self:forList(action.exp, function (exp) - self:searchExp(exp) - end) - self.env:push() - self:forList(action.arg, function (arg) - self:markLocal(arg) - end) - self:searchActions(action) - self.env:pop() -end - -function mt:searchDo(action) - self.env:push() - self:searchActions(action) - self.env:pop() -end - -function mt:searchWhile(action) - self:searchExp(action.filter) - self.env:push() - self:searchActions(action) - self.env:pop() -end - -function mt:searchRepeat(action) - self.env:push() - self:searchActions(action) - self:searchExp(action.filter) - self.env:pop() -end - -function mt:searchFunction(func) - self.env:push() - self.env:cut 'dots' - self.env.label = {} - if func.name then - self:markSet(func.name) - end - self:forList(func.arg, function (arg) - self:markLocal(arg) - end) - self:searchActions(func) - self.env:pop() -end - -function mt:searchLocalFunction(func) - self:markLocal(func.name) - self.env:push() - self:forList(func.arg, function (arg) - self:markLocal(arg) - end) - self:searchActions(func) - self.env:pop() -end - -function mt:markLabel(label) - local str = label[1] - if not self.env.label[str] then - self.env.label[str] = self:createLabel(str) - end - self:addInfo(self.env.label[str], 'set', label) -end - -function mt:searchGoTo(obj) - local str = obj[1] - if not self.env.label[str] then - self.env.label[str] = self:createLabel(str) - end - self:addInfo(self.env.label[str], 'goto', obj) -end - -function mt:searchAction(action) - local tp = action.type - if tp == 'do' then - self:searchDo(action) - elseif tp == 'break' then - elseif tp == 'return' then - self:searchReturn(action) - elseif tp == 'label' then - self:markLabel(action) - elseif tp == 'goto' then - self:searchGoTo(action) - elseif tp == 'set' then - self:markSets(action) - elseif tp == 'local' then - self:markLocals(action) - elseif tp == 'simple' then - self:searchSimple(action) - elseif tp == 'if' then - self:searchIfs(action) - elseif tp == 'loop' then - self:searchLoop(action) - elseif tp == 'in' then - self:searchIn(action) - elseif tp == 'while' then - self:searchWhile(action) - elseif tp == 'repeat' then - self:searchRepeat(action) - elseif tp == 'function' then - self:searchFunction(action) - elseif tp == 'localfunction' then - self:searchLocalFunction(action) - end -end - -function mt:searchActions(actions) - for _, action in ipairs(actions) do - self:searchAction(action) - end - return nil -end - -function mt:definition() - for _, var in ipairs(self.results.vars) do +local function findResult(results, pos) + for _, var in ipairs(results.vars) do for _, info in ipairs(var) do - if self:isContainPos(info.source) then + if isContainPos(info.source, pos) then return { type = 'var', var = var, @@ -505,9 +14,9 @@ function mt:definition() end end end - for _, dots in ipairs(self.results.dots) do + for _, dots in ipairs(results.dots) do for _, info in ipairs(dots) do - if self:isContainPos(info.source) then + if isContainPos(info.source, pos) then return { type = 'dots', dots = dots, @@ -515,9 +24,9 @@ function mt:definition() end end end - for _, label in ipairs(self.results.labels) do + for _, label in ipairs(results.labels) do for _, info in ipairs(label) do - if self:isContainPos(info.source) then + if isContainPos(info.source, pos) then return { type = 'label', label = label, @@ -529,7 +38,7 @@ function mt:definition() end local function parseResult(result) - local results = {} + local positions = {} local tp = result.type if tp == 'var' then local var = result.var @@ -538,11 +47,11 @@ local function parseResult(result) if not source then return false end - results[1] = {source.start, source.finish} + positions[1] = {source.start, source.finish} elseif var.type == 'field' then for _, info in ipairs(var) do if info.type == 'set' then - results[#results+1] = {info.source.start, info.source.finish} + positions[#positions+1] = {info.source.start, info.source.finish} end end else @@ -552,44 +61,27 @@ local function parseResult(result) local dots = result.dots for _, info in ipairs(dots) do if info.type == 'local' then - results[#results+1] = {info.source.start, info.source.finish} + positions[#positions+1] = {info.source.start, info.source.finish} end end elseif tp == 'label' then local label = result.label for _, info in ipairs(label) do if info.type == 'set' then - results[#results+1] = {info.source.start, info.source.finish} + positions[#positions+1] = {info.source.start, info.source.finish} end end else error('unknow result.type:' .. result.type) end - return true, results + return positions end -return function (ast, pos) - local searcher = setmetatable({ - pos = pos, - env = env { - var = {}, - usable = {} - }, - results = { - labels = {}, - vars = {}, - dots = {}, - } - }, mt) - searcher.env.label = {} - searcher:createLocal('_ENV') - searcher:searchActions(ast) - - local result = searcher:definition() - +return function (results, pos) + local result = findResult(results, pos) if not result then - return false + return nil end - - return parseResult(result) + local positions = parseResult(result) + return positions end |