模块:遗迹目录

此模块的文档可以在模块:遗迹目录/doc创建

-- Module:遗迹目录
--- 本模板为遗迹生成目录，样式为卡片集合。详见[[Template:遗迹目录]]
local p = {}
local fstr = mw.ustring.format -- shortcut for formattig a string
local utils = require([[Module:Utils]])
local po = require([[Module:Po]]).po
local getArgs = require('Module:Dev/Arguments').getArgs
local grData = mw.loadData([[Module:Data/GravitasEntities]])

local specialArgs = {
    sort = true,
    width = true,
    debug = true
}
local argvMap = {
    ["true"] = true,
    ["false"] = false
}

local function isMatchFilter(args, itemData)
    -- 根据数据模块的字段筛选
    for argk, argv in pairs(args) do
        local valid = specialArgs[argk] ~= true
        local mappedArgv = argvMap[argv] or argv
        if valid and (itemData[argk] == nil or itemData[argk] ~= mappedArgv) then
            return false
        end
    end
    return true
end

local function getDlcIcon(dlcs)
    -- DLC图标
    local out = {}
    local isBase = false
    for _, dlc in ipairs(dlcs) do
        if "" == dlc then
            isBase = true
        end
    end
    for _, dlc in ipairs(dlcs) do
        if "DLC2_ID" == dlc then
            table.insert(out, utils.DLC_ICONS[dlc])
        elseif not isBase and "EXPANSION1_ID" == dlc then
            table.insert(out, utils.DLC_ICONS[dlc])
        end
    end
    return table.concat(out, "")
end

function p.getInfo(args)
    local infos = {}

    for k, itemData in pairs(grData) do
        if itemData.Deprecated then
            -- do nothing
        elseif isMatchFilter(args, itemData) then
            table.insert(infos, itemData)
        end
    end

    -- 排序类型
    if args.sort ~= nil and #infos > 0 then
        local sortField = args.sort
        if infos[1][sortField] == nil then
            error(fstr('"args.sort = %s" is not supported', sortField))
        end
        table.sort(infos, function(a, b)
            return a[sortField] < b[sortField]
        end)
    end

    return infos
end

-- test by: p.main(require("Module:debug").frame({},{debug=1}))
-- test by: p.main(require("Module:debug").frame({},{sort="id", debug=1}))
function p.main(frame)
    local args = getArgs(frame)

    -- 筛选数据
    local infos = p.getInfo(args)

    local out = {}
    for _, itemData in ipairs(infos) do
        local name = po(itemData.Name)
        table.insert(out, frame:expandTemplate{
            title = "Template:卡片",
            args = {
                name,
                dlc = getDlcIcon(itemData.dlcIds),
                width = args.width
            }
        })
    end

    if args.debug then
        mw.logObject(out, "out")
    end

    return mw.getCurrentFrame():preprocess(table.concat(out, ""))
end

return p