diff options
Diffstat (limited to 'script/core/linker.lua')
| -rw-r--r-- | script/core/linker.lua | 68 |
1 files changed, 52 insertions, 16 deletions
diff --git a/script/core/linker.lua b/script/core/linker.lua index ef6dd010..e31d4eb9 100644 --- a/script/core/linker.lua +++ b/script/core/linker.lua @@ -6,7 +6,8 @@ local Linkers local LastIDCache = {} local SPLIT_CHAR = '\x1F' local SPLIT_REGEX = SPLIT_CHAR .. '[^' .. SPLIT_CHAR .. ']+$' -local INDEX_CHAR = '\x1E' +local RETURN_INDEX_CHAR = '#' +local PARAM_INDEX_CHAR = '@' ---是否是全局变量（包括 _G.XXX 形式） ---@param source parser.guide.object @@ -85,14 +86,15 @@ local function getKey(source) elseif source.type == 'function' then return source.start, nil elseif source.type == 'select' then - return ('%d%s%s%d'):format(source.start, SPLIT_CHAR, INDEX_CHAR, source.index) + return ('%d%s%s%d'):format(source.start, SPLIT_CHAR, RETURN_INDEX_CHAR, source.index) elseif source.type == 'doc.class.name' or source.type == 'doc.type.name' or source.type == 'doc.alias.name' then return source[1], nil elseif source.type == 'doc.class' or source.type == 'doc.type' - or source.type == 'doc.alias' then + or source.type == 'doc.alias' + or source.type == 'doc.param' then return source.start, nil end return nil, nil @@ -120,6 +122,9 @@ local function checkMode(source) if source.type == 'doc.type' then return 'dt:' end + if source.type == 'doc.param' then + return 'dp:' + end if source.type == 'doc.alias' then return 'da:' end @@ -185,7 +190,10 @@ local TempList = {} ---前进 ---@param source parser.guide.object ---@return parser.guide.object[] -local function checkForward(source, id) +local function checkForward(source) + if not source then + return + end local list = TempList local parent = source.parent if source.value then @@ -235,7 +243,10 @@ end ---后退 ---@param source parser.guide.object ---@return parser.guide.object[] -local function checkBackward(source, id) +local function checkBackward(source) + if not source then + return + end local list = TempList local parent = source.parent if parent.value == source then @@ -266,7 +277,7 @@ local function checkBackward(source, id) if source.returnIndex then for _, src in ipairs(parent.bindSources) do if src.type == 'function' then - local fullID = ('%s%s%s%s'):format(getID(src), SPLIT_CHAR, INDEX_CHAR, source.returnIndex) + local fullID = ('%s%s%s%s'):format(getID(src), SPLIT_CHAR, RETURN_INDEX_CHAR, source.returnIndex) list[#list+1] = fullID end end @@ -279,6 +290,29 @@ local function checkBackward(source, id) list[#list+1] = ('s:%d'):format(sel.start) end end + -- 将调用参数映射到函数调用上 + if parent.type == 'callargs' then + for i = 1, #parent do + if parent[i] == source then + local call = parent.parent + local node = call.node + local nodeID = getID(node) + if not nodeID then + break + end + list[#list+1] = ('%s%s%s%s'):format( + nodeID, + SPLIT_CHAR, + PARAM_INDEX_CHAR, + i + ) + break + end + end + end + if source.type == 'doc.param' then + print(source) + end if #list == 0 then return nil else @@ -298,18 +332,15 @@ end ---@field backward parser.guide.object[] ---创建source的链接信息 ----@param source parser.guide.object +---@param id string +---@param source? parser.guide.object ---@return link -local function createLink(source) - local id = getID(source) - if not id then - return nil - end +local function createLink(id, source) return { id = id, source = source, - forward = checkForward(source, id), - backward = checkBackward(source, id), + forward = checkForward(source), + backward = checkBackward(source), } end @@ -325,7 +356,8 @@ end local m = {} m.SPLIT_CHAR = SPLIT_CHAR -m.INDEX_CHAR = INDEX_CHAR +m.RETURN_INDEX_CHAR = RETURN_INDEX_CHAR +m.PARAM_INDEX_CHAR = PARAM_INDEX_CHAR ---根据ID来获取所有的link ---@param root parser.guide.object @@ -360,8 +392,12 @@ end ---@param source parser.guide.object ---@return link function m.getLink(source) + local id = getID(source) + if not id then + return nil + end if source._link == nil then - source._link = createLink(source) or false + source._link = createLink(id, source) or false end return source._link or nil end |