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
|
local searcher = require 'core.searcher'
local guide = require 'parser.guide'
local files = require 'files'
local vm = require 'vm'
local findSource = require 'core.find-source'
local function sortResults(results)
-- 先按照顺序排序
table.sort(results, function (a, b)
local u1 = searcher.getUri(a.target)
local u2 = searcher.getUri(b.target)
if u1 == u2 then
return a.target.start < b.target.start
else
return u1 < u2
end
end)
-- 如果2个结果处于嵌套状态，则取范围小的那个
local lf, lu
for i = #results, 1, -1 do
local res = results[i].target
local f = res.finish
local uri = searcher.getUri(res)
if lf and f > lf and uri == lu then
table.remove(results, i)
else
lu = uri
lf = f
end
end
end
local accept = {
['local'] = true,
['setlocal'] = true,
['getlocal'] = true,
['label'] = true,
['goto'] = true,
['field'] = true,
['method'] = true,
['setindex'] = true,
['getindex'] = true,
['tableindex'] = true,
['setglobal'] = true,
['getglobal'] = true,
['function'] = true,
['...'] = true,
['doc.type.name'] = true,
['doc.class.name'] = true,
['doc.extends.name'] = true,
['doc.alias.name'] = true,
}
return function (uri, offset)
local ast = files.getAst(uri)
if not ast then
return nil
end
local source = findSource(ast, offset, accept)
if not source then
return nil
end
local metaSource = vm.isMetaFile(uri)
local results = {}
for _, src in ipairs(vm.getRefs(source, 5)) do
if src.dummy then
goto CONTINUE
end
local root = guide.getRoot(src)
if not root then
goto CONTINUE
end
if not metaSource and vm.isMetaFile(root.uri) then
goto CONTINUE
end
if ( src.type == 'doc.class.name'
or src.type == 'doc.type.name'
)
and source.type ~= 'doc.type.name'
and source.type ~= 'doc.class.name' then
goto CONTINUE
end
if src.type == 'setfield'
or src.type == 'getfield'
or src.type == 'tablefield' then
src = src.field
elseif src.type == 'setindex'
or src.type == 'getindex'
or src.type == 'tableindex' then
src = src.index
elseif src.type == 'getmethod'
or src.type == 'setmethod' then
src = src.method
elseif src.type == 'table' and src.parent.type ~= 'return' then
goto CONTINUE
end
local ouri = files.getOriginUri(root.uri)
if not ouri then
goto CONTINUE
end
results[#results+1] = {
target = src,
uri = ouri,
}
::CONTINUE::
end
if #results == 0 then
return nil
end
sortResults(results)
return results
end
|
