summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
author最萌小汐 <sumneko@hotmail.com>2021-04-14 20:13:00 +0800
committer最萌小汐 <sumneko@hotmail.com>2021-04-14 20:13:00 +0800
commitede808fe32a66ac828762818bf43d18fc343c159 (patch)
treea4cc8808fff798ea3bfd7103fb56bfb1d74f9c9c
parentbabe8e194576b0cd2ede04d98417a5f0e2a23647 (diff)
downloadlua-language-server-ede808fe32a66ac828762818bf43d18fc343c159.zip
stash
-rw-r--r--script/core/guide.lua4744
-rw-r--r--script/core/guide2.lua4837
-rw-r--r--script/core/linker.lua100
-rw-r--r--script/parser/guide.lua9
-rw-r--r--test/basic/linker.lua16
-rw-r--r--test/basic/linker.txt16
6 files changed, 5048 insertions, 4674 deletions
diff --git a/script/core/guide.lua b/script/core/guide.lua
index 64192297..e0123987 100644
--- a/script/core/guide.lua
+++ b/script/core/guide.lua
@@ -1,2754 +1,27 @@
-local util = require 'utility'
-local error = error
-local type = type
-local next = next
-local tostring = tostring
-local print = print
-local ipairs = ipairs
-local tableInsert = table.insert
-local tableUnpack = table.unpack
-local tableRemove = table.remove
-local tableMove = table.move
-local tableSort = table.sort
-local tableConcat = table.concat
-local mathType = math.type
-local pairs = pairs
-local setmetatable = setmetatable
-local assert = assert
-local select = select
-local osClock = os.clock
-local tonumber = tonumber
-local tointeger = math.tointeger
-local DEVELOP = _G.DEVELOP
-local log = log
-local _G = _G
+local linker = require 'core.linker'
+local guide = require 'parser.guide'
-local function logWarn(...)
- log.warn(...)
-end
+local osClock = os.clock
+local pairs = pairs
local m = {}
-m.ANY = {"<ANY>"}
-
-local blockTypes = {
- ['while'] = true,
- ['in'] = true,
- ['loop'] = true,
- ['repeat'] = true,
- ['do'] = true,
- ['function'] = true,
- ['ifblock'] = true,
- ['elseblock'] = true,
- ['elseifblock'] = true,
- ['main'] = true,
-}
-
-local breakBlockTypes = {
- ['while'] = true,
- ['in'] = true,
- ['loop'] = true,
- ['repeat'] = true,
-}
-
-m.childMap = {
- ['main'] = {'#', 'docs'},
- ['repeat'] = {'#', 'filter'},
- ['while'] = {'filter', '#'},
- ['in'] = {'keys', '#'},
- ['loop'] = {'loc', 'max', 'step', '#'},
- ['if'] = {'#'},
- ['ifblock'] = {'filter', '#'},
- ['elseifblock'] = {'filter', '#'},
- ['elseblock'] = {'#'},
- ['setfield'] = {'node', 'field', 'value'},
- ['setglobal'] = {'value'},
- ['local'] = {'attrs', 'value'},
- ['setlocal'] = {'value'},
- ['return'] = {'#'},
- ['do'] = {'#'},
- ['select'] = {'vararg'},
- ['table'] = {'#'},
- ['tableindex'] = {'index', 'value'},
- ['tablefield'] = {'field', 'value'},
- ['function'] = {'args', '#'},
- ['funcargs'] = {'#'},
- ['setmethod'] = {'node', 'method', 'value'},
- ['getmethod'] = {'node', 'method'},
- ['setindex'] = {'node', 'index', 'value'},
- ['getindex'] = {'node', 'index'},
- ['paren'] = {'exp'},
- ['call'] = {'node', 'args'},
- ['callargs'] = {'#'},
- ['getfield'] = {'node', 'field'},
- ['list'] = {'#'},
- ['binary'] = {1, 2},
- ['unary'] = {1},
-
- ['doc'] = {'#'},
- ['doc.class'] = {'class', '#extends', 'comment'},
- ['doc.type'] = {'#types', '#enums', 'name', 'comment'},
- ['doc.alias'] = {'alias', 'extends', 'comment'},
- ['doc.param'] = {'param', 'extends', 'comment'},
- ['doc.return'] = {'#returns', 'comment'},
- ['doc.field'] = {'field', 'extends', 'comment'},
- ['doc.generic'] = {'#generics', 'comment'},
- ['doc.generic.object'] = {'generic', 'extends', 'comment'},
- ['doc.vararg'] = {'vararg', 'comment'},
- ['doc.type.array'] = {'node'},
- ['doc.type.table'] = {'node', 'key', 'value', 'comment'},
- ['doc.type.function'] = {'#args', '#returns', 'comment'},
- ['doc.type.typeliteral'] = {'node'},
- ['doc.type.arg'] = {'extends'},
- ['doc.overload'] = {'overload', 'comment'},
- ['doc.see'] = {'name', 'field', 'comment'},
- ['doc.diagnostic'] = {'#names', 'comment'},
-}
-
-m.actionMap = {
- ['main'] = {'#'},
- ['repeat'] = {'#'},
- ['while'] = {'#'},
- ['in'] = {'#'},
- ['loop'] = {'#'},
- ['if'] = {'#'},
- ['ifblock'] = {'#'},
- ['elseifblock'] = {'#'},
- ['elseblock'] = {'#'},
- ['do'] = {'#'},
- ['function'] = {'#'},
- ['funcargs'] = {'#'},
-}
-
-local TypeSort = {
- ['boolean'] = 1,
- ['string'] = 2,
- ['integer'] = 3,
- ['number'] = 4,
- ['table'] = 5,
- ['function'] = 6,
- ['true'] = 101,
- ['false'] = 102,
- ['nil'] = 999,
-}
-
-local NIL = setmetatable({'<nil>'}, { __tostring = function () return 'nil' end })
-
---- 是否是字面量
----@param obj parser.guide.object
----@return boolean
-function m.isLiteral(obj)
- local tp = obj.type
- return tp == 'nil'
- or tp == 'boolean'
- or tp == 'string'
- or tp == 'number'
- or tp == 'table'
- or tp == 'function'
-end
-
---- 获取字面量
----@param obj parser.guide.object
----@return any
-function m.getLiteral(obj)
- local tp = obj.type
- if tp == 'boolean' then
- return obj[1]
- elseif tp == 'string' then
- return obj[1]
- elseif tp == 'number' then
- return obj[1]
- end
- return nil
-end
-
---- 寻找父函数
----@param obj parser.guide.object
----@return parser.guide.object
-function m.getParentFunction(obj)
- for _ = 1, 1000 do
- obj = obj.parent
- if not obj then
- break
- end
- local tp = obj.type
- if tp == 'function' or tp == 'main' then
- return obj
- end
- end
- return nil
-end
-
---- 寻找父的table类型 doc.type.table
----@param obj parser.guide.object
----@return parser.guide.object
-function m.getParentDocTypeTable(obj)
- for _ = 1, 1000 do
- local parent = obj.parent
- if not parent then
- return nil
- end
- if parent.type == 'doc.type.table' then
- return obj
- end
- obj = parent
- end
- error('guide.getParentDocTypeTable overstack')
-end
-
---- 寻找所在区块
----@param obj parser.guide.object
----@return parser.guide.object
-function m.getBlock(obj)
- for _ = 1, 1000 do
- if not obj then
- return nil
- end
- local tp = obj.type
- if blockTypes[tp] then
- return obj
- end
- if obj == obj.parent then
- error('obj == obj.parent?', obj.type)
- end
- obj = obj.parent
- end
- error('guide.getBlock overstack')
-end
-
---- 寻找所在父区块
----@param obj parser.guide.object
----@return parser.guide.object
-function m.getParentBlock(obj)
- for _ = 1, 1000 do
- obj = obj.parent
- if not obj then
- return nil
- end
- local tp = obj.type
- if blockTypes[tp] then
- return obj
- end
- end
- error('guide.getParentBlock overstack')
-end
-
---- 寻找所在可break的父区块
----@param obj parser.guide.object
----@return parser.guide.object
-function m.getBreakBlock(obj)
- for _ = 1, 1000 do
- obj = obj.parent
- if not obj then
- return nil
- end
- local tp = obj.type
- if breakBlockTypes[tp] then
- return obj
- end
- if tp == 'function' then
- return nil
- end
- end
- error('guide.getBreakBlock overstack')
-end
-
---- 寻找doc的主体
----@param obj parser.guide.object
----@return parser.guide.object
-function m.getDocState(obj)
- for _ = 1, 1000 do
- local parent = obj.parent
- if not parent then
- return obj
- end
- if parent.type == 'doc' then
- return obj
- end
- obj = parent
- end
- error('guide.getDocState overstack')
-end
-
---- 寻找所在父类型
----@param obj parser.guide.object
----@return parser.guide.object
-function m.getParentType(obj, want)
- for _ = 1, 1000 do
- obj = obj.parent
- if not obj then
- return nil
- end
- if want == obj.type then
- return obj
- end
- end
- error('guide.getParentType overstack')
-end
-
---- 寻找根区块
----@param obj parser.guide.object
----@return parser.guide.object
-function m.getRoot(obj)
- local source = obj
- if source._root then
- return source._root
- end
- for _ = 1, 1000 do
- if obj.type == 'main' then
- source._root = obj
- return obj
- end
- local parent = obj.parent
- if not parent then
- return nil
- end
- obj = parent
- end
- error('guide.getRoot overstack')
-end
-
----@param obj parser.guide.object
----@return string
-function m.getUri(obj)
- if obj.uri then
- return obj.uri
- end
- local root = m.getRoot(obj)
- if root then
- return root.uri
- end
- return ''
-end
-
-function m.getENV(source, start)
- if not start then
- start = 1
- end
- return m.getLocal(source, '_ENV', start)
- or m.getLocal(source, '@fenv', start)
-end
-
---- 寻找函数的不定参数,返回不定参在第几个参数上,以及该参数对象。
---- 如果函数是主函数,则返回`0, nil`。
----@return table
----@return integer
-function m.getFunctionVarArgs(func)
- if func.type == 'main' then
- return 0, nil
- end
- if func.type ~= 'function' then
- return nil, nil
- end
- local args = func.args
- if not args then
- return nil, nil
- end
- for i = 1, #args do
- local arg = args[i]
- if arg.type == '...' then
- return i, arg
- end
- end
- return nil, nil
-end
-
---- 获取指定区块中可见的局部变量
----@param block table
----@param name string {comment = '变量名'}
----@param pos integer {comment = '可见位置'}
-function m.getLocal(block, name, pos)
- block = m.getBlock(block)
- for _ = 1, 1000 do
- if not block then
- return nil
- end
- local locals = block.locals
- local res
- if not locals then
- goto CONTINUE
- end
- for i = 1, #locals do
- local loc = locals[i]
- if loc.effect > pos then
- break
- end
- if loc[1] == name then
- if not res or res.effect < loc.effect then
- res = loc
- end
- end
- end
- if res then
- return res, res
- end
- ::CONTINUE::
- block = m.getParentBlock(block)
- end
- error('guide.getLocal overstack')
-end
-
---- 获取指定区块中所有的可见局部变量名称
-function m.getVisibleLocals(block, pos)
- local result = {}
- m.eachSourceContain(m.getRoot(block), pos, function (source)
- local locals = source.locals
- if locals then
- for i = 1, #locals do
- local loc = locals[i]
- local name = loc[1]
- if loc.effect <= pos then
- result[name] = loc
- end
- end
- end
- end)
- return result
-end
-
---- 获取指定区块中可见的标签
----@param block table
----@param name string {comment = '标签名'}
-function m.getLabel(block, name)
- block = m.getBlock(block)
- for _ = 1, 1000 do
- if not block then
- return nil
- end
- local labels = block.labels
- if labels then
- local label = labels[name]
- if label then
- return label
- end
- end
- if block.type == 'function' then
- return nil
- end
- block = m.getParentBlock(block)
- end
- error('guide.getLocal overstack')
-end
-
-function m.getStartFinish(source)
- local start = source.start
- local finish = source.finish
- if not start then
- local first = source[1]
- if not first then
- return nil, nil
- end
- local last = source[#source]
- start = first.start
- finish = last.finish
- end
- return start, finish
-end
-
-function m.getRange(source)
- local start = source.vstart or source.start
- local finish = source.range or source.finish
- if not start then
- local first = source[1]
- if not first then
- return nil, nil
- end
- local last = source[#source]
- start = first.vstart or first.start
- finish = last.range or last.finish
- end
- return start, finish
-end
-
---- 判断source是否包含offset
-function m.isContain(source, offset)
- local start, finish = m.getStartFinish(source)
- if not start then
- return false
- end
- return start <= offset and finish >= offset
-end
-
---- 判断offset在source的影响范围内
----
---- 主要针对赋值等语句时,key包含value
-function m.isInRange(source, offset)
- local start, finish = m.getRange(source)
- if not start then
- return false
- end
- return start <= offset and finish >= offset
-end
-
-function m.isBetween(source, tStart, tFinish)
- local start, finish = m.getStartFinish(source)
- if not start then
- return false
- end
- return start <= tFinish and finish >= tStart
-end
-
-function m.isBetweenRange(source, tStart, tFinish)
- local start, finish = m.getRange(source)
- if not start then
- return false
- end
- return start <= tFinish and finish >= tStart
-end
-
---- 添加child
-function m.addChilds(list, obj, map)
- local keys = map[obj.type]
- if keys then
- for i = 1, #keys do
- local key = keys[i]
- if key == '#' then
- for i = 1, #obj do
- list[#list+1] = obj[i]
- end
- elseif obj[key] then
- list[#list+1] = obj[key]
- elseif type(key) == 'string'
- and key:sub(1, 1) == '#' then
- key = key:sub(2)
- if obj[key] then
- for i = 1, #obj[key] do
- list[#list+1] = obj[key][i]
- end
- end
- end
- end
- end
-end
-
---- 遍历所有包含offset的source
-function m.eachSourceContain(ast, offset, callback)
- local list = { ast }
- local mark = {}
- while true do
- local len = #list
- if len == 0 then
- return
- end
- local obj = list[len]
- list[len] = nil
- if not mark[obj] then
- mark[obj] = true
- if m.isInRange(obj, offset) then
- if m.isContain(obj, offset) then
- local res = callback(obj)
- if res ~= nil then
- return res
- end
- end
- m.addChilds(list, obj, m.childMap)
- end
- end
- end
-end
-
---- 遍历所有在某个范围内的source
-function m.eachSourceBetween(ast, start, finish, callback)
- local list = { ast }
- local mark = {}
- while true do
- local len = #list
- if len == 0 then
- return
- end
- local obj = list[len]
- list[len] = nil
- if not mark[obj] then
- mark[obj] = true
- if m.isBetweenRange(obj, start, finish) then
- if m.isBetween(obj, start, finish) then
- local res = callback(obj)
- if res ~= nil then
- return res
- end
- end
- m.addChilds(list, obj, m.childMap)
- end
- end
- end
-end
-
---- 遍历所有指定类型的source
-function m.eachSourceType(ast, type, callback)
- local cache = ast._typeCache
- if not cache then
- cache = {}
- ast._typeCache = cache
- m.eachSource(ast, function (source)
- local tp = source.type
- if not tp then
- return
- end
- local myCache = cache[tp]
- if not myCache then
- myCache = {}
- cache[tp] = myCache
- end
- myCache[#myCache+1] = source
- end)
- end
- local myCache = cache[type]
- if not myCache then
- return
- end
- for i = 1, #myCache do
- callback(myCache[i])
- end
-end
-
---- 遍历所有的source
-function m.eachSource(ast, callback)
- local list = { ast }
- local mark = {}
- local index = 1
- while true do
- local obj = list[index]
- if not obj then
- return
- end
- list[index] = false
- index = index + 1
- if not mark[obj] then
- mark[obj] = true
- callback(obj)
- m.addChilds(list, obj, m.childMap)
- end
- end
-end
-
---- 获取指定的 special
-function m.eachSpecialOf(ast, name, callback)
- local root = m.getRoot(ast)
- if not root.specials then
- return
- end
- local specials = root.specials[name]
- if not specials then
- return
- end
- for i = 1, #specials do
- callback(specials[i])
- end
-end
-
---- 获取偏移对应的坐标
----@param lines table
----@return integer row
----@return integer col
-function m.positionOf(lines, offset)
- if offset < 1 then
- return 0, 0
- end
- local lastLine = lines[#lines]
- if offset > lastLine.finish then
- return #lines, offset - lastLine.start
- end
- local min = 1
- local max = #lines
- for _ = 1, 100 do
- if max <= min then
- local line = lines[min]
- return min, offset - line.start + 1
- end
- local row = (max - min) // 2 + min
- local line = lines[row]
- if offset < line.start then
- max = row - 1
- elseif offset > line.finish then
- min = row + 1
- else
- return row, offset - line.start + 1
- end
- end
- error('Stack overflow!')
-end
-
---- 获取坐标对应的偏移
----@param lines table
----@param row integer
----@param col integer
----@return integer {name = 'offset'}
-function m.offsetOf(lines, row, col)
- if row < 1 then
- return 0
- end
- if row > #lines then
- local lastLine = lines[#lines]
- return lastLine.finish
- end
- local line = lines[row]
- local len = line.finish - line.start + 1
- if col < 0 then
- return line.start
- elseif col > len then
- return line.finish
- else
- return line.start + col - 1
- end
-end
-
-function m.lineContent(lines, text, row, ignoreNL)
- local line = lines[row]
- if not line then
- return ''
- end
- if ignoreNL then
- return text:sub(line.start, line.range)
- else
- return text:sub(line.start, line.finish)
- end
-end
-
-function m.lineRange(lines, row, ignoreNL)
- local line = lines[row]
- if not line then
- return 0, 0
- end
- if ignoreNL then
- return line.start, line.range
- else
- return line.start, line.finish
- end
-end
-
-function m.lineData(lines, row)
- return lines[row]
-end
-
-function m.getKeyTypeOfLiteral(obj)
- if not obj then
- return nil
- end
- local tp = obj.type
- if tp == 'field'
- or tp == 'method' then
- return 'string'
- elseif tp == 'string' then
- return 'string'
- elseif tp == 'number' then
- return 'number'
- elseif tp == 'boolean' then
- return 'boolean'
- end
-end
-
-function m.getKeyType(obj)
- if not obj then
- return nil
- end
- local tp = obj.type
- if tp == 'getglobal'
- or tp == 'setglobal' then
- return 'string'
- elseif tp == 'local'
- or tp == 'getlocal'
- or tp == 'setlocal' then
- return 'local'
- elseif tp == 'getfield'
- or tp == 'setfield'
- or tp == 'tablefield' then
- return 'string'
- elseif tp == 'getmethod'
- or tp == 'setmethod' then
- return 'string'
- elseif tp == 'getindex'
- or tp == 'setindex'
- or tp == 'tableindex' then
- return m.getKeyTypeOfLiteral(obj.index)
- elseif tp == 'field'
- or tp == 'method'
- or tp == 'doc.see.field' then
- return 'string'
- elseif tp == 'doc.class' then
- return 'string'
- elseif tp == 'doc.alias' then
- return 'string'
- elseif tp == 'doc.field' then
- return 'string'
- elseif tp == 'dummy' then
- return 'string'
- end
- return m.getKeyTypeOfLiteral(obj)
-end
-
-function m.getKeyNameOfLiteral(obj)
- if not obj then
- return nil
- end
- local tp = obj.type
- if tp == 'field'
- or tp == 'method' then
- return obj[1]
- elseif tp == 'string' then
- local s = obj[1]
- if s then
- return s
- end
- elseif tp == 'number' then
- local n = obj[1]
- if n then
- return ('%s'):format(util.viewLiteral(obj[1]))
- end
- elseif tp == 'boolean' then
- local b = obj[1]
- if b then
- return tostring(b)
- end
- end
-end
-
-function m.getKeyName(obj)
- if not obj then
- return nil
- end
- local tp = obj.type
- if tp == 'getglobal'
- or tp == 'setglobal' then
- return obj[1]
- elseif tp == 'local'
- or tp == 'getlocal'
- or tp == 'setlocal' then
- return obj[1]
- elseif tp == 'getfield'
- or tp == 'setfield'
- or tp == 'tablefield' then
- if obj.field then
- return obj.field[1]
- end
- elseif tp == 'getmethod'
- or tp == 'setmethod' then
- if obj.method then
- return obj.method[1]
- end
- elseif tp == 'getindex'
- or tp == 'setindex'
- or tp == 'tableindex' then
- return m.getKeyNameOfLiteral(obj.index)
- elseif tp == 'field'
- or tp == 'method'
- or tp == 'doc.see.field' then
- return obj[1]
- elseif tp == 'doc.class' then
- return obj.class[1]
- elseif tp == 'doc.alias' then
- return obj.alias[1]
- elseif tp == 'doc.field' then
- return obj.field[1]
- elseif tp == 'dummy' then
- return obj[1]
- end
- return m.getKeyNameOfLiteral(obj)
-end
-
-function m.getSimpleName(obj)
- if obj.type == 'call' then
- local node = obj.node
- if not node then
- return
- end
- if node.special == 'rawset'
- or node.special == 'rawget' then
- local key = obj.args and obj.args[2]
- return m.getKeyName(key)
- end
- return ('%p'):format(obj)
- elseif obj.type == 'table' then
- return ('%p'):format(obj)
- elseif obj.type == 'select' then
- return ('%p'):format(obj)
- elseif obj.type == 'string' then
- return ('%p'):format(obj)
- elseif obj.type == 'doc.class.name'
- or obj.type == 'doc.type.name'
- or obj.type == 'doc.see.name' then
- return ('%s'):format(obj[1])
- elseif obj.type == 'doc.class' then
- return ('%s'):format(obj.class[1])
- end
- return m.getKeyName(obj)
-end
-
---- 测试 a 到 b 的路径(不经过函数,不考虑 goto),
---- 每个路径是一个 block 。
----
---- 如果 a 在 b 的前面,返回 `"before"` 加上 2个`list<block>`
----
---- 如果 a 在 b 的后面,返回 `"after"` 加上 2个`list<block>`
----
---- 否则返回 `false`
----
---- 返回的2个 `list` 分别为基准block到达 a 与 b 的路径。
----@param a table
----@param b table
----@return string|boolean mode
----@return table pathA?
----@return table pathB?
-function m.getPath(a, b, sameFunction)
- --- 首先测试双方在同一个函数内
- if sameFunction and m.getParentFunction(a) ~= m.getParentFunction(b) then
- return false
- end
- local mode
- local objA
- local objB
- if a.finish < b.start then
- mode = 'before'
- objA = a
- objB = b
- elseif a.start > b.finish then
- mode = 'after'
- objA = b
- objB = a
- else
- return 'equal', {}, {}
- end
- local pathA = {}
- local pathB = {}
- for _ = 1, 1000 do
- objA = m.getParentBlock(objA)
- pathA[#pathA+1] = objA
- if (not sameFunction and objA.type == 'function') or objA.type == 'main' then
- break
- end
- end
- for _ = 1, 1000 do
- objB = m.getParentBlock(objB)
- pathB[#pathB+1] = objB
- if (not sameFunction and objA.type == 'function') or objB.type == 'main' then
- break
- end
- end
- -- pathA: {1, 2, 3, 4, 5}
- -- pathB: {5, 6, 2, 3}
- local top = #pathB
- local start
- for i = #pathA, 1, -1 do
- local currentBlock = pathA[i]
- if currentBlock == pathB[top] then
- start = i
- break
- end
- end
- if not start then
- return nil
- end
- -- pathA: { 1, 2, 3}
- -- pathB: {5, 6, 2, 3}
- local extra = 0
- local align = top - start
- for i = start, 1, -1 do
- local currentA = pathA[i]
- local currentB = pathB[i+align]
- if currentA ~= currentB then
- extra = i
- break
- end
- end
- -- pathA: {1}
- local resultA = {}
- for i = extra, 1, -1 do
- resultA[#resultA+1] = pathA[i]
- end
- -- pathB: {5, 6}
- local resultB = {}
- for i = extra + align, 1, -1 do
- resultB[#resultB+1] = pathB[i]
- end
- return mode, resultA, resultB
-end
-
--- 根据语法,单步搜索定义
-local function stepRefOfLocal(loc, mode)
- local results = {}
- if loc.start ~= 0 then
- results[#results+1] = loc
- end
- local refs = loc.ref
- if not refs then
- return results
- end
- for i = 1, #refs do
- local ref = refs[i]
- if ref.start == 0 then
- goto CONTINUE
- end
- if mode == 'def' then
- if ref.type == 'local'
- or ref.type == 'setlocal' then
- results[#results+1] = ref
- end
- else
- if ref.type == 'local'
- or ref.type == 'setlocal'
- or ref.type == 'getlocal' then
- results[#results+1] = ref
- end
- end
- ::CONTINUE::
- end
- return results
-end
-
-local function stepRefOfLabel(label, mode)
- local results = { label }
- if not label or mode == 'def' then
- return results
- end
- local refs = label.ref
- if not refs then
- return results
- end
- for i = 1, #refs do
- local ref = refs[i]
- results[#results+1] = ref
- end
- return results
-end
-
-local function stepRefOfDocType(status, obj, mode)
- local results = {}
- if obj.type == 'doc.class.name'
- or obj.type == 'doc.type.name'
- or obj.type == 'doc.alias.name'
- or obj.type == 'doc.extends.name'
- or obj.type == 'doc.see.name' then
- local name = obj[1]
- if not name or not status.interface.docType then
- return results
- end
- if name == 'nil'
- or name == 'any'
- or name == 'boolean'
- or name == 'string'
- or name == 'table'
- or name == 'number'
- or name == 'integer'
- or name == 'function'
- or name == 'table'
- or name == 'thread'
- or name == 'userdata'
- or name == 'lightuserdata' then
- mode = 'def'
- end
- local docs = status.interface.docType(name)
- for i = 1, #docs do
- local doc = docs[i]
- if mode == 'def' then
- if doc.type == 'doc.class.name'
- or doc.type == 'doc.alias.name' then
- results[#results+1] = doc
- end
- else
- results[#results+1] = doc
- end
- end
- else
- results[#results+1] = obj
- end
- return results
-end
-
-function m.getStepRef(status, obj, mode)
- if obj.type == 'getlocal'
- or obj.type == 'setlocal' then
- return stepRefOfLocal(obj.node, mode)
- end
- if obj.type == 'local' then
- return stepRefOfLocal(obj, mode)
- end
- if obj.type == 'label' then
- return stepRefOfLabel(obj, mode)
- end
- if obj.type == 'goto' then
- return stepRefOfLabel(obj.node, mode)
- end
- if obj.type == 'doc.class.name'
- or obj.type == 'doc.type.name'
- or obj.type == 'doc.extends.name'
- or obj.type == 'doc.alias.name' then
- return stepRefOfDocType(status, obj, mode)
- end
- if obj.type == 'function' then
- return { obj }
- end
- return nil
-end
-
--- 根据语法,单步搜索field
-local function stepFieldOfLocal(loc)
- local results = {}
- local refs = loc.ref
- for i = 1, #refs do
- local ref = refs[i]
- if ref.type == 'setglobal'
- or ref.type == 'getglobal' then
- results[#results+1] = ref
- elseif ref.type == 'getlocal' then
- local nxt = ref.next
- if nxt then
- if nxt.type == 'setfield'
- or nxt.type == 'getfield'
- or nxt.type == 'setmethod'
- or nxt.type == 'getmethod'
- or nxt.type == 'setindex'
- or nxt.type == 'getindex' then
- results[#results+1] = nxt
- end
- end
- end
- end
- return results
-end
-local function stepFieldOfTable(tbl)
- local result = {}
- for i = 1, #tbl do
- result[i] = tbl[i]
- end
- return result
-end
-function m.getStepField(obj)
- if obj.type == 'getlocal'
- or obj.type == 'setlocal' then
- return stepFieldOfLocal(obj.node)
- end
- if obj.type == 'local' then
- return stepFieldOfLocal(obj)
- end
- if obj.type == 'table' then
- return stepFieldOfTable(obj)
- end
-end
-
-local function convertSimpleList(list)
- local simple = {}
- for i = #list, 1, -1 do
- local c = list[i]
- if c.type == 'getglobal'
- or c.type == 'setglobal' then
- if c.special == '_G' then
- simple.mode = 'global'
- goto CONTINUE
- end
- local loc = c.node
- if loc.special == '_G' then
- simple.mode = 'global'
- if not simple.node then
- simple.node = c
- end
- else
- simple.mode = 'local'
- simple[#simple+1] = m.getSimpleName(loc)
- if not simple.node then
- simple.node = loc
- end
- end
- elseif c.type == 'getlocal'
- or c.type == 'setlocal' then
- if c.special == '_G' then
- simple.mode = 'global'
- goto CONTINUE
- end
- simple.mode = 'local'
- if not simple.node then
- simple.node = c.node
- end
- elseif c.type == 'local' then
- simple.mode = 'local'
- if not simple.node then
- simple.node = c
- end
- else
- if not simple.node then
- simple.node = c
- end
- end
- simple[#simple+1] = m.getSimpleName(c) or m.ANY
- ::CONTINUE::
- end
- if simple.mode == 'global' and #simple == 0 then
- simple[1] = '_G'
- simple.node = list[#list]
- end
- return simple
-end
-
--- 搜索 `a.b.c` 的等价表达式
-local function buildSimpleList(obj, max)
- local list = {}
- local cur = obj
- local limit = max and (max + 1) or 11
- for i = 1, max or limit do
- if i == limit then
- return nil
- end
- while cur.type == 'paren' do
- cur = cur.exp
- if not cur then
- return nil
- end
- end
- if cur.type == 'setfield'
- or cur.type == 'getfield'
- or cur.type == 'setmethod'
- or cur.type == 'getmethod'
- or cur.type == 'setindex'
- or cur.type == 'getindex' then
- list[i] = cur
- cur = cur.node
- elseif cur.type == 'tablefield'
- or cur.type == 'tableindex' then
- list[i] = cur
- cur = cur.parent.parent
- if cur.type == 'return' then
- list[i+1] = list[i].parent
- break
- end
- elseif cur.type == 'getlocal'
- or cur.type == 'setlocal'
- or cur.type == 'local' then
- list[i] = cur
- break
- elseif cur.type == 'setglobal'
- or cur.type == 'getglobal' then
- list[i] = cur
- break
- elseif cur.type == 'select'
- or cur.type == 'table'
- or cur.type == 'call' then
- list[i] = cur
- break
- elseif cur.type == 'string' then
- list[i] = cur
- break
- elseif cur.type == '...' then
- list[i] = cur
- break
- elseif cur.type == 'doc.class.name'
- or cur.type == 'doc.type.name'
- or cur.type == 'doc.class'
- or cur.type == 'doc.see.name' then
- list[i] = cur
- break
- elseif cur.type == 'doc.see.field' then
- list[i] = cur
- cur = cur.parent.name
- elseif cur.type == 'function'
- or cur.type == 'main' then
- break
- else
- return nil
- end
- end
- return convertSimpleList(list)
-end
-
-function m.getSimple(obj, max)
- local simpleList
- if obj.type == 'getfield'
- or obj.type == 'setfield'
- or obj.type == 'getmethod'
- or obj.type == 'setmethod'
- or obj.type == 'getindex'
- or obj.type == 'setindex'
- or obj.type == 'local'
- or obj.type == 'getlocal'
- or obj.type == 'setlocal'
- or obj.type == 'setglobal'
- or obj.type == 'getglobal'
- or obj.type == 'tablefield'
- or obj.type == 'tableindex'
- or obj.type == 'select'
- or obj.type == 'call'
- or obj.type == 'table'
- or obj.type == 'string'
- or obj.type == '...'
- or obj.type == 'doc.class.name'
- or obj.type == 'doc.class'
- or obj.type == 'doc.type.name'
- or obj.type == 'doc.see.name'
- or obj.type == 'doc.see.field' then
- simpleList = buildSimpleList(obj, max)
- elseif obj.type == 'field'
- or obj.type == 'method' then
- simpleList = buildSimpleList(obj.parent, max)
- end
- return simpleList
-end
-
----Create a new status
----@param parentStatus core.guide.status
----@param interface table
----@param deep boolean
----@return core.guide.status
-function m.status(parentStatus, interface, deep)
- ---@class core.guide.status
- local status = {
- share = parentStatus and parentStatus.share or {
- count = 0,
- cacheLock = {},
- },
- depth = parentStatus and (parentStatus.depth + 1) or 0,
- searchDeep= parentStatus and parentStatus.searchDeep or deep or -999,
- interface = parentStatus and parentStatus.interface or {},
- deep = parentStatus and parentStatus.deep,
- clock = parentStatus and parentStatus.clock or osClock(),
- results = {},
- }
- if interface then
- for k, v in pairs(interface) do
- status.interface[k] = v
- end
- end
- status.deep = status.depth <= status.searchDeep
- return status
-end
-
-function m.copyStatusResults(a, b)
- local ra = a.results
- local rb = b.results
- for i = 1, #rb do
- ra[#ra+1] = rb[i]
- end
-end
-
-function m.isGlobal(source)
- if source.type == 'setglobal'
- or source.type == 'getglobal' then
- if source.node and source.node.tag == '_ENV' then
- return true
- end
- end
- if source.type == 'field' then
- source = source.parent
- end
- if source.type == 'getfield'
- or source.type == 'setfield' then
- local node = source.node
- if node and node.special == '_G' then
- return true
- end
- end
- return false
-end
-
-function m.isDoc(source)
- return source.type:sub(1, 4) == 'doc.'
-end
-
-function m.isDocClass(source)
- return source.type == 'doc.class'
-end
-
-function m.isSet(source)
- local tp = source.type
- if tp == 'setglobal'
- or tp == 'local'
- or tp == 'setlocal'
- or tp == 'setfield'
- or tp == 'setmethod'
- or tp == 'setindex'
- or tp == 'tablefield'
- or tp == 'tableindex' then
- return true
- end
- if tp == 'call' then
- local special = m.getSpecial(source.node)
- if special == 'rawset' then
- return true
- end
- end
- return false
-end
-
-function m.isGet(source)
- local tp = source.type
- if tp == 'getglobal'
- or tp == 'getlocal'
- or tp == 'getfield'
- or tp == 'getmethod'
- or tp == 'getindex' then
- return true
- end
- if tp == 'call' then
- local special = m.getSpecial(source.node)
- if special == 'rawget' then
- return true
- end
- end
- return false
-end
-
-function m.getSpecial(source)
- if not source then
- return nil
- end
- return source.special
-end
-
---- 根据函数的调用参数,获取:调用,参数索引
-function m.getCallAndArgIndex(callarg)
- local callargs = callarg.parent
- if not callargs or callargs.type ~= 'callargs' then
- return nil
- end
- local index
- for i = 1, #callargs do
- if callargs[i] == callarg then
- index = i
- break
- end
- end
- local call = callargs.parent
- return call, index
-end
-
---- 根据函数调用的返回值,获取:调用的函数,参数列表,自己是第几个返回值
-function m.getCallValue(source)
- local value = m.getObjectValue(source) or source
- if not value then
- return
- end
- local call, index
- if value.type == 'call' then
- call = value
- index = 1
- elseif value.type == 'select' then
- call = value.vararg
- index = value.index
- if call.type ~= 'call' then
- return
- end
- else
- return
- end
- if call.node.special == 'pcall'
- or call.node.special == 'xpcall' then
- return call.args and call.args[1], call.args, index - 1
- end
- return call.node, call.args, index
-end
-
-function m.getNextRef(ref)
- local nextRef = ref.next
- if nextRef then
- if nextRef.type == 'setfield'
- or nextRef.type == 'getfield'
- or nextRef.type == 'setmethod'
- or nextRef.type == 'getmethod'
- or nextRef.type == 'setindex'
- or nextRef.type == 'getindex' then
- return nextRef
- end
- end
- -- 穿透 rawget 与 rawset
- local call, index = m.getCallAndArgIndex(ref)
- if call then
- if call.node.special == 'rawset' and index == 1 then
- return call
- end
- if call.node.special == 'rawget' and index == 1 then
- return call
- end
- end
-
- return nil
-end
-
-function m.checkSameSimpleInValueOfTable(status, value, start, pushQueue)
- if value.type ~= 'table' then
- return
- end
- for i = 1, #value do
- local field = value[i]
- pushQueue(field, start + 1)
- end
-end
-
-function m.checkStatusDepth(status)
- if status.depth <= 20 then
- return true
- end
- if m.debugMode then
- error('status.depth overflow')
- elseif DEVELOP then
- --log.warn(debug.traceback('status.depth overflow'))
- logWarn('status.depth overflow')
- end
- return false
-end
-
-function m.searchFields(status, obj, key)
- if not m.checkStatusDepth(status) then
- return
- end
- local simple = m.getSimple(obj)
- if not simple then
- return
- end
- simple[#simple+1] = key or m.ANY
- m.searchSameFields(status, simple, 'field')
- m.cleanResults(status.results)
-end
-
-function m.searchDefFields(status, obj, key)
- if not m.checkStatusDepth(status) then
- return
- end
- local simple = m.getSimple(obj)
- if not simple then
- return
- end
- simple[#simple+1] = key or m.ANY
- m.searchSameFields(status, simple, 'deffield')
- m.cleanResults(status.results)
-end
-
----@param obj parser.guide.object
----@return parser.guide.object
-function m.getObjectValue(obj)
- while obj.type == 'paren' do
- obj = obj.exp
- if not obj then
- return nil
- end
- end
- if obj.type == 'boolean'
- or obj.type == 'number'
- or obj.type == 'integer'
- or obj.type == 'string'
- or obj.type == 'doc.type.table'
- or obj.type == 'doc.type.arrary' then
- return obj
- end
- if obj.value then
- return obj.value
- end
- if obj.type == 'field'
- or obj.type == 'method' then
- return obj.parent and obj.parent.value
- end
- if obj.type == 'call' then
- if obj.node.special == 'rawset' then
- return obj.args and obj.args[3]
- else
- return obj
- end
- end
- if obj.type == 'select' then
- return obj
- end
- return nil
-end
-
-function m.checkSameSimpleInValueInMetaTable(status, mt, start, pushQueue)
- local cache, makeCache = m.getRefCache(status, mt, '__index')
- if cache then
- for _, obj in ipairs(cache) do
- pushQueue(obj, start, true)
- end
- return
- end
- cache = {}
- local newStatus = m.status(status)
- m.searchDefFields(newStatus, mt, '__index')
- local refsStatus = m.status(status)
- for i = 1, #newStatus.results do
- local indexValue = m.getObjectValue(newStatus.results[i])
- if indexValue then
- m.searchRefs(refsStatus, indexValue, 'ref')
- end
- end
- for i = 1, #refsStatus.results do
- local obj = refsStatus.results[i]
- pushQueue(obj, start, true)
- cache[i] = obj
- end
- if makeCache then
- makeCache(cache)
- end
-end
-function m.checkSameSimpleInValueOfSetMetaTable(status, func, start, pushQueue)
- if not func or func.special ~= 'setmetatable' then
- return
- end
- local call = func.parent
- local args = call.args
- if not args then
- return
- end
- local obj = args[1]
- local mt = args[2]
- if obj then
- pushQueue(obj, start, true)
- local newStatus = m.status(status)
- m.searchRefs(newStatus, obj, 'def')
- for _, def in ipairs(newStatus.results) do
- pushQueue(def, start, true)
- end
- end
- if mt then
- if not status.share.markMetaTable then
- status.share.markMetaTable = {}
- end
- if status.share.markMetaTable[mt] then
- return
- end
- status.share.markMetaTable[mt] = true
- m.checkSameSimpleInValueInMetaTable(status, mt, start, pushQueue)
- status.share.markMetaTable[mt] = nil
- end
-end
-
-function m.checkSameSimpleInValueOfCallMetaTable(status, call, start, pushQueue)
- if status.crossMetaTableMark then
- return
- end
- status.crossMetaTableMark = true
- if call.type == 'call' then
- m.checkSameSimpleInValueOfSetMetaTable(status, call.node, start, pushQueue)
- end
- status.crossMetaTableMark = false
-end
-
-function m.checkSameSimpleInSpecialBranch(status, obj, start, pushQueue)
- if status.interface.index then
- local results = status.interface.index(obj)
- if not results then
- return
- end
- for _, res in ipairs(results) do
- pushQueue(obj, start + 1)
- end
- end
-end
-
-function m.checkSameSimpleInParamSelf(status, obj, start, pushQueue)
- if obj.type ~= 'getlocal' or obj[1] ~= 'self' then
- return
- end
- local node = obj.node
- if node.tag == 'self' then
- return
- end
- if node.parent.type ~= 'funcargs' then
- return
- end
- local func = node.parent.parent
- if func.type ~= 'function' or func.parent.type ~= 'setfield' then
- return
- end
- local fieldNode = func.parent.node
- local newStatus = m.status(status)
- m.searchRefs(newStatus, fieldNode, 'ref')
- for _, ref in ipairs(newStatus.results) do
- pushQueue(ref, start, true)
- end
-end
-
-local function appendValidGenericType(results, status, typeName, obj)
- if typeName.parent.type == 'doc.type.typeliteral' then
- if obj.type == 'string' and status.interface.docType then
- local docs = status.interface.docType(obj[1])
- for i = 1, #docs do
- local doc = docs[i]
- if doc.type == 'doc.class.name'
- or doc.type == 'doc.alias.name' then
- results[#results+1] = doc
- break
- end
- end
- end
- else
- -- 发现没有使用 `T`,则沿用既有逻辑直接返回实参
- results[#results+1] = obj
- end
-end
-
-local function stepRefOfGenericCrossTable(status, doc, typeName)
- for _, typeUnit in ipairs(doc.extends.types) do
- if typeUnit.type == 'doc.type.table' then
- for _, where in ipairs {'key', 'value'} do
- local childTypes = typeUnit[where].types
- for _, childName in ipairs(childTypes) do
- if childName[1] == typeName[1] then
- return function (obj)
- local childStatus = m.status(status)
- m.searchRefs(childStatus, obj, 'def')
- for _, res in ipairs(childStatus.results) do
- if res.type == 'doc.type.table' then
- return res[where]
- end
- if res.type == 'doc.type.array' then
- if where == 'key' then
- return status.interface and status.interface.docType('integer')[1]
- end
- if where == 'value' then
- return res.node
- end
- end
- end
- end
- end
- end
- end
- return function (obj)
- return nil
- end
- elseif typeUnit.type == 'doc.type.array' then
- return function (obj)
- local childStatus = m.status(status)
- m.searchRefs(childStatus, obj, 'def')
- for _, res in ipairs(childStatus.results) do
- if res.type == 'doc.type.array' then
- return res.node
- end
- end
- end
- end
- end
- return nil
-end
-
-local function getIteratorArg(status, args, index)
- local call = args.parent
- local node = call.node
- if not node.iterator then
- return nil
- end
- if node.type ~= 'call' then
- return nil
- end
- local results = m.checkSameSimpleInCallInSameFile(status, node.node, node.args, index + 1)
- return results[1]
-end
-
-local function stepRefOfGeneric(status, typeUnit, args, mode)
- local results = {}
- if not args then
- return results
- end
- local myName = typeUnit[1]
- for _, typeName in ipairs(typeUnit.typeGeneric[myName]) do
- if typeName == typeUnit then
- goto CONTINUE
- end
- local docArg = m.getParentType(typeName, 'doc.type.arg')
- or m.getParentType(typeName, 'doc.param')
- or m.getParentType(typeName, 'doc.type.array')
- if not docArg then
- goto CONTINUE
- end
- local doc = m.getDocState(docArg)
- if not doc.bindSources then
- goto CONTINUE
- end
- local crossTable = stepRefOfGenericCrossTable(status, docArg, typeName)
-
- -- find out param index
- local genericIndex
- if docArg.type == 'doc.param' then
- local paramName = docArg.param[1]
- for _, source in ipairs(doc.bindSources) do
- if source.type == 'function'
- and source.args then
- for i, arg in ipairs(source.args) do
- if arg[1] == paramName then
- genericIndex = i
- break
- end
- end
- end
- end
- elseif docArg.type == 'doc.type.arg' then
- for index, arg in ipairs(docArg.parent.args) do
- if arg == docArg then
- genericIndex = index
- break
- end
- end
- end
-
- local callArg = args[genericIndex]
- or getIteratorArg(status, args, genericIndex)
-
- if not callArg then
- goto CONTINUE
- end
- if crossTable then
- callArg = crossTable(callArg)
- if not callArg then
- goto CONTINUE
- end
- end
- appendValidGenericType(results, status, typeName, callArg)
- ::CONTINUE::
- end
- return results
-end
-
-function m.checkSameSimpleByDocType(status, doc, args)
- if status.share.searchingBindedDoc then
- return
- end
- if doc.type ~= 'doc.type' then
- return
- end
- local results = {}
- for _, piece in ipairs(doc.types) do
- if piece.typeGeneric then
- local pieceResult = stepRefOfGeneric(status, piece, args, 'def')
- for _, res in ipairs(pieceResult) do
- results[#results+1] = res
- end
- else
- local pieceResult = stepRefOfDocType(status, piece, 'def')
- for _, res in ipairs(pieceResult) do
- results[#results+1] = res
- end
- end
- end
- return results
-end
-
-function m.checkSameSimpleByBindDocs(status, obj, start, pushQueue, mode)
- if not obj.bindDocs then
- return
- end
- if status.share.searchingBindedDoc then
- return
- end
- local skipInfer = false
- local results = {}
- for _, doc in ipairs(obj.bindDocs) do
- if doc.type == 'doc.class' then
- results[#results+1] = doc
- elseif doc.type == 'doc.type' then
- results[#results+1] = doc
- elseif doc.type == 'doc.param' then
- -- function (x) 的情况
- if obj.type == 'local'
- and m.getKeyName(obj) == doc.param[1] then
- if obj.parent.type == 'funcargs'
- or obj.parent.type == 'in'
- or obj.parent.type == 'loop' then
- results[#results+1] = doc.extends
- end
- end
- elseif doc.type == 'doc.field' then
- results[#results+1] = doc
- elseif doc.type == 'doc.vararg' then
- if obj.type == '...' then
- results[#results+1] = doc
- end
- end
- end
- for _, res in ipairs(results) do
- if res.type == 'doc.class'
- or res.type == 'doc.type'
- or res.type == 'doc.vararg' then
- pushQueue(res, start, true)
- skipInfer = true
- end
- if res.type == 'doc.type.function' then
- pushQueue(res, start, true)
- elseif res.type == 'doc.field' then
- pushQueue(res, start + 1)
- end
- end
- return skipInfer
-end
-
-function m.checkSameSimpleOfRefByDocSource(status, obj, start, pushQueue, mode)
- if status.share.searchingBindedDoc then
- return
- end
- if not obj.bindSources then
- return
- end
- status.share.searchingBindedDoc = true
- local mark = {}
- local newStatus = m.status(status)
- for _, ref in ipairs(obj.bindSources) do
- if not mark[ref] then
- mark[ref] = true
- m.searchRefs(newStatus, ref, mode)
- end
- end
- status.share.searchingBindedDoc = nil
- for _, res in ipairs(newStatus.results) do
- pushQueue(res, start, true)
- end
-end
-
-function m.checkSameSimpleOfRefByDocReturn(status, obj, start, pushQueue, mode)
- if status.share.searchingBindedDoc then
- return
- end
- if not obj.bindSources then
- return
- end
- local index = 0
- for _, doc in ipairs(obj.bindGroup) do
- if doc.type == 'doc.return' then
- index = index + 1
- if doc == obj then
- break
- end
- end
- end
- status.share.searchingBindedDoc = true
- local mark = {}
- local newStatus = m.status(status)
- for _, ref in ipairs(obj.bindSources) do
- if not mark[ref] then
- mark[ref] = true
- m.searchRefs(newStatus, ref, mode)
- end
- end
- status.share.searchingBindedDoc = nil
- for _, res in ipairs(newStatus.results) do
- local call = res.parent
- if call.type == 'call' then
- if index == 1 then
- local sel = call.parent
- if sel.type == 'select' and sel.index == index then
- pushQueue(sel.parent, start, true)
- end
- else
- if call.extParent then
- for _, sel in ipairs(call.extParent) do
- if sel.type == 'select' and sel.index == index then
- pushQueue(sel.parent, start, true)
- end
- end
- end
- end
- end
- end
-end
-
-local function getArrayOrTableLevel(obj)
- local level = 0
- while true do
- local parent = obj.parent
- if parent.type == 'doc.type.array' then
- level = level + 1
- elseif parent.type == 'doc.type.table' then
- if obj.type == 'doc.type' then
- level = level + 1
- -- else 只存在 obj.type == 'doc.type.name' 的情况,即 table<k,v> 中的 table,这种是不需要再增加层级的
- end
- elseif parent.type == 'doc.type' and parent.parent and parent.parent.type == 'doc.type.table' then
- level = level + 1
- parent = parent.parent
- else
- break
- end
- obj = parent
- end
- return level
-end
-
-function m.checkSameSimpleByDoc(status, obj, start, pushQueue, mode)
- if obj.type == 'doc.class.name'
- or obj.type == 'doc.class' then
- if obj.type == 'doc.class.name' then
- obj = m.getDocState(obj)
- end
- local classStart
- for _, doc in ipairs(obj.bindGroup) do
- if doc == obj then
- classStart = true
- elseif doc.type == 'doc.class' then
- classStart = false
- end
- if classStart and doc.type == 'doc.field' then
- pushQueue(doc, start + 1)
- end
- end
- m.checkSameSimpleOfRefByDocSource(status, obj, start, pushQueue, mode)
- if mode == 'ref' then
- local pieceResult = stepRefOfDocType(status, obj.class, 'ref')
- for _, res in ipairs(pieceResult) do
- pushQueue(res, start, true)
- end
- if obj.extends then
- for _, ext in ipairs(obj.extends) do
- local pieceResult = stepRefOfDocType(status, ext, 'def')
- for _, res in ipairs(pieceResult) do
- pushQueue(res, start, true)
- end
- end
- end
- end
- return true
- elseif obj.type == 'doc.type' then
- for _, piece in ipairs(obj.types) do
- local pieceResult = stepRefOfDocType(status, piece, 'def')
- for _, res in ipairs(pieceResult) do
- pushQueue(res, start, true)
- end
- end
- if mode == 'ref' then
- m.checkSameSimpleOfRefByDocSource(status, obj, start, pushQueue, mode)
- end
- return true
- elseif obj.type == 'doc.type.name'
- or obj.type == 'doc.see.name' then
- local pieceResult = stepRefOfDocType(status, obj, 'def')
- for _, res in ipairs(pieceResult) do
- pushQueue(res, start, true)
- end
-
- if mode == 'ref' then
- local state = m.getDocState(obj)
- if state.type == 'doc.type' then
- m.checkSameSimpleOfRefByDocSource(status, state, start - getArrayOrTableLevel(obj), pushQueue, mode)
- end
- if state.type == 'doc.return' then
- m.checkSameSimpleOfRefByDocReturn(status, state, start - getArrayOrTableLevel(obj), pushQueue, mode)
- end
- end
- return true
- elseif obj.type == 'doc.field' then
- if mode ~= 'field'
- and mode ~= 'deffield' then
- return m.checkSameSimpleByDoc(status, obj.extends, start, pushQueue, mode)
- end
- elseif obj.type == 'doc.type.array' then
- pushQueue(obj.node, start + 1, true)
- return true
- elseif obj.type == 'doc.type.table' then
- pushQueue(obj.node, start, true)
- pushQueue(obj.value, start + 1, true)
- return true
- elseif obj.type == 'doc.vararg' then
- pushQueue(obj.vararg, start, true)
- end
-end
-
-function m.checkSameSimpleInArg1OfSetMetaTable(status, obj, start, pushQueue)
- local args = obj.parent
- if not args or args.type ~= 'callargs' then
- return
- end
- local callNode = args.parent.node
- if callNode.special ~= 'setmetatable' then
- return
- end
- if args[1] ~= obj then
- return
- end
- local mt = args[2]
- if mt then
- if m.hasValueMark(status, mt) then
- return
- end
- m.checkSameSimpleInValueInMetaTable(status, mt, start, pushQueue)
- end
-end
-
-function m.searchSameMethodOutSelf(ref, mark)
- local selfNode
- if ref.tag == 'self' then
- selfNode = ref
- else
- if ref.type == 'getlocal'
- or ref.type == 'setlocal' then
- local node = ref.node
- if node.tag == 'self' then
- selfNode = node
- end
- end
- end
- if selfNode then
- if mark[selfNode] then
- return nil
- end
- mark[selfNode] = true
- local method = selfNode.method.node
- if mark[method] then
- return nil
- end
- mark[method] = true
- return method
- end
-end
-
-function m.searchSameMethodIntoSelf(ref, mark)
- local nxt = ref.next
- if not nxt then
- return nil
- end
- if nxt.type ~= 'setmethod' then
- return nil
- end
- if mark[ref] then
- return nil
- end
- mark[ref] = true
- local value = nxt.value
- if not value or value.type ~= 'function' then
- return nil
- end
- local selfRef = value.locals and value.locals[1]
- if not selfRef or selfRef.tag ~= 'self' then
- return nil
- end
- if mark[selfRef] then
- return nil
- end
- mark[selfRef] = true
- return selfRef
-end
-
-function m.searchSameFieldsCrossMethod(status, ref, start, pushQueue, mode)
- if status.share.crossMethodMark2 then
- return
- end
- local mark = status.crossMethodMark
- if not mark then
- mark = {}
- status.crossMethodMark = mark
- end
- if mark[ref] then
- return
- end
- local selfRef = m.searchSameMethodIntoSelf(ref, mark)
- if selfRef then
- tracy.ZoneBeginN 'searchSameFieldsCrossMethod'
- local _ <close> = tracy.ZoneEnd
- -- 如果自己是method,则只检查自己内部的self引用
- status.share.inBeSetValue = (status.share.inBeSetValue or 0) + 1
- status.share.crossMethodMark2 = true
- local newStatus = m.status(status)
- m.searchRefs(newStatus, selfRef, mode)
- for _, res in ipairs(newStatus.results) do
- pushQueue(res, start, true)
- end
- status.share.inBeSetValue = (status.share.inBeSetValue or 0) - 1
- status.share.crossMethodMark2 = nil
- return
- end
- local method = m.searchSameMethodOutSelf(ref, mark)
- if method then
- pushQueue(method, start, true)
- return
- end
-end
-
-local function checkSameSimpleAndMergeFunctionReturnsByDoc(status, results, source, index, args)
- source = m.getObjectValue(source) or source
- if not source or source.type ~= 'function' then
- return
- end
- if not source.bindDocs then
- return
- end
- local returns = {}
- for _, doc in ipairs(source.bindDocs) do
- if doc.type == 'doc.return' then
- for _, rtn in ipairs(doc.returns) do
- returns[#returns+1] = rtn
- end
- end
- end
- local rtn = returns[index]
- if not rtn then
- return
- end
- local types = m.checkSameSimpleByDocType(status, rtn, args)
- if not types then
- return
- end
- for _, res in ipairs(types) do
- results[#results+1] = res
- end
- return true
-end
-
-local function checkSameSimpleAndMergeDocFunctionReturn(status, results, docFunc, index, args)
- if docFunc.type ~= 'doc.type.function' then
- return
- end
- local rtn = docFunc.returns[index]
- if rtn then
- local types = m.checkSameSimpleByDocType(status, rtn, args)
- if types then
- for _, res in ipairs(types) do
- results[#results+1] = res
- end
- return true
- end
- end
-end
-
-local function checkSameSimpleAndMergeDocTypeFunctionReturns(status, results, source, index)
- if not source.bindDocs then
- return
- end
- for _, doc in ipairs(source.bindDocs) do
- if doc.type == 'doc.type' then
- for _, typeUnit in ipairs(doc.types) do
- if checkSameSimpleAndMergeDocFunctionReturn(status, results, typeUnit, index) then
- return true
- end
- end
- end
- end
-end
-
-function m.checkSameSimpleInCallInSameFile(status, func, args, index)
- if not status.share.callResultsCache then
- status.share.callResultsCache = {}
- end
- local cache = status.share.callResultsCache[func]
- if not cache then
- cache = {}
- status.share.callResultsCache[func] = cache
- end
- local results = cache[index]
- if results then
- return results
- end
- results = {}
- if func.special then
- --return results
- end
- local newStatus = m.status(status)
- m.searchRefs(newStatus, func, 'def')
- local hasDocReturn
- for _, def in ipairs(newStatus.results) do
- hasDocReturn = checkSameSimpleAndMergeDocTypeFunctionReturns(status, results, def, index)
- or checkSameSimpleAndMergeFunctionReturnsByDoc(status, results, def, index, args)
- or checkSameSimpleAndMergeDocFunctionReturn(status, results, def, index, args)
- or hasDocReturn
- end
- if not hasDocReturn then
- for _, def in ipairs(newStatus.results) do
- local value = m.getObjectValue(def) or def
- if value.type == 'function' then
- local returns = value.returns
- if returns then
- for _, ret in ipairs(returns) do
- local exp = ret[index]
- if exp then
- results[#results+1] = exp
- end
- end
- end
- end
- end
- cache[index] = results
- end
- return results
-end
-
-function m.checkSameSimpleInCall(status, ref, start, pushQueue, mode)
- if status.share.inBeSetValue and status.share.inBeSetValue > 0 then
- return
- end
- if status.share.inSetValue and status.share.inSetValue > 5 then
- return
- end
- local func, args, index = m.getCallValue(ref)
- if not func then
- return
- end
- if m.checkCallMark(status, func.parent, true) then
- return
- end
- status.share.inSetValue = (status.share.inSetValue or 0) + 1
- -- 检查赋值是 semetatable() 的情况
- m.checkSameSimpleInValueOfSetMetaTable(status, func, start, pushQueue)
- local objs = m.checkSameSimpleInCallInSameFile(status, func, args, index)
- if status.interface.call then
- local cobjs = status.interface.call(func, args, index)
- if cobjs then
- for _, obj in ipairs(cobjs) do
- if not m.checkReturnMark(status, obj) then
- objs[#objs+1] = obj
- end
- end
- end
- end
- m.cleanResults(objs)
- if not status.share.callFuncMark then
- status.share.callFuncMark = {}
- end
- local mark = {}
- for _, obj in ipairs(objs) do
- if mark[obj] then
- goto CONTINUE
- end
- if status.share.callFuncMark[obj] then
- goto CONTINUE
- end
- status.share.callFuncMark[obj] = true
- local newStatus = m.status(status)
- m.searchRefs(newStatus, obj, mode)
- pushQueue(obj, start, true)
- mark[obj] = true
- for _, obj in ipairs(newStatus.results) do
- pushQueue(obj, start, true)
- mark[obj] = true
- end
- status.share.callFuncMark[obj] = nil
- ::CONTINUE::
- end
- status.share.inSetValue = (status.share.inSetValue or 0) - 1
-end
-
-local function searchRawset(ref, results)
- if m.getKeyName(ref) ~= 'rawset' then
- return
- end
- local call = ref.parent
- if call.type ~= 'call' or call.node ~= ref then
- return
- end
- if not call.args then
- return
- end
- local arg1 = call.args[1]
- if arg1.special ~= '_G' then
- -- 不会吧不会吧,不会真的有人写成 `rawset(_G._G._G, 'xxx', value)` 吧
- return
- end
- results[#results+1] = call
-end
-
-local function searchG(ref, results)
- while ref and m.getKeyName(ref) == '_G' do
- results[#results+1] = ref
- ref = ref.next
- end
- if ref then
- results[#results+1] = ref
- searchRawset(ref, results)
- end
-end
-
-local function searchEnvRef(ref, results)
- if ref.type == 'setglobal'
- or ref.type == 'getglobal' then
- results[#results+1] = ref
- searchG(ref, results)
- elseif ref.type == 'getlocal' then
- results[#results+1] = ref.next
- searchG(ref.next, results)
- end
-end
-
-function m.findGlobals(ast)
- local root = m.getRoot(ast)
- local results = {}
- local env = m.getENV(root)
- if env.ref then
- for _, ref in ipairs(env.ref) do
- searchEnvRef(ref, results)
- end
- end
- return results
-end
-
-function m.findGlobalsOfName(ast, name)
- local root = m.getRoot(ast)
- local results = {}
- local globals = m.findGlobals(root)
- for _, global in ipairs(globals) do
- if m.getKeyName(global) == name then
- results[#results+1] = global
- end
- end
- return results
-end
-
-function m.checkSameSimpleInGlobal(status, source)
- local name = m.getKeyName(source)
- if not name then
- return
- end
- local objs
- if status.interface.global then
- objs = status.interface.global(name)
- else
- objs = m.findGlobalsOfName(source, name)
- end
- return objs
-end
-
-function m.hasValueMark(status, value)
- if not status.share.valueMark then
- status.share.valueMark = {}
- end
- if status.share.valueMark[value] then
- return true
- end
- status.share.valueMark[value] = true
- return false
-end
-
-function m.checkCallMark(status, a, mark)
- if not status.share.callMark then
- status.share.callMark = {}
- end
- if mark then
- status.share.callMark[a] = mark
- else
- return status.share.callMark[a]
- end
- return false
-end
-
-function m.checkReturnMark(status, a, mark)
- if not status.share.returnMark then
- status.share.returnMark = {}
- end
- local result = status.share.returnMark[a]
- if mark then
- status.share.returnMark[a] = mark
- end
- return result
-end
-
-function m.searchSameFieldsInValue(status, ref, start, pushQueue, mode)
- if status.share.inBeSetValue and status.share.inBeSetValue > 0 then
- return
- end
- if status.share.inSetValue and status.share.inSetValue > 5 then
- return
- end
- local value = m.getObjectValue(ref)
- if not value then
- return
- end
- if m.hasValueMark(status, value) then
- return
- end
- status.share.inSetValue = (status.share.inSetValue or 0) + 1
- if not status.share.tempValueMark then
- status.share.tempValueMark = {}
- end
- if not status.share.tempValueMark[value] then
- status.share.tempValueMark[value] = true
- local newStatus = m.status(status)
- m.searchRefs(newStatus, value, mode)
- status.share.tempValueMark[value] = nil
- for _, res in ipairs(newStatus.results) do
- pushQueue(res, start, true)
- end
- pushQueue(value, start, true)
- end
- status.share.inSetValue = (status.share.inSetValue or 0) - 1
- -- 检查形如 a = f() 的分支情况
- m.checkSameSimpleInCall(status, value, start, pushQueue, mode)
-end
-
-function m.checkSameSimpleAsTableField(status, ref, start, pushQueue)
- if not status.deep then
- --return
- end
- local parent = ref.parent
- if not parent or parent.type ~= 'tablefield' then
- return
- end
- if m.hasValueMark(status, ref) then
- return
- end
- local newStatus = m.status(status)
- m.searchRefs(newStatus, parent.field, 'ref')
- for _, res in ipairs(newStatus.results) do
- pushQueue(res, start, true)
- end
-end
-
-function m.checkSearchLevel(status)
- status.share.back = status.share.back or 0
- if status.share.back >= (status.interface.searchLevel or 0) then
- -- TODO 限制向前搜索的次数
- --return true
- end
- status.share.back = status.share.back + 1
- return false
-end
-
-function m.checkSameSimpleAsReturn(status, ref, start, pushQueue)
- if not status.deep then
- return
- end
- if not ref.parent or ref.parent.type ~= 'return' then
- return
- end
- if ref.parent.parent.type ~= 'main' then
- return
- end
- if m.checkSearchLevel(status) then
- return
- end
- local newStatus = m.status(status)
- m.searchRefsAsFunctionReturn(newStatus, ref, 'ref')
- for _, res in ipairs(newStatus.results) do
- if not m.checkCallMark(status, res) then
- pushQueue(res, start, true)
- end
- end
-end
-
-function m.checkSameSimpleAsSetValue(status, ref, start, pushQueue)
- if not status.deep then
- --return
- end
- if status.share.inSetValue and status.share.inSetValue > 0 then
- return
- end
- if status.share.inBeSetValue and status.share.inBeSetValue > 5 then
- return
- end
- if ref.type == 'select' then
- return
- end
- local parent = ref.parent
- if not parent then
- return
- end
- if m.getObjectValue(parent) ~= ref then
- return
- end
- if m.hasValueMark(status, ref) then
- return
- end
- if m.checkSearchLevel(status) then
- return
- end
- local obj
- if parent.type == 'local'
- or parent.type == 'setglobal'
- or parent.type == 'setlocal' then
- obj = parent
- elseif parent.type == 'setfield' then
- obj = parent.field
- elseif parent.type == 'setmethod' then
- obj = parent.method
- end
- if not obj then
- return
- end
- status.share.inBeSetValue = (status.share.inBeSetValue or 0) + 1
- local newStatus = m.status(status)
- m.searchRefs(newStatus, obj, 'ref')
- for _, res in ipairs(newStatus.results) do
- pushQueue(res, start, true)
- end
- status.share.inBeSetValue = (status.share.inBeSetValue or 0) - 1
-end
-
-local function getTableAndIndexIfIsForPairsKeyOrValue(ref)
- if ref.type ~= 'local' then
- return
- end
-
- if not ref.parent or ref.parent.type ~= 'in' then
- return
- end
-
- if not ref.value or ref.value.type ~= 'select' then
- return
- end
-
- local rootSelectObj = ref.value
- if rootSelectObj.index ~= 1 and rootSelectObj.index ~= 2 then
- return
- end
-
- if not rootSelectObj.vararg or rootSelectObj.vararg.type ~= 'call' then
- return
- end
- local rootCallObj = rootSelectObj.vararg
-
- if not rootCallObj.node or rootCallObj.node.type ~= 'call' then
- return
- end
- local pairsCallObj = rootCallObj.node
-
- if not pairsCallObj.node
- or (pairsCallObj.node.special ~= 'pairs' and pairsCallObj.node.special ~= 'ipairs') then
- return
- end
-
- if not pairsCallObj.args or not pairsCallObj.args[1] then
- return
- end
- local tableObj = pairsCallObj.args[1]
-
- return tableObj, rootSelectObj.index
-end
-
-function m.checkSameSimpleAsKeyOrValueInForParis(status, ref, start, pushQueue)
- local tableObj, index = getTableAndIndexIfIsForPairsKeyOrValue(ref)
- if not tableObj then
- return
- end
-
- local newStatus = m.status(status)
- m.searchRefs(newStatus, tableObj, 'def')
- for _, def in ipairs(newStatus.results) do
- if def.bindDocs then
- for _, binddoc in ipairs(def.bindDocs) do
- if binddoc.type == 'doc.type' then
- if binddoc.types[1] and binddoc.types[1].type == 'doc.type.table' then
- if index == 1 then
- pushQueue(binddoc.types[1].key, start, true)
- elseif index == 2 then
- pushQueue(binddoc.types[1].value, start, true)
- end
- end
- end
- end
- end
- end
-end
-
----
----@param func parser.guide.object
----@param argIndex integer
----@return integer?
-local function findGenericFromArgIndexToReturnIndex(func, argIndex)
- if not func.bindDocs then
- return nil
- end
- local paramType
- for _, doc in ipairs(func.bindDocs) do
- if doc.type == 'doc.param' then
- if doc.extends and doc.extends.paramIndex == argIndex then
- paramType = doc.extends
- break
- end
- end
- end
- if not paramType then
- return nil
- end
- for _, typeUnit in ipairs(paramType.types) do
- if typeUnit.typeGeneric then
- local generic = typeUnit.typeGeneric[typeUnit[1]]
- if generic then
- for _, typeName in ipairs(generic) do
- local docType = typeName.parent
- if docType.returnIndex then
- return docType.returnIndex
- end
- end
- end
- end
- end
- return nil
-end
-
-function m.checkSameSimpleAsCallArg(status, ref, start, pushQueue)
- if not status.deep then
- return
- end
- local call, index = m.getCallAndArgIndex(ref)
- if not call then
- return
- end
- if call.parent.type ~= 'select' then
- return
- end
- if (status.share.inSetValue or 0) > 0 then
- return
- end
- if status.share.inBeSetValue and status.share.inBeSetValue > 5 then
- return
- end
- status.share.inBeSetValue = (status.share.inBeSetValue or 0) + 1
- local newStatus = m.status(status)
- m.searchRefs(newStatus, call.node, 'def')
- for _, func in ipairs(newStatus.results) do
- local rindex = findGenericFromArgIndexToReturnIndex(func, index)
- if rindex then
- if rindex == 1 then
- pushQueue(call.parent.parent, start, true)
- else
- if call.extParent then
- for _, slt in ipairs(call.extParent) do
- if slt.index == rindex then
- pushQueue(slt.parent, start, true)
- end
- end
- end
- end
- end
- end
- status.share.inBeSetValue = (status.share.inBeSetValue or 0) - 1
-end
-
-local function hasTypeName(doc, name)
- if doc.type == 'doc.type' then
- for _, tunit in ipairs(doc.types) do
- if tunit.type == 'doc.type.name'
- and tunit[1] == name then
- return true
- end
- end
- end
- if doc.type == 'doc.type.name'
- or doc.type == 'doc.class.name' then
- if doc[1] == name then
- return true
- end
- end
- return false
-end
-
-function m.checkSameSimpleInString(status, ref, start, pushQueue, mode)
- -- 特殊处理 ('xxx').xxx 的形式
- if ref.type ~= 'string'
- and not hasTypeName(ref, 'string') then
- return
- end
- if not status.interface.docType then
- return
- end
- if status.share.searchingBindedDoc then
- return
- end
- if not status.share.markString then
- status.share.markString = {}
- end
- local marked = ref
- if ref.type == 'string' then
- marked = 'string'
- end
- if status.share.markString[marked] then
- return
- end
- status.share.markString[marked] = true
- local newStatus = m.status(status)
- local docs = status.interface.docType('stringlib')
- local mark = {}
- for i = 1, #docs do
- local doc = docs[i]
- m.searchFields(newStatus, doc)
- end
- for _, res in ipairs(newStatus.results) do
- if mark[res] then
- goto CONTINUE
- end
- mark[res] = true
- pushQueue(res, start + 1)
- ::CONTINUE::
- end
- status.share.markString[marked] = nil
- return true
-end
-
+-- TODO: compatible
+m.getRoot = guide.getRoot
+m.eachSource = guide.eachSource
+m.eachSourceType = guide.eachSourceType
+m.eachSourceContain = guide.eachSourceContain
+m.eachSourceBetween = guide.eachSourceBetween
+m.getStartFinish = guide.getStartFinish
+m.isLiteral = guide.isLiteral
+
+---@alias guide.searchmode '"ref"'|'"def"'|'"field"'
+
+---添加结果
+---@param status guide.status
+---@param mode guide.searchmode
+---@param ref parser.guide.object
+---@param simple table
function m.pushResult(status, mode, ref, simple)
local results = status.results
if mode == 'def' then
@@ -2836,7 +109,7 @@ function m.pushResult(status, mode, ref, simple)
if ref.parent and ref.parent.type == 'return' then
results[#results+1] = ref
end
- if m.isLiteral(ref)
+ if guide.isLiteral(ref)
and ref.parent
and ref.parent.type == 'callargs'
and ref ~= simple.node then
@@ -2890,1948 +163,91 @@ function m.pushResult(status, mode, ref, simple)
end
end
-function m.checkSameSimpleName(ref, sm)
- if sm == m.ANY then
- return true
- end
-
- if m.getSimpleName(ref) == sm then
- return true
- end
- if ref.type == 'doc.type'
- and ref.array == true then
- return true
- end
- return false
-end
-
-function m.isValidSetRef(ref)
- if m.isSet(ref) then
- return true
- end
-end
-
-function m.checkSameSimple(status, simple, ref, start, force, mode, pushQueue)
- if start > #simple then
- return
- end
- for i = start, #simple do
- local sm = simple[i]
- if not force and not m.checkSameSimpleName(ref, sm) then
- return
- end
- force = false
- local cmode = mode
- local skipSearch
- if i < #simple then
- cmode = 'ref'
- else
- if mode == 'deffield' then
- if not m.isSet(ref) then
- skipSearch = true
- end
- end
- end
- -- 检查 doc
- local skipInfer = m.checkSameSimpleByBindDocs(status, ref, i, pushQueue, cmode)
- or m.checkSameSimpleByDoc(status, ref, i, pushQueue, cmode)
- -- 检查自己是字符串的分支情况
- m.checkSameSimpleInString(status, ref, i, pushQueue, cmode)
- if not skipInfer and not skipSearch then
- -- 穿透 self:func 与 mt:func
- m.searchSameFieldsCrossMethod(status, ref, i, pushQueue, cmode)
- -- 穿透赋值
- m.searchSameFieldsInValue(status, ref, i, pushQueue, cmode)
- -- 检查自己是字面量表的情况
- m.checkSameSimpleInValueOfTable(status, ref, i, pushQueue)
- -- 检查自己作为 setmetatable 第一个参数的情况
- m.checkSameSimpleInArg1OfSetMetaTable(status, ref, i, pushQueue)
- -- 检查自己作为 setmetatable 调用的情况
- m.checkSameSimpleInValueOfCallMetaTable(status, ref, i, pushQueue)
- -- 检查自己是特殊变量的分支的情况
- m.checkSameSimpleInSpecialBranch(status, ref, i, pushQueue)
- -- self 的特殊处理
- m.checkSameSimpleInParamSelf(status, ref, i, pushQueue)
- -- 自己是 call 的情况
- m.checkSameSimpleInCall(status, ref, i, pushQueue, cmode)
- if cmode == 'ref' then
- -- 检查形如 { a = f } 的情况
- m.checkSameSimpleAsTableField(status, ref, i, pushQueue)
- -- 检查形如 return m 的情况
- m.checkSameSimpleAsReturn(status, ref, i, pushQueue)
- -- 检查形如 a = f 的情况
- m.checkSameSimpleAsSetValue(status, ref, i, pushQueue)
- -- 检查形如 for k,v in pairs()/ipairs() do end 的情况
- m.checkSameSimpleAsKeyOrValueInForParis(status, ref, i, pushQueue)
- -- 检查自己是函数参数的情况(泛型) local x = call(V)
- m.checkSameSimpleAsCallArg(status, ref, i, pushQueue)
- end
- end
- if i == #simple then
- break
- end
- ref = m.getNextRef(ref)
- if not ref then
- return
- end
- end
- m.pushResult(status, mode, ref, simple)
- local value = m.getObjectValue(ref)
- if value then
- m.pushResult(status, mode, value, simple)
- end
-end
-
-local queuesPool = {}
-local startsPool = {}
-local forcesPool = {}
-local poolSize = 0
-
-local function allocQueue()
- if poolSize <= 0 then
- return {}, {}, {}
- else
- local queues = queuesPool[poolSize]
- local starts = startsPool[poolSize]
- local forces = forcesPool[poolSize]
- poolSize = poolSize - 1
- return queues, starts, forces
- end
-end
-
-local function deallocQueue(queues, starts, forces)
- poolSize = poolSize + 1
- queuesPool[poolSize] = queues
- startsPool[poolSize] = starts
- forcesPool[poolSize] = forces
-end
-
-function m.searchSameFields(status, simple, mode)
- local queues, starts, forces = allocQueue()
- local queueLen = 0
- local locks = {}
- local function appendQueue(obj, start, force)
- local lock = locks[start]
- if not lock then
- lock = {}
- locks[start] = lock
- end
- if lock[obj] then
- return
- end
- lock[obj] = true
- queueLen = queueLen + 1
- queues[queueLen] = obj
- starts[queueLen] = start
- forces[queueLen] = force
- if obj.mirror then
- if not lock[obj.mirror] then
- lock[obj.mirror] = true
- queueLen = queueLen + 1
- queues[queueLen] = obj.mirror
- starts[queueLen] = start
- forces[queueLen] = force
- end
- end
- end
- local function pushQueue(obj, start, force)
- if obj.type == 'getlocal'
- or obj.type == 'setlocal' then
- obj = obj.node
- end
- appendQueue(obj, start, force)
- if obj.type == 'local' and obj.ref then
- for _, ref in ipairs(obj.ref) do
- appendQueue(ref, start, force)
- end
- end
- if m.isGlobal(obj) then
- local refs = m.checkSameSimpleInGlobal(status, obj)
- if refs then
- for _, ref in ipairs(refs) do
- appendQueue(ref, start, force)
- end
- end
- end
- end
- if simple.mode == 'global' then
- -- 全局变量开头
- pushQueue(simple.node, 1)
- elseif simple.mode == 'local' then
- -- 局部变量开头
- pushQueue(simple.node, 1)
- else
- pushQueue(simple.node, 1)
- end
- local max = 0
- for i = 1, 1e6 do
- if queueLen <= 0 then
- break
- end
- local obj = queues[queueLen]
- local start = starts[queueLen]
- local force = forces[queueLen]
- queues[queueLen] = nil
- starts[queueLen] = nil
- forces[queueLen] = nil
- queueLen = queueLen - 1
- max = max + 1
- status.share.count = status.share.count + 1
- if status.share.count % 10000 == 0 then
- --if TEST then
- -- print('####', status.share.count, osClock() - status.clock)
- --end
- if status.interface and status.interface.pulse then
- status.interface.pulse()
- end
- end
- --if status.share.count >= 100000 then
- -- logWarn('Count too large!')
- -- break
- --end
- m.checkSameSimple(status, simple, obj, start, force, mode, pushQueue)
- if max >= 10000 then
- logWarn('Queue too large!')
- break
- end
- end
- --deallocQueue(queues, starts, forces)
-end
-
-function m.getCallerInSameFile(status, func)
- -- 搜索所有所在函数的调用者
- local funcRefs = m.status(status)
- m.searchRefOfValue(funcRefs, func)
-
- local calls = {}
- if #funcRefs.results == 0 then
- return calls
+---获取uri
+---@param obj parser.guide.object
+---@return uri
+function m.getUri(obj)
+ if obj.uri then
+ return obj.uri
end
- for _, res in ipairs(funcRefs.results) do
- local call = res.parent
- if call.type == 'call' then
- calls[#calls+1] = call
- end
+ local root = m.getRoot(obj)
+ if root then
+ return root.uri
end
- return calls
+ return ''
end
-function m.getCallerCrossFiles(status, main)
- if status.interface.link then
- return status.interface.link(main.uri)
- end
+-- TODO
+function m.findGlobals(root)
+ linker.compileLinks(root)
+ -- TODO
return {}
end
-function m.searchRefsAsFunctionReturn(status, obj, mode)
- if not status.deep then
- return
- end
- if mode == 'def' then
- return
- end
- if m.checkReturnMark(status, obj, true) then
- return
- end
- status.results[#status.results+1] = obj
- -- 搜索所在函数
- local currentFunc = m.getParentFunction(obj)
- local rtn = obj.parent
- if rtn.type ~= 'return' then
- return
- end
- -- 看看他是第几个返回值
- local index
- for i = 1, #rtn do
- if obj == rtn[i] then
- index = i
- break
- end
- end
- if not index then
- return
- end
- local calls
- if currentFunc.type == 'main' then
- calls = m.getCallerCrossFiles(status, currentFunc)
- else
- calls = m.getCallerInSameFile(status, currentFunc)
- end
- -- 搜索调用者的返回值
- if #calls == 0 then
- return
- end
- local selects = {}
- for i = 1, #calls do
- local parent = calls[i].parent
- if parent.type == 'select' and parent.index == index then
- selects[#selects+1] = parent.parent
- end
- local extParent = calls[i].extParent
- if extParent then
- for j = 1, #extParent do
- local ext = extParent[j]
- if ext.type == 'select' and ext.index == index then
- selects[#selects+1] = ext.parent
- end
- end
- end
- end
- -- 搜索调用者的引用
- for i = 1, #selects do
- m.searchRefs(status, selects[i], 'ref')
- end
-end
-
-function m.searchRefsAsFunctionSet(status, obj, mode)
- local parent = obj.parent
- if not parent then
- return
- end
- if parent.type == 'local'
- or parent.type == 'setlocal'
- or parent.type == 'setglobal'
- or parent.type == 'setfield'
- or parent.type == 'setmethod'
- or parent.type == 'tablefield' then
- m.searchRefs(status, parent, mode)
- elseif parent.type == 'setindex'
- or parent.type == 'tableindex' then
- if parent.index == obj then
- m.searchRefs(status, parent, mode)
- end
- end
-end
-
-function m.searchRefsAsFunction(status, obj, mode)
- if obj.type ~= 'function'
- and obj.type ~= 'table' then
- return
- end
- m.searchRefsAsFunctionSet(status, obj, mode)
- -- 检查自己作为返回函数时的引用
- m.searchRefsAsFunctionReturn(status, obj, mode)
-end
-
-function m.cleanResults(results)
- local mark = {}
- for i = #results, 1, -1 do
- local res = results[i]
- if res.tag == 'self'
- or mark[res] then
- results[i] = results[#results]
- results[#results] = nil
- else
- mark[res] = true
- end
- end
-end
-
---function m.getRefCache(status, obj, mode)
--- local cache = status.interface.cache and status.interface.cache()
--- if not cache then
--- return
--- end
--- if m.isGlobal(obj) then
--- obj = m.getKeyName(obj)
--- end
--- if not cache[mode] then
--- cache[mode] = {}
--- end
--- local sourceCache = cache[mode][obj]
--- if sourceCache then
--- return sourceCache
--- end
--- sourceCache = {}
--- cache[mode][obj] = sourceCache
--- return nil, function (results)
--- for i = 1, #results do
--- sourceCache[i] = results[i]
--- end
--- end
---end
-
-function m.getRefCache(status, obj, mode)
- local isDeep = status.deep
- if mode == 'infer' then
- if not isDeep then
- return nil, nil
- end
- end
- local globalCache = status.interface.cache and status.interface.cache() or {}
- if m.isGlobal(obj) then
- obj = m.getKeyName(obj)
- end
- if not obj then
- return {}
- end
- if not globalCache[mode] then
- globalCache[mode] = {}
- end
- local sourceCache = globalCache[mode][obj]
- if sourceCache then
- return sourceCache
- end
- if not status.share.cacheLock[mode] then
- status.share.cacheLock[mode] = {}
- end
- if status.share.cacheLock[mode][obj] then
- return {}
- end
- status.share.cacheLock[mode][obj] = {}
- return nil, function (results)
- sourceCache = {}
- for i = 1, #results do
- sourceCache[i] = results[i]
- end
- globalCache[mode][obj] = sourceCache
- if not isDeep then
- return
- end
- if mode == 'ref'
- or mode == 'def' then
- for i = 1, #results do
- local res = results[i]
- if not globalCache[mode][res] then
- globalCache[mode][res] = sourceCache
- end
- end
- end
- end
+-- TODO
+function m.isGlobal(source)
+ return false
end
+---搜索对象的引用
+---@param status guide.status
+---@param obj parser.guide.object
+---@param mode guide.searchmode
function m.searchRefs(status, obj, mode)
- if not obj then
- return
- end
- local cache, makeCache = m.getRefCache(status, obj, mode)
- if cache then
- for i = 1, #cache do
- status.results[#status.results+1] = cache[i]
- end
- return
- end
-
- -- 检查单步引用
- tracy.ZoneBeginN('searchRefs getStepRef')
- local res = m.getStepRef(status, obj, mode)
- if res then
- for i = 1, #res do
- status.results[#status.results+1] = res[i]
- end
- end
- tracy.ZoneEnd()
- -- 检查simple
- tracy.ZoneBeginN('searchRefs searchSameFields')
- if m.checkStatusDepth(status) then
- local simple = m.getSimple(obj)
- if simple then
- m.searchSameFields(status, simple, mode)
- end
- end
- tracy.ZoneEnd()
-
- m.cleanResults(status.results)
-
- if makeCache then
- makeCache(status.results)
- end
-end
+ local root = guide.getRoot(obj)
+ linker.compileLinks(root)
-function m.searchRefOfValue(status, obj)
- local var = obj.parent
- if var.type == 'local'
- or var.type == 'set' then
- return m.searchRefs(status, var, 'ref')
- end
-end
-
-function m.allocInfer(o)
- if type(o.type) == 'table' then
- local infers = {}
- for i = 1, #o.type do
- infers[i] = {
- type = o.type[i],
- value = o.value,
- source = o.source,
- level = o.level
- }
+ local links = linker.getLinkersBySource(obj)
+ if links then
+ for _, link in ipairs(links) do
+ m.pushResult(status, mode, link.source)
end
- return infers
- else
- return {
- [1] = o,
- }
end
end
-function m.mergeTypes(types)
- local hasAny = types['any']
-
- types['any'] = nil
-
- if not next(types) then
- return 'any'
- end
- -- 同时包含 number 与 integer 时,去掉 integer
- if types['number'] and types['integer'] then
- types['integer'] = nil
- end
-
- local results = {}
- for tp in pairs(types) do
- results[#results+1] = tp
- end
- -- 只有显性的 nil 与 any 时,取 any
- if #results == 1 then
- if results[1] == 'nil' and hasAny then
- return 'any'
- else
- return results[1]
- end
- end
-
- tableSort(results, function (a, b)
- local sa = TypeSort[a] or 100
- local sb = TypeSort[b] or 100
- if sa == sb then
- return a < b
- else
- return sa < sb
- end
- end)
+---@class guide.status
+---搜索结果
+---@field results parser.guide.object[]
- return tableConcat(results, '|')
-end
-
-function m.getClassExtends(class)
- if class.type == 'doc.class.name' then
- class = class.parent
- end
- if not class.extends then
- return nil
- end
- local names = {}
- for _, ext in ipairs(class.extends) do
- names[#names+1] = ext[1]
- end
- return names
-end
-
-function m.viewInferType(infers)
- if not infers then
- return 'any'
- end
- local types = {}
- local hasDoc
- local hasDocTable
- for i = 1, #infers do
- local infer = infers[i]
- local src = infer.source
- if src.type == 'doc.class'
- or src.type == 'doc.class.name'
- or src.type == 'doc.type.name'
- or src.type == 'doc.type.array'
- or src.type == 'doc.type.table' then
- if infer.type ~= 'any' then
- hasDoc = true
- end
- if src.type == 'doc.type.array'
- or src.type == 'doc.type.table' then
- hasDocTable = true
- end
- end
- end
- if hasDoc then
- for i = 1, #infers do
- local infer = infers[i]
- local src = infer.source
- if src.type == 'doc.class'
- or src.type == 'doc.class.name'
- or src.type == 'doc.type.name'
- or src.type == 'doc.type.array'
- or src.type == 'doc.type.table'
- or src.type == 'doc.type.enum'
- or src.type == 'doc.resume' then
- local tp = infer.type or 'any'
- if hasDocTable and tp == 'table' then
- goto CONTINUE
- end
- if types[tp] == nil then
- types[tp] = true
- end
- end
- if src.type == 'doc.class'
- or src.type == 'doc.class.name' then
- local extends = m.getClassExtends(src)
- if extends then
- for _, tp in ipairs(extends) do
- types[tp] = false
- end
- end
- end
- ::CONTINUE::
- end
- for k, v in pairs(types) do
- if not v then
- types[k] = nil
- end
- end
- else
- for i = 1, #infers do
- local infer = infers[i]
- if infer.source.typeGeneric then
- goto CONTINUE
- end
- local tp = infer.type or 'any'
- types[tp] = true
- ::CONTINUE::
- end
- end
- return m.mergeTypes(types)
-end
-
-function m.checkTrue(status, source)
- local newStatus = m.status(status)
- m.searchInfer(newStatus, source)
- -- 当前认为的结果
- local current
- for _, infer in ipairs(newStatus.results) do
- -- 新的结果
- local new
- if infer.type == 'nil' then
- new = false
- elseif infer.type == 'boolean' then
- if infer.value == true then
- new = true
- elseif infer.value == false then
- new = false
- end
- end
- if new ~= nil then
- if current == nil then
- current = new
- else
- -- 如果2个结果完全相反,则返回 nil 表示不确定
- if new ~= current then
- return nil
- end
- end
- end
- end
- return current
-end
-
---- 获取特定类型的字面量值
-function m.getInferLiteral(status, source, type)
- local newStatus = m.status(status)
- m.searchInfer(newStatus, source)
- for _, infer in ipairs(newStatus.results) do
- if infer.value ~= nil then
- if type == nil or infer.type == type then
- return infer.value
- end
- end
- end
- return nil
-end
-
---- 是否包含某种类型
-function m.hasType(status, source, type)
- m.searchInfer(status, source)
- for _, infer in ipairs(status.results) do
- if infer.type == type then
- return true
- end
- end
- return false
-end
-
-function m.isSameValue(status, a, b)
- local statusA = m.status(status)
- m.searchInfer(statusA, a)
- local statusB = m.status(status)
- m.searchInfer(statusB, b)
- local infers = {}
- for _, infer in ipairs(statusA.results) do
- local literal = infer.value
- if literal then
- infers[literal] = false
- end
- end
- for _, infer in ipairs(statusB.results) do
- local literal = infer.value
- if literal then
- if infers[literal] == nil then
- return false
- end
- infers[literal] = true
- end
- end
- for k, v in pairs(infers) do
- if v == false then
- return false
- end
- end
- return true
-end
-
-function m.inferCheckLiteralTableWithDocVararg(status, source)
- if #source ~= 1 then
- return
- end
- local vararg = source[1]
- if vararg.type ~= 'varargs' then
- return
- end
- local results = m.getVarargDocType(status, source)
- status.results[#status.results+1] = {
- type = m.viewInferType(results) .. '[]',
- source = source,
- level = 100,
+---创建搜索状态
+---@param parentStatus guide.status
+---@param interface table
+---@param deep integer
+---@return guide.status
+function m.status(parentStatus, interface, deep)
+ local status = {
+ share = parentStatus and parentStatus.share or {
+ count = 0,
+ cacheLock = {},
+ },
+ depth = parentStatus and (parentStatus.depth + 1) or 0,
+ searchDeep= parentStatus and parentStatus.searchDeep or deep or -999,
+ interface = parentStatus and parentStatus.interface or {},
+ deep = parentStatus and parentStatus.deep,
+ clock = parentStatus and parentStatus.clock or osClock(),
+ results = {},
}
- return true
-end
-
-function m.inferCheckLiteral(status, source)
- if source.type == 'string' then
- status.results = m.allocInfer {
- type = 'string',
- value = source[1],
- source = source,
- level = 100,
- }
- return true
- elseif source.type == 'nil' then
- status.results = m.allocInfer {
- type = 'nil',
- value = NIL,
- source = source,
- level = 100,
- }
- return true
- elseif source.type == 'boolean' then
- status.results = m.allocInfer {
- type = 'boolean',
- value = source[1],
- source = source,
- level = 100,
- }
- return true
- elseif source.type == 'number' then
- if mathType(source[1]) == 'integer' then
- status.results = m.allocInfer {
- type = 'integer',
- value = source[1],
- source = source,
- level = 100,
- }
- return true
- else
- status.results = m.allocInfer {
- type = 'number',
- value = source[1],
- source = source,
- level = 100,
- }
- return true
- end
- elseif source.type == 'integer' then
- status.results = m.allocInfer {
- type = 'integer',
- source = source,
- level = 100,
- }
- return true
- elseif source.type == 'table' then
- if m.inferCheckLiteralTableWithDocVararg(status, source) then
- return true
- end
- status.results = m.allocInfer {
- type = 'table',
- source = source,
- level = 100,
- }
- return true
- elseif source.type == 'function' then
- status.results = m.allocInfer {
- type = 'function',
- source = source,
- level = 100,
- }
- return true
- elseif source.type == '...' then
- status.results = m.allocInfer {
- type = '...',
- source = source,
- level = 100,
- }
- return true
- end
-end
-
-local function getDocAliasExtends(status, typeUnit)
- if not status.interface.docType then
- return nil
- end
- if typeUnit.type ~= 'doc.type.name' then
- return nil
- end
- for _, doc in ipairs(status.interface.docType(typeUnit[1])) do
- if doc.type == 'doc.alias.name' then
- return doc.parent.extends
- end
- end
- return nil
-end
-
-function m.getDocTypeUnitName(status, unit)
- local typeName
- if unit.type == 'doc.type.name' then
- typeName = unit[1]
- elseif unit.type == 'doc.type.function' then
- typeName = 'function'
- elseif unit.type == 'doc.type.array' then
- typeName = m.getDocTypeUnitName(status, unit.node) .. '[]'
- elseif unit.type == 'doc.type.table' then
- typeName = ('%s<%s, %s>'):format(
- m.getDocTypeUnitName(status, unit.node),
- m.viewInferType(m.getDocTypeNames(status, unit.key)),
- m.viewInferType(m.getDocTypeNames(status, unit.value))
- )
- end
- if unit.typeGeneric then
- typeName = ('<%s>'):format(typeName)
- end
- return typeName
-end
-
-function m.getDocTypeNames(status, doc)
- local results = {}
- if not doc then
- return results
- end
- for _, unit in ipairs(doc.types) do
- local alias = getDocAliasExtends(status, unit)
- if alias then
- local aliasResults = m.getDocTypeNames(status, alias)
- for _, res in ipairs(aliasResults) do
- results[#results+1] = res
- end
- else
- local typeName = m.getDocTypeUnitName(status, unit)
- results[#results+1] = {
- type = typeName,
- source = unit,
- level = 100,
- }
- end
- end
- for _, enum in ipairs(doc.enums) do
- results[#results+1] = {
- type = enum[1],
- source = enum,
- level = 100,
- }
- end
- for _, resume in ipairs(doc.resumes) do
- if not resume.additional then
- results[#results+1] = {
- type = resume[1],
- source = resume,
- level = 100,
- }
- end
- end
- return results
-end
-
-function m.inferCheckDoc(status, source)
- if source.type == 'doc.class.name' then
- status.results[#status.results+1] = {
- type = source[1],
- source = source,
- level = 100,
- }
- return true
- end
- if source.type == 'doc.class' then
- status.results[#status.results+1] = {
- type = source.class[1],
- source = source,
- level = 100,
- }
- return true
- end
- if source.type == 'doc.type' then
- local results = m.getDocTypeNames(status, source)
- for _, res in ipairs(results) do
- status.results[#status.results+1] = res
- end
- return true
- end
- if source.type == 'doc.type.function'
- or source.type == 'doc.type.table'
- or source.type == 'doc.type.array' then
- local typeName = m.getDocTypeUnitName(status, source)
- status.results[#status.results+1] = {
- type = typeName,
- source = source,
- level = 100,
- }
- return true
- end
- if source.type == 'doc.field' then
- local results = m.getDocTypeNames(status, source.extends)
- for _, res in ipairs(results) do
- status.results[#status.results+1] = res
- end
- return true
- end
- if source.type == 'doc.alias.name' then
- local results = m.getDocTypeNames(status, m.getDocState(source).extends)
- for _, res in ipairs(results) do
- status.results[#status.results+1] = res
- end
- return true
- end
-end
-
-function m.getVarargDocType(status, source)
- local func = m.getParentFunction(source)
- if not func then
- return
- end
- if not func.args then
- return
- end
- for _, arg in ipairs(func.args) do
- if arg.type == '...' then
- if arg.bindDocs then
- for _, doc in ipairs(arg.bindDocs) do
- if doc.type == 'doc.vararg' then
- return m.getDocTypeNames(status, doc.vararg)
- end
- end
- end
- end
- end
-end
-
-function m.inferCheckUpDocOfVararg(status, source)
- if not source.vararg then
- return
- end
- local results = m.getVarargDocType(status, source)
- if not results then
- return
- end
- for _, res in ipairs(results) do
- status.results[#status.results+1] = res
- end
- return true
-end
-
-function m.inferCheckUpDoc(status, source)
- if m.inferCheckUpDocOfVararg(status, source) then
- return true
- end
- local parent = source.parent
- if parent then
- if parent.type == 'local'
- or parent.type == 'setlocal'
- or parent.type == 'setglobal' then
- source = parent
- end
- if parent.type == 'setfield'
- or parent.type == 'tablefield' then
- if parent.field == source
- or parent.value == source then
- source = parent
- end
- end
- if parent.type == 'setmethod' then
- if parent.method == source
- or parent.value == source then
- source = parent
- end
- end
- if parent.type == 'setindex'
- or parent.type == 'tableindex' then
- if parent.index == source
- or parent.value == source then
- source = parent
- end
- end
- end
- local binds = source.bindDocs
- if not binds then
- return
- end
- status.results = {}
- for i = #binds, 1, -1 do
- local doc = binds[i]
- if doc.type == 'doc.class' then
- status.results[#status.results+1] = {
- type = doc.class[1],
- source = doc,
- level = 100,
- }
- -- ---@class Class
- -- local x = { field = 1 }
- -- 这种情况下,将字面量表接受为Class的定义
- if source.value and source.value.type == 'table' then
- status.results[#status.results+1] = {
- type = source.value.type,
- source = source.value,
- level = 100,
- }
- end
- return true
- elseif doc.type == 'doc.type' then
- local results = m.getDocTypeNames(status, doc)
- for _, res in ipairs(results) do
- status.results[#status.results+1] = res
- end
- return true
- elseif doc.type == 'doc.param' then
- -- function (x) 的情况
- if source.type == 'local'
- and m.getKeyName(source) == doc.param[1] then
- if source.parent.type == 'funcargs'
- or source.parent.type == 'in'
- or source.parent.type == 'loop' then
- local results = m.getDocTypeNames(status, doc.extends)
- for _, res in ipairs(results) do
- status.results[#status.results+1] = res
- end
- return true
- end
- end
- end
- end
-end
-
-function m.inferCheckUnary(status, source)
- if source.type ~= 'unary' then
- return
- end
- local op = source.op
- if op.type == 'not' then
- local checkTrue = m.checkTrue(status, source[1])
- local value = nil
- if checkTrue == true then
- value = false
- elseif checkTrue == false then
- value = true
- end
- status.results = m.allocInfer {
- type = 'boolean',
- value = value,
- source = source,
- level = 100,
- }
- return true
- elseif op.type == '#' then
- status.results = m.allocInfer {
- type = 'integer',
- source = source,
- level = 100,
- }
- return true
- elseif op.type == '~' then
- local l = m.getInferLiteral(status, source[1], 'integer')
- status.results = m.allocInfer {
- type = 'integer',
- value = l and ~l or nil,
- source = source,
- level = 100,
- }
- return true
- elseif op.type == '-' then
- local v = m.getInferLiteral(status, source[1], 'integer')
- if v then
- status.results = m.allocInfer {
- type = 'integer',
- value = - v,
- source = source,
- level = 100,
- }
- return true
- end
- v = m.getInferLiteral(status, source[1], 'number')
- status.results = m.allocInfer {
- type = 'number',
- value = v and -v or nil,
- source = source,
- level = 100,
- }
- return true
- end
-end
-
-local function mathCheck(status, a, b)
- local v1 = m.getInferLiteral(status, a, 'integer')
- or m.getInferLiteral(status, a, 'number')
- local v2 = m.getInferLiteral(status, b, 'integer')
- or m.getInferLiteral(status, a, 'number')
- v1 = tonumber(v1)
- v2 = tonumber(v2)
- local int = m.hasType(status, a, 'integer')
- and m.hasType(status, b, 'integer')
- and not m.hasType(status, a, 'number')
- and not m.hasType(status, b, 'number')
- return int and 'integer' or 'number', v1, v2
-end
-
-function m.inferCheckBinary(status, source)
- if source.type ~= 'binary' then
- return
- end
- local op = source.op
- if op.type == 'and' then
- local isTrue = m.checkTrue(status, source[1])
- if isTrue == true then
- m.searchInfer(status, source[2])
- return true
- elseif isTrue == false then
- m.searchInfer(status, source[1])
- return true
- else
- m.searchInfer(status, source[1])
- m.searchInfer(status, source[2])
- return true
- end
- elseif op.type == 'or' then
- local isTrue = m.checkTrue(status, source[1])
- if isTrue == true then
- m.searchInfer(status, source[1])
- return true
- elseif isTrue == false then
- m.searchInfer(status, source[2])
- return true
- else
- m.searchInfer(status, source[1])
- m.searchInfer(status, source[2])
- return true
- end
- elseif op.type == '==' then
- local value = m.isSameValue(status, source[1], source[2])
- if value ~= nil then
- status.results = m.allocInfer {
- type = 'boolean',
- value = value,
- source = source,
- level = 100,
- }
- return true
- end
- --local isSame = m.isSameDef(status, source[1], source[2])
- --if isSame == true then
- -- value = true
- --else
- -- value = nil
- --end
- status.results = m.allocInfer {
- type = 'boolean',
- value = value,
- source = source,
- level = 100,
- }
- return true
- elseif op.type == '~=' then
- local value = m.isSameValue(status, source[1], source[2])
- if value ~= nil then
- status.results = m.allocInfer {
- type = 'boolean',
- value = not value,
- source = source,
- level = 100,
- }
- return true
- end
- --local isSame = m.isSameDef(status, source[1], source[2])
- --if isSame == true then
- -- value = false
- --else
- -- value = nil
- --end
- status.results = m.allocInfer {
- type = 'boolean',
- value = value,
- source = source,
- level = 100,
- }
- return true
- elseif op.type == '<=' then
- local v1 = m.getInferLiteral(status, source[1], 'integer')
- or m.getInferLiteral(status, source[1], 'number')
- local v2 = m.getInferLiteral(status, source[2], 'integer')
- or m.getInferLiteral(status, source[2], 'number')
- v1 = tonumber(v1)
- v2 = tonumber(v2)
- local v
- if v1 and v2 then
- v = v1 <= v2
- end
- status.results = m.allocInfer {
- type = 'boolean',
- value = v,
- source = source,
- level = 100,
- }
- return true
- elseif op.type == '>=' then
- local v1 = m.getInferLiteral(status, source[1], 'integer')
- or m.getInferLiteral(status, source[1], 'number')
- local v2 = m.getInferLiteral(status, source[2], 'integer')
- or m.getInferLiteral(status, source[2], 'number')
- v1 = tonumber(v1)
- v2 = tonumber(v2)
- local v
- if v1 and v2 then
- v = v1 >= v2
- end
- status.results = m.allocInfer {
- type = 'boolean',
- value = v,
- source = source,
- level = 100,
- }
- return true
- elseif op.type == '<' then
- local v1 = m.getInferLiteral(status, source[1], 'integer')
- or m.getInferLiteral(status, source[1], 'number')
- local v2 = m.getInferLiteral(status, source[2], 'integer')
- or m.getInferLiteral(status, source[2], 'number')
- v1 = tonumber(v1)
- v2 = tonumber(v2)
- local v
- if v1 and v2 then
- v = v1 < v2
- end
- status.results = m.allocInfer {
- type = 'boolean',
- value = v,
- source = source,
- level = 100,
- }
- return true
- elseif op.type == '>' then
- local v1 = m.getInferLiteral(status, source[1], 'integer')
- or m.getInferLiteral(status, source[1], 'number')
- local v2 = m.getInferLiteral(status, source[2], 'integer')
- or m.getInferLiteral(status, source[2], 'number')
- v1 = tonumber(v1)
- v2 = tonumber(v2)
- local v
- if v1 and v2 then
- v = v1 > v2
- end
- status.results = m.allocInfer {
- type = 'boolean',
- value = v,
- source = source,
- level = 100,
- }
- return true
- elseif op.type == '|' then
- local v1 = m.getInferLiteral(status, source[1], 'integer')
- local v2 = m.getInferLiteral(status, source[2], 'integer')
- v1 = tointeger(v1)
- v2 = tointeger(v2)
- local v
- if v1 and v2 then
- v = v1 | v2
- end
- status.results = m.allocInfer {
- type = 'integer',
- value = v,
- source = source,
- level = 100,
- }
- return true
- elseif op.type == '~' then
- local v1 = m.getInferLiteral(status, source[1], 'integer')
- local v2 = m.getInferLiteral(status, source[2], 'integer')
- v1 = tointeger(v1)
- v2 = tointeger(v2)
- local v
- if v1 and v2 then
- v = v1 ~ v2
- end
- status.results = m.allocInfer {
- type = 'integer',
- value = v,
- source = source,
- level = 100,
- }
- return true
- elseif op.type == '&' then
- local v1 = m.getInferLiteral(status, source[1], 'integer')
- local v2 = m.getInferLiteral(status, source[2], 'integer')
- v1 = tointeger(v1)
- v2 = tointeger(v2)
- local v
- if v1 and v2 then
- v = v1 & v2
- end
- status.results = m.allocInfer {
- type = 'integer',
- value = v,
- source = source,
- level = 100,
- }
- return true
- elseif op.type == '<<' then
- local v1 = m.getInferLiteral(status, source[1], 'integer')
- local v2 = m.getInferLiteral(status, source[2], 'integer')
- v1 = tointeger(v1)
- v2 = tointeger(v2)
- local v
- if v1 and v2 then
- v = v1 << v2
- end
- status.results = m.allocInfer {
- type = 'integer',
- value = v,
- source = source,
- level = 100,
- }
- return true
- elseif op.type == '>>' then
- local v1 = m.getInferLiteral(status, source[1], 'integer')
- local v2 = m.getInferLiteral(status, source[2], 'integer')
- v1 = tointeger(v1)
- v2 = tointeger(v2)
- local v
- if v1 and v2 then
- v = v1 >> v2
- end
- status.results = m.allocInfer {
- type = 'integer',
- value = v,
- source = source,
- level = 100,
- }
- return true
- elseif op.type == '..' then
- local v1 = m.getInferLiteral(status, source[1], 'string')
- local v2 = m.getInferLiteral(status, source[2], 'string')
- v1 = type(v1) == 'string' and v1 or nil
- v2 = type(v2) == 'string' and v2 or nil
- local v
- if v1 and v2 then
- v = v1 .. v2
- end
- status.results = m.allocInfer {
- type = 'string',
- value = v,
- source = source,
- level = 100,
- }
- return true
- elseif op.type == '^' then
- local v1 = m.getInferLiteral(status, source[1], 'integer')
- or m.getInferLiteral(status, source[1], 'number')
- local v2 = m.getInferLiteral(status, source[2], 'integer')
- or m.getInferLiteral(status, source[2], 'number')
- v1 = tonumber(v1)
- v2 = tonumber(v2)
- local v
- if v1 and v2 then
- v = v1 ^ v2
- end
- status.results = m.allocInfer {
- type = 'number',
- value = v,
- source = source,
- level = 100,
- }
- return true
- elseif op.type == '/' then
- local v1 = m.getInferLiteral(status, source[1], 'integer')
- or m.getInferLiteral(status, source[1], 'number')
- local v2 = m.getInferLiteral(status, source[2], 'integer')
- or m.getInferLiteral(status, source[2], 'number')
- v1 = tonumber(v1)
- v2 = tonumber(v2)
- local v
- if v1 and v2 and v2 ~= 0 then
- v = v1 / v2
- end
- status.results = m.allocInfer {
- type = 'number',
- value = v,
- source = source,
- level = 100,
- }
- return true
- -- 其他数学运算根据2侧的值决定,当2侧的值均为整数时返回整数
- elseif op.type == '+' then
- local int, v1, v2 = mathCheck(status, source[1], source[2])
- status.results = m.allocInfer{
- type = int,
- value = (v1 and v2) and (v1 + v2) or nil,
- source = source,
- level = 100,
- }
- return true
- elseif op.type == '-' then
- local int, v1, v2 = mathCheck(status, source[1], source[2])
- status.results = m.allocInfer{
- type = int,
- value = (v1 and v2) and (v1 - v2) or nil,
- source = source,
- level = 100,
- }
- return true
- elseif op.type == '*' then
- local int, v1, v2 = mathCheck(status, source[1], source[2])
- status.results = m.allocInfer {
- type = int,
- value = (v1 and v2) and (v1 * v2) or nil,
- source = source,
- level = 100,
- }
- return true
- elseif op.type == '%' then
- local int, v1, v2 = mathCheck(status, source[1], source[2])
- status.results = m.allocInfer {
- type = int,
- value = (v1 and v2 and v2 ~= 0) and (v1 % v2) or nil,
- source = source,
- level = 100,
- }
- return true
- elseif op.type == '//' then
- local int, v1, v2 = mathCheck(status, source[1], source[2])
- status.results = m.allocInfer {
- type = int,
- value = (v1 and v2 and v2 ~= 0) and (v1 // v2) or nil,
- source = source,
- level = 100,
- }
- return true
- end
-end
-
-function m.inferByDef(status, obj)
- if not status.share.inferedDef then
- status.share.inferedDef = {}
- end
- if status.share.inferedDef[obj] then
- return
- end
- status.share.inferedDef[obj] = true
- local mark = {}
- local newStatus = m.status(status, status.interface)
- tracy.ZoneBeginN('inferByDef searchRefs')
- m.searchRefs(newStatus, obj, 'def')
- tracy.ZoneEnd()
- for _, src in ipairs(newStatus.results) do
- local inferStatus = m.status(newStatus)
- m.searchInfer(inferStatus, src)
- if #inferStatus.results == 0 then
- status.results[#status.results+1] = {
- type = 'any',
- source = src,
- level = 0,
- }
- else
- for _, infer in ipairs(inferStatus.results) do
- if not mark[infer.source] then
- mark[infer.source] = true
- status.results[#status.results+1] = infer
- end
- end
- end
- end
-end
-
-local function inferBySetOfLocal(status, source)
- if status.share[source] then
- return
- end
- status.share[source] = true
- local newStatus = m.status(status)
- if source.value then
- m.searchInfer(newStatus, source.value)
- end
- if source.ref then
- for _, ref in ipairs(source.ref) do
- if ref.type == 'setlocal' then
- break
- end
- m.searchInfer(newStatus, ref)
- end
- for _, infer in ipairs(newStatus.results) do
- status.results[#status.results+1] = infer
- end
- end
-end
-
-function m.inferByLocalRef(status, source)
- if #status.results ~= 0 then
- return
- end
- if source.type == 'local' then
- inferBySetOfLocal(status, source)
- elseif source.type == 'setlocal'
- or source.type == 'getlocal' then
- inferBySetOfLocal(status, source.node)
- end
-end
-
-function m.inferByCall(status, source)
- if not source.parent then
- return
- end
- if source.parent.type ~= 'call' then
- return
- end
- if source.parent.node == source then
- status.results[#status.results+1] = {
- type = 'function',
- source = source,
- level = 10,
- }
- return
- end
-end
-
-function m.inferByGetTable(status, source)
- if source.type == 'field'
- or source.type == 'method' then
- source = source.parent
- end
- local next = source.next
- if not next then
- return
- end
- if next.type == 'getfield'
- or next.type == 'getindex'
- or next.type == 'setmethod'
- or next.type == 'setfield'
- or next.type == 'setindex' then
- status.results[#status.results+1] = {
- type = 'table',
- source = source,
- level = 10,
- }
- elseif next.type == 'getmethod' then
- status.results[#status.results+1] = {
- type = 'table',
- source = source,
- level = 5,
- }
- status.results[#status.results+1] = {
- type = 'string',
- source = source,
- level = 5,
- }
- end
-end
-
-function m.inferByUnary(status, source)
- local parent = source.parent
- if not parent or parent.type ~= 'unary' then
- return
- end
- local op = parent.op
- if op.type == '#' then
- status.results[#status.results+1] = {
- type = 'string',
- source = source,
- level = 5,
- }
- status.results[#status.results+1] = {
- type = 'table',
- source = source,
- level = 5,
- }
- elseif op.type == '~' then
- status.results[#status.results+1] = {
- type = 'integer',
- source = source,
- level = 10,
- }
- elseif op.type == '-' then
- status.results[#status.results+1] = {
- type = 'number',
- source = source,
- level = 10,
- }
- end
-end
-
-function m.inferByBinary(status, source)
- local parent = source.parent
- if not parent or parent.type ~= 'binary' then
- return
- end
- local op = parent.op
- if op.type == '<='
- or op.type == '>='
- or op.type == '<'
- or op.type == '>'
- or op.type == '^'
- or op.type == '/'
- or op.type == '+'
- or op.type == '-'
- or op.type == '*'
- or op.type == '%' then
- status.results[#status.results+1] = {
- type = 'number',
- source = source,
- level = 10,
- }
- elseif op.type == '|'
- or op.type == '~'
- or op.type == '&'
- or op.type == '<<'
- or op.type == '>>'
- -- 整数的可能性比较高
- or op.type == '//' then
- status.results[#status.results+1] = {
- type = 'integer',
- source = source,
- level = 10,
- }
- elseif op.type == '..' then
- status.results[#status.results+1] = {
- type = 'string',
- source = source,
- level = 10,
- }
- end
-end
-
-local function mergeFunctionReturnsByDoc(status, source, index, call)
- if not source or source.type ~= 'function' then
- return
- end
- if not source.bindDocs then
- return
- end
- local returns = {}
- for _, doc in ipairs(source.bindDocs) do
- if doc.type == 'doc.return' then
- for _, rtn in ipairs(doc.returns) do
- returns[#returns+1] = rtn
- end
- end
- end
- local rtn = returns[index]
- if not rtn then
- return
- end
- local results = m.getDocTypeNames(status, rtn)
- if #results == 0 then
- return
- end
- for _, res in ipairs(results) do
- status.results[#status.results+1] = res
- end
- return true
-end
-
-local function mergeDocTypeFunctionReturns(status, source, index)
- if not source.bindDocs then
- return
- end
- for _, doc in ipairs(source.bindDocs) do
- if doc.type == 'doc.type' then
- for _, typeUnit in ipairs(doc.types) do
- if typeUnit.type == 'doc.type.function' then
- local rtn = typeUnit.returns[index]
- if rtn then
- local results = m.getDocTypeNames(status, rtn)
- for _, res in ipairs(results) do
- status.results[#status.results+1] = res
- end
- end
- end
- end
- end
- end
-end
-
-local function mergeFunctionReturns(status, source, index, call)
- local returns = source.returns
- if not returns then
- return
- end
- for i = 1, #returns do
- local rtn = returns[i]
- if rtn[index] then
- if rtn[index].type == 'call' then
- if not m.checkReturnMark(status, rtn[index], true) then
- m.inferByCallReturnAndIndex(status, rtn[index], index)
- end
- else
- local newStatus = m.status(status)
- m.searchInfer(newStatus, rtn[index])
- if #newStatus.results == 0 then
- status.results[#status.results+1] = {
- type = 'any',
- source = rtn[index],
- level = 0,
- }
- else
- for _, infer in ipairs(newStatus.results) do
- status.results[#status.results+1] = infer
- end
- end
- end
- end
- end
-end
-
-function m.inferByCallReturnAndIndex(status, call, index)
- local node = call.node
- local newStatus = m.status(status, status.interface)
- m.searchRefs(newStatus, node, 'def')
- local hasDocReturn
- for _, src in ipairs(newStatus.results) do
- if mergeDocTypeFunctionReturns(status, src, index) then
- hasDocReturn = true
- elseif mergeFunctionReturnsByDoc(status, src.value, index, call) then
- hasDocReturn = true
- end
- end
- if not hasDocReturn then
- for _, src in ipairs(newStatus.results) do
- if src.value and src.value.type == 'function' then
- if not m.checkReturnMark(status, src.value, true) then
- mergeFunctionReturns(status, src.value, index, call)
- end
- end
- end
- end
-end
-
-function m.inferByCallReturn(status, source)
- if source.type == 'call' then
- m.inferByCallReturnAndIndex(status, source, 1)
- return
- end
- if source.type ~= 'select' then
- if source.value and source.value.type == 'select' then
- source = source.value
- else
- return
- end
- end
- if not source.vararg or source.vararg.type ~= 'call' then
- return
- end
- m.inferByCallReturnAndIndex(status, source.vararg, source.index)
-end
-
-function m.inferByPCallReturn(status, source)
- if source.type ~= 'select' then
- if source.value and source.value.type == 'select' then
- source = source.value
- else
- return
- end
- end
- local call = source.vararg
- if not call or call.type ~= 'call' then
- return
- end
- if not call.args then
- return
- end
- local node = call.node
- local specialName = node.special
- local func, index
- if specialName == 'pcall' then
- func = call.args[1]
- index = source.index - 1
- elseif specialName == 'xpcall' then
- func = call.args[1]
- index = source.index - 2
- else
- return
- end
- local newStatus = m.status(status, status.interface)
- m.searchRefs(newStatus, func, 'def')
- for _, src in ipairs(newStatus.results) do
- if src.value and src.value.type == 'function' then
- mergeFunctionReturns(status, src.value, index)
- end
- end
-end
-
-function m.cleanInfers(infers, obj)
- -- kick lower level infers
- local level = 0
- if obj.type ~= 'select' then
- for i = 1, #infers do
- local infer = infers[i]
- if infer.level > level then
- level = infer.level
- end
- end
- end
- -- merge infers
- local mark = {}
- for i = #infers, 1, -1 do
- local infer = infers[i]
- if infer.level < level then
- infers[i] = infers[#infers]
- infers[#infers] = nil
- goto CONTINUE
- end
- local key = ('%p'):format(infer.type, infer.source)
- if mark[key] then
- infers[i] = infers[#infers]
- infers[#infers] = nil
- else
- mark[key] = true
- end
- ::CONTINUE::
- end
- -- kick doc.generic
- if #infers > 1 then
- for i = #infers, 1, -1 do
- local infer = infers[i]
- if infer.source.typeGeneric then
- infers[i] = infers[#infers]
- infers[#infers] = nil
- end
- end
- end
-end
-
-function m.searchInfer(status, obj)
- while obj.type == 'paren' do
- obj = obj.exp
- if not obj then
- return
- end
- end
- while true do
- local value = m.getObjectValue(obj)
- if not value or value == obj then
- break
- end
- obj = value
- end
- if not obj then
- return
- end
-
- local cache, makeCache = m.getRefCache(status, obj, 'infer')
- if cache then
- for i = 1, #cache do
- status.results[#status.results+1] = cache[i]
- end
- return
- end
-
- if DEVELOP then
- status.share.clock = status.share.clock or osClock()
- end
-
- if not status.share.lockInfer then
- status.share.lockInfer = {}
- end
- if status.share.lockInfer[obj] then
- return
- end
- status.share.lockInfer[obj] = true
-
- local checked = m.inferCheckDoc(status, obj)
- or m.inferCheckUpDoc(status, obj)
- or m.inferCheckLiteral(status, obj)
- or m.inferCheckUnary(status, obj)
- or m.inferCheckBinary(status, obj)
- if checked then
- m.cleanInfers(status.results, obj)
- if makeCache then
- makeCache(status.results)
+ if interface then
+ for k, v in pairs(interface) do
+ status.interface[k] = v
end
- return
- end
-
- m.inferByLocalRef(status, obj)
- if status.deep then
- tracy.ZoneBeginN('inferByDef')
- m.inferByDef(status, obj)
- tracy.ZoneEnd()
- end
- --m.inferByCall(status, obj)
- m.inferByGetTable(status, obj)
- m.inferByUnary(status, obj)
- m.inferByBinary(status, obj)
- m.cleanInfers(status.results, obj)
- if makeCache then
- makeCache(status.results)
end
+ status.deep = status.depth <= status.searchDeep
+ return status
end
---- 请求对象的引用,包括 `a.b.c` 形式
---- 与 `return function` 形式。
---- 不穿透 `setmetatable` ,考虑由
---- 业务层进行反向 def 搜索。
+--- 请求对象的引用
+---@param obj parser.guide.object
+---@param interface table
+---@param deep integer
+---@return parser.guide.object[]
+---@return integer
function m.requestReference(obj, interface, deep)
local status = m.status(nil, interface, deep)
-- 根据 field 搜索引用
m.searchRefs(status, obj, 'ref')
- m.searchRefsAsFunction(status, obj, 'ref')
-
- if m.debugMode then
- --print('count:', status.share.count)
- end
-
- return status.results, status.share.count
-end
-
---- 请求对象的定义,包括 `a.b.c` 形式
---- 与 `return function` 形式。
---- 穿透 `setmetatable` 。
-function m.requestDefinition(obj, interface, deep)
- local status = m.status(nil, interface, deep)
- -- 根据 field 搜索定义
- m.searchRefs(status, obj, 'def')
-
- return status.results, status.share.count
-end
-
---- 请求对象的字段
----@param filterKey nil|string|table
-function m.requestFields(obj, interface, deep, filterKey)
- local status = m.status(nil, interface, deep)
-
- m.searchFields(status, obj, filterKey)
-
- return status.results, status.share.count
-end
-
---- 请求对象的定义字段
----@param filterKey nil|string|table
-function m.requestDefFields(obj, interface, deep, filterKey)
- local status = m.status(nil, interface, deep)
-
- m.searchDefFields(status, obj, filterKey)
-
- return status.results, status.share.count
-end
-
---- 请求对象的类型推测
-function m.requestInfer(obj, interface, deep)
- local status = m.status(nil, interface, deep)
- m.searchInfer(status, obj)
-
- return status.results, status.share.count
-end
-
-function m.debugView(obj)
- return require 'files'.position(m.getUri(obj), obj.start), m.getUri(obj)
+ return status.results, 0
end
return m
diff --git a/script/core/guide2.lua b/script/core/guide2.lua
new file mode 100644
index 00000000..64192297
--- /dev/null
+++ b/script/core/guide2.lua
@@ -0,0 +1,4837 @@
+local util = require 'utility'
+local error = error
+local type = type
+local next = next
+local tostring = tostring
+local print = print
+local ipairs = ipairs
+local tableInsert = table.insert
+local tableUnpack = table.unpack
+local tableRemove = table.remove
+local tableMove = table.move
+local tableSort = table.sort
+local tableConcat = table.concat
+local mathType = math.type
+local pairs = pairs
+local setmetatable = setmetatable
+local assert = assert
+local select = select
+local osClock = os.clock
+local tonumber = tonumber
+local tointeger = math.tointeger
+local DEVELOP = _G.DEVELOP
+local log = log
+local _G = _G
+
+local function logWarn(...)
+ log.warn(...)
+end
+
+local m = {}
+
+m.ANY = {"<ANY>"}
+
+local blockTypes = {
+ ['while'] = true,
+ ['in'] = true,
+ ['loop'] = true,
+ ['repeat'] = true,
+ ['do'] = true,
+ ['function'] = true,
+ ['ifblock'] = true,
+ ['elseblock'] = true,
+ ['elseifblock'] = true,
+ ['main'] = true,
+}
+
+local breakBlockTypes = {
+ ['while'] = true,
+ ['in'] = true,
+ ['loop'] = true,
+ ['repeat'] = true,
+}
+
+m.childMap = {
+ ['main'] = {'#', 'docs'},
+ ['repeat'] = {'#', 'filter'},
+ ['while'] = {'filter', '#'},
+ ['in'] = {'keys', '#'},
+ ['loop'] = {'loc', 'max', 'step', '#'},
+ ['if'] = {'#'},
+ ['ifblock'] = {'filter', '#'},
+ ['elseifblock'] = {'filter', '#'},
+ ['elseblock'] = {'#'},
+ ['setfield'] = {'node', 'field', 'value'},
+ ['setglobal'] = {'value'},
+ ['local'] = {'attrs', 'value'},
+ ['setlocal'] = {'value'},
+ ['return'] = {'#'},
+ ['do'] = {'#'},
+ ['select'] = {'vararg'},
+ ['table'] = {'#'},
+ ['tableindex'] = {'index', 'value'},
+ ['tablefield'] = {'field', 'value'},
+ ['function'] = {'args', '#'},
+ ['funcargs'] = {'#'},
+ ['setmethod'] = {'node', 'method', 'value'},
+ ['getmethod'] = {'node', 'method'},
+ ['setindex'] = {'node', 'index', 'value'},
+ ['getindex'] = {'node', 'index'},
+ ['paren'] = {'exp'},
+ ['call'] = {'node', 'args'},
+ ['callargs'] = {'#'},
+ ['getfield'] = {'node', 'field'},
+ ['list'] = {'#'},
+ ['binary'] = {1, 2},
+ ['unary'] = {1},
+
+ ['doc'] = {'#'},
+ ['doc.class'] = {'class', '#extends', 'comment'},
+ ['doc.type'] = {'#types', '#enums', 'name', 'comment'},
+ ['doc.alias'] = {'alias', 'extends', 'comment'},
+ ['doc.param'] = {'param', 'extends', 'comment'},
+ ['doc.return'] = {'#returns', 'comment'},
+ ['doc.field'] = {'field', 'extends', 'comment'},
+ ['doc.generic'] = {'#generics', 'comment'},
+ ['doc.generic.object'] = {'generic', 'extends', 'comment'},
+ ['doc.vararg'] = {'vararg', 'comment'},
+ ['doc.type.array'] = {'node'},
+ ['doc.type.table'] = {'node', 'key', 'value', 'comment'},
+ ['doc.type.function'] = {'#args', '#returns', 'comment'},
+ ['doc.type.typeliteral'] = {'node'},
+ ['doc.type.arg'] = {'extends'},
+ ['doc.overload'] = {'overload', 'comment'},
+ ['doc.see'] = {'name', 'field', 'comment'},
+ ['doc.diagnostic'] = {'#names', 'comment'},
+}
+
+m.actionMap = {
+ ['main'] = {'#'},
+ ['repeat'] = {'#'},
+ ['while'] = {'#'},
+ ['in'] = {'#'},
+ ['loop'] = {'#'},
+ ['if'] = {'#'},
+ ['ifblock'] = {'#'},
+ ['elseifblock'] = {'#'},
+ ['elseblock'] = {'#'},
+ ['do'] = {'#'},
+ ['function'] = {'#'},
+ ['funcargs'] = {'#'},
+}
+
+local TypeSort = {
+ ['boolean'] = 1,
+ ['string'] = 2,
+ ['integer'] = 3,
+ ['number'] = 4,
+ ['table'] = 5,
+ ['function'] = 6,
+ ['true'] = 101,
+ ['false'] = 102,
+ ['nil'] = 999,
+}
+
+local NIL = setmetatable({'<nil>'}, { __tostring = function () return 'nil' end })
+
+--- 是否是字面量
+---@param obj parser.guide.object
+---@return boolean
+function m.isLiteral(obj)
+ local tp = obj.type
+ return tp == 'nil'
+ or tp == 'boolean'
+ or tp == 'string'
+ or tp == 'number'
+ or tp == 'table'
+ or tp == 'function'
+end
+
+--- 获取字面量
+---@param obj parser.guide.object
+---@return any
+function m.getLiteral(obj)
+ local tp = obj.type
+ if tp == 'boolean' then
+ return obj[1]
+ elseif tp == 'string' then
+ return obj[1]
+ elseif tp == 'number' then
+ return obj[1]
+ end
+ return nil
+end
+
+--- 寻找父函数
+---@param obj parser.guide.object
+---@return parser.guide.object
+function m.getParentFunction(obj)
+ for _ = 1, 1000 do
+ obj = obj.parent
+ if not obj then
+ break
+ end
+ local tp = obj.type
+ if tp == 'function' or tp == 'main' then
+ return obj
+ end
+ end
+ return nil
+end
+
+--- 寻找父的table类型 doc.type.table
+---@param obj parser.guide.object
+---@return parser.guide.object
+function m.getParentDocTypeTable(obj)
+ for _ = 1, 1000 do
+ local parent = obj.parent
+ if not parent then
+ return nil
+ end
+ if parent.type == 'doc.type.table' then
+ return obj
+ end
+ obj = parent
+ end
+ error('guide.getParentDocTypeTable overstack')
+end
+
+--- 寻找所在区块
+---@param obj parser.guide.object
+---@return parser.guide.object
+function m.getBlock(obj)
+ for _ = 1, 1000 do
+ if not obj then
+ return nil
+ end
+ local tp = obj.type
+ if blockTypes[tp] then
+ return obj
+ end
+ if obj == obj.parent then
+ error('obj == obj.parent?', obj.type)
+ end
+ obj = obj.parent
+ end
+ error('guide.getBlock overstack')
+end
+
+--- 寻找所在父区块
+---@param obj parser.guide.object
+---@return parser.guide.object
+function m.getParentBlock(obj)
+ for _ = 1, 1000 do
+ obj = obj.parent
+ if not obj then
+ return nil
+ end
+ local tp = obj.type
+ if blockTypes[tp] then
+ return obj
+ end
+ end
+ error('guide.getParentBlock overstack')
+end
+
+--- 寻找所在可break的父区块
+---@param obj parser.guide.object
+---@return parser.guide.object
+function m.getBreakBlock(obj)
+ for _ = 1, 1000 do
+ obj = obj.parent
+ if not obj then
+ return nil
+ end
+ local tp = obj.type
+ if breakBlockTypes[tp] then
+ return obj
+ end
+ if tp == 'function' then
+ return nil
+ end
+ end
+ error('guide.getBreakBlock overstack')
+end
+
+--- 寻找doc的主体
+---@param obj parser.guide.object
+---@return parser.guide.object
+function m.getDocState(obj)
+ for _ = 1, 1000 do
+ local parent = obj.parent
+ if not parent then
+ return obj
+ end
+ if parent.type == 'doc' then
+ return obj
+ end
+ obj = parent
+ end
+ error('guide.getDocState overstack')
+end
+
+--- 寻找所在父类型
+---@param obj parser.guide.object
+---@return parser.guide.object
+function m.getParentType(obj, want)
+ for _ = 1, 1000 do
+ obj = obj.parent
+ if not obj then
+ return nil
+ end
+ if want == obj.type then
+ return obj
+ end
+ end
+ error('guide.getParentType overstack')
+end
+
+--- 寻找根区块
+---@param obj parser.guide.object
+---@return parser.guide.object
+function m.getRoot(obj)
+ local source = obj
+ if source._root then
+ return source._root
+ end
+ for _ = 1, 1000 do
+ if obj.type == 'main' then
+ source._root = obj
+ return obj
+ end
+ local parent = obj.parent
+ if not parent then
+ return nil
+ end
+ obj = parent
+ end
+ error('guide.getRoot overstack')
+end
+
+---@param obj parser.guide.object
+---@return string
+function m.getUri(obj)
+ if obj.uri then
+ return obj.uri
+ end
+ local root = m.getRoot(obj)
+ if root then
+ return root.uri
+ end
+ return ''
+end
+
+function m.getENV(source, start)
+ if not start then
+ start = 1
+ end
+ return m.getLocal(source, '_ENV', start)
+ or m.getLocal(source, '@fenv', start)
+end
+
+--- 寻找函数的不定参数,返回不定参在第几个参数上,以及该参数对象。
+--- 如果函数是主函数,则返回`0, nil`。
+---@return table
+---@return integer
+function m.getFunctionVarArgs(func)
+ if func.type == 'main' then
+ return 0, nil
+ end
+ if func.type ~= 'function' then
+ return nil, nil
+ end
+ local args = func.args
+ if not args then
+ return nil, nil
+ end
+ for i = 1, #args do
+ local arg = args[i]
+ if arg.type == '...' then
+ return i, arg
+ end
+ end
+ return nil, nil
+end
+
+--- 获取指定区块中可见的局部变量
+---@param block table
+---@param name string {comment = '变量名'}
+---@param pos integer {comment = '可见位置'}
+function m.getLocal(block, name, pos)
+ block = m.getBlock(block)
+ for _ = 1, 1000 do
+ if not block then
+ return nil
+ end
+ local locals = block.locals
+ local res
+ if not locals then
+ goto CONTINUE
+ end
+ for i = 1, #locals do
+ local loc = locals[i]
+ if loc.effect > pos then
+ break
+ end
+ if loc[1] == name then
+ if not res or res.effect < loc.effect then
+ res = loc
+ end
+ end
+ end
+ if res then
+ return res, res
+ end
+ ::CONTINUE::
+ block = m.getParentBlock(block)
+ end
+ error('guide.getLocal overstack')
+end
+
+--- 获取指定区块中所有的可见局部变量名称
+function m.getVisibleLocals(block, pos)
+ local result = {}
+ m.eachSourceContain(m.getRoot(block), pos, function (source)
+ local locals = source.locals
+ if locals then
+ for i = 1, #locals do
+ local loc = locals[i]
+ local name = loc[1]
+ if loc.effect <= pos then
+ result[name] = loc
+ end
+ end
+ end
+ end)
+ return result
+end
+
+--- 获取指定区块中可见的标签
+---@param block table
+---@param name string {comment = '标签名'}
+function m.getLabel(block, name)
+ block = m.getBlock(block)
+ for _ = 1, 1000 do
+ if not block then
+ return nil
+ end
+ local labels = block.labels
+ if labels then
+ local label = labels[name]
+ if label then
+ return label
+ end
+ end
+ if block.type == 'function' then
+ return nil
+ end
+ block = m.getParentBlock(block)
+ end
+ error('guide.getLocal overstack')
+end
+
+function m.getStartFinish(source)
+ local start = source.start
+ local finish = source.finish
+ if not start then
+ local first = source[1]
+ if not first then
+ return nil, nil
+ end
+ local last = source[#source]
+ start = first.start
+ finish = last.finish
+ end
+ return start, finish
+end
+
+function m.getRange(source)
+ local start = source.vstart or source.start
+ local finish = source.range or source.finish
+ if not start then
+ local first = source[1]
+ if not first then
+ return nil, nil
+ end
+ local last = source[#source]
+ start = first.vstart or first.start
+ finish = last.range or last.finish
+ end
+ return start, finish
+end
+
+--- 判断source是否包含offset
+function m.isContain(source, offset)
+ local start, finish = m.getStartFinish(source)
+ if not start then
+ return false
+ end
+ return start <= offset and finish >= offset
+end
+
+--- 判断offset在source的影响范围内
+---
+--- 主要针对赋值等语句时,key包含value
+function m.isInRange(source, offset)
+ local start, finish = m.getRange(source)
+ if not start then
+ return false
+ end
+ return start <= offset and finish >= offset
+end
+
+function m.isBetween(source, tStart, tFinish)
+ local start, finish = m.getStartFinish(source)
+ if not start then
+ return false
+ end
+ return start <= tFinish and finish >= tStart
+end
+
+function m.isBetweenRange(source, tStart, tFinish)
+ local start, finish = m.getRange(source)
+ if not start then
+ return false
+ end
+ return start <= tFinish and finish >= tStart
+end
+
+--- 添加child
+function m.addChilds(list, obj, map)
+ local keys = map[obj.type]
+ if keys then
+ for i = 1, #keys do
+ local key = keys[i]
+ if key == '#' then
+ for i = 1, #obj do
+ list[#list+1] = obj[i]
+ end
+ elseif obj[key] then
+ list[#list+1] = obj[key]
+ elseif type(key) == 'string'
+ and key:sub(1, 1) == '#' then
+ key = key:sub(2)
+ if obj[key] then
+ for i = 1, #obj[key] do
+ list[#list+1] = obj[key][i]
+ end
+ end
+ end
+ end
+ end
+end
+
+--- 遍历所有包含offset的source
+function m.eachSourceContain(ast, offset, callback)
+ local list = { ast }
+ local mark = {}
+ while true do
+ local len = #list
+ if len == 0 then
+ return
+ end
+ local obj = list[len]
+ list[len] = nil
+ if not mark[obj] then
+ mark[obj] = true
+ if m.isInRange(obj, offset) then
+ if m.isContain(obj, offset) then
+ local res = callback(obj)
+ if res ~= nil then
+ return res
+ end
+ end
+ m.addChilds(list, obj, m.childMap)
+ end
+ end
+ end
+end
+
+--- 遍历所有在某个范围内的source
+function m.eachSourceBetween(ast, start, finish, callback)
+ local list = { ast }
+ local mark = {}
+ while true do
+ local len = #list
+ if len == 0 then
+ return
+ end
+ local obj = list[len]
+ list[len] = nil
+ if not mark[obj] then
+ mark[obj] = true
+ if m.isBetweenRange(obj, start, finish) then
+ if m.isBetween(obj, start, finish) then
+ local res = callback(obj)
+ if res ~= nil then
+ return res
+ end
+ end
+ m.addChilds(list, obj, m.childMap)
+ end
+ end
+ end
+end
+
+--- 遍历所有指定类型的source
+function m.eachSourceType(ast, type, callback)
+ local cache = ast._typeCache
+ if not cache then
+ cache = {}
+ ast._typeCache = cache
+ m.eachSource(ast, function (source)
+ local tp = source.type
+ if not tp then
+ return
+ end
+ local myCache = cache[tp]
+ if not myCache then
+ myCache = {}
+ cache[tp] = myCache
+ end
+ myCache[#myCache+1] = source
+ end)
+ end
+ local myCache = cache[type]
+ if not myCache then
+ return
+ end
+ for i = 1, #myCache do
+ callback(myCache[i])
+ end
+end
+
+--- 遍历所有的source
+function m.eachSource(ast, callback)
+ local list = { ast }
+ local mark = {}
+ local index = 1
+ while true do
+ local obj = list[index]
+ if not obj then
+ return
+ end
+ list[index] = false
+ index = index + 1
+ if not mark[obj] then
+ mark[obj] = true
+ callback(obj)
+ m.addChilds(list, obj, m.childMap)
+ end
+ end
+end
+
+--- 获取指定的 special
+function m.eachSpecialOf(ast, name, callback)
+ local root = m.getRoot(ast)
+ if not root.specials then
+ return
+ end
+ local specials = root.specials[name]
+ if not specials then
+ return
+ end
+ for i = 1, #specials do
+ callback(specials[i])
+ end
+end
+
+--- 获取偏移对应的坐标
+---@param lines table
+---@return integer row
+---@return integer col
+function m.positionOf(lines, offset)
+ if offset < 1 then
+ return 0, 0
+ end
+ local lastLine = lines[#lines]
+ if offset > lastLine.finish then
+ return #lines, offset - lastLine.start
+ end
+ local min = 1
+ local max = #lines
+ for _ = 1, 100 do
+ if max <= min then
+ local line = lines[min]
+ return min, offset - line.start + 1
+ end
+ local row = (max - min) // 2 + min
+ local line = lines[row]
+ if offset < line.start then
+ max = row - 1
+ elseif offset > line.finish then
+ min = row + 1
+ else
+ return row, offset - line.start + 1
+ end
+ end
+ error('Stack overflow!')
+end
+
+--- 获取坐标对应的偏移
+---@param lines table
+---@param row integer
+---@param col integer
+---@return integer {name = 'offset'}
+function m.offsetOf(lines, row, col)
+ if row < 1 then
+ return 0
+ end
+ if row > #lines then
+ local lastLine = lines[#lines]
+ return lastLine.finish
+ end
+ local line = lines[row]
+ local len = line.finish - line.start + 1
+ if col < 0 then
+ return line.start
+ elseif col > len then
+ return line.finish
+ else
+ return line.start + col - 1
+ end
+end
+
+function m.lineContent(lines, text, row, ignoreNL)
+ local line = lines[row]
+ if not line then
+ return ''
+ end
+ if ignoreNL then
+ return text:sub(line.start, line.range)
+ else
+ return text:sub(line.start, line.finish)
+ end
+end
+
+function m.lineRange(lines, row, ignoreNL)
+ local line = lines[row]
+ if not line then
+ return 0, 0
+ end
+ if ignoreNL then
+ return line.start, line.range
+ else
+ return line.start, line.finish
+ end
+end
+
+function m.lineData(lines, row)
+ return lines[row]
+end
+
+function m.getKeyTypeOfLiteral(obj)
+ if not obj then
+ return nil
+ end
+ local tp = obj.type
+ if tp == 'field'
+ or tp == 'method' then
+ return 'string'
+ elseif tp == 'string' then
+ return 'string'
+ elseif tp == 'number' then
+ return 'number'
+ elseif tp == 'boolean' then
+ return 'boolean'
+ end
+end
+
+function m.getKeyType(obj)
+ if not obj then
+ return nil
+ end
+ local tp = obj.type
+ if tp == 'getglobal'
+ or tp == 'setglobal' then
+ return 'string'
+ elseif tp == 'local'
+ or tp == 'getlocal'
+ or tp == 'setlocal' then
+ return 'local'
+ elseif tp == 'getfield'
+ or tp == 'setfield'
+ or tp == 'tablefield' then
+ return 'string'
+ elseif tp == 'getmethod'
+ or tp == 'setmethod' then
+ return 'string'
+ elseif tp == 'getindex'
+ or tp == 'setindex'
+ or tp == 'tableindex' then
+ return m.getKeyTypeOfLiteral(obj.index)
+ elseif tp == 'field'
+ or tp == 'method'
+ or tp == 'doc.see.field' then
+ return 'string'
+ elseif tp == 'doc.class' then
+ return 'string'
+ elseif tp == 'doc.alias' then
+ return 'string'
+ elseif tp == 'doc.field' then
+ return 'string'
+ elseif tp == 'dummy' then
+ return 'string'
+ end
+ return m.getKeyTypeOfLiteral(obj)
+end
+
+function m.getKeyNameOfLiteral(obj)
+ if not obj then
+ return nil
+ end
+ local tp = obj.type
+ if tp == 'field'
+ or tp == 'method' then
+ return obj[1]
+ elseif tp == 'string' then
+ local s = obj[1]
+ if s then
+ return s
+ end
+ elseif tp == 'number' then
+ local n = obj[1]
+ if n then
+ return ('%s'):format(util.viewLiteral(obj[1]))
+ end
+ elseif tp == 'boolean' then
+ local b = obj[1]
+ if b then
+ return tostring(b)
+ end
+ end
+end
+
+function m.getKeyName(obj)
+ if not obj then
+ return nil
+ end
+ local tp = obj.type
+ if tp == 'getglobal'
+ or tp == 'setglobal' then
+ return obj[1]
+ elseif tp == 'local'
+ or tp == 'getlocal'
+ or tp == 'setlocal' then
+ return obj[1]
+ elseif tp == 'getfield'
+ or tp == 'setfield'
+ or tp == 'tablefield' then
+ if obj.field then
+ return obj.field[1]
+ end
+ elseif tp == 'getmethod'
+ or tp == 'setmethod' then
+ if obj.method then
+ return obj.method[1]
+ end
+ elseif tp == 'getindex'
+ or tp == 'setindex'
+ or tp == 'tableindex' then
+ return m.getKeyNameOfLiteral(obj.index)
+ elseif tp == 'field'
+ or tp == 'method'
+ or tp == 'doc.see.field' then
+ return obj[1]
+ elseif tp == 'doc.class' then
+ return obj.class[1]
+ elseif tp == 'doc.alias' then
+ return obj.alias[1]
+ elseif tp == 'doc.field' then
+ return obj.field[1]
+ elseif tp == 'dummy' then
+ return obj[1]
+ end
+ return m.getKeyNameOfLiteral(obj)
+end
+
+function m.getSimpleName(obj)
+ if obj.type == 'call' then
+ local node = obj.node
+ if not node then
+ return
+ end
+ if node.special == 'rawset'
+ or node.special == 'rawget' then
+ local key = obj.args and obj.args[2]
+ return m.getKeyName(key)
+ end
+ return ('%p'):format(obj)
+ elseif obj.type == 'table' then
+ return ('%p'):format(obj)
+ elseif obj.type == 'select' then
+ return ('%p'):format(obj)
+ elseif obj.type == 'string' then
+ return ('%p'):format(obj)
+ elseif obj.type == 'doc.class.name'
+ or obj.type == 'doc.type.name'
+ or obj.type == 'doc.see.name' then
+ return ('%s'):format(obj[1])
+ elseif obj.type == 'doc.class' then
+ return ('%s'):format(obj.class[1])
+ end
+ return m.getKeyName(obj)
+end
+
+--- 测试 a 到 b 的路径(不经过函数,不考虑 goto),
+--- 每个路径是一个 block 。
+---
+--- 如果 a 在 b 的前面,返回 `"before"` 加上 2个`list<block>`
+---
+--- 如果 a 在 b 的后面,返回 `"after"` 加上 2个`list<block>`
+---
+--- 否则返回 `false`
+---
+--- 返回的2个 `list` 分别为基准block到达 a 与 b 的路径。
+---@param a table
+---@param b table
+---@return string|boolean mode
+---@return table pathA?
+---@return table pathB?
+function m.getPath(a, b, sameFunction)
+ --- 首先测试双方在同一个函数内
+ if sameFunction and m.getParentFunction(a) ~= m.getParentFunction(b) then
+ return false
+ end
+ local mode
+ local objA
+ local objB
+ if a.finish < b.start then
+ mode = 'before'
+ objA = a
+ objB = b
+ elseif a.start > b.finish then
+ mode = 'after'
+ objA = b
+ objB = a
+ else
+ return 'equal', {}, {}
+ end
+ local pathA = {}
+ local pathB = {}
+ for _ = 1, 1000 do
+ objA = m.getParentBlock(objA)
+ pathA[#pathA+1] = objA
+ if (not sameFunction and objA.type == 'function') or objA.type == 'main' then
+ break
+ end
+ end
+ for _ = 1, 1000 do
+ objB = m.getParentBlock(objB)
+ pathB[#pathB+1] = objB
+ if (not sameFunction and objA.type == 'function') or objB.type == 'main' then
+ break
+ end
+ end
+ -- pathA: {1, 2, 3, 4, 5}
+ -- pathB: {5, 6, 2, 3}
+ local top = #pathB
+ local start
+ for i = #pathA, 1, -1 do
+ local currentBlock = pathA[i]
+ if currentBlock == pathB[top] then
+ start = i
+ break
+ end
+ end
+ if not start then
+ return nil
+ end
+ -- pathA: { 1, 2, 3}
+ -- pathB: {5, 6, 2, 3}
+ local extra = 0
+ local align = top - start
+ for i = start, 1, -1 do
+ local currentA = pathA[i]
+ local currentB = pathB[i+align]
+ if currentA ~= currentB then
+ extra = i
+ break
+ end
+ end
+ -- pathA: {1}
+ local resultA = {}
+ for i = extra, 1, -1 do
+ resultA[#resultA+1] = pathA[i]
+ end
+ -- pathB: {5, 6}
+ local resultB = {}
+ for i = extra + align, 1, -1 do
+ resultB[#resultB+1] = pathB[i]
+ end
+ return mode, resultA, resultB
+end
+
+-- 根据语法,单步搜索定义
+local function stepRefOfLocal(loc, mode)
+ local results = {}
+ if loc.start ~= 0 then
+ results[#results+1] = loc
+ end
+ local refs = loc.ref
+ if not refs then
+ return results
+ end
+ for i = 1, #refs do
+ local ref = refs[i]
+ if ref.start == 0 then
+ goto CONTINUE
+ end
+ if mode == 'def' then
+ if ref.type == 'local'
+ or ref.type == 'setlocal' then
+ results[#results+1] = ref
+ end
+ else
+ if ref.type == 'local'
+ or ref.type == 'setlocal'
+ or ref.type == 'getlocal' then
+ results[#results+1] = ref
+ end
+ end
+ ::CONTINUE::
+ end
+ return results
+end
+
+local function stepRefOfLabel(label, mode)
+ local results = { label }
+ if not label or mode == 'def' then
+ return results
+ end
+ local refs = label.ref
+ if not refs then
+ return results
+ end
+ for i = 1, #refs do
+ local ref = refs[i]
+ results[#results+1] = ref
+ end
+ return results
+end
+
+local function stepRefOfDocType(status, obj, mode)
+ local results = {}
+ if obj.type == 'doc.class.name'
+ or obj.type == 'doc.type.name'
+ or obj.type == 'doc.alias.name'
+ or obj.type == 'doc.extends.name'
+ or obj.type == 'doc.see.name' then
+ local name = obj[1]
+ if not name or not status.interface.docType then
+ return results
+ end
+ if name == 'nil'
+ or name == 'any'
+ or name == 'boolean'
+ or name == 'string'
+ or name == 'table'
+ or name == 'number'
+ or name == 'integer'
+ or name == 'function'
+ or name == 'table'
+ or name == 'thread'
+ or name == 'userdata'
+ or name == 'lightuserdata' then
+ mode = 'def'
+ end
+ local docs = status.interface.docType(name)
+ for i = 1, #docs do
+ local doc = docs[i]
+ if mode == 'def' then
+ if doc.type == 'doc.class.name'
+ or doc.type == 'doc.alias.name' then
+ results[#results+1] = doc
+ end
+ else
+ results[#results+1] = doc
+ end
+ end
+ else
+ results[#results+1] = obj
+ end
+ return results
+end
+
+function m.getStepRef(status, obj, mode)
+ if obj.type == 'getlocal'
+ or obj.type == 'setlocal' then
+ return stepRefOfLocal(obj.node, mode)
+ end
+ if obj.type == 'local' then
+ return stepRefOfLocal(obj, mode)
+ end
+ if obj.type == 'label' then
+ return stepRefOfLabel(obj, mode)
+ end
+ if obj.type == 'goto' then
+ return stepRefOfLabel(obj.node, mode)
+ end
+ if obj.type == 'doc.class.name'
+ or obj.type == 'doc.type.name'
+ or obj.type == 'doc.extends.name'
+ or obj.type == 'doc.alias.name' then
+ return stepRefOfDocType(status, obj, mode)
+ end
+ if obj.type == 'function' then
+ return { obj }
+ end
+ return nil
+end
+
+-- 根据语法,单步搜索field
+local function stepFieldOfLocal(loc)
+ local results = {}
+ local refs = loc.ref
+ for i = 1, #refs do
+ local ref = refs[i]
+ if ref.type == 'setglobal'
+ or ref.type == 'getglobal' then
+ results[#results+1] = ref
+ elseif ref.type == 'getlocal' then
+ local nxt = ref.next
+ if nxt then
+ if nxt.type == 'setfield'
+ or nxt.type == 'getfield'
+ or nxt.type == 'setmethod'
+ or nxt.type == 'getmethod'
+ or nxt.type == 'setindex'
+ or nxt.type == 'getindex' then
+ results[#results+1] = nxt
+ end
+ end
+ end
+ end
+ return results
+end
+local function stepFieldOfTable(tbl)
+ local result = {}
+ for i = 1, #tbl do
+ result[i] = tbl[i]
+ end
+ return result
+end
+function m.getStepField(obj)
+ if obj.type == 'getlocal'
+ or obj.type == 'setlocal' then
+ return stepFieldOfLocal(obj.node)
+ end
+ if obj.type == 'local' then
+ return stepFieldOfLocal(obj)
+ end
+ if obj.type == 'table' then
+ return stepFieldOfTable(obj)
+ end
+end
+
+local function convertSimpleList(list)
+ local simple = {}
+ for i = #list, 1, -1 do
+ local c = list[i]
+ if c.type == 'getglobal'
+ or c.type == 'setglobal' then
+ if c.special == '_G' then
+ simple.mode = 'global'
+ goto CONTINUE
+ end
+ local loc = c.node
+ if loc.special == '_G' then
+ simple.mode = 'global'
+ if not simple.node then
+ simple.node = c
+ end
+ else
+ simple.mode = 'local'
+ simple[#simple+1] = m.getSimpleName(loc)
+ if not simple.node then
+ simple.node = loc
+ end
+ end
+ elseif c.type == 'getlocal'
+ or c.type == 'setlocal' then
+ if c.special == '_G' then
+ simple.mode = 'global'
+ goto CONTINUE
+ end
+ simple.mode = 'local'
+ if not simple.node then
+ simple.node = c.node
+ end
+ elseif c.type == 'local' then
+ simple.mode = 'local'
+ if not simple.node then
+ simple.node = c
+ end
+ else
+ if not simple.node then
+ simple.node = c
+ end
+ end
+ simple[#simple+1] = m.getSimpleName(c) or m.ANY
+ ::CONTINUE::
+ end
+ if simple.mode == 'global' and #simple == 0 then
+ simple[1] = '_G'
+ simple.node = list[#list]
+ end
+ return simple
+end
+
+-- 搜索 `a.b.c` 的等价表达式
+local function buildSimpleList(obj, max)
+ local list = {}
+ local cur = obj
+ local limit = max and (max + 1) or 11
+ for i = 1, max or limit do
+ if i == limit then
+ return nil
+ end
+ while cur.type == 'paren' do
+ cur = cur.exp
+ if not cur then
+ return nil
+ end
+ end
+ if cur.type == 'setfield'
+ or cur.type == 'getfield'
+ or cur.type == 'setmethod'
+ or cur.type == 'getmethod'
+ or cur.type == 'setindex'
+ or cur.type == 'getindex' then
+ list[i] = cur
+ cur = cur.node
+ elseif cur.type == 'tablefield'
+ or cur.type == 'tableindex' then
+ list[i] = cur
+ cur = cur.parent.parent
+ if cur.type == 'return' then
+ list[i+1] = list[i].parent
+ break
+ end
+ elseif cur.type == 'getlocal'
+ or cur.type == 'setlocal'
+ or cur.type == 'local' then
+ list[i] = cur
+ break
+ elseif cur.type == 'setglobal'
+ or cur.type == 'getglobal' then
+ list[i] = cur
+ break
+ elseif cur.type == 'select'
+ or cur.type == 'table'
+ or cur.type == 'call' then
+ list[i] = cur
+ break
+ elseif cur.type == 'string' then
+ list[i] = cur
+ break
+ elseif cur.type == '...' then
+ list[i] = cur
+ break
+ elseif cur.type == 'doc.class.name'
+ or cur.type == 'doc.type.name'
+ or cur.type == 'doc.class'
+ or cur.type == 'doc.see.name' then
+ list[i] = cur
+ break
+ elseif cur.type == 'doc.see.field' then
+ list[i] = cur
+ cur = cur.parent.name
+ elseif cur.type == 'function'
+ or cur.type == 'main' then
+ break
+ else
+ return nil
+ end
+ end
+ return convertSimpleList(list)
+end
+
+function m.getSimple(obj, max)
+ local simpleList
+ if obj.type == 'getfield'
+ or obj.type == 'setfield'
+ or obj.type == 'getmethod'
+ or obj.type == 'setmethod'
+ or obj.type == 'getindex'
+ or obj.type == 'setindex'
+ or obj.type == 'local'
+ or obj.type == 'getlocal'
+ or obj.type == 'setlocal'
+ or obj.type == 'setglobal'
+ or obj.type == 'getglobal'
+ or obj.type == 'tablefield'
+ or obj.type == 'tableindex'
+ or obj.type == 'select'
+ or obj.type == 'call'
+ or obj.type == 'table'
+ or obj.type == 'string'
+ or obj.type == '...'
+ or obj.type == 'doc.class.name'
+ or obj.type == 'doc.class'
+ or obj.type == 'doc.type.name'
+ or obj.type == 'doc.see.name'
+ or obj.type == 'doc.see.field' then
+ simpleList = buildSimpleList(obj, max)
+ elseif obj.type == 'field'
+ or obj.type == 'method' then
+ simpleList = buildSimpleList(obj.parent, max)
+ end
+ return simpleList
+end
+
+---Create a new status
+---@param parentStatus core.guide.status
+---@param interface table
+---@param deep boolean
+---@return core.guide.status
+function m.status(parentStatus, interface, deep)
+ ---@class core.guide.status
+ local status = {
+ share = parentStatus and parentStatus.share or {
+ count = 0,
+ cacheLock = {},
+ },
+ depth = parentStatus and (parentStatus.depth + 1) or 0,
+ searchDeep= parentStatus and parentStatus.searchDeep or deep or -999,
+ interface = parentStatus and parentStatus.interface or {},
+ deep = parentStatus and parentStatus.deep,
+ clock = parentStatus and parentStatus.clock or osClock(),
+ results = {},
+ }
+ if interface then
+ for k, v in pairs(interface) do
+ status.interface[k] = v
+ end
+ end
+ status.deep = status.depth <= status.searchDeep
+ return status
+end
+
+function m.copyStatusResults(a, b)
+ local ra = a.results
+ local rb = b.results
+ for i = 1, #rb do
+ ra[#ra+1] = rb[i]
+ end
+end
+
+function m.isGlobal(source)
+ if source.type == 'setglobal'
+ or source.type == 'getglobal' then
+ if source.node and source.node.tag == '_ENV' then
+ return true
+ end
+ end
+ if source.type == 'field' then
+ source = source.parent
+ end
+ if source.type == 'getfield'
+ or source.type == 'setfield' then
+ local node = source.node
+ if node and node.special == '_G' then
+ return true
+ end
+ end
+ return false
+end
+
+function m.isDoc(source)
+ return source.type:sub(1, 4) == 'doc.'
+end
+
+function m.isDocClass(source)
+ return source.type == 'doc.class'
+end
+
+function m.isSet(source)
+ local tp = source.type
+ if tp == 'setglobal'
+ or tp == 'local'
+ or tp == 'setlocal'
+ or tp == 'setfield'
+ or tp == 'setmethod'
+ or tp == 'setindex'
+ or tp == 'tablefield'
+ or tp == 'tableindex' then
+ return true
+ end
+ if tp == 'call' then
+ local special = m.getSpecial(source.node)
+ if special == 'rawset' then
+ return true
+ end
+ end
+ return false
+end
+
+function m.isGet(source)
+ local tp = source.type
+ if tp == 'getglobal'
+ or tp == 'getlocal'
+ or tp == 'getfield'
+ or tp == 'getmethod'
+ or tp == 'getindex' then
+ return true
+ end
+ if tp == 'call' then
+ local special = m.getSpecial(source.node)
+ if special == 'rawget' then
+ return true
+ end
+ end
+ return false
+end
+
+function m.getSpecial(source)
+ if not source then
+ return nil
+ end
+ return source.special
+end
+
+--- 根据函数的调用参数,获取:调用,参数索引
+function m.getCallAndArgIndex(callarg)
+ local callargs = callarg.parent
+ if not callargs or callargs.type ~= 'callargs' then
+ return nil
+ end
+ local index
+ for i = 1, #callargs do
+ if callargs[i] == callarg then
+ index = i
+ break
+ end
+ end
+ local call = callargs.parent
+ return call, index
+end
+
+--- 根据函数调用的返回值,获取:调用的函数,参数列表,自己是第几个返回值
+function m.getCallValue(source)
+ local value = m.getObjectValue(source) or source
+ if not value then
+ return
+ end
+ local call, index
+ if value.type == 'call' then
+ call = value
+ index = 1
+ elseif value.type == 'select' then
+ call = value.vararg
+ index = value.index
+ if call.type ~= 'call' then
+ return
+ end
+ else
+ return
+ end
+ if call.node.special == 'pcall'
+ or call.node.special == 'xpcall' then
+ return call.args and call.args[1], call.args, index - 1
+ end
+ return call.node, call.args, index
+end
+
+function m.getNextRef(ref)
+ local nextRef = ref.next
+ if nextRef then
+ if nextRef.type == 'setfield'
+ or nextRef.type == 'getfield'
+ or nextRef.type == 'setmethod'
+ or nextRef.type == 'getmethod'
+ or nextRef.type == 'setindex'
+ or nextRef.type == 'getindex' then
+ return nextRef
+ end
+ end
+ -- 穿透 rawget 与 rawset
+ local call, index = m.getCallAndArgIndex(ref)
+ if call then
+ if call.node.special == 'rawset' and index == 1 then
+ return call
+ end
+ if call.node.special == 'rawget' and index == 1 then
+ return call
+ end
+ end
+
+ return nil
+end
+
+function m.checkSameSimpleInValueOfTable(status, value, start, pushQueue)
+ if value.type ~= 'table' then
+ return
+ end
+ for i = 1, #value do
+ local field = value[i]
+ pushQueue(field, start + 1)
+ end
+end
+
+function m.checkStatusDepth(status)
+ if status.depth <= 20 then
+ return true
+ end
+ if m.debugMode then
+ error('status.depth overflow')
+ elseif DEVELOP then
+ --log.warn(debug.traceback('status.depth overflow'))
+ logWarn('status.depth overflow')
+ end
+ return false
+end
+
+function m.searchFields(status, obj, key)
+ if not m.checkStatusDepth(status) then
+ return
+ end
+ local simple = m.getSimple(obj)
+ if not simple then
+ return
+ end
+ simple[#simple+1] = key or m.ANY
+ m.searchSameFields(status, simple, 'field')
+ m.cleanResults(status.results)
+end
+
+function m.searchDefFields(status, obj, key)
+ if not m.checkStatusDepth(status) then
+ return
+ end
+ local simple = m.getSimple(obj)
+ if not simple then
+ return
+ end
+ simple[#simple+1] = key or m.ANY
+ m.searchSameFields(status, simple, 'deffield')
+ m.cleanResults(status.results)
+end
+
+---@param obj parser.guide.object
+---@return parser.guide.object
+function m.getObjectValue(obj)
+ while obj.type == 'paren' do
+ obj = obj.exp
+ if not obj then
+ return nil
+ end
+ end
+ if obj.type == 'boolean'
+ or obj.type == 'number'
+ or obj.type == 'integer'
+ or obj.type == 'string'
+ or obj.type == 'doc.type.table'
+ or obj.type == 'doc.type.arrary' then
+ return obj
+ end
+ if obj.value then
+ return obj.value
+ end
+ if obj.type == 'field'
+ or obj.type == 'method' then
+ return obj.parent and obj.parent.value
+ end
+ if obj.type == 'call' then
+ if obj.node.special == 'rawset' then
+ return obj.args and obj.args[3]
+ else
+ return obj
+ end
+ end
+ if obj.type == 'select' then
+ return obj
+ end
+ return nil
+end
+
+function m.checkSameSimpleInValueInMetaTable(status, mt, start, pushQueue)
+ local cache, makeCache = m.getRefCache(status, mt, '__index')
+ if cache then
+ for _, obj in ipairs(cache) do
+ pushQueue(obj, start, true)
+ end
+ return
+ end
+ cache = {}
+ local newStatus = m.status(status)
+ m.searchDefFields(newStatus, mt, '__index')
+ local refsStatus = m.status(status)
+ for i = 1, #newStatus.results do
+ local indexValue = m.getObjectValue(newStatus.results[i])
+ if indexValue then
+ m.searchRefs(refsStatus, indexValue, 'ref')
+ end
+ end
+ for i = 1, #refsStatus.results do
+ local obj = refsStatus.results[i]
+ pushQueue(obj, start, true)
+ cache[i] = obj
+ end
+ if makeCache then
+ makeCache(cache)
+ end
+end
+function m.checkSameSimpleInValueOfSetMetaTable(status, func, start, pushQueue)
+ if not func or func.special ~= 'setmetatable' then
+ return
+ end
+ local call = func.parent
+ local args = call.args
+ if not args then
+ return
+ end
+ local obj = args[1]
+ local mt = args[2]
+ if obj then
+ pushQueue(obj, start, true)
+ local newStatus = m.status(status)
+ m.searchRefs(newStatus, obj, 'def')
+ for _, def in ipairs(newStatus.results) do
+ pushQueue(def, start, true)
+ end
+ end
+ if mt then
+ if not status.share.markMetaTable then
+ status.share.markMetaTable = {}
+ end
+ if status.share.markMetaTable[mt] then
+ return
+ end
+ status.share.markMetaTable[mt] = true
+ m.checkSameSimpleInValueInMetaTable(status, mt, start, pushQueue)
+ status.share.markMetaTable[mt] = nil
+ end
+end
+
+function m.checkSameSimpleInValueOfCallMetaTable(status, call, start, pushQueue)
+ if status.crossMetaTableMark then
+ return
+ end
+ status.crossMetaTableMark = true
+ if call.type == 'call' then
+ m.checkSameSimpleInValueOfSetMetaTable(status, call.node, start, pushQueue)
+ end
+ status.crossMetaTableMark = false
+end
+
+function m.checkSameSimpleInSpecialBranch(status, obj, start, pushQueue)
+ if status.interface.index then
+ local results = status.interface.index(obj)
+ if not results then
+ return
+ end
+ for _, res in ipairs(results) do
+ pushQueue(obj, start + 1)
+ end
+ end
+end
+
+function m.checkSameSimpleInParamSelf(status, obj, start, pushQueue)
+ if obj.type ~= 'getlocal' or obj[1] ~= 'self' then
+ return
+ end
+ local node = obj.node
+ if node.tag == 'self' then
+ return
+ end
+ if node.parent.type ~= 'funcargs' then
+ return
+ end
+ local func = node.parent.parent
+ if func.type ~= 'function' or func.parent.type ~= 'setfield' then
+ return
+ end
+ local fieldNode = func.parent.node
+ local newStatus = m.status(status)
+ m.searchRefs(newStatus, fieldNode, 'ref')
+ for _, ref in ipairs(newStatus.results) do
+ pushQueue(ref, start, true)
+ end
+end
+
+local function appendValidGenericType(results, status, typeName, obj)
+ if typeName.parent.type == 'doc.type.typeliteral' then
+ if obj.type == 'string' and status.interface.docType then
+ local docs = status.interface.docType(obj[1])
+ for i = 1, #docs do
+ local doc = docs[i]
+ if doc.type == 'doc.class.name'
+ or doc.type == 'doc.alias.name' then
+ results[#results+1] = doc
+ break
+ end
+ end
+ end
+ else
+ -- 发现没有使用 `T`,则沿用既有逻辑直接返回实参
+ results[#results+1] = obj
+ end
+end
+
+local function stepRefOfGenericCrossTable(status, doc, typeName)
+ for _, typeUnit in ipairs(doc.extends.types) do
+ if typeUnit.type == 'doc.type.table' then
+ for _, where in ipairs {'key', 'value'} do
+ local childTypes = typeUnit[where].types
+ for _, childName in ipairs(childTypes) do
+ if childName[1] == typeName[1] then
+ return function (obj)
+ local childStatus = m.status(status)
+ m.searchRefs(childStatus, obj, 'def')
+ for _, res in ipairs(childStatus.results) do
+ if res.type == 'doc.type.table' then
+ return res[where]
+ end
+ if res.type == 'doc.type.array' then
+ if where == 'key' then
+ return status.interface and status.interface.docType('integer')[1]
+ end
+ if where == 'value' then
+ return res.node
+ end
+ end
+ end
+ end
+ end
+ end
+ end
+ return function (obj)
+ return nil
+ end
+ elseif typeUnit.type == 'doc.type.array' then
+ return function (obj)
+ local childStatus = m.status(status)
+ m.searchRefs(childStatus, obj, 'def')
+ for _, res in ipairs(childStatus.results) do
+ if res.type == 'doc.type.array' then
+ return res.node
+ end
+ end
+ end
+ end
+ end
+ return nil
+end
+
+local function getIteratorArg(status, args, index)
+ local call = args.parent
+ local node = call.node
+ if not node.iterator then
+ return nil
+ end
+ if node.type ~= 'call' then
+ return nil
+ end
+ local results = m.checkSameSimpleInCallInSameFile(status, node.node, node.args, index + 1)
+ return results[1]
+end
+
+local function stepRefOfGeneric(status, typeUnit, args, mode)
+ local results = {}
+ if not args then
+ return results
+ end
+ local myName = typeUnit[1]
+ for _, typeName in ipairs(typeUnit.typeGeneric[myName]) do
+ if typeName == typeUnit then
+ goto CONTINUE
+ end
+ local docArg = m.getParentType(typeName, 'doc.type.arg')
+ or m.getParentType(typeName, 'doc.param')
+ or m.getParentType(typeName, 'doc.type.array')
+ if not docArg then
+ goto CONTINUE
+ end
+ local doc = m.getDocState(docArg)
+ if not doc.bindSources then
+ goto CONTINUE
+ end
+ local crossTable = stepRefOfGenericCrossTable(status, docArg, typeName)
+
+ -- find out param index
+ local genericIndex
+ if docArg.type == 'doc.param' then
+ local paramName = docArg.param[1]
+ for _, source in ipairs(doc.bindSources) do
+ if source.type == 'function'
+ and source.args then
+ for i, arg in ipairs(source.args) do
+ if arg[1] == paramName then
+ genericIndex = i
+ break
+ end
+ end
+ end
+ end
+ elseif docArg.type == 'doc.type.arg' then
+ for index, arg in ipairs(docArg.parent.args) do
+ if arg == docArg then
+ genericIndex = index
+ break
+ end
+ end
+ end
+
+ local callArg = args[genericIndex]
+ or getIteratorArg(status, args, genericIndex)
+
+ if not callArg then
+ goto CONTINUE
+ end
+ if crossTable then
+ callArg = crossTable(callArg)
+ if not callArg then
+ goto CONTINUE
+ end
+ end
+ appendValidGenericType(results, status, typeName, callArg)
+ ::CONTINUE::
+ end
+ return results
+end
+
+function m.checkSameSimpleByDocType(status, doc, args)
+ if status.share.searchingBindedDoc then
+ return
+ end
+ if doc.type ~= 'doc.type' then
+ return
+ end
+ local results = {}
+ for _, piece in ipairs(doc.types) do
+ if piece.typeGeneric then
+ local pieceResult = stepRefOfGeneric(status, piece, args, 'def')
+ for _, res in ipairs(pieceResult) do
+ results[#results+1] = res
+ end
+ else
+ local pieceResult = stepRefOfDocType(status, piece, 'def')
+ for _, res in ipairs(pieceResult) do
+ results[#results+1] = res
+ end
+ end
+ end
+ return results
+end
+
+function m.checkSameSimpleByBindDocs(status, obj, start, pushQueue, mode)
+ if not obj.bindDocs then
+ return
+ end
+ if status.share.searchingBindedDoc then
+ return
+ end
+ local skipInfer = false
+ local results = {}
+ for _, doc in ipairs(obj.bindDocs) do
+ if doc.type == 'doc.class' then
+ results[#results+1] = doc
+ elseif doc.type == 'doc.type' then
+ results[#results+1] = doc
+ elseif doc.type == 'doc.param' then
+ -- function (x) 的情况
+ if obj.type == 'local'
+ and m.getKeyName(obj) == doc.param[1] then
+ if obj.parent.type == 'funcargs'
+ or obj.parent.type == 'in'
+ or obj.parent.type == 'loop' then
+ results[#results+1] = doc.extends
+ end
+ end
+ elseif doc.type == 'doc.field' then
+ results[#results+1] = doc
+ elseif doc.type == 'doc.vararg' then
+ if obj.type == '...' then
+ results[#results+1] = doc
+ end
+ end
+ end
+ for _, res in ipairs(results) do
+ if res.type == 'doc.class'
+ or res.type == 'doc.type'
+ or res.type == 'doc.vararg' then
+ pushQueue(res, start, true)
+ skipInfer = true
+ end
+ if res.type == 'doc.type.function' then
+ pushQueue(res, start, true)
+ elseif res.type == 'doc.field' then
+ pushQueue(res, start + 1)
+ end
+ end
+ return skipInfer
+end
+
+function m.checkSameSimpleOfRefByDocSource(status, obj, start, pushQueue, mode)
+ if status.share.searchingBindedDoc then
+ return
+ end
+ if not obj.bindSources then
+ return
+ end
+ status.share.searchingBindedDoc = true
+ local mark = {}
+ local newStatus = m.status(status)
+ for _, ref in ipairs(obj.bindSources) do
+ if not mark[ref] then
+ mark[ref] = true
+ m.searchRefs(newStatus, ref, mode)
+ end
+ end
+ status.share.searchingBindedDoc = nil
+ for _, res in ipairs(newStatus.results) do
+ pushQueue(res, start, true)
+ end
+end
+
+function m.checkSameSimpleOfRefByDocReturn(status, obj, start, pushQueue, mode)
+ if status.share.searchingBindedDoc then
+ return
+ end
+ if not obj.bindSources then
+ return
+ end
+ local index = 0
+ for _, doc in ipairs(obj.bindGroup) do
+ if doc.type == 'doc.return' then
+ index = index + 1
+ if doc == obj then
+ break
+ end
+ end
+ end
+ status.share.searchingBindedDoc = true
+ local mark = {}
+ local newStatus = m.status(status)
+ for _, ref in ipairs(obj.bindSources) do
+ if not mark[ref] then
+ mark[ref] = true
+ m.searchRefs(newStatus, ref, mode)
+ end
+ end
+ status.share.searchingBindedDoc = nil
+ for _, res in ipairs(newStatus.results) do
+ local call = res.parent
+ if call.type == 'call' then
+ if index == 1 then
+ local sel = call.parent
+ if sel.type == 'select' and sel.index == index then
+ pushQueue(sel.parent, start, true)
+ end
+ else
+ if call.extParent then
+ for _, sel in ipairs(call.extParent) do
+ if sel.type == 'select' and sel.index == index then
+ pushQueue(sel.parent, start, true)
+ end
+ end
+ end
+ end
+ end
+ end
+end
+
+local function getArrayOrTableLevel(obj)
+ local level = 0
+ while true do
+ local parent = obj.parent
+ if parent.type == 'doc.type.array' then
+ level = level + 1
+ elseif parent.type == 'doc.type.table' then
+ if obj.type == 'doc.type' then
+ level = level + 1
+ -- else 只存在 obj.type == 'doc.type.name' 的情况,即 table<k,v> 中的 table,这种是不需要再增加层级的
+ end
+ elseif parent.type == 'doc.type' and parent.parent and parent.parent.type == 'doc.type.table' then
+ level = level + 1
+ parent = parent.parent
+ else
+ break
+ end
+ obj = parent
+ end
+ return level
+end
+
+function m.checkSameSimpleByDoc(status, obj, start, pushQueue, mode)
+ if obj.type == 'doc.class.name'
+ or obj.type == 'doc.class' then
+ if obj.type == 'doc.class.name' then
+ obj = m.getDocState(obj)
+ end
+ local classStart
+ for _, doc in ipairs(obj.bindGroup) do
+ if doc == obj then
+ classStart = true
+ elseif doc.type == 'doc.class' then
+ classStart = false
+ end
+ if classStart and doc.type == 'doc.field' then
+ pushQueue(doc, start + 1)
+ end
+ end
+ m.checkSameSimpleOfRefByDocSource(status, obj, start, pushQueue, mode)
+ if mode == 'ref' then
+ local pieceResult = stepRefOfDocType(status, obj.class, 'ref')
+ for _, res in ipairs(pieceResult) do
+ pushQueue(res, start, true)
+ end
+ if obj.extends then
+ for _, ext in ipairs(obj.extends) do
+ local pieceResult = stepRefOfDocType(status, ext, 'def')
+ for _, res in ipairs(pieceResult) do
+ pushQueue(res, start, true)
+ end
+ end
+ end
+ end
+ return true
+ elseif obj.type == 'doc.type' then
+ for _, piece in ipairs(obj.types) do
+ local pieceResult = stepRefOfDocType(status, piece, 'def')
+ for _, res in ipairs(pieceResult) do
+ pushQueue(res, start, true)
+ end
+ end
+ if mode == 'ref' then
+ m.checkSameSimpleOfRefByDocSource(status, obj, start, pushQueue, mode)
+ end
+ return true
+ elseif obj.type == 'doc.type.name'
+ or obj.type == 'doc.see.name' then
+ local pieceResult = stepRefOfDocType(status, obj, 'def')
+ for _, res in ipairs(pieceResult) do
+ pushQueue(res, start, true)
+ end
+
+ if mode == 'ref' then
+ local state = m.getDocState(obj)
+ if state.type == 'doc.type' then
+ m.checkSameSimpleOfRefByDocSource(status, state, start - getArrayOrTableLevel(obj), pushQueue, mode)
+ end
+ if state.type == 'doc.return' then
+ m.checkSameSimpleOfRefByDocReturn(status, state, start - getArrayOrTableLevel(obj), pushQueue, mode)
+ end
+ end
+ return true
+ elseif obj.type == 'doc.field' then
+ if mode ~= 'field'
+ and mode ~= 'deffield' then
+ return m.checkSameSimpleByDoc(status, obj.extends, start, pushQueue, mode)
+ end
+ elseif obj.type == 'doc.type.array' then
+ pushQueue(obj.node, start + 1, true)
+ return true
+ elseif obj.type == 'doc.type.table' then
+ pushQueue(obj.node, start, true)
+ pushQueue(obj.value, start + 1, true)
+ return true
+ elseif obj.type == 'doc.vararg' then
+ pushQueue(obj.vararg, start, true)
+ end
+end
+
+function m.checkSameSimpleInArg1OfSetMetaTable(status, obj, start, pushQueue)
+ local args = obj.parent
+ if not args or args.type ~= 'callargs' then
+ return
+ end
+ local callNode = args.parent.node
+ if callNode.special ~= 'setmetatable' then
+ return
+ end
+ if args[1] ~= obj then
+ return
+ end
+ local mt = args[2]
+ if mt then
+ if m.hasValueMark(status, mt) then
+ return
+ end
+ m.checkSameSimpleInValueInMetaTable(status, mt, start, pushQueue)
+ end
+end
+
+function m.searchSameMethodOutSelf(ref, mark)
+ local selfNode
+ if ref.tag == 'self' then
+ selfNode = ref
+ else
+ if ref.type == 'getlocal'
+ or ref.type == 'setlocal' then
+ local node = ref.node
+ if node.tag == 'self' then
+ selfNode = node
+ end
+ end
+ end
+ if selfNode then
+ if mark[selfNode] then
+ return nil
+ end
+ mark[selfNode] = true
+ local method = selfNode.method.node
+ if mark[method] then
+ return nil
+ end
+ mark[method] = true
+ return method
+ end
+end
+
+function m.searchSameMethodIntoSelf(ref, mark)
+ local nxt = ref.next
+ if not nxt then
+ return nil
+ end
+ if nxt.type ~= 'setmethod' then
+ return nil
+ end
+ if mark[ref] then
+ return nil
+ end
+ mark[ref] = true
+ local value = nxt.value
+ if not value or value.type ~= 'function' then
+ return nil
+ end
+ local selfRef = value.locals and value.locals[1]
+ if not selfRef or selfRef.tag ~= 'self' then
+ return nil
+ end
+ if mark[selfRef] then
+ return nil
+ end
+ mark[selfRef] = true
+ return selfRef
+end
+
+function m.searchSameFieldsCrossMethod(status, ref, start, pushQueue, mode)
+ if status.share.crossMethodMark2 then
+ return
+ end
+ local mark = status.crossMethodMark
+ if not mark then
+ mark = {}
+ status.crossMethodMark = mark
+ end
+ if mark[ref] then
+ return
+ end
+ local selfRef = m.searchSameMethodIntoSelf(ref, mark)
+ if selfRef then
+ tracy.ZoneBeginN 'searchSameFieldsCrossMethod'
+ local _ <close> = tracy.ZoneEnd
+ -- 如果自己是method,则只检查自己内部的self引用
+ status.share.inBeSetValue = (status.share.inBeSetValue or 0) + 1
+ status.share.crossMethodMark2 = true
+ local newStatus = m.status(status)
+ m.searchRefs(newStatus, selfRef, mode)
+ for _, res in ipairs(newStatus.results) do
+ pushQueue(res, start, true)
+ end
+ status.share.inBeSetValue = (status.share.inBeSetValue or 0) - 1
+ status.share.crossMethodMark2 = nil
+ return
+ end
+ local method = m.searchSameMethodOutSelf(ref, mark)
+ if method then
+ pushQueue(method, start, true)
+ return
+ end
+end
+
+local function checkSameSimpleAndMergeFunctionReturnsByDoc(status, results, source, index, args)
+ source = m.getObjectValue(source) or source
+ if not source or source.type ~= 'function' then
+ return
+ end
+ if not source.bindDocs then
+ return
+ end
+ local returns = {}
+ for _, doc in ipairs(source.bindDocs) do
+ if doc.type == 'doc.return' then
+ for _, rtn in ipairs(doc.returns) do
+ returns[#returns+1] = rtn
+ end
+ end
+ end
+ local rtn = returns[index]
+ if not rtn then
+ return
+ end
+ local types = m.checkSameSimpleByDocType(status, rtn, args)
+ if not types then
+ return
+ end
+ for _, res in ipairs(types) do
+ results[#results+1] = res
+ end
+ return true
+end
+
+local function checkSameSimpleAndMergeDocFunctionReturn(status, results, docFunc, index, args)
+ if docFunc.type ~= 'doc.type.function' then
+ return
+ end
+ local rtn = docFunc.returns[index]
+ if rtn then
+ local types = m.checkSameSimpleByDocType(status, rtn, args)
+ if types then
+ for _, res in ipairs(types) do
+ results[#results+1] = res
+ end
+ return true
+ end
+ end
+end
+
+local function checkSameSimpleAndMergeDocTypeFunctionReturns(status, results, source, index)
+ if not source.bindDocs then
+ return
+ end
+ for _, doc in ipairs(source.bindDocs) do
+ if doc.type == 'doc.type' then
+ for _, typeUnit in ipairs(doc.types) do
+ if checkSameSimpleAndMergeDocFunctionReturn(status, results, typeUnit, index) then
+ return true
+ end
+ end
+ end
+ end
+end
+
+function m.checkSameSimpleInCallInSameFile(status, func, args, index)
+ if not status.share.callResultsCache then
+ status.share.callResultsCache = {}
+ end
+ local cache = status.share.callResultsCache[func]
+ if not cache then
+ cache = {}
+ status.share.callResultsCache[func] = cache
+ end
+ local results = cache[index]
+ if results then
+ return results
+ end
+ results = {}
+ if func.special then
+ --return results
+ end
+ local newStatus = m.status(status)
+ m.searchRefs(newStatus, func, 'def')
+ local hasDocReturn
+ for _, def in ipairs(newStatus.results) do
+ hasDocReturn = checkSameSimpleAndMergeDocTypeFunctionReturns(status, results, def, index)
+ or checkSameSimpleAndMergeFunctionReturnsByDoc(status, results, def, index, args)
+ or checkSameSimpleAndMergeDocFunctionReturn(status, results, def, index, args)
+ or hasDocReturn
+ end
+ if not hasDocReturn then
+ for _, def in ipairs(newStatus.results) do
+ local value = m.getObjectValue(def) or def
+ if value.type == 'function' then
+ local returns = value.returns
+ if returns then
+ for _, ret in ipairs(returns) do
+ local exp = ret[index]
+ if exp then
+ results[#results+1] = exp
+ end
+ end
+ end
+ end
+ end
+ cache[index] = results
+ end
+ return results
+end
+
+function m.checkSameSimpleInCall(status, ref, start, pushQueue, mode)
+ if status.share.inBeSetValue and status.share.inBeSetValue > 0 then
+ return
+ end
+ if status.share.inSetValue and status.share.inSetValue > 5 then
+ return
+ end
+ local func, args, index = m.getCallValue(ref)
+ if not func then
+ return
+ end
+ if m.checkCallMark(status, func.parent, true) then
+ return
+ end
+ status.share.inSetValue = (status.share.inSetValue or 0) + 1
+ -- 检查赋值是 semetatable() 的情况
+ m.checkSameSimpleInValueOfSetMetaTable(status, func, start, pushQueue)
+ local objs = m.checkSameSimpleInCallInSameFile(status, func, args, index)
+ if status.interface.call then
+ local cobjs = status.interface.call(func, args, index)
+ if cobjs then
+ for _, obj in ipairs(cobjs) do
+ if not m.checkReturnMark(status, obj) then
+ objs[#objs+1] = obj
+ end
+ end
+ end
+ end
+ m.cleanResults(objs)
+ if not status.share.callFuncMark then
+ status.share.callFuncMark = {}
+ end
+ local mark = {}
+ for _, obj in ipairs(objs) do
+ if mark[obj] then
+ goto CONTINUE
+ end
+ if status.share.callFuncMark[obj] then
+ goto CONTINUE
+ end
+ status.share.callFuncMark[obj] = true
+ local newStatus = m.status(status)
+ m.searchRefs(newStatus, obj, mode)
+ pushQueue(obj, start, true)
+ mark[obj] = true
+ for _, obj in ipairs(newStatus.results) do
+ pushQueue(obj, start, true)
+ mark[obj] = true
+ end
+ status.share.callFuncMark[obj] = nil
+ ::CONTINUE::
+ end
+ status.share.inSetValue = (status.share.inSetValue or 0) - 1
+end
+
+local function searchRawset(ref, results)
+ if m.getKeyName(ref) ~= 'rawset' then
+ return
+ end
+ local call = ref.parent
+ if call.type ~= 'call' or call.node ~= ref then
+ return
+ end
+ if not call.args then
+ return
+ end
+ local arg1 = call.args[1]
+ if arg1.special ~= '_G' then
+ -- 不会吧不会吧,不会真的有人写成 `rawset(_G._G._G, 'xxx', value)` 吧
+ return
+ end
+ results[#results+1] = call
+end
+
+local function searchG(ref, results)
+ while ref and m.getKeyName(ref) == '_G' do
+ results[#results+1] = ref
+ ref = ref.next
+ end
+ if ref then
+ results[#results+1] = ref
+ searchRawset(ref, results)
+ end
+end
+
+local function searchEnvRef(ref, results)
+ if ref.type == 'setglobal'
+ or ref.type == 'getglobal' then
+ results[#results+1] = ref
+ searchG(ref, results)
+ elseif ref.type == 'getlocal' then
+ results[#results+1] = ref.next
+ searchG(ref.next, results)
+ end
+end
+
+function m.findGlobals(ast)
+ local root = m.getRoot(ast)
+ local results = {}
+ local env = m.getENV(root)
+ if env.ref then
+ for _, ref in ipairs(env.ref) do
+ searchEnvRef(ref, results)
+ end
+ end
+ return results
+end
+
+function m.findGlobalsOfName(ast, name)
+ local root = m.getRoot(ast)
+ local results = {}
+ local globals = m.findGlobals(root)
+ for _, global in ipairs(globals) do
+ if m.getKeyName(global) == name then
+ results[#results+1] = global
+ end
+ end
+ return results
+end
+
+function m.checkSameSimpleInGlobal(status, source)
+ local name = m.getKeyName(source)
+ if not name then
+ return
+ end
+ local objs
+ if status.interface.global then
+ objs = status.interface.global(name)
+ else
+ objs = m.findGlobalsOfName(source, name)
+ end
+ return objs
+end
+
+function m.hasValueMark(status, value)
+ if not status.share.valueMark then
+ status.share.valueMark = {}
+ end
+ if status.share.valueMark[value] then
+ return true
+ end
+ status.share.valueMark[value] = true
+ return false
+end
+
+function m.checkCallMark(status, a, mark)
+ if not status.share.callMark then
+ status.share.callMark = {}
+ end
+ if mark then
+ status.share.callMark[a] = mark
+ else
+ return status.share.callMark[a]
+ end
+ return false
+end
+
+function m.checkReturnMark(status, a, mark)
+ if not status.share.returnMark then
+ status.share.returnMark = {}
+ end
+ local result = status.share.returnMark[a]
+ if mark then
+ status.share.returnMark[a] = mark
+ end
+ return result
+end
+
+function m.searchSameFieldsInValue(status, ref, start, pushQueue, mode)
+ if status.share.inBeSetValue and status.share.inBeSetValue > 0 then
+ return
+ end
+ if status.share.inSetValue and status.share.inSetValue > 5 then
+ return
+ end
+ local value = m.getObjectValue(ref)
+ if not value then
+ return
+ end
+ if m.hasValueMark(status, value) then
+ return
+ end
+ status.share.inSetValue = (status.share.inSetValue or 0) + 1
+ if not status.share.tempValueMark then
+ status.share.tempValueMark = {}
+ end
+ if not status.share.tempValueMark[value] then
+ status.share.tempValueMark[value] = true
+ local newStatus = m.status(status)
+ m.searchRefs(newStatus, value, mode)
+ status.share.tempValueMark[value] = nil
+ for _, res in ipairs(newStatus.results) do
+ pushQueue(res, start, true)
+ end
+ pushQueue(value, start, true)
+ end
+ status.share.inSetValue = (status.share.inSetValue or 0) - 1
+ -- 检查形如 a = f() 的分支情况
+ m.checkSameSimpleInCall(status, value, start, pushQueue, mode)
+end
+
+function m.checkSameSimpleAsTableField(status, ref, start, pushQueue)
+ if not status.deep then
+ --return
+ end
+ local parent = ref.parent
+ if not parent or parent.type ~= 'tablefield' then
+ return
+ end
+ if m.hasValueMark(status, ref) then
+ return
+ end
+ local newStatus = m.status(status)
+ m.searchRefs(newStatus, parent.field, 'ref')
+ for _, res in ipairs(newStatus.results) do
+ pushQueue(res, start, true)
+ end
+end
+
+function m.checkSearchLevel(status)
+ status.share.back = status.share.back or 0
+ if status.share.back >= (status.interface.searchLevel or 0) then
+ -- TODO 限制向前搜索的次数
+ --return true
+ end
+ status.share.back = status.share.back + 1
+ return false
+end
+
+function m.checkSameSimpleAsReturn(status, ref, start, pushQueue)
+ if not status.deep then
+ return
+ end
+ if not ref.parent or ref.parent.type ~= 'return' then
+ return
+ end
+ if ref.parent.parent.type ~= 'main' then
+ return
+ end
+ if m.checkSearchLevel(status) then
+ return
+ end
+ local newStatus = m.status(status)
+ m.searchRefsAsFunctionReturn(newStatus, ref, 'ref')
+ for _, res in ipairs(newStatus.results) do
+ if not m.checkCallMark(status, res) then
+ pushQueue(res, start, true)
+ end
+ end
+end
+
+function m.checkSameSimpleAsSetValue(status, ref, start, pushQueue)
+ if not status.deep then
+ --return
+ end
+ if status.share.inSetValue and status.share.inSetValue > 0 then
+ return
+ end
+ if status.share.inBeSetValue and status.share.inBeSetValue > 5 then
+ return
+ end
+ if ref.type == 'select' then
+ return
+ end
+ local parent = ref.parent
+ if not parent then
+ return
+ end
+ if m.getObjectValue(parent) ~= ref then
+ return
+ end
+ if m.hasValueMark(status, ref) then
+ return
+ end
+ if m.checkSearchLevel(status) then
+ return
+ end
+ local obj
+ if parent.type == 'local'
+ or parent.type == 'setglobal'
+ or parent.type == 'setlocal' then
+ obj = parent
+ elseif parent.type == 'setfield' then
+ obj = parent.field
+ elseif parent.type == 'setmethod' then
+ obj = parent.method
+ end
+ if not obj then
+ return
+ end
+ status.share.inBeSetValue = (status.share.inBeSetValue or 0) + 1
+ local newStatus = m.status(status)
+ m.searchRefs(newStatus, obj, 'ref')
+ for _, res in ipairs(newStatus.results) do
+ pushQueue(res, start, true)
+ end
+ status.share.inBeSetValue = (status.share.inBeSetValue or 0) - 1
+end
+
+local function getTableAndIndexIfIsForPairsKeyOrValue(ref)
+ if ref.type ~= 'local' then
+ return
+ end
+
+ if not ref.parent or ref.parent.type ~= 'in' then
+ return
+ end
+
+ if not ref.value or ref.value.type ~= 'select' then
+ return
+ end
+
+ local rootSelectObj = ref.value
+ if rootSelectObj.index ~= 1 and rootSelectObj.index ~= 2 then
+ return
+ end
+
+ if not rootSelectObj.vararg or rootSelectObj.vararg.type ~= 'call' then
+ return
+ end
+ local rootCallObj = rootSelectObj.vararg
+
+ if not rootCallObj.node or rootCallObj.node.type ~= 'call' then
+ return
+ end
+ local pairsCallObj = rootCallObj.node
+
+ if not pairsCallObj.node
+ or (pairsCallObj.node.special ~= 'pairs' and pairsCallObj.node.special ~= 'ipairs') then
+ return
+ end
+
+ if not pairsCallObj.args or not pairsCallObj.args[1] then
+ return
+ end
+ local tableObj = pairsCallObj.args[1]
+
+ return tableObj, rootSelectObj.index
+end
+
+function m.checkSameSimpleAsKeyOrValueInForParis(status, ref, start, pushQueue)
+ local tableObj, index = getTableAndIndexIfIsForPairsKeyOrValue(ref)
+ if not tableObj then
+ return
+ end
+
+ local newStatus = m.status(status)
+ m.searchRefs(newStatus, tableObj, 'def')
+ for _, def in ipairs(newStatus.results) do
+ if def.bindDocs then
+ for _, binddoc in ipairs(def.bindDocs) do
+ if binddoc.type == 'doc.type' then
+ if binddoc.types[1] and binddoc.types[1].type == 'doc.type.table' then
+ if index == 1 then
+ pushQueue(binddoc.types[1].key, start, true)
+ elseif index == 2 then
+ pushQueue(binddoc.types[1].value, start, true)
+ end
+ end
+ end
+ end
+ end
+ end
+end
+
+---
+---@param func parser.guide.object
+---@param argIndex integer
+---@return integer?
+local function findGenericFromArgIndexToReturnIndex(func, argIndex)
+ if not func.bindDocs then
+ return nil
+ end
+ local paramType
+ for _, doc in ipairs(func.bindDocs) do
+ if doc.type == 'doc.param' then
+ if doc.extends and doc.extends.paramIndex == argIndex then
+ paramType = doc.extends
+ break
+ end
+ end
+ end
+ if not paramType then
+ return nil
+ end
+ for _, typeUnit in ipairs(paramType.types) do
+ if typeUnit.typeGeneric then
+ local generic = typeUnit.typeGeneric[typeUnit[1]]
+ if generic then
+ for _, typeName in ipairs(generic) do
+ local docType = typeName.parent
+ if docType.returnIndex then
+ return docType.returnIndex
+ end
+ end
+ end
+ end
+ end
+ return nil
+end
+
+function m.checkSameSimpleAsCallArg(status, ref, start, pushQueue)
+ if not status.deep then
+ return
+ end
+ local call, index = m.getCallAndArgIndex(ref)
+ if not call then
+ return
+ end
+ if call.parent.type ~= 'select' then
+ return
+ end
+ if (status.share.inSetValue or 0) > 0 then
+ return
+ end
+ if status.share.inBeSetValue and status.share.inBeSetValue > 5 then
+ return
+ end
+ status.share.inBeSetValue = (status.share.inBeSetValue or 0) + 1
+ local newStatus = m.status(status)
+ m.searchRefs(newStatus, call.node, 'def')
+ for _, func in ipairs(newStatus.results) do
+ local rindex = findGenericFromArgIndexToReturnIndex(func, index)
+ if rindex then
+ if rindex == 1 then
+ pushQueue(call.parent.parent, start, true)
+ else
+ if call.extParent then
+ for _, slt in ipairs(call.extParent) do
+ if slt.index == rindex then
+ pushQueue(slt.parent, start, true)
+ end
+ end
+ end
+ end
+ end
+ end
+ status.share.inBeSetValue = (status.share.inBeSetValue or 0) - 1
+end
+
+local function hasTypeName(doc, name)
+ if doc.type == 'doc.type' then
+ for _, tunit in ipairs(doc.types) do
+ if tunit.type == 'doc.type.name'
+ and tunit[1] == name then
+ return true
+ end
+ end
+ end
+ if doc.type == 'doc.type.name'
+ or doc.type == 'doc.class.name' then
+ if doc[1] == name then
+ return true
+ end
+ end
+ return false
+end
+
+function m.checkSameSimpleInString(status, ref, start, pushQueue, mode)
+ -- 特殊处理 ('xxx').xxx 的形式
+ if ref.type ~= 'string'
+ and not hasTypeName(ref, 'string') then
+ return
+ end
+ if not status.interface.docType then
+ return
+ end
+ if status.share.searchingBindedDoc then
+ return
+ end
+ if not status.share.markString then
+ status.share.markString = {}
+ end
+ local marked = ref
+ if ref.type == 'string' then
+ marked = 'string'
+ end
+ if status.share.markString[marked] then
+ return
+ end
+ status.share.markString[marked] = true
+ local newStatus = m.status(status)
+ local docs = status.interface.docType('stringlib')
+ local mark = {}
+ for i = 1, #docs do
+ local doc = docs[i]
+ m.searchFields(newStatus, doc)
+ end
+ for _, res in ipairs(newStatus.results) do
+ if mark[res] then
+ goto CONTINUE
+ end
+ mark[res] = true
+ pushQueue(res, start + 1)
+ ::CONTINUE::
+ end
+ status.share.markString[marked] = nil
+ return true
+end
+
+function m.pushResult(status, mode, ref, simple)
+ local results = status.results
+ if mode == 'def' then
+ if ref.type == 'setglobal'
+ or ref.type == 'setlocal'
+ or ref.type == 'local' then
+ results[#results+1] = ref
+ elseif ref.type == 'setfield'
+ or ref.type == 'tablefield' then
+ results[#results+1] = ref
+ elseif ref.type == 'setmethod' then
+ results[#results+1] = ref
+ elseif ref.type == 'setindex'
+ or ref.type == 'tableindex' then
+ results[#results+1] = ref
+ elseif ref.type == 'call' then
+ if ref.node.special == 'rawset' then
+ results[#results+1] = ref
+ end
+ elseif ref.type == 'function' then
+ results[#results+1] = ref
+ elseif ref.type == 'table' then
+ results[#results+1] = ref
+ elseif ref.type == 'doc.type.function'
+ or ref.type == 'doc.class.name'
+ or ref.type == 'doc.alias.name'
+ or ref.type == 'doc.field'
+ or ref.type == 'doc.type.table'
+ or ref.type == 'doc.type.array' then
+ results[#results+1] = ref
+ elseif ref.type == 'doc.type' then
+ if #ref.enums > 0 or #ref.resumes > 0 then
+ results[#results+1] = ref
+ end
+ end
+ if ref.parent and ref.parent.type == 'return' then
+ if m.getParentFunction(ref) ~= m.getParentFunction(simple.node) then
+ results[#results+1] = ref
+ end
+ end
+ if m.isLiteral(ref)
+ and ref.parent and ref.parent.type == 'callargs'
+ and ref ~= simple.node then
+ results[#results+1] = ref
+ end
+ elseif mode == 'ref' then
+ if ref.type == 'setfield'
+ or ref.type == 'getfield'
+ or ref.type == 'tablefield' then
+ results[#results+1] = ref
+ elseif ref.type == 'setmethod'
+ or ref.type == 'getmethod' then
+ results[#results+1] = ref
+ elseif ref.type == 'setindex'
+ or ref.type == 'getindex'
+ or ref.type == 'tableindex' then
+ results[#results+1] = ref
+ elseif ref.type == 'setglobal'
+ or ref.type == 'getglobal'
+ or ref.type == 'local'
+ or ref.type == 'setlocal'
+ or ref.type == 'getlocal' then
+ results[#results+1] = ref
+ elseif ref.type == 'function' then
+ results[#results+1] = ref
+ elseif ref.type == 'table' then
+ results[#results+1] = ref
+ elseif ref.type == 'call' then
+ if ref.node.special == 'rawset'
+ or ref.node.special == 'rawget' then
+ results[#results+1] = ref
+ end
+ elseif ref.type == 'doc.type.function'
+ or ref.type == 'doc.class.name'
+ or ref.type == 'doc.alias.name'
+ or ref.type == 'doc.field'
+ or ref.type == 'doc.type.table'
+ or ref.type == 'doc.type.array' then
+ results[#results+1] = ref
+ elseif ref.type == 'doc.type' then
+ if #ref.enums > 0 or #ref.resumes > 0 then
+ results[#results+1] = ref
+ end
+ end
+ if ref.parent and ref.parent.type == 'return' then
+ results[#results+1] = ref
+ end
+ if m.isLiteral(ref)
+ and ref.parent
+ and ref.parent.type == 'callargs'
+ and ref ~= simple.node then
+ results[#results+1] = ref
+ end
+ elseif mode == 'field' then
+ if ref.type == 'setfield'
+ or ref.type == 'getfield'
+ or ref.type == 'tablefield' then
+ results[#results+1] = ref
+ elseif ref.type == 'setmethod'
+ or ref.type == 'getmethod' then
+ results[#results+1] = ref
+ elseif ref.type == 'setindex'
+ or ref.type == 'tableindex' then
+ results[#results+1] = ref
+ elseif ref.type == 'getindex' then
+ -- do not trust `t[1]`
+ if ref.index and ref.index.type == 'string' then
+ results[#results+1] = ref
+ end
+ elseif ref.type == 'setglobal'
+ or ref.type == 'getglobal' then
+ results[#results+1] = ref
+ elseif ref.type == 'call' then
+ if ref.node.special == 'rawset'
+ or ref.node.special == 'rawget' then
+ results[#results+1] = ref
+ end
+ elseif ref.type == 'doc.field' then
+ results[#results+1] = ref
+ end
+ elseif mode == 'deffield' then
+ if ref.type == 'setfield'
+ or ref.type == 'tablefield' then
+ results[#results+1] = ref
+ elseif ref.type == 'setmethod' then
+ results[#results+1] = ref
+ elseif ref.type == 'setindex'
+ or ref.type == 'tableindex' then
+ results[#results+1] = ref
+ elseif ref.type == 'setglobal' then
+ results[#results+1] = ref
+ elseif ref.type == 'call' then
+ if ref.node.special == 'rawset' then
+ results[#results+1] = ref
+ end
+ elseif ref.type == 'doc.field' then
+ results[#results+1] = ref
+ end
+ end
+end
+
+function m.checkSameSimpleName(ref, sm)
+ if sm == m.ANY then
+ return true
+ end
+
+ if m.getSimpleName(ref) == sm then
+ return true
+ end
+ if ref.type == 'doc.type'
+ and ref.array == true then
+ return true
+ end
+ return false
+end
+
+function m.isValidSetRef(ref)
+ if m.isSet(ref) then
+ return true
+ end
+end
+
+function m.checkSameSimple(status, simple, ref, start, force, mode, pushQueue)
+ if start > #simple then
+ return
+ end
+ for i = start, #simple do
+ local sm = simple[i]
+ if not force and not m.checkSameSimpleName(ref, sm) then
+ return
+ end
+ force = false
+ local cmode = mode
+ local skipSearch
+ if i < #simple then
+ cmode = 'ref'
+ else
+ if mode == 'deffield' then
+ if not m.isSet(ref) then
+ skipSearch = true
+ end
+ end
+ end
+ -- 检查 doc
+ local skipInfer = m.checkSameSimpleByBindDocs(status, ref, i, pushQueue, cmode)
+ or m.checkSameSimpleByDoc(status, ref, i, pushQueue, cmode)
+ -- 检查自己是字符串的分支情况
+ m.checkSameSimpleInString(status, ref, i, pushQueue, cmode)
+ if not skipInfer and not skipSearch then
+ -- 穿透 self:func 与 mt:func
+ m.searchSameFieldsCrossMethod(status, ref, i, pushQueue, cmode)
+ -- 穿透赋值
+ m.searchSameFieldsInValue(status, ref, i, pushQueue, cmode)
+ -- 检查自己是字面量表的情况
+ m.checkSameSimpleInValueOfTable(status, ref, i, pushQueue)
+ -- 检查自己作为 setmetatable 第一个参数的情况
+ m.checkSameSimpleInArg1OfSetMetaTable(status, ref, i, pushQueue)
+ -- 检查自己作为 setmetatable 调用的情况
+ m.checkSameSimpleInValueOfCallMetaTable(status, ref, i, pushQueue)
+ -- 检查自己是特殊变量的分支的情况
+ m.checkSameSimpleInSpecialBranch(status, ref, i, pushQueue)
+ -- self 的特殊处理
+ m.checkSameSimpleInParamSelf(status, ref, i, pushQueue)
+ -- 自己是 call 的情况
+ m.checkSameSimpleInCall(status, ref, i, pushQueue, cmode)
+ if cmode == 'ref' then
+ -- 检查形如 { a = f } 的情况
+ m.checkSameSimpleAsTableField(status, ref, i, pushQueue)
+ -- 检查形如 return m 的情况
+ m.checkSameSimpleAsReturn(status, ref, i, pushQueue)
+ -- 检查形如 a = f 的情况
+ m.checkSameSimpleAsSetValue(status, ref, i, pushQueue)
+ -- 检查形如 for k,v in pairs()/ipairs() do end 的情况
+ m.checkSameSimpleAsKeyOrValueInForParis(status, ref, i, pushQueue)
+ -- 检查自己是函数参数的情况(泛型) local x = call(V)
+ m.checkSameSimpleAsCallArg(status, ref, i, pushQueue)
+ end
+ end
+ if i == #simple then
+ break
+ end
+ ref = m.getNextRef(ref)
+ if not ref then
+ return
+ end
+ end
+ m.pushResult(status, mode, ref, simple)
+ local value = m.getObjectValue(ref)
+ if value then
+ m.pushResult(status, mode, value, simple)
+ end
+end
+
+local queuesPool = {}
+local startsPool = {}
+local forcesPool = {}
+local poolSize = 0
+
+local function allocQueue()
+ if poolSize <= 0 then
+ return {}, {}, {}
+ else
+ local queues = queuesPool[poolSize]
+ local starts = startsPool[poolSize]
+ local forces = forcesPool[poolSize]
+ poolSize = poolSize - 1
+ return queues, starts, forces
+ end
+end
+
+local function deallocQueue(queues, starts, forces)
+ poolSize = poolSize + 1
+ queuesPool[poolSize] = queues
+ startsPool[poolSize] = starts
+ forcesPool[poolSize] = forces
+end
+
+function m.searchSameFields(status, simple, mode)
+ local queues, starts, forces = allocQueue()
+ local queueLen = 0
+ local locks = {}
+ local function appendQueue(obj, start, force)
+ local lock = locks[start]
+ if not lock then
+ lock = {}
+ locks[start] = lock
+ end
+ if lock[obj] then
+ return
+ end
+ lock[obj] = true
+ queueLen = queueLen + 1
+ queues[queueLen] = obj
+ starts[queueLen] = start
+ forces[queueLen] = force
+ if obj.mirror then
+ if not lock[obj.mirror] then
+ lock[obj.mirror] = true
+ queueLen = queueLen + 1
+ queues[queueLen] = obj.mirror
+ starts[queueLen] = start
+ forces[queueLen] = force
+ end
+ end
+ end
+ local function pushQueue(obj, start, force)
+ if obj.type == 'getlocal'
+ or obj.type == 'setlocal' then
+ obj = obj.node
+ end
+ appendQueue(obj, start, force)
+ if obj.type == 'local' and obj.ref then
+ for _, ref in ipairs(obj.ref) do
+ appendQueue(ref, start, force)
+ end
+ end
+ if m.isGlobal(obj) then
+ local refs = m.checkSameSimpleInGlobal(status, obj)
+ if refs then
+ for _, ref in ipairs(refs) do
+ appendQueue(ref, start, force)
+ end
+ end
+ end
+ end
+ if simple.mode == 'global' then
+ -- 全局变量开头
+ pushQueue(simple.node, 1)
+ elseif simple.mode == 'local' then
+ -- 局部变量开头
+ pushQueue(simple.node, 1)
+ else
+ pushQueue(simple.node, 1)
+ end
+ local max = 0
+ for i = 1, 1e6 do
+ if queueLen <= 0 then
+ break
+ end
+ local obj = queues[queueLen]
+ local start = starts[queueLen]
+ local force = forces[queueLen]
+ queues[queueLen] = nil
+ starts[queueLen] = nil
+ forces[queueLen] = nil
+ queueLen = queueLen - 1
+ max = max + 1
+ status.share.count = status.share.count + 1
+ if status.share.count % 10000 == 0 then
+ --if TEST then
+ -- print('####', status.share.count, osClock() - status.clock)
+ --end
+ if status.interface and status.interface.pulse then
+ status.interface.pulse()
+ end
+ end
+ --if status.share.count >= 100000 then
+ -- logWarn('Count too large!')
+ -- break
+ --end
+ m.checkSameSimple(status, simple, obj, start, force, mode, pushQueue)
+ if max >= 10000 then
+ logWarn('Queue too large!')
+ break
+ end
+ end
+ --deallocQueue(queues, starts, forces)
+end
+
+function m.getCallerInSameFile(status, func)
+ -- 搜索所有所在函数的调用者
+ local funcRefs = m.status(status)
+ m.searchRefOfValue(funcRefs, func)
+
+ local calls = {}
+ if #funcRefs.results == 0 then
+ return calls
+ end
+ for _, res in ipairs(funcRefs.results) do
+ local call = res.parent
+ if call.type == 'call' then
+ calls[#calls+1] = call
+ end
+ end
+ return calls
+end
+
+function m.getCallerCrossFiles(status, main)
+ if status.interface.link then
+ return status.interface.link(main.uri)
+ end
+ return {}
+end
+
+function m.searchRefsAsFunctionReturn(status, obj, mode)
+ if not status.deep then
+ return
+ end
+ if mode == 'def' then
+ return
+ end
+ if m.checkReturnMark(status, obj, true) then
+ return
+ end
+ status.results[#status.results+1] = obj
+ -- 搜索所在函数
+ local currentFunc = m.getParentFunction(obj)
+ local rtn = obj.parent
+ if rtn.type ~= 'return' then
+ return
+ end
+ -- 看看他是第几个返回值
+ local index
+ for i = 1, #rtn do
+ if obj == rtn[i] then
+ index = i
+ break
+ end
+ end
+ if not index then
+ return
+ end
+ local calls
+ if currentFunc.type == 'main' then
+ calls = m.getCallerCrossFiles(status, currentFunc)
+ else
+ calls = m.getCallerInSameFile(status, currentFunc)
+ end
+ -- 搜索调用者的返回值
+ if #calls == 0 then
+ return
+ end
+ local selects = {}
+ for i = 1, #calls do
+ local parent = calls[i].parent
+ if parent.type == 'select' and parent.index == index then
+ selects[#selects+1] = parent.parent
+ end
+ local extParent = calls[i].extParent
+ if extParent then
+ for j = 1, #extParent do
+ local ext = extParent[j]
+ if ext.type == 'select' and ext.index == index then
+ selects[#selects+1] = ext.parent
+ end
+ end
+ end
+ end
+ -- 搜索调用者的引用
+ for i = 1, #selects do
+ m.searchRefs(status, selects[i], 'ref')
+ end
+end
+
+function m.searchRefsAsFunctionSet(status, obj, mode)
+ local parent = obj.parent
+ if not parent then
+ return
+ end
+ if parent.type == 'local'
+ or parent.type == 'setlocal'
+ or parent.type == 'setglobal'
+ or parent.type == 'setfield'
+ or parent.type == 'setmethod'
+ or parent.type == 'tablefield' then
+ m.searchRefs(status, parent, mode)
+ elseif parent.type == 'setindex'
+ or parent.type == 'tableindex' then
+ if parent.index == obj then
+ m.searchRefs(status, parent, mode)
+ end
+ end
+end
+
+function m.searchRefsAsFunction(status, obj, mode)
+ if obj.type ~= 'function'
+ and obj.type ~= 'table' then
+ return
+ end
+ m.searchRefsAsFunctionSet(status, obj, mode)
+ -- 检查自己作为返回函数时的引用
+ m.searchRefsAsFunctionReturn(status, obj, mode)
+end
+
+function m.cleanResults(results)
+ local mark = {}
+ for i = #results, 1, -1 do
+ local res = results[i]
+ if res.tag == 'self'
+ or mark[res] then
+ results[i] = results[#results]
+ results[#results] = nil
+ else
+ mark[res] = true
+ end
+ end
+end
+
+--function m.getRefCache(status, obj, mode)
+-- local cache = status.interface.cache and status.interface.cache()
+-- if not cache then
+-- return
+-- end
+-- if m.isGlobal(obj) then
+-- obj = m.getKeyName(obj)
+-- end
+-- if not cache[mode] then
+-- cache[mode] = {}
+-- end
+-- local sourceCache = cache[mode][obj]
+-- if sourceCache then
+-- return sourceCache
+-- end
+-- sourceCache = {}
+-- cache[mode][obj] = sourceCache
+-- return nil, function (results)
+-- for i = 1, #results do
+-- sourceCache[i] = results[i]
+-- end
+-- end
+--end
+
+function m.getRefCache(status, obj, mode)
+ local isDeep = status.deep
+ if mode == 'infer' then
+ if not isDeep then
+ return nil, nil
+ end
+ end
+ local globalCache = status.interface.cache and status.interface.cache() or {}
+ if m.isGlobal(obj) then
+ obj = m.getKeyName(obj)
+ end
+ if not obj then
+ return {}
+ end
+ if not globalCache[mode] then
+ globalCache[mode] = {}
+ end
+ local sourceCache = globalCache[mode][obj]
+ if sourceCache then
+ return sourceCache
+ end
+ if not status.share.cacheLock[mode] then
+ status.share.cacheLock[mode] = {}
+ end
+ if status.share.cacheLock[mode][obj] then
+ return {}
+ end
+ status.share.cacheLock[mode][obj] = {}
+ return nil, function (results)
+ sourceCache = {}
+ for i = 1, #results do
+ sourceCache[i] = results[i]
+ end
+ globalCache[mode][obj] = sourceCache
+ if not isDeep then
+ return
+ end
+ if mode == 'ref'
+ or mode == 'def' then
+ for i = 1, #results do
+ local res = results[i]
+ if not globalCache[mode][res] then
+ globalCache[mode][res] = sourceCache
+ end
+ end
+ end
+ end
+end
+
+function m.searchRefs(status, obj, mode)
+ if not obj then
+ return
+ end
+ local cache, makeCache = m.getRefCache(status, obj, mode)
+ if cache then
+ for i = 1, #cache do
+ status.results[#status.results+1] = cache[i]
+ end
+ return
+ end
+
+ -- 检查单步引用
+ tracy.ZoneBeginN('searchRefs getStepRef')
+ local res = m.getStepRef(status, obj, mode)
+ if res then
+ for i = 1, #res do
+ status.results[#status.results+1] = res[i]
+ end
+ end
+ tracy.ZoneEnd()
+ -- 检查simple
+ tracy.ZoneBeginN('searchRefs searchSameFields')
+ if m.checkStatusDepth(status) then
+ local simple = m.getSimple(obj)
+ if simple then
+ m.searchSameFields(status, simple, mode)
+ end
+ end
+ tracy.ZoneEnd()
+
+ m.cleanResults(status.results)
+
+ if makeCache then
+ makeCache(status.results)
+ end
+end
+
+function m.searchRefOfValue(status, obj)
+ local var = obj.parent
+ if var.type == 'local'
+ or var.type == 'set' then
+ return m.searchRefs(status, var, 'ref')
+ end
+end
+
+function m.allocInfer(o)
+ if type(o.type) == 'table' then
+ local infers = {}
+ for i = 1, #o.type do
+ infers[i] = {
+ type = o.type[i],
+ value = o.value,
+ source = o.source,
+ level = o.level
+ }
+ end
+ return infers
+ else
+ return {
+ [1] = o,
+ }
+ end
+end
+
+function m.mergeTypes(types)
+ local hasAny = types['any']
+
+ types['any'] = nil
+
+ if not next(types) then
+ return 'any'
+ end
+ -- 同时包含 number 与 integer 时,去掉 integer
+ if types['number'] and types['integer'] then
+ types['integer'] = nil
+ end
+
+ local results = {}
+ for tp in pairs(types) do
+ results[#results+1] = tp
+ end
+ -- 只有显性的 nil 与 any 时,取 any
+ if #results == 1 then
+ if results[1] == 'nil' and hasAny then
+ return 'any'
+ else
+ return results[1]
+ end
+ end
+
+ tableSort(results, function (a, b)
+ local sa = TypeSort[a] or 100
+ local sb = TypeSort[b] or 100
+ if sa == sb then
+ return a < b
+ else
+ return sa < sb
+ end
+ end)
+
+ return tableConcat(results, '|')
+end
+
+function m.getClassExtends(class)
+ if class.type == 'doc.class.name' then
+ class = class.parent
+ end
+ if not class.extends then
+ return nil
+ end
+ local names = {}
+ for _, ext in ipairs(class.extends) do
+ names[#names+1] = ext[1]
+ end
+ return names
+end
+
+function m.viewInferType(infers)
+ if not infers then
+ return 'any'
+ end
+ local types = {}
+ local hasDoc
+ local hasDocTable
+ for i = 1, #infers do
+ local infer = infers[i]
+ local src = infer.source
+ if src.type == 'doc.class'
+ or src.type == 'doc.class.name'
+ or src.type == 'doc.type.name'
+ or src.type == 'doc.type.array'
+ or src.type == 'doc.type.table' then
+ if infer.type ~= 'any' then
+ hasDoc = true
+ end
+ if src.type == 'doc.type.array'
+ or src.type == 'doc.type.table' then
+ hasDocTable = true
+ end
+ end
+ end
+ if hasDoc then
+ for i = 1, #infers do
+ local infer = infers[i]
+ local src = infer.source
+ if src.type == 'doc.class'
+ or src.type == 'doc.class.name'
+ or src.type == 'doc.type.name'
+ or src.type == 'doc.type.array'
+ or src.type == 'doc.type.table'
+ or src.type == 'doc.type.enum'
+ or src.type == 'doc.resume' then
+ local tp = infer.type or 'any'
+ if hasDocTable and tp == 'table' then
+ goto CONTINUE
+ end
+ if types[tp] == nil then
+ types[tp] = true
+ end
+ end
+ if src.type == 'doc.class'
+ or src.type == 'doc.class.name' then
+ local extends = m.getClassExtends(src)
+ if extends then
+ for _, tp in ipairs(extends) do
+ types[tp] = false
+ end
+ end
+ end
+ ::CONTINUE::
+ end
+ for k, v in pairs(types) do
+ if not v then
+ types[k] = nil
+ end
+ end
+ else
+ for i = 1, #infers do
+ local infer = infers[i]
+ if infer.source.typeGeneric then
+ goto CONTINUE
+ end
+ local tp = infer.type or 'any'
+ types[tp] = true
+ ::CONTINUE::
+ end
+ end
+ return m.mergeTypes(types)
+end
+
+function m.checkTrue(status, source)
+ local newStatus = m.status(status)
+ m.searchInfer(newStatus, source)
+ -- 当前认为的结果
+ local current
+ for _, infer in ipairs(newStatus.results) do
+ -- 新的结果
+ local new
+ if infer.type == 'nil' then
+ new = false
+ elseif infer.type == 'boolean' then
+ if infer.value == true then
+ new = true
+ elseif infer.value == false then
+ new = false
+ end
+ end
+ if new ~= nil then
+ if current == nil then
+ current = new
+ else
+ -- 如果2个结果完全相反,则返回 nil 表示不确定
+ if new ~= current then
+ return nil
+ end
+ end
+ end
+ end
+ return current
+end
+
+--- 获取特定类型的字面量值
+function m.getInferLiteral(status, source, type)
+ local newStatus = m.status(status)
+ m.searchInfer(newStatus, source)
+ for _, infer in ipairs(newStatus.results) do
+ if infer.value ~= nil then
+ if type == nil or infer.type == type then
+ return infer.value
+ end
+ end
+ end
+ return nil
+end
+
+--- 是否包含某种类型
+function m.hasType(status, source, type)
+ m.searchInfer(status, source)
+ for _, infer in ipairs(status.results) do
+ if infer.type == type then
+ return true
+ end
+ end
+ return false
+end
+
+function m.isSameValue(status, a, b)
+ local statusA = m.status(status)
+ m.searchInfer(statusA, a)
+ local statusB = m.status(status)
+ m.searchInfer(statusB, b)
+ local infers = {}
+ for _, infer in ipairs(statusA.results) do
+ local literal = infer.value
+ if literal then
+ infers[literal] = false
+ end
+ end
+ for _, infer in ipairs(statusB.results) do
+ local literal = infer.value
+ if literal then
+ if infers[literal] == nil then
+ return false
+ end
+ infers[literal] = true
+ end
+ end
+ for k, v in pairs(infers) do
+ if v == false then
+ return false
+ end
+ end
+ return true
+end
+
+function m.inferCheckLiteralTableWithDocVararg(status, source)
+ if #source ~= 1 then
+ return
+ end
+ local vararg = source[1]
+ if vararg.type ~= 'varargs' then
+ return
+ end
+ local results = m.getVarargDocType(status, source)
+ status.results[#status.results+1] = {
+ type = m.viewInferType(results) .. '[]',
+ source = source,
+ level = 100,
+ }
+ return true
+end
+
+function m.inferCheckLiteral(status, source)
+ if source.type == 'string' then
+ status.results = m.allocInfer {
+ type = 'string',
+ value = source[1],
+ source = source,
+ level = 100,
+ }
+ return true
+ elseif source.type == 'nil' then
+ status.results = m.allocInfer {
+ type = 'nil',
+ value = NIL,
+ source = source,
+ level = 100,
+ }
+ return true
+ elseif source.type == 'boolean' then
+ status.results = m.allocInfer {
+ type = 'boolean',
+ value = source[1],
+ source = source,
+ level = 100,
+ }
+ return true
+ elseif source.type == 'number' then
+ if mathType(source[1]) == 'integer' then
+ status.results = m.allocInfer {
+ type = 'integer',
+ value = source[1],
+ source = source,
+ level = 100,
+ }
+ return true
+ else
+ status.results = m.allocInfer {
+ type = 'number',
+ value = source[1],
+ source = source,
+ level = 100,
+ }
+ return true
+ end
+ elseif source.type == 'integer' then
+ status.results = m.allocInfer {
+ type = 'integer',
+ source = source,
+ level = 100,
+ }
+ return true
+ elseif source.type == 'table' then
+ if m.inferCheckLiteralTableWithDocVararg(status, source) then
+ return true
+ end
+ status.results = m.allocInfer {
+ type = 'table',
+ source = source,
+ level = 100,
+ }
+ return true
+ elseif source.type == 'function' then
+ status.results = m.allocInfer {
+ type = 'function',
+ source = source,
+ level = 100,
+ }
+ return true
+ elseif source.type == '...' then
+ status.results = m.allocInfer {
+ type = '...',
+ source = source,
+ level = 100,
+ }
+ return true
+ end
+end
+
+local function getDocAliasExtends(status, typeUnit)
+ if not status.interface.docType then
+ return nil
+ end
+ if typeUnit.type ~= 'doc.type.name' then
+ return nil
+ end
+ for _, doc in ipairs(status.interface.docType(typeUnit[1])) do
+ if doc.type == 'doc.alias.name' then
+ return doc.parent.extends
+ end
+ end
+ return nil
+end
+
+function m.getDocTypeUnitName(status, unit)
+ local typeName
+ if unit.type == 'doc.type.name' then
+ typeName = unit[1]
+ elseif unit.type == 'doc.type.function' then
+ typeName = 'function'
+ elseif unit.type == 'doc.type.array' then
+ typeName = m.getDocTypeUnitName(status, unit.node) .. '[]'
+ elseif unit.type == 'doc.type.table' then
+ typeName = ('%s<%s, %s>'):format(
+ m.getDocTypeUnitName(status, unit.node),
+ m.viewInferType(m.getDocTypeNames(status, unit.key)),
+ m.viewInferType(m.getDocTypeNames(status, unit.value))
+ )
+ end
+ if unit.typeGeneric then
+ typeName = ('<%s>'):format(typeName)
+ end
+ return typeName
+end
+
+function m.getDocTypeNames(status, doc)
+ local results = {}
+ if not doc then
+ return results
+ end
+ for _, unit in ipairs(doc.types) do
+ local alias = getDocAliasExtends(status, unit)
+ if alias then
+ local aliasResults = m.getDocTypeNames(status, alias)
+ for _, res in ipairs(aliasResults) do
+ results[#results+1] = res
+ end
+ else
+ local typeName = m.getDocTypeUnitName(status, unit)
+ results[#results+1] = {
+ type = typeName,
+ source = unit,
+ level = 100,
+ }
+ end
+ end
+ for _, enum in ipairs(doc.enums) do
+ results[#results+1] = {
+ type = enum[1],
+ source = enum,
+ level = 100,
+ }
+ end
+ for _, resume in ipairs(doc.resumes) do
+ if not resume.additional then
+ results[#results+1] = {
+ type = resume[1],
+ source = resume,
+ level = 100,
+ }
+ end
+ end
+ return results
+end
+
+function m.inferCheckDoc(status, source)
+ if source.type == 'doc.class.name' then
+ status.results[#status.results+1] = {
+ type = source[1],
+ source = source,
+ level = 100,
+ }
+ return true
+ end
+ if source.type == 'doc.class' then
+ status.results[#status.results+1] = {
+ type = source.class[1],
+ source = source,
+ level = 100,
+ }
+ return true
+ end
+ if source.type == 'doc.type' then
+ local results = m.getDocTypeNames(status, source)
+ for _, res in ipairs(results) do
+ status.results[#status.results+1] = res
+ end
+ return true
+ end
+ if source.type == 'doc.type.function'
+ or source.type == 'doc.type.table'
+ or source.type == 'doc.type.array' then
+ local typeName = m.getDocTypeUnitName(status, source)
+ status.results[#status.results+1] = {
+ type = typeName,
+ source = source,
+ level = 100,
+ }
+ return true
+ end
+ if source.type == 'doc.field' then
+ local results = m.getDocTypeNames(status, source.extends)
+ for _, res in ipairs(results) do
+ status.results[#status.results+1] = res
+ end
+ return true
+ end
+ if source.type == 'doc.alias.name' then
+ local results = m.getDocTypeNames(status, m.getDocState(source).extends)
+ for _, res in ipairs(results) do
+ status.results[#status.results+1] = res
+ end
+ return true
+ end
+end
+
+function m.getVarargDocType(status, source)
+ local func = m.getParentFunction(source)
+ if not func then
+ return
+ end
+ if not func.args then
+ return
+ end
+ for _, arg in ipairs(func.args) do
+ if arg.type == '...' then
+ if arg.bindDocs then
+ for _, doc in ipairs(arg.bindDocs) do
+ if doc.type == 'doc.vararg' then
+ return m.getDocTypeNames(status, doc.vararg)
+ end
+ end
+ end
+ end
+ end
+end
+
+function m.inferCheckUpDocOfVararg(status, source)
+ if not source.vararg then
+ return
+ end
+ local results = m.getVarargDocType(status, source)
+ if not results then
+ return
+ end
+ for _, res in ipairs(results) do
+ status.results[#status.results+1] = res
+ end
+ return true
+end
+
+function m.inferCheckUpDoc(status, source)
+ if m.inferCheckUpDocOfVararg(status, source) then
+ return true
+ end
+ local parent = source.parent
+ if parent then
+ if parent.type == 'local'
+ or parent.type == 'setlocal'
+ or parent.type == 'setglobal' then
+ source = parent
+ end
+ if parent.type == 'setfield'
+ or parent.type == 'tablefield' then
+ if parent.field == source
+ or parent.value == source then
+ source = parent
+ end
+ end
+ if parent.type == 'setmethod' then
+ if parent.method == source
+ or parent.value == source then
+ source = parent
+ end
+ end
+ if parent.type == 'setindex'
+ or parent.type == 'tableindex' then
+ if parent.index == source
+ or parent.value == source then
+ source = parent
+ end
+ end
+ end
+ local binds = source.bindDocs
+ if not binds then
+ return
+ end
+ status.results = {}
+ for i = #binds, 1, -1 do
+ local doc = binds[i]
+ if doc.type == 'doc.class' then
+ status.results[#status.results+1] = {
+ type = doc.class[1],
+ source = doc,
+ level = 100,
+ }
+ -- ---@class Class
+ -- local x = { field = 1 }
+ -- 这种情况下,将字面量表接受为Class的定义
+ if source.value and source.value.type == 'table' then
+ status.results[#status.results+1] = {
+ type = source.value.type,
+ source = source.value,
+ level = 100,
+ }
+ end
+ return true
+ elseif doc.type == 'doc.type' then
+ local results = m.getDocTypeNames(status, doc)
+ for _, res in ipairs(results) do
+ status.results[#status.results+1] = res
+ end
+ return true
+ elseif doc.type == 'doc.param' then
+ -- function (x) 的情况
+ if source.type == 'local'
+ and m.getKeyName(source) == doc.param[1] then
+ if source.parent.type == 'funcargs'
+ or source.parent.type == 'in'
+ or source.parent.type == 'loop' then
+ local results = m.getDocTypeNames(status, doc.extends)
+ for _, res in ipairs(results) do
+ status.results[#status.results+1] = res
+ end
+ return true
+ end
+ end
+ end
+ end
+end
+
+function m.inferCheckUnary(status, source)
+ if source.type ~= 'unary' then
+ return
+ end
+ local op = source.op
+ if op.type == 'not' then
+ local checkTrue = m.checkTrue(status, source[1])
+ local value = nil
+ if checkTrue == true then
+ value = false
+ elseif checkTrue == false then
+ value = true
+ end
+ status.results = m.allocInfer {
+ type = 'boolean',
+ value = value,
+ source = source,
+ level = 100,
+ }
+ return true
+ elseif op.type == '#' then
+ status.results = m.allocInfer {
+ type = 'integer',
+ source = source,
+ level = 100,
+ }
+ return true
+ elseif op.type == '~' then
+ local l = m.getInferLiteral(status, source[1], 'integer')
+ status.results = m.allocInfer {
+ type = 'integer',
+ value = l and ~l or nil,
+ source = source,
+ level = 100,
+ }
+ return true
+ elseif op.type == '-' then
+ local v = m.getInferLiteral(status, source[1], 'integer')
+ if v then
+ status.results = m.allocInfer {
+ type = 'integer',
+ value = - v,
+ source = source,
+ level = 100,
+ }
+ return true
+ end
+ v = m.getInferLiteral(status, source[1], 'number')
+ status.results = m.allocInfer {
+ type = 'number',
+ value = v and -v or nil,
+ source = source,
+ level = 100,
+ }
+ return true
+ end
+end
+
+local function mathCheck(status, a, b)
+ local v1 = m.getInferLiteral(status, a, 'integer')
+ or m.getInferLiteral(status, a, 'number')
+ local v2 = m.getInferLiteral(status, b, 'integer')
+ or m.getInferLiteral(status, a, 'number')
+ v1 = tonumber(v1)
+ v2 = tonumber(v2)
+ local int = m.hasType(status, a, 'integer')
+ and m.hasType(status, b, 'integer')
+ and not m.hasType(status, a, 'number')
+ and not m.hasType(status, b, 'number')
+ return int and 'integer' or 'number', v1, v2
+end
+
+function m.inferCheckBinary(status, source)
+ if source.type ~= 'binary' then
+ return
+ end
+ local op = source.op
+ if op.type == 'and' then
+ local isTrue = m.checkTrue(status, source[1])
+ if isTrue == true then
+ m.searchInfer(status, source[2])
+ return true
+ elseif isTrue == false then
+ m.searchInfer(status, source[1])
+ return true
+ else
+ m.searchInfer(status, source[1])
+ m.searchInfer(status, source[2])
+ return true
+ end
+ elseif op.type == 'or' then
+ local isTrue = m.checkTrue(status, source[1])
+ if isTrue == true then
+ m.searchInfer(status, source[1])
+ return true
+ elseif isTrue == false then
+ m.searchInfer(status, source[2])
+ return true
+ else
+ m.searchInfer(status, source[1])
+ m.searchInfer(status, source[2])
+ return true
+ end
+ elseif op.type == '==' then
+ local value = m.isSameValue(status, source[1], source[2])
+ if value ~= nil then
+ status.results = m.allocInfer {
+ type = 'boolean',
+ value = value,
+ source = source,
+ level = 100,
+ }
+ return true
+ end
+ --local isSame = m.isSameDef(status, source[1], source[2])
+ --if isSame == true then
+ -- value = true
+ --else
+ -- value = nil
+ --end
+ status.results = m.allocInfer {
+ type = 'boolean',
+ value = value,
+ source = source,
+ level = 100,
+ }
+ return true
+ elseif op.type == '~=' then
+ local value = m.isSameValue(status, source[1], source[2])
+ if value ~= nil then
+ status.results = m.allocInfer {
+ type = 'boolean',
+ value = not value,
+ source = source,
+ level = 100,
+ }
+ return true
+ end
+ --local isSame = m.isSameDef(status, source[1], source[2])
+ --if isSame == true then
+ -- value = false
+ --else
+ -- value = nil
+ --end
+ status.results = m.allocInfer {
+ type = 'boolean',
+ value = value,
+ source = source,
+ level = 100,
+ }
+ return true
+ elseif op.type == '<=' then
+ local v1 = m.getInferLiteral(status, source[1], 'integer')
+ or m.getInferLiteral(status, source[1], 'number')
+ local v2 = m.getInferLiteral(status, source[2], 'integer')
+ or m.getInferLiteral(status, source[2], 'number')
+ v1 = tonumber(v1)
+ v2 = tonumber(v2)
+ local v
+ if v1 and v2 then
+ v = v1 <= v2
+ end
+ status.results = m.allocInfer {
+ type = 'boolean',
+ value = v,
+ source = source,
+ level = 100,
+ }
+ return true
+ elseif op.type == '>=' then
+ local v1 = m.getInferLiteral(status, source[1], 'integer')
+ or m.getInferLiteral(status, source[1], 'number')
+ local v2 = m.getInferLiteral(status, source[2], 'integer')
+ or m.getInferLiteral(status, source[2], 'number')
+ v1 = tonumber(v1)
+ v2 = tonumber(v2)
+ local v
+ if v1 and v2 then
+ v = v1 >= v2
+ end
+ status.results = m.allocInfer {
+ type = 'boolean',
+ value = v,
+ source = source,
+ level = 100,
+ }
+ return true
+ elseif op.type == '<' then
+ local v1 = m.getInferLiteral(status, source[1], 'integer')
+ or m.getInferLiteral(status, source[1], 'number')
+ local v2 = m.getInferLiteral(status, source[2], 'integer')
+ or m.getInferLiteral(status, source[2], 'number')
+ v1 = tonumber(v1)
+ v2 = tonumber(v2)
+ local v
+ if v1 and v2 then
+ v = v1 < v2
+ end
+ status.results = m.allocInfer {
+ type = 'boolean',
+ value = v,
+ source = source,
+ level = 100,
+ }
+ return true
+ elseif op.type == '>' then
+ local v1 = m.getInferLiteral(status, source[1], 'integer')
+ or m.getInferLiteral(status, source[1], 'number')
+ local v2 = m.getInferLiteral(status, source[2], 'integer')
+ or m.getInferLiteral(status, source[2], 'number')
+ v1 = tonumber(v1)
+ v2 = tonumber(v2)
+ local v
+ if v1 and v2 then
+ v = v1 > v2
+ end
+ status.results = m.allocInfer {
+ type = 'boolean',
+ value = v,
+ source = source,
+ level = 100,
+ }
+ return true
+ elseif op.type == '|' then
+ local v1 = m.getInferLiteral(status, source[1], 'integer')
+ local v2 = m.getInferLiteral(status, source[2], 'integer')
+ v1 = tointeger(v1)
+ v2 = tointeger(v2)
+ local v
+ if v1 and v2 then
+ v = v1 | v2
+ end
+ status.results = m.allocInfer {
+ type = 'integer',
+ value = v,
+ source = source,
+ level = 100,
+ }
+ return true
+ elseif op.type == '~' then
+ local v1 = m.getInferLiteral(status, source[1], 'integer')
+ local v2 = m.getInferLiteral(status, source[2], 'integer')
+ v1 = tointeger(v1)
+ v2 = tointeger(v2)
+ local v
+ if v1 and v2 then
+ v = v1 ~ v2
+ end
+ status.results = m.allocInfer {
+ type = 'integer',
+ value = v,
+ source = source,
+ level = 100,
+ }
+ return true
+ elseif op.type == '&' then
+ local v1 = m.getInferLiteral(status, source[1], 'integer')
+ local v2 = m.getInferLiteral(status, source[2], 'integer')
+ v1 = tointeger(v1)
+ v2 = tointeger(v2)
+ local v
+ if v1 and v2 then
+ v = v1 & v2
+ end
+ status.results = m.allocInfer {
+ type = 'integer',
+ value = v,
+ source = source,
+ level = 100,
+ }
+ return true
+ elseif op.type == '<<' then
+ local v1 = m.getInferLiteral(status, source[1], 'integer')
+ local v2 = m.getInferLiteral(status, source[2], 'integer')
+ v1 = tointeger(v1)
+ v2 = tointeger(v2)
+ local v
+ if v1 and v2 then
+ v = v1 << v2
+ end
+ status.results = m.allocInfer {
+ type = 'integer',
+ value = v,
+ source = source,
+ level = 100,
+ }
+ return true
+ elseif op.type == '>>' then
+ local v1 = m.getInferLiteral(status, source[1], 'integer')
+ local v2 = m.getInferLiteral(status, source[2], 'integer')
+ v1 = tointeger(v1)
+ v2 = tointeger(v2)
+ local v
+ if v1 and v2 then
+ v = v1 >> v2
+ end
+ status.results = m.allocInfer {
+ type = 'integer',
+ value = v,
+ source = source,
+ level = 100,
+ }
+ return true
+ elseif op.type == '..' then
+ local v1 = m.getInferLiteral(status, source[1], 'string')
+ local v2 = m.getInferLiteral(status, source[2], 'string')
+ v1 = type(v1) == 'string' and v1 or nil
+ v2 = type(v2) == 'string' and v2 or nil
+ local v
+ if v1 and v2 then
+ v = v1 .. v2
+ end
+ status.results = m.allocInfer {
+ type = 'string',
+ value = v,
+ source = source,
+ level = 100,
+ }
+ return true
+ elseif op.type == '^' then
+ local v1 = m.getInferLiteral(status, source[1], 'integer')
+ or m.getInferLiteral(status, source[1], 'number')
+ local v2 = m.getInferLiteral(status, source[2], 'integer')
+ or m.getInferLiteral(status, source[2], 'number')
+ v1 = tonumber(v1)
+ v2 = tonumber(v2)
+ local v
+ if v1 and v2 then
+ v = v1 ^ v2
+ end
+ status.results = m.allocInfer {
+ type = 'number',
+ value = v,
+ source = source,
+ level = 100,
+ }
+ return true
+ elseif op.type == '/' then
+ local v1 = m.getInferLiteral(status, source[1], 'integer')
+ or m.getInferLiteral(status, source[1], 'number')
+ local v2 = m.getInferLiteral(status, source[2], 'integer')
+ or m.getInferLiteral(status, source[2], 'number')
+ v1 = tonumber(v1)
+ v2 = tonumber(v2)
+ local v
+ if v1 and v2 and v2 ~= 0 then
+ v = v1 / v2
+ end
+ status.results = m.allocInfer {
+ type = 'number',
+ value = v,
+ source = source,
+ level = 100,
+ }
+ return true
+ -- 其他数学运算根据2侧的值决定,当2侧的值均为整数时返回整数
+ elseif op.type == '+' then
+ local int, v1, v2 = mathCheck(status, source[1], source[2])
+ status.results = m.allocInfer{
+ type = int,
+ value = (v1 and v2) and (v1 + v2) or nil,
+ source = source,
+ level = 100,
+ }
+ return true
+ elseif op.type == '-' then
+ local int, v1, v2 = mathCheck(status, source[1], source[2])
+ status.results = m.allocInfer{
+ type = int,
+ value = (v1 and v2) and (v1 - v2) or nil,
+ source = source,
+ level = 100,
+ }
+ return true
+ elseif op.type == '*' then
+ local int, v1, v2 = mathCheck(status, source[1], source[2])
+ status.results = m.allocInfer {
+ type = int,
+ value = (v1 and v2) and (v1 * v2) or nil,
+ source = source,
+ level = 100,
+ }
+ return true
+ elseif op.type == '%' then
+ local int, v1, v2 = mathCheck(status, source[1], source[2])
+ status.results = m.allocInfer {
+ type = int,
+ value = (v1 and v2 and v2 ~= 0) and (v1 % v2) or nil,
+ source = source,
+ level = 100,
+ }
+ return true
+ elseif op.type == '//' then
+ local int, v1, v2 = mathCheck(status, source[1], source[2])
+ status.results = m.allocInfer {
+ type = int,
+ value = (v1 and v2 and v2 ~= 0) and (v1 // v2) or nil,
+ source = source,
+ level = 100,
+ }
+ return true
+ end
+end
+
+function m.inferByDef(status, obj)
+ if not status.share.inferedDef then
+ status.share.inferedDef = {}
+ end
+ if status.share.inferedDef[obj] then
+ return
+ end
+ status.share.inferedDef[obj] = true
+ local mark = {}
+ local newStatus = m.status(status, status.interface)
+ tracy.ZoneBeginN('inferByDef searchRefs')
+ m.searchRefs(newStatus, obj, 'def')
+ tracy.ZoneEnd()
+ for _, src in ipairs(newStatus.results) do
+ local inferStatus = m.status(newStatus)
+ m.searchInfer(inferStatus, src)
+ if #inferStatus.results == 0 then
+ status.results[#status.results+1] = {
+ type = 'any',
+ source = src,
+ level = 0,
+ }
+ else
+ for _, infer in ipairs(inferStatus.results) do
+ if not mark[infer.source] then
+ mark[infer.source] = true
+ status.results[#status.results+1] = infer
+ end
+ end
+ end
+ end
+end
+
+local function inferBySetOfLocal(status, source)
+ if status.share[source] then
+ return
+ end
+ status.share[source] = true
+ local newStatus = m.status(status)
+ if source.value then
+ m.searchInfer(newStatus, source.value)
+ end
+ if source.ref then
+ for _, ref in ipairs(source.ref) do
+ if ref.type == 'setlocal' then
+ break
+ end
+ m.searchInfer(newStatus, ref)
+ end
+ for _, infer in ipairs(newStatus.results) do
+ status.results[#status.results+1] = infer
+ end
+ end
+end
+
+function m.inferByLocalRef(status, source)
+ if #status.results ~= 0 then
+ return
+ end
+ if source.type == 'local' then
+ inferBySetOfLocal(status, source)
+ elseif source.type == 'setlocal'
+ or source.type == 'getlocal' then
+ inferBySetOfLocal(status, source.node)
+ end
+end
+
+function m.inferByCall(status, source)
+ if not source.parent then
+ return
+ end
+ if source.parent.type ~= 'call' then
+ return
+ end
+ if source.parent.node == source then
+ status.results[#status.results+1] = {
+ type = 'function',
+ source = source,
+ level = 10,
+ }
+ return
+ end
+end
+
+function m.inferByGetTable(status, source)
+ if source.type == 'field'
+ or source.type == 'method' then
+ source = source.parent
+ end
+ local next = source.next
+ if not next then
+ return
+ end
+ if next.type == 'getfield'
+ or next.type == 'getindex'
+ or next.type == 'setmethod'
+ or next.type == 'setfield'
+ or next.type == 'setindex' then
+ status.results[#status.results+1] = {
+ type = 'table',
+ source = source,
+ level = 10,
+ }
+ elseif next.type == 'getmethod' then
+ status.results[#status.results+1] = {
+ type = 'table',
+ source = source,
+ level = 5,
+ }
+ status.results[#status.results+1] = {
+ type = 'string',
+ source = source,
+ level = 5,
+ }
+ end
+end
+
+function m.inferByUnary(status, source)
+ local parent = source.parent
+ if not parent or parent.type ~= 'unary' then
+ return
+ end
+ local op = parent.op
+ if op.type == '#' then
+ status.results[#status.results+1] = {
+ type = 'string',
+ source = source,
+ level = 5,
+ }
+ status.results[#status.results+1] = {
+ type = 'table',
+ source = source,
+ level = 5,
+ }
+ elseif op.type == '~' then
+ status.results[#status.results+1] = {
+ type = 'integer',
+ source = source,
+ level = 10,
+ }
+ elseif op.type == '-' then
+ status.results[#status.results+1] = {
+ type = 'number',
+ source = source,
+ level = 10,
+ }
+ end
+end
+
+function m.inferByBinary(status, source)
+ local parent = source.parent
+ if not parent or parent.type ~= 'binary' then
+ return
+ end
+ local op = parent.op
+ if op.type == '<='
+ or op.type == '>='
+ or op.type == '<'
+ or op.type == '>'
+ or op.type == '^'
+ or op.type == '/'
+ or op.type == '+'
+ or op.type == '-'
+ or op.type == '*'
+ or op.type == '%' then
+ status.results[#status.results+1] = {
+ type = 'number',
+ source = source,
+ level = 10,
+ }
+ elseif op.type == '|'
+ or op.type == '~'
+ or op.type == '&'
+ or op.type == '<<'
+ or op.type == '>>'
+ -- 整数的可能性比较高
+ or op.type == '//' then
+ status.results[#status.results+1] = {
+ type = 'integer',
+ source = source,
+ level = 10,
+ }
+ elseif op.type == '..' then
+ status.results[#status.results+1] = {
+ type = 'string',
+ source = source,
+ level = 10,
+ }
+ end
+end
+
+local function mergeFunctionReturnsByDoc(status, source, index, call)
+ if not source or source.type ~= 'function' then
+ return
+ end
+ if not source.bindDocs then
+ return
+ end
+ local returns = {}
+ for _, doc in ipairs(source.bindDocs) do
+ if doc.type == 'doc.return' then
+ for _, rtn in ipairs(doc.returns) do
+ returns[#returns+1] = rtn
+ end
+ end
+ end
+ local rtn = returns[index]
+ if not rtn then
+ return
+ end
+ local results = m.getDocTypeNames(status, rtn)
+ if #results == 0 then
+ return
+ end
+ for _, res in ipairs(results) do
+ status.results[#status.results+1] = res
+ end
+ return true
+end
+
+local function mergeDocTypeFunctionReturns(status, source, index)
+ if not source.bindDocs then
+ return
+ end
+ for _, doc in ipairs(source.bindDocs) do
+ if doc.type == 'doc.type' then
+ for _, typeUnit in ipairs(doc.types) do
+ if typeUnit.type == 'doc.type.function' then
+ local rtn = typeUnit.returns[index]
+ if rtn then
+ local results = m.getDocTypeNames(status, rtn)
+ for _, res in ipairs(results) do
+ status.results[#status.results+1] = res
+ end
+ end
+ end
+ end
+ end
+ end
+end
+
+local function mergeFunctionReturns(status, source, index, call)
+ local returns = source.returns
+ if not returns then
+ return
+ end
+ for i = 1, #returns do
+ local rtn = returns[i]
+ if rtn[index] then
+ if rtn[index].type == 'call' then
+ if not m.checkReturnMark(status, rtn[index], true) then
+ m.inferByCallReturnAndIndex(status, rtn[index], index)
+ end
+ else
+ local newStatus = m.status(status)
+ m.searchInfer(newStatus, rtn[index])
+ if #newStatus.results == 0 then
+ status.results[#status.results+1] = {
+ type = 'any',
+ source = rtn[index],
+ level = 0,
+ }
+ else
+ for _, infer in ipairs(newStatus.results) do
+ status.results[#status.results+1] = infer
+ end
+ end
+ end
+ end
+ end
+end
+
+function m.inferByCallReturnAndIndex(status, call, index)
+ local node = call.node
+ local newStatus = m.status(status, status.interface)
+ m.searchRefs(newStatus, node, 'def')
+ local hasDocReturn
+ for _, src in ipairs(newStatus.results) do
+ if mergeDocTypeFunctionReturns(status, src, index) then
+ hasDocReturn = true
+ elseif mergeFunctionReturnsByDoc(status, src.value, index, call) then
+ hasDocReturn = true
+ end
+ end
+ if not hasDocReturn then
+ for _, src in ipairs(newStatus.results) do
+ if src.value and src.value.type == 'function' then
+ if not m.checkReturnMark(status, src.value, true) then
+ mergeFunctionReturns(status, src.value, index, call)
+ end
+ end
+ end
+ end
+end
+
+function m.inferByCallReturn(status, source)
+ if source.type == 'call' then
+ m.inferByCallReturnAndIndex(status, source, 1)
+ return
+ end
+ if source.type ~= 'select' then
+ if source.value and source.value.type == 'select' then
+ source = source.value
+ else
+ return
+ end
+ end
+ if not source.vararg or source.vararg.type ~= 'call' then
+ return
+ end
+ m.inferByCallReturnAndIndex(status, source.vararg, source.index)
+end
+
+function m.inferByPCallReturn(status, source)
+ if source.type ~= 'select' then
+ if source.value and source.value.type == 'select' then
+ source = source.value
+ else
+ return
+ end
+ end
+ local call = source.vararg
+ if not call or call.type ~= 'call' then
+ return
+ end
+ if not call.args then
+ return
+ end
+ local node = call.node
+ local specialName = node.special
+ local func, index
+ if specialName == 'pcall' then
+ func = call.args[1]
+ index = source.index - 1
+ elseif specialName == 'xpcall' then
+ func = call.args[1]
+ index = source.index - 2
+ else
+ return
+ end
+ local newStatus = m.status(status, status.interface)
+ m.searchRefs(newStatus, func, 'def')
+ for _, src in ipairs(newStatus.results) do
+ if src.value and src.value.type == 'function' then
+ mergeFunctionReturns(status, src.value, index)
+ end
+ end
+end
+
+function m.cleanInfers(infers, obj)
+ -- kick lower level infers
+ local level = 0
+ if obj.type ~= 'select' then
+ for i = 1, #infers do
+ local infer = infers[i]
+ if infer.level > level then
+ level = infer.level
+ end
+ end
+ end
+ -- merge infers
+ local mark = {}
+ for i = #infers, 1, -1 do
+ local infer = infers[i]
+ if infer.level < level then
+ infers[i] = infers[#infers]
+ infers[#infers] = nil
+ goto CONTINUE
+ end
+ local key = ('%p'):format(infer.type, infer.source)
+ if mark[key] then
+ infers[i] = infers[#infers]
+ infers[#infers] = nil
+ else
+ mark[key] = true
+ end
+ ::CONTINUE::
+ end
+ -- kick doc.generic
+ if #infers > 1 then
+ for i = #infers, 1, -1 do
+ local infer = infers[i]
+ if infer.source.typeGeneric then
+ infers[i] = infers[#infers]
+ infers[#infers] = nil
+ end
+ end
+ end
+end
+
+function m.searchInfer(status, obj)
+ while obj.type == 'paren' do
+ obj = obj.exp
+ if not obj then
+ return
+ end
+ end
+ while true do
+ local value = m.getObjectValue(obj)
+ if not value or value == obj then
+ break
+ end
+ obj = value
+ end
+ if not obj then
+ return
+ end
+
+ local cache, makeCache = m.getRefCache(status, obj, 'infer')
+ if cache then
+ for i = 1, #cache do
+ status.results[#status.results+1] = cache[i]
+ end
+ return
+ end
+
+ if DEVELOP then
+ status.share.clock = status.share.clock or osClock()
+ end
+
+ if not status.share.lockInfer then
+ status.share.lockInfer = {}
+ end
+ if status.share.lockInfer[obj] then
+ return
+ end
+ status.share.lockInfer[obj] = true
+
+ local checked = m.inferCheckDoc(status, obj)
+ or m.inferCheckUpDoc(status, obj)
+ or m.inferCheckLiteral(status, obj)
+ or m.inferCheckUnary(status, obj)
+ or m.inferCheckBinary(status, obj)
+ if checked then
+ m.cleanInfers(status.results, obj)
+ if makeCache then
+ makeCache(status.results)
+ end
+ return
+ end
+
+ m.inferByLocalRef(status, obj)
+ if status.deep then
+ tracy.ZoneBeginN('inferByDef')
+ m.inferByDef(status, obj)
+ tracy.ZoneEnd()
+ end
+ --m.inferByCall(status, obj)
+ m.inferByGetTable(status, obj)
+ m.inferByUnary(status, obj)
+ m.inferByBinary(status, obj)
+ m.cleanInfers(status.results, obj)
+ if makeCache then
+ makeCache(status.results)
+ end
+end
+
+--- 请求对象的引用,包括 `a.b.c` 形式
+--- 与 `return function` 形式。
+--- 不穿透 `setmetatable` ,考虑由
+--- 业务层进行反向 def 搜索。
+function m.requestReference(obj, interface, deep)
+ local status = m.status(nil, interface, deep)
+ -- 根据 field 搜索引用
+ m.searchRefs(status, obj, 'ref')
+
+ m.searchRefsAsFunction(status, obj, 'ref')
+
+ if m.debugMode then
+ --print('count:', status.share.count)
+ end
+
+ return status.results, status.share.count
+end
+
+--- 请求对象的定义,包括 `a.b.c` 形式
+--- 与 `return function` 形式。
+--- 穿透 `setmetatable` 。
+function m.requestDefinition(obj, interface, deep)
+ local status = m.status(nil, interface, deep)
+ -- 根据 field 搜索定义
+ m.searchRefs(status, obj, 'def')
+
+ return status.results, status.share.count
+end
+
+--- 请求对象的字段
+---@param filterKey nil|string|table
+function m.requestFields(obj, interface, deep, filterKey)
+ local status = m.status(nil, interface, deep)
+
+ m.searchFields(status, obj, filterKey)
+
+ return status.results, status.share.count
+end
+
+--- 请求对象的定义字段
+---@param filterKey nil|string|table
+function m.requestDefFields(obj, interface, deep, filterKey)
+ local status = m.status(nil, interface, deep)
+
+ m.searchDefFields(status, obj, filterKey)
+
+ return status.results, status.share.count
+end
+
+--- 请求对象的类型推测
+function m.requestInfer(obj, interface, deep)
+ local status = m.status(nil, interface, deep)
+ m.searchInfer(status, obj)
+
+ return status.results, status.share.count
+end
+
+function m.debugView(obj)
+ return require 'files'.position(m.getUri(obj), obj.start), m.getUri(obj)
+end
+
+return m
diff --git a/script/core/linker.lua b/script/core/linker.lua
index d7f53fa4..6002b22c 100644
--- a/script/core/linker.lua
+++ b/script/core/linker.lua
@@ -1,4 +1,5 @@
local util = require 'utility'
+local guide = require 'parser.guide'
local function getKey(source)
if source.type == 'local' then
@@ -33,6 +34,15 @@ local function checkGlobal(source)
return nil
end
+local function checkLocal(source)
+ if source.type == 'local'
+ or source.type == 'setlocal'
+ or source.type == 'getlocal' then
+ return true
+ end
+ return nil
+end
+
local function checkTableField(source)
if source.type == 'table' then
return true
@@ -56,35 +66,61 @@ local function checkFunctionReturn(source)
return nil
end
+local IDList = {}
local function getID(source)
if source.type == 'field'
or source.type == 'method' then
source = source.parent
end
local current = source
- local idList = {}
+ local index = 0
while true do
local id, node = getKey(current)
if not id then
break
end
- idList[#idList+1] = id
+ index = index + 1
+ IDList[index] = id
source = current
if not node then
break
end
current = node
end
- util.revertTable(idList)
- local id = table.concat(idList, '|')
+ for i = index + 1, #IDList do
+ IDList[i] = nil
+ end
+ util.revertTable(IDList)
+ local id = table.concat(IDList, '|')
return id, current
end
+---@class link
+---@field id string
+-- 语法树单元
+---@field source parser.guide.object
+-- 是否是局部变量
+---@field loc boolean
+-- 是否是全局变量
+---@field global boolean
+-- 是否是字面量表中的字段
+---@field tfield boolean
+-- 返回值,文件返回值总是0,函数返回值为第几个返回值
+---@field freturn integer
+
---创建source的链接信息
+---@param source parser.guide.object
+---@return link
local function createLink(source)
local id, node = getID(source)
+ if not id then
+ return nil
+ end
return {
- id = id,
+ id = id,
+ source = source,
+ -- 局部变量
+ loc = checkLocal(node),
-- 全局变量
global = checkGlobal(node),
-- 字面量表中的字段
@@ -94,9 +130,31 @@ local function createLink(source)
}
end
+local function insertLinker(linkers, tp, link)
+ local list = linkers[tp]
+ local id = link.id
+ if not list[id] then
+ list[id] = {}
+ end
+ list[id][#list[id]+1] = link
+ link._linker = list[id]
+end
+
local m = {}
+---根据语法树单元获取关联的link列表
+---@param source parser.guide.object
+---@return link[]?
+function m.getLinkersBySource(source)
+ if not source._link then
+ source._link = createLink(source)
+ end
+ return source._link and source._link._linker
+end
+
---获取source的链接信息
+---@param source parser.guide.object
+---@return link
function m.getLink(source)
if not source._link then
source._link = createLink(source)
@@ -104,4 +162,36 @@ function m.getLink(source)
return source._link
end
+---编译整个文件的link
+---@param source parser.guide.object
+---@return table
+function m.compileLinks(source)
+ local root = guide.getRoot(source)
+ if root._linkers then
+ return root._linkers
+ end
+ local linkers = {
+ loc = {},
+ global = {},
+ tfield = {},
+ }
+ guide.eachSource(root, function (src)
+ local link = m.getLink(src)
+ if not link then
+ return
+ end
+ if link.global then
+ insertLinker(linkers, 'global', link)
+ end
+ if link.loc then
+ insertLinker(linkers, 'loc', link)
+ end
+ if link.tfield then
+ insertLinker(linkers, 'tfield', link)
+ end
+ end)
+ root._linkers = linkers
+ return linkers
+end
+
return m
diff --git a/script/parser/guide.lua b/script/parser/guide.lua
index 2369e84f..46c32f3a 100644
--- a/script/parser/guide.lua
+++ b/script/parser/guide.lua
@@ -293,10 +293,19 @@ end
---@param obj parser.guide.object
---@return parser.guide.object
function m.getRoot(obj)
+ local source = obj
+ if source._root then
+ return source._root
+ end
for _ = 1, 1000 do
if obj.type == 'main' then
+ source._root = obj
return obj
end
+ if obj._root then
+ source._root = obj._root
+ return source._root
+ end
local parent = obj.parent
if not parent then
return nil
diff --git a/test/basic/linker.lua b/test/basic/linker.lua
index bf4e50bb..0be93e8d 100644
--- a/test/basic/linker.lua
+++ b/test/basic/linker.lua
@@ -37,24 +37,28 @@ local function TEST(script)
end
CARE['id'] = true
+CARE['loc'] = true
TEST [[
local <?x?>
]] {
- id = '9',
+ id = '9',
+ loc = true,
}
TEST [[
local x
print(<?x?>)
]] {
- id = '7',
+ id = '7',
+ loc = true,
}
TEST [[
local x
<?x?> = 1
]] {
- id = '7',
+ id = '7',
+ loc = true,
}
CARE['global'] = true
@@ -76,14 +80,16 @@ TEST [[
local x
print(x.y.<?z?>)
]] {
- id = '7|"y"|"z"',
+ id = '7|"y"|"z"',
+ loc = true,
}
TEST [[
local x
function x:<?f?>() end
]] {
- id = '7|"f"',
+ id = '7|"f"',
+ loc = true,
}
TEST [[
diff --git a/test/basic/linker.txt b/test/basic/linker.txt
new file mode 100644
index 00000000..18176369
--- /dev/null
+++ b/test/basic/linker.txt
@@ -0,0 +1,16 @@
+ast -> linkers = {
+ globals = {
+ ['"X"|"Y"|"Z"'] = {src1, src2, src3},
+ ['"X"|"Y"'] = {src4, src5, src6},
+ ['"X"'] = {src7, src8, src9},
+ },
+ locals = {
+ ['7'] = {src10},
+ ['7|"x"'] = {src11},
+ },
+ tfield = {
+ ['11|"k"'] = {src12},
+ },
+}
+
+