cleanup

2 files changed, 19 insertions, 24 deletions

diff --git a/script/core/linker.lua b/script/core/linker.lua
index b97fc9bf..20bf6b34 100644
--- a/script/core/linker.lua
+++ b/script/core/linker.lua
@@ -95,6 +95,9 @@ local function checkMode(source)
     if source.type == 'select' then
         return 's'
     end
+    if source.type == 'function' then
+        return 'f'
+    end
     if source.type == 'doc.class.name' then
         return 'dc'
     end
diff --git a/script/core/searcher.lua b/script/core/searcher.lua
index da00f46a..a29407e7 100644
--- a/script/core/searcher.lua
+++ b/script/core/searcher.lua
@@ -128,7 +128,12 @@ function m.searchRefsByID(status, uri, expect, mode)
         return nil
     end
 
-    local function searchFunction(obj)
+    local function searchFunction(id)
+        local funcs = linker.getLinksByID(root, id)
+        if not funcs then
+            return
+        end
+        local obj = funcs[1].source
         if obj.type ~= 'function' then
             return
         end
@@ -157,34 +162,20 @@ function m.searchRefsByID(status, uri, expect, mode)
     end
 
     local function checkForward(link, field)
-        if link.forward then
-            for _, forwardSources in ipairs(link.forward) do
-                searchSource(forwardSources, field)
-            end
+        if not link.forward then
+            return
         end
-        if link.fforward then
-            for _, func in ipairs(link.fforward) do
-                local forwards = func()
-                for _, forwardSources in ipairs(forwards) do
-                    searchSource(forwardSources, field)
-                end
-            end
+        for _, forwardSources in ipairs(link.forward) do
+            searchSource(forwardSources, field)
         end
     end
 
     local function checkBackward(link, field)
-        if link.backward then
-            for _, backSources in ipairs(link.backward) do
-                searchSource(backSources, field)
-            end
+        if not link.backward then
+            return
         end
-        if link.fbackward then
-            for _, func in ipairs(link.fbackward) do
-                local backwards = func()
-                for _, backSources in ipairs(backwards) do
-                    searchSource(backSources, field)
-                end
-            end
+        for _, backSources in ipairs(link.backward) do
+            searchSource(backSources, field)
         end
     end
 
@@ -214,7 +205,8 @@ function m.searchRefsByID(status, uri, expect, mode)
     end
 
     search(expect)
-
+    checkLastID(expect)
+    searchFunction(expect)
 end
 
 ---搜索对象的引用