特殊处理getLibrary

1 files changed, 7 insertions, 3 deletions

diff --git a/script-beta/vm/getLibrary.lua b/script-beta/vm/getLibrary.lua
index 69bc3f2b..e9c3b78a 100644
--- a/script-beta/vm/getLibrary.lua
+++ b/script-beta/vm/getLibrary.lua
@@ -6,6 +6,10 @@ local function checkStdLibrary(source)
     if source.library then
         return source
     end
+    local value = guide.getObjectValue(source)
+    if value and value.type == 'select' then
+        source = value
+    end
     if source.type == 'getglobal'
     or source.type == 'setglobal' then
         local name = guide.getName(source)
@@ -79,8 +83,8 @@ local function checkNode(source)
     end
 end
 
-local function checkRef(source)
-    local results = guide.requestReference(source)
+local function checkDef(source)
+    local results = guide.requestDefinition(source)
     for _, src in ipairs(results) do
         local lib = checkStdLibrary(src) or checkNode(src)
         if lib then
@@ -93,7 +97,7 @@ end
 local function getLibrary(source)
     return checkNode(source)
         or checkStdLibrary(source)
-        or checkRef(source)
+        or checkDef(source)
 end
 
 function vm.getLibrary(source)