70 lines
1.7 KiB
Lua
70 lines
1.7 KiB
Lua
local vm = require 'vm'
|
|
|
|
local function getLiterals(arg)
|
|
local literals = vm.getLiterals(arg)
|
|
local res = {}
|
|
if not literals then
|
|
return res
|
|
end
|
|
for k, v in pairs(literals) do
|
|
if type(k) == 'string' then
|
|
res[#res+1] = k
|
|
end
|
|
end
|
|
return res
|
|
end
|
|
|
|
---@return string[]?
|
|
local function getCode(CdefReference)
|
|
local target = CdefReference.target
|
|
if not (target.type == 'field' and target.parent.type == 'getfield') then
|
|
return
|
|
end
|
|
target = target.parent.parent
|
|
if target.type == 'call' then
|
|
return getLiterals(target.args and target.args[1])
|
|
elseif target.type == 'local' then
|
|
local res = {}
|
|
for _, o in ipairs(target.ref) do
|
|
if o.parent.type ~= 'call' then
|
|
goto CONTINUE
|
|
end
|
|
local target = o.parent
|
|
local literals = vm.getLiterals(target.args and target.args[1])
|
|
if not literals then
|
|
goto CONTINUE
|
|
end
|
|
for k, v in pairs(literals) do
|
|
if type(k) == 'string' then
|
|
res[#res+1] = k
|
|
end
|
|
end
|
|
::CONTINUE::
|
|
end
|
|
return res
|
|
end
|
|
end
|
|
|
|
---@async
|
|
return function (CdefReference, target_uri)
|
|
if not CdefReference then
|
|
return nil
|
|
end
|
|
local codeResults
|
|
for i, v in ipairs(CdefReference) do
|
|
if v.uri ~= target_uri then
|
|
goto continue
|
|
end
|
|
local codes = getCode(v)
|
|
if not codes then
|
|
goto continue
|
|
end
|
|
for i, v in ipairs(codes) do
|
|
codeResults = codeResults or {}
|
|
codeResults[#codeResults+1] = v
|
|
end
|
|
::continue::
|
|
end
|
|
return codeResults
|
|
end
|