查看“Module:Memory Archive”的源代码

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]={both={}, mobile={}, 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]=1
		end
	end
	for i,v in ipairs(songlist) do
		if v.set=='single' then
			mobileId[v.id]=1
			local cate=getCategory[v.category]
			if nsId[v.id] then
				songs[cate].both[#songs[cate].both+1]=v
			else
				songs[cate].mobile[#songs[cate].mobile+1]=v
			end
		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=''
		local first=true
		if #songs[i].both ~= 0 then
			if not first then list = list .. dot end
			list = list .. expand(songs[i].both)
			first=false
		end
		if #songs[i].mobile ~= 0 then
			if not first then list = list .. dot end
			list = list .. frame:expandTemplate{title = '仅',args={"移动版",expand(songs[i].mobile)}}
			first=false
		end
		if #songs[i].ns ~= 0 then
			if not first then list = list .. dot end
			list = list .. frame:expandTemplate{title = '仅',args={"NS版",expand(songs[i].ns)}}
			first=false
		end
		final['group' .. i] = categoryNames[i]
		final['list' .. i] = list
	end
	return frame:expandTemplate{title = '曲目索引',args=final}
end
return p