summaryrefslogtreecommitdiff
path: root/server/src/matcher/find_lib.lua
blob: e91568fc0a3222f7e96f62254ff1f88953cbcf06 (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
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
local lni = require 'lni'
local fs = require 'bee.filesystem'

local function mergeEnum(lib, locale)
    if not lib or not locale then
        return
    end
    local pack = {}
    for _, enum in ipairs(lib) do
        if enum.enum then
            pack[enum.enum] = enum
        end
    end
    for _, enum in ipairs(locale) do
        if pack[enum.enum] then
            pack[enum.enum].description = enum.description
        end
    end
end

local function mergeField(lib, locale)
    if not lib or not locale then
        return
    end
    local pack = {}
    for _, field in ipairs(lib) do
        if field.field then
            pack[field.field] = field
        end
    end
    for _, field in ipairs(locale) do
        if pack[field.field] then
            pack[field.field].description = field.description
        end
    end
end

local function mergeLocale(libs, locale)
    if not libs or not locale then
        return
    end
    for name in pairs(locale) do
        if libs[name] then
            libs[name].description = locale[name].description
            mergeEnum(libs[name].enums, locale[name].enums)
            mergeField(libs[name].fields, locale[name].fields)
        end
    end
end

local function mergeLibs(target, libs)
    if not libs then
        return
    end
    for name, lib in pairs(libs) do
        target.names[#target.names+1] = name
        target.libs[#target.libs+1] = lib
    end
end

local function loadLocale(language, relative)
    local localePath = ROOT / 'locale' / language / relative
    local localeBuf = io.load(localePath)
    if localeBuf then
        local locale = table.container()
        xpcall(lni.classics, log.error, localeBuf, localePath:string(), {locale})
        return locale
    end
    return nil
end

local Libs
local function getLibs()
    if Libs then
        return Libs
    end
    local language = require 'language'
    Libs = setmetatable({
        names = {},
        libs = {},
    }, {
        __pairs = function (self)
            local i = 0
            return function ()
                i = i + 1
                return self.names[i], self.libs[i]
            end
        end,
    })
    for path in io.scan(ROOT / 'libs') do
        local libs
        local buf = io.load(path)
        if buf then
            libs = table.container()
            xpcall(lni.classics, log.error, buf, path:string(), {libs})
        end
        local relative = fs.relative(path, ROOT)

        local locale = loadLocale('en-US', relative)
        mergeLocale(libs, locale)
        if language ~= 'en-US' then
            locale = loadLocale(language, relative)
            mergeLocale(libs, locale)
        end
        mergeLibs(Libs, libs)
    end

    return Libs
end

local function isGlobal(var)
    if var.type ~= 'field' then
        return false
    end
    if not var.parent then
        return false
    end
    return var.parent.key == '_ENV' or var.parent.key == '_G'
end

local function checkSourceAsGlobal(value, name)
    if value.key == name and isGlobal(value) then
        return name
    end
    return nil
end

local function checkSourceAsLibrary(value, name)
    if value.type ~= 'lib' then
        return nil
    end
    if value.name == name then
        return name
    end
    return nil
end

local function checkSource(value, name, lib)
    if not lib.source then
        return checkSourceAsGlobal(value, name)
    end
    for _, source in ipairs(lib.source) do
        if source.type == 'global' then
            local fullKey = checkSourceAsGlobal(value, name)
            if fullKey then
                return fullKey
            end
        elseif source.type == 'library' then
            local fullKey = checkSourceAsLibrary(value, name)
            if fullKey then
                return fullKey
            end
        end
    end
    return nil
end

local function checkParentAsGlobal(parentValue, name, parent)
    local parentName = checkSourceAsGlobal(parentValue, parent.name)
    if not parentName then
        return nil
    end
    return ('%s.%s'):format(parentName, name)
end

local function checkParentAsLibrary(parentValue, name, parent)
    local parentName = checkSourceAsLibrary(parentValue, parent.name)
    if not parentName then
        return nil
    end
    return ('%s.%s'):format(parentName, name)
end

local function checkParent(value, name, lib)
    if not lib.parent then
        return nil
    end
    if name ~= value.key then
        return nil
    end
    local parentValue = value.parent
    if not parentValue then
        return nil
    end
    parentValue = parentValue.value or parentValue
    for _, parent in ipairs(lib.parent) do
        if parent.type == 'global' then
            local fullKey = checkParentAsGlobal(parentValue, name, parent)
            if fullKey then
                return fullKey
            end
        elseif parent.type == 'library' then
            local fullKey = checkParentAsLibrary(parentValue, name, parent)
            if fullKey then
                return fullKey
            end
        end
    end
    return nil
end

local function findLib(var, libs)
    local value = var.value or var
    for name, lib in pairs(libs) do
        local fullKey = checkSource(value, name, lib)
                     or checkParent(value, name, lib)
        if fullKey then
            return lib, fullKey
        end
    end
    return nil, nil
end

return function (var)
    local libs = getLibs()
    local lib, fullKey = findLib(var, libs)
    return lib, fullKey
end