summaryrefslogtreecommitdiff
path: root/script/core/collector.lua
blob: 2f29fc37ab3f168e59e8fde043c4ca0dad525cd3 (plain)
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
local scope = require 'workspace.scope'
local ws    = require 'workspace'

local collect    = {}
local subscribed = {}

local m = {}

--- 订阅一个名字
---@param uri uri
---@param name string
---@param value any
function m.subscribe(uri, name, value)
    -- 订阅部分
    local uriSubscribed = subscribed[uri]
    if not uriSubscribed then
        uriSubscribed = {}
        subscribed[uri] = uriSubscribed
    end
    uriSubscribed[name] = true
    -- 收集部分
    local nameCollect = collect[name]
    if not nameCollect then
        nameCollect = {}
        collect[name] = nameCollect
    end
    if value == nil then
        value = true
    end
    nameCollect[uri] = value
end

--- 丢弃掉某个 uri 中收集的所有信息
---@param uri uri
function m.dropUri(uri)
    local uriSubscribed = subscribed[uri]
    if not uriSubscribed then
        return
    end
    subscribed[uri] = nil
    for name in pairs(uriSubscribed) do
        collect[name][uri] = nil
    end
end

--- 是否包含某个名字的订阅
---@param name string
---@return boolean
function m.has(name)
    local nameCollect = collect[name]
    if not nameCollect then
        return false
    end
    if next(nameCollect) == nil then
        collect[name] = nil
        return false
    end
    return true
end

local DUMMY_FUNCTION = function () end

--- 迭代某个名字的订阅
---@param uri  uri
---@param name string
function m.each(uri, name)
    local nameCollect = collect[name]
    if not nameCollect then
        return DUMMY_FUNCTION
    end
    ---@type scope
    local scp = scope.getFolder(uri)
            or  scope.getLinkedScope(uri)
            or  scope.fallback

    local curi, value
    local function getNext()
        curi, value = next(nameCollect, curi)
        if not curi then
            return nil, nil
        end
        if  not scp:isChildUri(curi)
        and not scp:isLinkedUri(curi) then
            return getNext()
        end

        return value, curi
    end
    return getNext
end

--- 迭代某个名字的引用订阅
---@param uri  uri
---@param name string
function m.eachRef(uri, name)
    local nameCollect = collect[name]
    if not nameCollect then
        return DUMMY_FUNCTION
    end
    ---@type scope
    if scope.getFolder(uri) then
        return m.each(uri, name)
    end

    local curi, value
    local function getNext()
        curi, value = next(nameCollect, curi)
        if not curi then
            return nil, nil
        end

        return value, curi
    end
    return getNext
end
return m