script/vm/getLinks.lua


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

local guide = require 'parser.guide'
---@diagnostic disable-next-line
---@class vm
local vm    = require 'vm.vm'
local files = require 'files'

local function getFileLinks(uri)
    local ws    = require 'workspace'
    local links = {}
    local ast = files.getState(uri)
    if not ast then
        return links
    end
    tracy.ZoneBeginN('getFileLinks')
    guide.eachSpecialOf(ast.ast, 'require', function (source)
        local call = source.parent
        if not call or call.type ~= 'call' then
            return
        end
        local args = call.args
        if not args or not args[1] or args[1].type ~= 'string' then
            return
        end
        local uris = ws.findUrisByRequirePath(args[1][1])
        for _, u in ipairs(uris) do
            u = files.asKey(u)
            if not links[u] then
                links[u] = {}
            end
            links[u][#links[u]+1] = call
        end
    end)
    tracy.ZoneEnd()
    return links
end

local function getFileLinksOrCache(uri)
    local cache = files.getCache(uri)
    cache.links = cache.links or getFileLinks(uri)
    return cache.links
end

local function getLinksTo(uri)
    uri = files.asKey(uri)
    local links = {}
    for u in files.eachFile() do
        local ls = getFileLinksOrCache(u)
        if ls[uri] then
            for _, l in ipairs(ls[uri]) do
                links[#links+1] = l
            end
        end
    end
    return links
end

-- 获取所有 require(uri) 的文件
function vm.getLinksTo(uri)
    local cache = vm.getCache('getLinksTo')[uri]
    if cache ~= nil then
        return cache
    end
    cache = getLinksTo(uri)
    vm.getCache('getLinksTo')[uri] = cache
    return cache
end