查看“Module:Unlock”的源代码

local mad = require 'Module:AnotherData'
local p = {}

local linkPool = {}
local mLink = {}
local mSong = {}
for _, song in ipairs(mad.listOf 'songs') do
	local query = mad.songQueryWrap(song)
	local link = query.linkTitle
	linkPool[link] = true
	mSong[song.id] = song
	mLink[song.id] = link
end

function p.world(frame)
	local world = mw.title.new('世界模式地图详表 (移动版' .. frame.args[1] .. ')'):getContent()
	local res = {}
	for name, wtb in world:gmatch '== *([^B].-) *==[^={]+(%b{})' do
		local tot
		for stair, reward in wtb:gmatch '| *(%d+)[0-9()%-| ]+(%b[])' do
			if linkPool[reward:sub(3, -3)] then
				tot = tot or tonumber(wtb:match '| *(%d+)[^\n]+\n|%-\n| *总计') - 1
				table.insert(res, ('|-\n|%s||%s||%d/%d'):format(reward, name, tonumber(stair) - 1, tot))
			end
		end
	end
	return table.concat(res, '\n')
end

local templates = {
	[0] = '$credit 残片',
	'$grade通关 $song_id $song_difficulty',
	'游玩 $song_id $song_difficulty',
	'$grade通关 $song_id $song_difficulty$times回',
	"$1 <br>'''或''' $2",
	'个人游玩潜力值 $rating 或以上',
	'通关$count首$rating$ratingPlus难度曲目',
}
local handlers = {}
local linkize = '[[%s]]'
function handlers.song_id(v) return linkize:format(mLink[v]) end
function handlers.song_difficulty(v) return ({[0] = '[PST]', '[PRS]', '[FTR]', '[BYD]'})[v] end
function handlers.grade(v) return v == 0 and '' or ('以 「%s」 或以上成绩'):format(({'C', 'B', 'A', 'AA', 'EX'})[v]) end
function handlers.rating(v) return v > 13 and ('%.2f'):format(v / 100) or v end
function handlers.ratingPlus(v) return v and '+' or '' end
local function stringify(cond)
	local res = templates[cond.type]:gsub('%$([a-zA-Z12_]+)', function(k)
		local nk = tonumber(k)
		if nk then return stringify(cond.conditions[nk]) end
		if handlers[k] then return handlers[k](cond[k]) end
		return cond[k]
	end)
	return res
end

local names = {'frag', 'prev', 'ptt'}
local function judgeCond(type)
	if type == 0 then
		return 'frag'
	elseif type == 5 then
		return 'ptt'
	elseif type <= 100 then
		return 'prev'
	end
end
local function createRow(res)
	for _, k in ipairs(names) do res[k] = 0 end
	return res
end
local function judgeRow(row)
	local res = {'ulk'}
	for _, k in ipairs(names) do if row[k] > 0 then table.insert(res, k) end end
	return res
end

function p.main()
	local view = {}
	for _, chartUlk in ipairs(mad.listOf 'unlocks') do
		local id, rc = chartUlk.songId, chartUlk.ratingClass + 1
		for _, cond in ipairs(chartUlk.conditions) do
			local className = judgeCond(cond.type)
			if className then
				local row = view[id]
				if not row then
					row = createRow {}
					view[id] = row
				end
				row[className] = row[className] + 1
				local cell = row[rc]
				if not cell then
					cell = {}
					row[rc] = cell
				end
				table.insert(cell, tostring(mw.html.create 'span':addClass 'ulk':addClass(className):wikitext(stringify(cond))))
			end
		end
	end

	local texts = {}
	local rowFormat = '|-%s\n|[[%s]]' .. ('||%s'):rep(3)
	local sectionCode = {unknown = 0, single = 1, free = 2, mainstory = 3, sidestory = 4, collab = 5}
	for id, row in pairs(view) do
		local song = mSong[id]
		local packItem = mad.packQueryWrap(song.set, 'mobile')
		local rowText = {}
		for i = 1, 3 do
			local cell = row[i]
			if not cell or #cell == 0 then
				rowText[i] = ' '
			else
				rowText[i] = table.concat(cell, '<br>')
			end
		end
		local classList = judgeRow(row)
		if id == 'lasteternity' then classList = {'ulk'} end
		table.insert(texts, {
			data = rowFormat:format(('class="%s"'):format(table.concat(classList, ' ')), mLink[id], unpack(rowText)),
			cls = classList,
			pack = packItem['name'],
			sort = {sectionCode[packItem['section']], packItem['numero'], song.date},
		})
	end

	table.sort(texts, function(a, b)
		a, b = a.sort, b.sort
		for i = 1, 3 do
			local d = a[i] - b[i]
			if d ~= 0 then return d < 0 end
		end
		return false
	end)

	local hybrid = {}
	local pack
	local idx
	local row = createRow {}
	for _, value in ipairs(texts) do
		if pack ~= value.pack then
			pack = value.pack
			table.insert(hybrid, '|-class="%s"\n| colspan="4" |' .. pack)
			if idx then
				hybrid[idx] = hybrid[idx]:format(table.concat(judgeRow(row), ' '))
			end
			idx = #hybrid
			row = createRow {}
		end
		table.insert(hybrid, value.data)
		for _, cls in ipairs(value.cls) do
			for _, k in ipairs(names) do
				if k == cls then
					row[k] = row[k] + 1
				end
			end
		end
	end
	if idx then
		hybrid[idx] = hybrid[idx]:format(table.concat(judgeRow(row), ' '))
	end

	return table.concat(hybrid, '\n')
end

return p