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
|
local rpc = require 'rpc'
local TokenTypes = require 'constant.TokenTypes'
local TokenModifiers = require 'constant.TokenModifiers'
local isEnable = false
local function toArray(map)
local array = {}
for k in pairs(map) do
array[#array+1] = k
end
table.sort(array, function (a, b)
return map[a] < map[b]
end)
return array
end
local function enable(lsp)
if isEnable then
return
end
if not lsp.client.capabilities.textDocument.semanticTokens then
return
end
isEnable = true
log.debug('Enable semantic.')
rpc:request('client/registerCapability', {
registrations = {
{
id = 'semantic',
method = 'textDocument/semanticTokens',
registerOptions = {
legend = {
tokenTypes = toArray(TokenTypes),
tokenModifiers = toArray(TokenModifiers),
},
range = false,
full = true,
},
},
}
})
end
local function disable()
if not isEnable then
return
end
isEnable = false
log.debug('Disable semantic.')
rpc:request('client/unregisterCapability', {
unregisterations = {
{
id = 'semantic',
method = 'textDocument/semanticTokens',
},
}
})
end
return {
enable = enable,
disable = disable,
}
|