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
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
|
local hoverFunction = require 'core.hover.function'
local getName = require 'core.name'
local hover = require 'core.hover'
local SymbolKind = {
File = 1,
Module = 2,
Namespace = 3,
Package = 4,
Class = 5,
Method = 6,
Property = 7,
Field = 8,
Constructor = 9,
Enum = 10,
Interface = 11,
Function = 12,
Variable = 13,
Constant = 14,
String = 15,
Number = 16,
Boolean = 17,
Array = 18,
Object = 19,
Key = 20,
Null = 21,
EnumMember = 22,
Struct = 23,
Event = 24,
Operator = 25,
TypeParameter = 26,
}
local function buildLocal(vm, source, used, callback)
local vars = source[1]
local exps = source[2]
if vars.type ~= 'list' then
vars = {vars}
end
if not exps or exps.type ~= 'list' then
exps = {exps}
end
for i, var in ipairs(vars) do
local exp = exps[i]
local data = {}
local loc = var:bindLocal()
data.name = loc:getName()
data.range = { var.start, var.finish }
data.selectionRange = { var.start, var.finish }
if exp then
local hvr = hover(var)
if exp.type == 'function' then
data.kind = SymbolKind.Function
else
data.kind = SymbolKind.Variable
end
data.detail = hvr.label:gsub('[\r\n]', '')
data.valueRange = { exp.start, exp.finish }
used[exp] = true
else
data.kind = SymbolKind.Variable
data.detail = ''
data.valueRange = { var.start, var.finish }
end
callback(data)
end
end
local function buildSet(vm, source, used, callback)
local vars = source[1]
local exps = source[2]
if vars.type ~= 'list' then
vars = {vars}
end
if not exps or exps.type ~= 'list' then
exps = {exps}
end
for i, var in ipairs(vars) do
if var:bindLocal() then
goto CONTINUE
end
local exp = exps[i]
local data = {}
data.name = getName(var)
data.range = { var.start, var.finish }
data.selectionRange = { var.start, var.finish }
if exp then
local hvr = hover(var)
if not hvr then
goto CONTINUE
end
if exp.type == 'function' then
data.kind = SymbolKind.Function
else
data.kind = SymbolKind.Property
end
data.detail = hvr.label:gsub('[\r\n]', '')
data.valueRange = { exp.start, exp.finish }
used[exp] = true
else
data.kind = SymbolKind.Property
data.detail = ''
data.valueRange = { var.start, var.finish }
end
callback(data)
:: CONTINUE ::
end
end
local function buildPair(vm, source, used, callback)
local var = source[1]
local exp = source[2]
local data = {}
data.name = getName(var)
data.range = { var.start, var.finish }
data.selectionRange = { var.start, var.finish }
if exp then
local hvr = hover(var)
if not hvr then
return
end
if exp.type == 'function' then
data.kind = SymbolKind.Function
else
data.kind = SymbolKind.Class
end
data.detail = hvr.label:gsub('[\r\n]', '')
data.valueRange = { exp.start, exp.finish }
used[exp] = true
else
data.kind = SymbolKind.Class
data.detail = ''
data.valueRange = { var.start, var.finish }
end
callback(data)
end
local function buildLocalFunction(vm, source, used, callback)
local value = source:bindFunction()
if not value then
return
end
local name = getName(source.name)
local hvr = hoverFunction(name, value:getFunction())
if not hvr then
return
end
local kind = SymbolKind.Function
callback {
name = name,
detail = hvr.label:gsub('[\r\n]', ''),
kind = kind,
range = { source.start, source.finish },
selectionRange = { source.name.start, source.name.finish },
valueRange = { source.start, source.finish },
}
end
local function buildFunction(vm, source, used, callback)
if used[source] then
return
end
local value = source:bindFunction()
if not value then
return
end
local name = getName(source.name)
local func = value:getFunction()
local hvr = hoverFunction(name, func, func:getObject())
if not hvr then
return
end
local data = {}
data.name = name
data.detail = hvr.label:gsub('[\r\n]', '')
data.range = { source.start, source.finish }
data.valueRange = { source.start, source.finish }
if source.name then
data.selectionRange = { source.name.start, source.name.finish }
else
data.selectionRange = { source.start, source.start }
end
if func:getObject() then
data.kind = SymbolKind.Field
else
data.kind = SymbolKind.Function
end
callback(data)
end
local function buildSource(vm, source, used, callback)
if source.type == 'local' then
buildLocal(vm, source, used, callback)
return
end
if source.type == 'set' then
buildSet(vm, source, used, callback)
return
end
if source.type == 'pair' then
buildPair(vm, source, used, callback)
return
end
if source.type == 'localfunction' then
buildLocalFunction(vm, source, used, callback)
return
end
if source.type == 'function' then
buildFunction(vm, source, used, callback)
return
end
end
local function packChild(symbols, finish, kind)
local t
while true do
local symbol = symbols[#symbols]
if not symbol then
break
end
if symbol.valueRange[1] > finish then
break
end
symbols[#symbols] = nil
symbol.children = packChild(symbols, symbol.valueRange[2], symbol.kind)
if not t then
t = {}
end
t[#t+1] = symbol
end
return t
end
local function packSymbols(symbols)
-- 按照start位置反向排序
table.sort(symbols, function (a, b)
return a.range[1] > b.range[1]
end)
-- 处理嵌套
return packChild(symbols, math.maxinteger, SymbolKind.Function)
end
return function (vm)
local symbols = {}
local used = {}
for _, source in ipairs(vm.sources) do
buildSource(vm, source, used, function (data)
symbols[#symbols+1] = data
end)
end
local packedSymbols = packSymbols(symbols)
return packedSymbols
end
|
