Module:沙盒/Economy666/Memory Archive NS

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

local mad = require 'Module:AnotherData'
local songlist = mad.listOf 'songs'
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, 5 do
        getCategory[categories[i]] = i
    end
    
    local songs = {}
    for i = 1, 5 do
        songs[i] = {ns = {}}
    end
    
    local nsId = {}
    local mobileId={}
    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(songlist) do
        if v.set == 'single' and nsId[v.id] then
            mobileId[v.id] = true
            local cate = getCategory[v.category]
            songs[cate].ns[#songs[cate].ns + 1] = v
        end
    end
    
    for i,v in ipairs(songlistNS) do
		if v.set=='single' or mad.packQueryWrap(v.set,'ns')._parentId_=="single" then 
			if not mobileId[v.id] then
				-- 硬编码 gengaozo 等
				songs[4].ns[#songs[4].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, 5 do
        local list = expand(songs[i].ns)
        final['group' .. i] = categoryNames[i]
        final['list' .. i] = list
    end
    
    return frame:expandTemplate{title = '曲目索引', args = final}
end
return p