查看“Module:ArtistSong”的源代码

local p = {}
local CALv2 = mw.loadJsonData 'Template:ComplexArtistsList.json'
local trans = mw.loadJsonData 'Template:Transition.json'
local SL = mw.loadJsonData 'Template:Song Length.json'

local mad = require 'Module:AnotherData'
local data = require 'Module:Arcaea Data'
local CL = require 'Module:ConvertLink'
local tt = require 'Module:TableTools'

local lang = mw.language.getContentLanguage()

local queryMO = data.allSongInformation('id', 'mobile')
local queryNS = data.allSongInformation('id', 'ns')

-- byd?=>id=>c-artist
local function readList()
	local pickList = {}
	local pickListBYD = {}
	for _, song in ipairs(mad.listOf('songs', 'mobile')) do
		if not song.deleted then
			local artist = song.artist
			pickList[song.id] = song.artist
			if song.difficulties[4] and (song.difficulties[4].artist or song.difficulties[4].title_localized) then
				local bydArtist = song['difficulties'][4]['artist'] or song.artist
				pickListBYD[song['id']] = bydArtist
			end
		end
	end

	for _, song in ipairs(mad.listOf('songs', 'ns')) do
		if not pickList[song.id] and not trans.multiId[song.id] then
			pickList[song.id] = song.artist
			if song.difficulties[4] and song.difficulties[4].artist or song.difficulties[4].title_localized then
				local bydArtist = song['difficulties'][4]['artist'] or song.artist
				pickListBYD[song['id']] = bydArtist
			end
		end
	end

	return pickList, pickListBYD
end

-- groupby: c-artist=>byd?=>[id]
local function convertPick(pickList, pickListBYD)
	convertList = {}
	for id, artist in pairs(pickList) do
		if not convertList[artist] then convertList[artist] = {} end
		if not convertList[artist].normal then convertList[artist]['normal'] = {} end
		table.insert(convertList[artist]['normal'], id)
	end

	for id, artist in pairs(pickListBYD) do
		if not convertList[artist] then convertList[artist] = {} end
		if not convertList[artist].beyond then convertList[artist]['beyond'] = {} end
		table.insert(convertList[artist]['beyond'], id)
	end

	return convertList
end

-- n-n map: s-artist=>byd?=>[id]
local zhName = {}
local function genList(convertList)
	-- TODO:既然convertList都闭包了那singleList也可以
	local function cate(singleList, cArtist, artist)
		local artist = artist or cArtist
		if not singleList[artist] then singleList[artist] = {} end
		if convertList[cArtist].beyond then
			if not singleList[artist].beyond then singleList[artist]['beyond'] = {} end
			tt.extend(singleList[artist]['beyond'], convertList[cArtist]['beyond'])
		end
		if convertList[cArtist].normal then
			if not singleList[artist].normal then singleList[artist]['normal'] = {} end
			tt.extend(singleList[artist]['normal'], convertList[cArtist]['normal'])
		end
		return singleList
	end


	local singleList = {}
	for cArtist in pairs(convertList) do
		local artistlist = {}
		if CALv2[cArtist] then
			if CALv2[cArtist].__FullData__ then
				for _, text in pairs(CALv2[cArtist].__FullData__) do
					if text.link then
						local artist = text.link
						if not artistlist[artist] then
							singleList = cate(singleList, cArtist, artist)
							artistlist[artist] = 1
							if text.zh then
								zhName[artist] = true
							end
						end
					end
				end
			else
				for link in pairs(CALv2[cArtist]) do
					local artist = link
					if not artistlist[artist] then
						singleList = cate(singleList, cArtist, artist)
						artistlist[artist] = 1
					end
				end
			end
		else
			singleList = cate(singleList, cArtist)
		end
	end
	return singleList
end

local function isLetter(char)
	return char:match '%a' ~= nil
end

-- 排序
local function customSort(a, b)
	local a1 = a:sub(1, 1)
	local b1 = b:sub(1, 1)
	local la1 = isLetter(a:sub(1, 1))
	local lb1 = isLetter(b:sub(1, 1))

	if la1 and not lb1 then
		return true
	elseif lb1 and not la1 then
		return false
	elseif string.lower(a1) == string.lower(b1) then
		local la2 = isLetter(a:sub(2, 2))
		local lb2 = isLetter(b:sub(2, 2))
		if la2 and not lb2 then
			return true
		elseif lb2 and not la2 then
			return false
		else
			return string.lower(a) < string.lower(b)
		end
	else
		return string.lower(a) < string.lower(b)
	end

	return string.lower(a) < string.lower(b)
end

local function songTitle(id, byd)
	local display
	local title = queryMO(id, 'title') or queryNS(id, 'title')
	if byd then
		display = queryMO(id, 'bydSongName') or queryNS(id, 'bydSongName') or 'cant find BYD'
	else
		display = title
	end
	local res = trans.songNameToDisplayName[title]
	if res then return res .. '|' .. display end
	res = trans.sameName[title]
	if res then return res[id] .. '|' .. display end
	if title then
		return title .. '|' .. display
	else
		return id
	end
end

local function songBPM(id, byd)
	local bpm
	if byd then
		bpm = queryMO(id, 'bydBPM') or queryNS(id, 'bydBPM')
		if bpm then return bpm end
	end
	bpm = queryMO(id, 'bpm') or queryNS(id, 'bpm')
	if bpm then
		return bpm
	else
		return nil
	end
end

local function songDate(id, byd)
	local date, ver
	if byd then
		date = queryMO(id, 'bydDate') or queryNS(id, 'bydDate')
		if date then return date end
	end
	date = queryMO(id, 'date') or queryNS(id, 'date')
	if date then return date else return 0 end
end

local function songVer(id, byd)
	local date, ver
	if byd then
		ver = queryMO(id, 'bydVersion') or queryNS(id, 'bydVersion')
		if ver then return ver end
	end
	ver = queryMO(id, 'version') or queryNS(id, 'version')
	if ver then return ver else return id end
end

local function songPack(id)
	local packid, plat, isNS
	-- if queryNS(id, 'set') then
	-- 	packid = queryNS(id, 'set')
	-- 	plat = 'ns'
	-- 	isNS = 1
	if queryMO(id, 'set') then
		packid = queryMO(id, 'set')
	else
		return nil
	end
	local packName = trans.packName[packid] or mad.packQueryWrap(packid, plat).name
	local packLink = CL.packLink(packName, 1, isNS)
	return packLink
end

local function songLength(id, byd)
	if byd then
		return SL.beyond[id]
	else
		return SL.normal[id]
	end
end



function p.main(frame)
	local parentArgs = frame:getParent().args
	local Args = frame.args

	local rl, rlb = readList()
	local cl = convertPick(rl, rlb)
	local gl = genList(cl)
	rl, rlb, cl = nil

	local appendList = Args['appendList']
	if appendList then
		local list = mw.text.split(appendList, '\n')
		for _, id in pairs(list) do
			local artists = Args[id .. '.' .. 'artist']
			if artists then
				local artistList = mw.text.split(artists, '/')
				for _, artist in pairs(artistList) do
					if not gl[artist] then gl[artist] = {} end
					if not gl[artist].normal then gl[artist].normal = {} end
					table.insert(gl[artist].normal, id)
				end
			end
		end
	end

	local render = mw.html.create()

	local a2z = {}
	local gl2 = {}
	for name, artistData in pairs(gl) do
		if mw.ustring.match(name, '[一-龠ぁ-ゔァ-ヴー々〆〤ヶ]+') then
			artistData.notAllLatin = 1
			local notJaName = parentArgs[name .. '.notJaName'] or zhName[name]
			if not notJaName then artistData.ja = 1 end
		end
		artistData.name = name
		artistData.enName = parentArgs[name .. '.enName'] or nil
		artistData.otherName = parentArgs[name .. '.otherName'] or nil
		artistData.description = parentArgs[name .. '.description'] or nil
		local sortName = mw.ustring.gsub(name, '[A-Z]+', string.lower)
		gl2[sortName] = artistData
		table.insert(a2z, sortName)
	end
	table.sort(a2z)

	-- 生成页首
	local function alpha()
		local ret, cla, cua = {}, string.byte 'a', string.byte 'A'
		for i = 0, 25 do
			table.insert(ret, {string.char(i + cua), string.char(i + cla)})
		end
		return ret
	end
	local menuItems = tt.merge(
		{{'#', ''}},
		alpha(),
		{{'か', 'ぁ'}, {'汉', '一'}}
	)
	local ia2z = 1
	for _, menuItem in ipairs(menuItems) do
		while a2z[ia2z] < menuItem[2] do
			ia2z = ia2z + 1
		end
		menuItem[3] = a2z[ia2z]
	end
	-- span id=toc
	render:wikitext('__NOTOC__',
		frame:expandTemplate {title = '通用框', args = {'normal narrow'}},
		"'''目录'''<br>")
	for _, menuItem in ipairs(menuItems) do
		render:wikitext('[[#', gl2[menuItem[3]].name, '|', menuItem[1], ']] ')
	end
	render:wikitext(frame:expandTemplate {title = '通用框-end'})
		:newline()

	local function songArg(artist, id, byd, arg)
		arg = '.' .. arg
		artist = artist .. '.'
		local function fun(a) return a and artist .. a .. arg end
		if byd then
			return parentArgs[fun(queryMO(id, 'bydSongName'))] or parentArgs[fun(id .. '.byd')]
		else
			return parentArgs[fun(queryMO(id, 'title'))] or parentArgs[fun(id)]
		end
	end

	local function aslAppend(artist, songid, date, byd)
		local theList = {}
		local function Input(arg) return songid .. (byd and '.byd' or '') .. (arg and '.' .. arg or '') end
		local function aInput(arg) return artist .. '.' .. Input(arg) end
		local function none(arg) return '<span title="' .. aInput(arg) .. '">' .. '--' .. '</span>' end
		theList.title = ('[[' .. (Args[Input 'title'] or songTitle(songid, byd)) .. ']]') or none()
		theList.length = Args[Input 'length'] or songLength(songid, byd) or none()
		theList.bpm = Args[Input 'bpm'] or songBPM(songid, byd) or none()
		theList.pack = Args[Input 'pack'] or songPack(songid) or none()
		local releaseVer, releaseDate = CL.fullVersionNum(songVer(songid, byd), date)
		local ver = Args[Input 'ver'] or releaseVer or none()
		theList.date = Args[Input 'date'] or releaseDate
		local AprilFoolVer = Args[Input 'AprilFoolVer']
		theList.AprilFoolDate = Args[Input 'AprilFoolDate']
		local afVer
		if AprilFoolVer and theList.AprilFoolDate then
			afVer = ("'''愚人节'''<br>" .. 'v' .. AprilFoolVer .. '<br />(' .. theList.AprilFoolDate .. ")<br>'''常驻'''<br>") or ''
		end
		theList.displayVer = (afVer or '') .. 'v' .. ver .. '<br />(' .. theList.date .. ')' or none()
		theList.text = songArg(artist, songid, byd, 'text') or none 'text'
		return theList
	end

	for _, name in ipairs(a2z) do
		local item = gl2[name]
		local usingName = item.name
		local displayName = mw.html.create()
		if item.ja then
			displayName:wikitext('-{<span lang="ja">', usingName, '</span>}-')
		else
			displayName:wikitext(usingName)
		end
		displayName:wikitext(item.otherName and (' (' .. item.otherName .. ')') or '')
		if item.notAllLatin then
			displayName:wikitext(item.enName and (' (' .. item.enName .. ')') or '')
		end
		displayName = tostring(displayName)

		render
			:wikitext(('<h2 id="%s"><span title="%s">%s</span></h2>')
				:format(usingName, usingName, displayName))
			:newline()
			:wikitext(item.description)
			:newline()
		local wt = render:tag 'table'
		wt:attr {
			class = 'wikitable', border = '1', cellspacing = '1', cellpadding = '5', style = 'text-align:center;width: 100%'
		}:tag 'tr'
			:tag 'th':attr {style = 'width: 30.5%;'}:wikitext '曲目名称':done()
			:tag 'th':attr {style = 'width: 6%;'}:wikitext '时长':done()
			:tag 'th':attr {style = 'width: 10%;'}:wikitext 'BPM':done()
			:tag 'th':attr {style = 'width: 18.5%;'}:wikitext '曲包':done()
			:tag 'th':attr {style = 'width: 15%;'}:wikitext '发布版本':done()
			:tag 'th':attr {style = 'width: 20%;'}:wikitext '备注':done()

		local artistSongList = {}
		local listSort = {}
		if item.normal then
			for _, songid in ipairs(item.normal) do
				local date = songDate(songid)
				local theList = aslAppend(usingName, songid, date)
				local aslId = (theList.AprilFoolDate or theList.date) .. songid
				artistSongList[aslId] = theList
				table.insert(listSort, aslId)
			end
		end
		if item.beyond then
			for _, songid in ipairs(item.beyond) do
				local date = songDate(songid, 1)
				local theList = aslAppend(usingName, songid, date, 1)
				local aslId = (theList.AprilFoolDate or theList.date) .. songid .. 'byd'
				artistSongList[aslId] = theList
				table.insert(listSort, aslId)
			end
		end
		table.sort(listSort)
		for _, id in ipairs(listSort) do
			local theList = artistSongList[id]
			wt:tag 'tr'
				:tag 'td':wikitext(theList.title):done()
				:tag 'td':wikitext(theList.length):done()
				:tag 'td':wikitext(theList.bpm):done()
				:tag 'td':wikitext(theList.pack):done()
				:tag 'td':wikitext(theList.displayVer):done()
				:tag 'td':wikitext(theList.text):done()
		end
	end
	return tostring(render)
end

return p