diff options
Diffstat (limited to 'server/src/method')
| -rw-r--r-- | server/src/method/textDocument/references.lua | 35 |
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 |