summaryrefslogtreecommitdiff
path: root/script-beta/library.lua
blob: 6776a796285f629757532d5df69512104a4ce4ea (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
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
local lni    = require 'lni'
local fs     = require 'bee.filesystem'
local config = require 'config'
local util   = require 'utility'

local m = {}

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
        if enum.code then
            pack[enum.code] = enum
        end
    end
    for _, enum in ipairs(locale) do
        if pack[enum.enum] then
            if enum.description then
                pack[enum.enum].description = enum.description
            end
        end
        if pack[enum.code] then
            if enum.description then
                pack[enum.code].description = enum.description
            end
        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
            if field.description then
                pack[field.field].description = field.description
            end
        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
            if locale[name].description then
                libs[name].description = locale[name].description
            end
            mergeEnum(libs[name].enums, locale[name].enums)
            mergeField(libs[name].fields, locale[name].fields)
        end
    end
end

local function isMatchVersion(version)
    if not version then
        return true
    end
    local runtimeVersion = config.config.runtime.version
    if type(version) == 'table' then
        for i = 1, #version do
            if version[i] == runtimeVersion then
                return true
            end
        end
    else
        if version == runtimeVersion then
            return true
        end
    end
    return false
end

local function insertGlobal(tbl, key, value)
    if not isMatchVersion(value.version) then
        return false
    end
    if not value.doc then
        value.doc = key
    end
    tbl[key] = {
        type  = 'library',
        name  = key,
        child = {},
        value = value,
    }
    return true
end

local function insertOther(tbl, key, value)
    if not value.version then
        return
    end
    if not tbl[key] then
        tbl[key] = {}
    end
    if type(value.version) == 'string' then
        tbl[key][#tbl[key]+1] = value.version
    elseif type(value.version) == 'table' then
        for _, version in ipairs(value.version) do
            if type(version) == 'string' then
                tbl[key][#tbl[key]+1] = version
            end
        end
    end
    table.sort(tbl[key])
end

local function insertCustom(tbl, key, value, libName)
    if not tbl[key] then
        tbl[key] = {}
    end
    tbl[key][#tbl[key]+1] = libName
    table.sort(tbl[key])
end

local function isEnableGlobal(libName)
    if config.config.runtime.library[libName] then
        return true
    end
    if libName:sub(1, 1) == '@' then
        return true
    end
    return false
end

local function mergeSource(alllibs, name, lib, libName)
    if not lib.source then
        if isEnableGlobal(libName) then
            local suc = insertGlobal(alllibs.global, name, lib)
            if not suc then
                insertOther(alllibs.other, name, lib)
            end
        else
            insertCustom(alllibs.custom, name, lib, libName)
        end
        return
    end
    for _, source in ipairs(lib.source) do
        local sourceName = source.name or name
        if source.type == 'global' then
            if isEnableGlobal(libName) then
                local suc = insertGlobal(alllibs.global, sourceName, lib)
                if not suc then
                    insertOther(alllibs.other, sourceName, lib)
                end
            else
                insertCustom(alllibs.custom, sourceName, lib, libName)
            end
        elseif source.type == 'library' then
            insertGlobal(alllibs.library, sourceName, lib)
        elseif source.type == 'object' then
            insertGlobal(alllibs.object, sourceName, lib)
        end
    end
end

local function copy(t)
    local new = {}
    for k, v in pairs(t) do
        new[k] = v
    end
    return new
end

local function insertChild(tbl, name, key, value)
    if not name or not key then
        return
    end
    if not isMatchVersion(value.version) then
        return
    end
    if not value.doc then
        value.doc = ('%s.%s'):format(name, key)
    end
    if not tbl[name] then
        tbl[name] = {
            type = 'library',
            name = name,
            child = {},
        }
    end
    tbl[name].child[key] = {
        type  = 'library',
        name  = key,
        value = value,
    }
end

local function mergeParent(alllibs, name, lib, libName)
    for _, parent in ipairs(lib.parent) do
        if parent.type == 'global' then
            if isEnableGlobal(libName) then
                insertChild(alllibs.global, parent.name, name, lib)
            end
        elseif parent.type == 'library' then
            insertChild(alllibs.library, parent.name, name, lib)
        elseif parent.type == 'object' then
            insertChild(alllibs.object,  parent.name, name, lib)
        end
    end
end

local function mergeLibs(alllibs, libs, libName)
    if not libs then
        return
    end
    for _, lib in pairs(libs) do
        if lib.parent then
            mergeParent(alllibs, lib.name, lib, libName)
        else
            mergeSource(alllibs, lib.name, lib, libName)
        end
    end
end

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

local function fix(libs)
    for name, lib in pairs(libs) do
        lib.name = lib.name or name
    end
end

local function scan(path)
    local result = {path}
    local i = 0
    return function ()
        i = i + 1
        local current = result[i]
        if not current then
            return nil
        end
        if fs.is_directory(current) then
            for path in current:list_directory() do
                result[#result+1] = path
            end
        end
        return current
    end
end

local function markLibrary(library)
    for _, lib in pairs(library) do
        lib.fields  = {}
        if lib.child then
            for _, child in util.sortPairs(lib.child) do
                table.insert(lib.fields, child)
                child.parent = library
            end
        end
    end
end

local function init()
    local lang = require 'language'
    local id = lang.id
    m.global  = util.container()
    m.library = util.container()
    m.object  = util.container()
    m.other   = util.container()
    m.custom  = util.container()

    for libPath in (ROOT / 'libs'):list_directory() do
        local libName = libPath:filename():string()
        for path in scan(libPath) do
            local libs
            local buf = util.loadFile(path:string())
            if buf then
                libs = util.container()
                xpcall(lni, log.error, buf, path:string(), {libs})
                fix(libs)
            end
            local relative = fs.relative(path, ROOT)

            local locale = loadLocale('en-US', relative)
            mergeLocale(libs, locale)
            if id ~= 'en-US' then
                locale = loadLocale(id, relative)
                mergeLocale(libs, locale)
            end
            mergeLibs(m, libs, libName)
        end
    end

    markLibrary(m.global)
    markLibrary(m.library)
    markLibrary(m.object)
    markLibrary(m.other)
    markLibrary(m.custom)
end

function m.reload()
    init()
end

init()

return m