1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
|
local core = require 'core'
local function posToRange(lines, start, finish)
local start_row, start_col = lines:rowcol(start)
local finish_row, finish_col = lines:rowcol(finish)
return {
start = {
line = start_row - 1,
character = start_col - 1,
},
['end'] = {
line = finish_row - 1,
character = finish_col,
},
}
end
local function convertRange(lines, symbol)
symbol.range = posToRange(lines, symbol.range[1], symbol.range[2])
symbol.selectionRange = posToRange(lines, symbol.selectionRange[1], symbol.selectionRange[2])
if symbol.children then
for _, child in ipairs(symbol.children) do
convertRange(lines, child)
end
end
end
return function (lsp, params)
local uri = params.textDocument.uri
local vm, lines = lsp:loadVM(uri)
if not vm then
return nil
end
local symbols = core.documentSymbol(vm)
for _, symbol in ipairs(symbols) do
convertRange(lines, symbol)
end
return symbols
end
|
