修正换算错误的问题

1 files changed, 19 insertions, 16 deletions

diff --git a/server/src/method/textDocument/references.lua b/server/src/method/textDocument/references.lua
index 47f7b370..ee7fa80c 100644
--- a/server/src/method/textDocument/references.lua
+++ b/server/src/method/textDocument/references.lua
@@ -17,23 +17,26 @@ return function (lsp, params)
 
     local locations = {}
     for i, position in ipairs(positions) do
-        local start, finish, uri = position[1], position[2], position[3]
-        local start_row,  start_col  = lines:rowcol(start)
-        local finish_row, finish_col = lines:rowcol(finish)
-        locations[i] = {
-            uri = uri,
-            range = {
-                start = {
-                    line = start_row - 1,
-                    character = start_col - 1,
-                },
-                ['end'] = {
-                    line = finish_row - 1,
-                    -- 这里不用-1，因为前端期待的是匹配完成后的位置
-                    character = finish_col,
-                },
+        local start, finish, valueUri = position[1], position[2], (position[3] or uri)
+        local _, valueLines = lsp:loadVM(valueUri)
+        if valueLines then
+            local start_row,  start_col  = valueLines:rowcol(start)
+            local finish_row, finish_col = valueLines:rowcol(finish)
+            locations[i] = {
+                uri =  valueUri,
+                range = {
+                    start = {
+                        line = start_row - 1,
+                        character = start_col - 1,
+                    },
+                    ['end'] = {
+                        line = finish_row - 1,
+                        -- 这里不用-1，因为前端期待的是匹配完成后的位置
+                        character = finish_col,
+                    },
+                }
             }
-        }
+        end
     end
 
     local response = locations