更新诊断

1 files changed, 27 insertions, 12 deletions

diff --git a/server-beta/src/searcher/eachRef.lua b/server-beta/src/searcher/eachRef.lua
index 8f6b5c2f..3ae020cd 100644
--- a/server-beta/src/searcher/eachRef.lua
+++ b/server-beta/src/searcher/eachRef.lua
@@ -227,20 +227,35 @@ local function ofLocal(loc, callback)
 end
 
 local function ofGlobal(source, callback)
-    local key   = guide.getKeyName(source)
-    for uri in files.eachFile() do
-        local globals = files.getGlobals(uri)
-        local ast = files.getAst(uri)
-        if ast and globals and globals[key] then
-            searcher.eachGlobal(ast.ast, function (info)
-                if key == info.key then
-                    callback(info)
-                    if info.value then
-                        ofValue(info.value, callback)
+    local key = guide.getKeyName(source)
+    local node = source.node
+    if node.tag == '_ENV' then
+        for uri in files.eachFile() do
+            local globals = files.getGlobals(uri)
+            local ast = files.getAst(uri)
+            if ast and globals and globals[key] then
+                searcher.eachGlobal(ast.ast, function (info)
+                    if key == info.key then
+                        callback(info)
+                        if info.value then
+                            ofValue(info.value, callback)
+                        end
                     end
-                end
-            end)
+                end)
+            end
         end
+    else
+        searcher.eachField(node, function (info)
+            if key == info.key then
+                callback {
+                    source   = info.source,
+                    mode     = info.mode,
+                }
+                if info.value then
+                    ofValue(info.value, callback)
+                end
+            end
+        end)
     end
 end