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
|
local core = require 'core'
local parser = require 'parser'
local buildVM = require 'vm'
local function catch_target(script)
local list = {}
local cur = 1
while true do
local start, finish = script:find('<[!?].-[!?]>', cur)
if not start then
break
end
list[#list+1] = { start + 2, finish - 2 }
cur = finish + 1
end
return list
end
local function founded(targets, results)
if #targets ~= #results then
return false
end
for _, target in ipairs(targets) do
for _, result in ipairs(results) do
if target[1] == result[1] and target[2] == result[2] then
goto NEXT
end
end
do return false end
::NEXT::
end
return true
end
function TEST(script)
local target = catch_target(script)
local start = script:find('<?', 1, true)
local finish = script:find('?>', 1, true)
local pos = (start + finish) // 2 + 1
local new_script = script:gsub('<[!?]', ' '):gsub('[!?]>', ' ')
local ast = parser:parse(new_script, 'lua', 'Lua 5.3')
assert(ast)
local vm = buildVM(ast)
assert(vm)
local positions = core.definition(vm, pos, 'reference')
if positions then
assert(founded(target, positions))
else
assert(#target == 0)
end
end
TEST [[
local <?a?> = 1
<!a!> = <!a!>
]]
TEST [[
t.<?a?> = 1
t.<!a!> = t.<!a!>
]]
TEST [[
:: <!LABEL!> ::
goto <?LABEL?>
]]
TEST [[
local a = 1
local <?a?> = 1
<!a!> = <!a!>
]]
TEST [[
local t = {
<?a?> = 1
}
print(t.<!a!>)
]]
TEST [[
local <!mt!> = {}
function <!mt!>:a()
<?self?>:remove()
end
]]
TEST [[
local function f()
return <?function ()
end?>
end
local <!f2!> = f()
]]
TEST [[
table.<!dump!>()
function table.<?dump?>()
end
]]
TEST [[
---@class <!Class!>
---@type <?Class?>
---@type <!Class!>
]]
TEST [[
---@class <?Class?>
---@type <!Class!>
---@type <!Class!>
]]
|
