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
|
local proto = require 'proto'
local define = require 'proto.define'
local client = require 'client'
local json = require "json"
local config = require 'config'
local lang = require 'language'
local nonil = require 'without-check-nil'
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 dontShowAgain = false
local function enable(uri)
if isEnable then
return
end
nonil.enable()
if not client.info.capabilities.textDocument.semanticTokens.dynamicRegistration then
return
end
nonil.disable()
isEnable = true
log.debug('Enable semantic tokens.')
proto.request('client/registerCapability', {
registrations = {
{
id = 'semantic-tokens',
method = 'textDocument/semanticTokens',
registerOptions = {
legend = {
tokenTypes = toArray(define.TokenTypes),
tokenModifiers = toArray(define.TokenModifiers),
},
range = true,
full = false,
},
},
}
})
if config.get(uri, 'editor.semanticHighlighting.enabled') == 'configuredByTheme' and not dontShowAgain then
proto.request('window/showMessageRequest', {
type = define.MessageType.Info,
message = lang.script.WINDOW_CHECK_SEMANTIC,
actions = {
{
title = lang.script.WINDOW_APPLY_SETTING,
},
{
title = lang.script.WINDOW_DONT_SHOW_AGAIN,
},
}
}, function (item)
if not item then
return
end
if item.title == lang.script.WINDOW_APPLY_SETTING then
client.setConfig {
{
key = 'editor.semanticHighlighting.enabled',
action = 'set',
value = true,
global = true,
}
}
end
if item.title == lang.script.WINDOW_DONT_SHOW_AGAIN then
dontShowAgain = true
end
end)
end
end
local function disable(uri)
if not isEnable then
return
end
nonil.enable()
if not client.info.capabilities.textDocument.semanticTokens.dynamicRegistration then
return
end
nonil.disable()
isEnable = false
log.debug('Disable semantic tokens.')
proto.request('client/unregisterCapability', {
unregisterations = {
{
id = 'semantic-tokens',
method = 'textDocument/semanticTokens',
},
}
})
end
local function refresh()
if not isEnable then
return
end
log.debug('Refresh semantic tokens.')
proto.request('workspace/semanticTokens/refresh', json.null)
end
config.watch(function (uri, key, value, oldValue)
if key == '' then
key = 'Lua.color.mode'
value = config.get(uri, key)
end
if key == 'Lua.color.mode' then
if value == 'Semantic' or value == 'SemanticEnhanced' then
if isEnable and value ~= oldValue then
refresh()
else
enable(uri)
end
else
disable(uri)
end
end
if key:find '^Lua.runtime'
or key:find '^Lua.workspace'
or key:find '^files' then
refresh()
end
end)
return {
enable = enable,
disable = disable,
refresh = refresh,
}
|
