blob: b55c1ab56b84fcaab44d7144058fc672b75c5b60 (
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
|
local union = require 'vm.union'
---@alias vm.node parser.object | vm.node.union | vm.node.global | vm.generic
---@class vm.node-manager
local m = {}
local DUMMY_FUNCTION = function () end
---@type table<parser.object, vm.node>
m.nodeCache = {}
---@param a vm.node
---@param b vm.node
function m.mergeNode(a, b)
if not b then
return a
end
if not a then
return b
end
return union(a, b)
end
function m.setNode(source, node)
if not node then
return
end
local me = m.nodeCache[source]
if not me then
m.nodeCache[source] = node
return
end
if me == node then
return
end
m.nodeCache[source] = m.mergeNode(me, node)
end
function m.getNode(source)
return m.nodeCache[source]
end
---@param node vm.node
---@return vm.node.union
function m.addOptional(node)
if not node or node.type ~= 'union' then
node = union(node)
end
node = node:addOptional()
return node
end
---@param node vm.node
---@return vm.node.union?
function m.removeOptional(node)
if not node then
return node
end
if node.type ~= 'union' then
node = union(node)
end
node = node:removeOptional()
return node
end
---@return fun():vm.node
function m.eachNode(node)
if not node then
return DUMMY_FUNCTION
end
if node.type == 'union' then
return node:eachNode()
end
local first = true
return function ()
if first then
first = false
return node
end
return nil
end
end
function m.clearNodeCache()
m.nodeCache = {}
end
return m
|