stash

2 files changed, 27 insertions, 12 deletions

diff --git a/script/core/guide.lua b/script/core/guide.lua
index eba1301b..dd7888c6 100644
--- a/script/core/guide.lua
+++ b/script/core/guide.lua
@@ -121,6 +121,24 @@ function m.searchRefs(status, source, mode)
 
     pushQueueWithID(source)
 
+    local function checkForward(link)
+        if not link.forward then
+            return
+        end
+        for _, forwardSources in ipairs(link.forward) do
+            pushQueueWithID(forwardSources)
+        end
+    end
+
+    local function checkBackward(link)
+        if not link.backward then
+            return
+        end
+        for _, backSources in ipairs(link.backward) do
+            pushQueueWithID(backSources)
+        end
+    end
+
     for _ = 1, 1000 do
         if index <= 0 then
             break
@@ -134,16 +152,8 @@ function m.searchRefs(status, source, mode)
         end
         for _, eachLink in ipairs(links) do
             m.pushResult(status, mode, eachLink.source)
-            if eachLink.forward then
-                for _, forwardSources in ipairs(eachLink.forward) do
-                    pushQueueWithID(forwardSources)
-                end
-            end
-            if eachLink.backward then
-                for _, backSources in ipairs(eachLink.backward) do
-                    pushQueueWithID(backSources)
-                end
-            end
+            checkForward(eachLink)
+            checkBackward(eachLink)
         end
         ::CONTINUE::
     end
diff --git a/script/core/linker.lua b/script/core/linker.lua
index a064deaa..64aaef1c 100644
--- a/script/core/linker.lua
+++ b/script/core/linker.lua
@@ -147,11 +147,15 @@ local function getID(source)
     end
     util.revertTable(IDList)
     local id = table.concat(IDList, '|')
-    return id, current
+    local parentID = table.concat(IDList, '|', 1, -2)
+    return id, current, parentID
 end
 
 ---@class link
+-- 当前节点的id
 ---@field id     string
+-- 父节点的id
+---@field parentID string
 -- 语法树单元
 ---@field source parser.guide.object
 -- 是否是局部变量
@@ -173,13 +177,14 @@ end
 ---@param source parser.guide.object
 ---@return link
 local function createLink(source)
-    local id, node = getID(source)
+    local id, node, parentID = getID(source)
     if not id then
         return nil
     end
     return {
         id       = id,
         source   = source,
+        parentID = parentID,
         loc      = checkLocal(node),
         global   = checkGlobal(node),
         tfield   = checkTableField(node),