script/vm/generic-manager.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

local createGeneric = require 'vm.generic'
local compiler      = require 'vm.compiler'
local globalMgr     = require 'vm.global-manager'
local guide         = require 'parser.guide'

---@class vm.node.generic-manager
---@field parent   parser.object
---@field signList parser.object[]
local mt = {}
mt.__index = mt
mt.type = 'generic-manager'

---@param key parser.object
function mt:addSign(key)
    self.signList[#self.signList+1] = key
end

---@param node vm.node
function mt:getChild(node)
    local generic = createGeneric(self, node)
    return generic
end

---@param argNodes vm.node[]
---@return table<string, vm.node>
function mt:resolve(argNodes)
    local resolved = {}
    for i, node in ipairs(argNodes) do
        local sign = self.signList[i]
        if not sign then
            break
        end
        for _, typeUnit in ipairs(sign.types) do
            if typeUnit.type == 'doc.generic.name' then
                local key = typeUnit[1]
                if typeUnit.literal then
                    for n in compiler.eachNode(node) do
                        if n.type == 'string' then
                            local type = globalMgr.declareGlobal('type', n[1], guide.getUri(n))
                            resolved[key] = compiler.mergeNode(type, resolved[key])
                        end
                    end
                else
                    resolved[key] = compiler.mergeNode(node, resolved[key])
                end
            end
        end
    end
    return resolved
end

---@return vm.node.generic-manager
return function (parent)
    local genericMgr = setmetatable({
        parent   = parent,
        signList = {},
    }, mt)
    return genericMgr
end