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
|
local findSource = require 'core.find_source'
local parser = require 'parser'
local DocumentHighlightKind = {
Text = 1,
Read = 2,
Write = 3,
}
local function parseResult(source)
local positions = {}
if source:bindLabel() then
source:bindLabel():eachInfo(function (info, src)
positions[#positions+1] = { src.start, src.finish, DocumentHighlightKind.Text }
end)
return positions
end
if source:bindLocal() then
local loc = source:bindLocal()
local mark = {}
loc:eachInfo(function (info, src)
if not mark[src] then
mark[src] = info
positions[#positions+1] = { src.start, src.finish, DocumentHighlightKind.Text }
end
end)
return positions
end
if source:bindValue() then
local parent = source:get 'parent'
local mark = {}
parent:eachInfo(function (info, src)
if not mark[src] then
mark[src] = info
if info.type == 'get child' or info.type == 'set child' then
if info[1] == source[1] then
positions[#positions+1] = {src.start, src.finish, DocumentHighlightKind.Text}
end
end
end
end)
return positions
end
return nil
end
return function (vm, pos)
local source = findSource(vm, pos)
if not source then
return nil
end
local positions = parseResult(source)
return positions
end
|
