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
|
local findResult = require 'matcher.find_result'
local hover = require 'matcher.hover'
local CompletionItemKind = {
Text = 1,
Method = 2,
Function = 3,
Constructor = 4,
Field = 5,
Variable = 6,
Class = 7,
Interface = 8,
Module = 9,
Property = 10,
Unit = 11,
Value = 12,
Enum = 13,
Keyword = 14,
Snippet = 15,
Color = 16,
File = 17,
Reference = 18,
Folder = 19,
EnumMember = 20,
Constant = 21,
Struct = 22,
Event = 23,
Operator = 24,
TypeParameter = 25,
}
local function matchKey(me, other)
if me == other then
return false
end
if #me > #other then
return false
end
local lMe = me:lower()
local lOther = other:lower()
if lMe:sub(1, 1) ~= lOther:sub(1, 1) then
return false
end
if lMe == lOther:sub(1, #lMe) then
return true
end
local used = {
[1] = true,
}
local cur = 2
local lookup
local researched
for i = 2, #lMe do
local c = lMe:sub(i, i)
-- 1. 看当前字符是否匹配
if c == lOther:sub(cur, cur) then
used[cur] = true
goto NEXT
end
-- 2. 看前一个字符是否匹配
if not used[cur-1] then
if c == lOther:sub(cur-1, cur-1) then
used[cur-1] = true
goto NEXT
end
end
-- 3. 向后找这个字
lookup = lOther:find(c, cur+1, true)
if lookup then
cur = lookup
used[cur] = true
goto NEXT
end
-- 4. 重新搜索整个字符串，但是只允许1次，否则失败.如果找不到也失败
if researched then
return false
else
researched = true
for j = 2, cur - 2 do
if c == lOther:sub(j, j) then
used[j] = true
goto NEXT
end
end
return false
end
-- 5. 找到下一个可用的字，如果超出长度就算成功
::NEXT::
repeat
cur = cur + 1
until not used[cur]
if cur > #lOther then
break
end
end
return true
end
local function searchLocals(vm, pos, name, callback)
for _, loc in ipairs(vm.results.locals) do
if loc.source.start == 0 then
goto CONTINUE
end
if loc.source.start <= pos and loc.close >= pos then
if matchKey(name, loc.key) then
callback(loc)
end
end
::CONTINUE::
end
end
local function searchFields(name, parent, callback)
if not parent then
return
end
for key, field in pairs(parent.value.child) do
if type(key) ~= 'string' then
goto CONTINUE
end
if matchKey(name, key) then
callback(field)
end
::CONTINUE::
end
end
local function getKind(var, default)
local value = var.value
if default == CompletionItemKind.Variable then
if value.type == 'function' then
return CompletionItemKind.Function
end
end
if default == CompletionItemKind.Field then
local tp = type(value.value)
if tp == 'number' or tp == 'integer' or tp == 'string' then
return CompletionItemKind.Enum
end
if value.type == 'function' then
if var.parent and var.parent.value and var.parent.value.ENV ~= true then
return CompletionItemKind.Method
else
return CompletionItemKind.Function
end
end
end
return default
end
local function getDetail(var)
local tp = type(var.value.value)
if tp == 'boolean' or tp == 'number' or tp == 'integer' or tp == 'string' then
return ('%s = %q'):format(var.key, var.value.value)
end
return nil
end
local function getDocument(var, source)
if var.value.type == 'function' then
return {
kind = 'markdown',
value = hover(var, source),
}
end
return nil
end
return function (vm, pos)
local result, source = findResult(vm, pos)
if not result then
return nil
end
local list = {}
local mark = {}
if result.type == 'local' then
searchLocals(vm, pos, result.key, function (var)
if mark[var.key] then
return
end
mark[var.key] = true
list[#list+1] = {
label = var.key,
kind = getKind(var, CompletionItemKind.Variable),
detail = getDetail(var),
documentation = getDocument(var, source),
}
end)
-- 也尝试搜索全局变量
searchFields(result.key, vm.results.locals[1], function (var)
if mark[var.key] then
return
end
mark[var.key] = true
list[#list+1] = {
label = var.key,
kind = getKind(var, CompletionItemKind.Field),
detail = getDetail(var),
documentation = getDocument(var, source),
}
end)
elseif result.type == 'field' then
if result.parent and result.parent.value and result.parent.value.ENV == true then
-- 全局变量也搜索是不是local
searchLocals(vm, pos, result.key, function (var)
if mark[var.key] then
return
end
mark[var.key] = true
list[#list+1] = {
label = var.key,
kind = getKind(var, CompletionItemKind.Variable),
detail = getDetail(var),
documentation = getDocument(var, source),
}
end)
end
searchFields(result.key, result.parent, function (var)
if mark[var.key] then
return
end
mark[var.key] = true
list[#list+1] = {
label = var.key,
kind = getKind(var, CompletionItemKind.Field),
detail = getDetail(var),
documentation = getDocument(var, source),
}
end)
end
return list
end
|
