Module:沙盒/Economy666/Memory Archive NS

可在Module:沙盒/Economy666/Memory Archive NS/doc创建此模块的帮助文档

local mad = require 'Module:AnotherData'
local songlistNS = mad.listOf('songs', 'ns')

local p = {}
function p.main(frame)
    local dot = frame:expandTemplate {title = '·'}
    
    local categories = {'poprec', 'partner', 'original', 'variety', 'musicgames', '其他'}
    local getCategory = {}
    for i = 1, 6 do
        getCategory[categories[i]] = i
    end
    
    local songs = {}
    for i = 1, 6 do
        songs[i] = {ns = {}}
    end
    
    local nsId = {}
    for i, v in ipairs(songlistNS) do
        if v.set == 'single' or mad.packQueryWrap(v.set, 'ns')._parentId_ == "single" then
            nsId[v.id] = true
        end
    end
    
    for i, v in ipairs(songlistNS) do
        if v.set == 'single' and nsId[v.id] then
            local cate = getCategory[v.category]
            if not cate then
                cate = 6  -- 将没有分类的歌曲放入 "其他" 分类
            end
            if cate then
                songs[cate].ns[#songs[cate].ns + 1] = v
            end
        end
    end
    
    local categoryNames = {'流行/推荐', '搭档', '原创', '综合', '音乐游戏', '其他'}
    local final = {}
    for i, v in pairs(frame.args) do
        final[i] = v
    end

    function expand(t)
        local res = ''
        for i, v in ipairs(t) do
            if i ~= 1 then
                res = res .. dot
            end
            res = res .. '[[' .. mad.linkTitle(v) .. ']]'
        end
        return res
    end

    for i = 1, 6 do
        local list = expand(songs[i].ns)
        final['group' .. i] = categoryNames[i]
        final['list' .. i] = list
    end
    
    return frame:expandTemplate{title = 'NS版曲目索引', args = final}
end
return p