过测试

1 files changed, 9 insertions, 2 deletions

diff --git a/script-beta/core/definition.lua b/script-beta/core/definition.lua
index 77bef643..8afdc2ba 100644
--- a/script-beta/core/definition.lua
+++ b/script-beta/core/definition.lua
@@ -22,8 +22,7 @@ local function findDef(source, callback)
             return
         end
         if info.mode == 'declare'
-        or info.mode == 'set'
-        or info.mode == 'return' then
+        or info.mode == 'set' then
             local src  = info.source
             local root = guide.getRoot(src)
             local uri  = root.uri
@@ -42,6 +41,14 @@ local function findDef(source, callback)
                 callback(src, uri)
             end
         end
+        if info.mode == 'value' then
+            local src  = info.source
+            local root = guide.getRoot(src)
+            local uri  = root.uri
+            if src.parent.type == 'return' then
+                callback(src, uri)
+            end
+        end
     end)
 end