查看“Module:Unlock”的源代码

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

local linkPool = {}
local mLink = {}
local mSong = {}
for _, song in ipairs(mad.listOf 'songs') do
  local link = mad.linkTitle(song)
  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><strong>或</strong> $2',
  '个人游玩潜力值 $rating 或以上',
  '通关$count首$rating$ratingPlus难度曲目',
  '解锁$song_id $song_difficulty',
  [101] = '<small><em>Gauge:$min%-$max%</em></small>',
  [103] = '选择搭档「[[$id]]」',
  [104] = '<small><em>Axiom</em></small>',
  [105] = '<small><em>Eternity</em></small>',
  [106] = '<small><em>Moment</em></small>',
  [107] = '<small><em>Magnolia</em></small>',
  [108] = '连接她们的故事（15-6｜16-6）',
  [109] = '于第$index曲$difficulty上刻下真名',
  [110] = '<small><em>Course</em></small>',
  [111] = '阅读剧情【18-6】',
  [112] = '得到你所渴求的',
}
local handlers = {}
function handlers.song_id(v) return ('[[%s]]'):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
handlers.difficulty = handlers.song_difficulty
function handlers.id(v) return ({[45] = '拉格兰', [71] = '摩耶'})[v] end
local function stringify(cond)
  local template = templates[cond.type] or ('不支持的解锁类型号：' .. cond.type)
  local res = template: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 function createView()
  local res = {}
  for _, chartUlk in ipairs(mad.listOf 'unlocks') do
    local id = chartUlk.songId
    local song = res[id]
    if not song then
      song = {group = mSong[id].set, link = mLink[id]}
      res[id] = song
    end
    local chart = {ratingClass = chartUlk.ratingClass}
    table.insert(song, chart)
    for _, cond in ipairs(chartUlk.conditions) do table.insert(chart, stringify(cond)) end
  end
  return res
end

local function createText(view)
  local sectionCode = {unknown = 0, single = 1, free = 2, mainstory = 3, sidestory = 4, collab = 5}
  local uis = {}
  for id, row in pairs(view) do
    local packItem = mad.packQueryWrap(row.group)
    table.insert(uis, {id = id, name = packItem['name'], sectionCode[packItem['section']], packItem['numero'], mSong[id].date})
  end
  table.sort(uis, function(a, b)
    for i = 1, 3 do
      local d = a[i] - b[i]
      if d ~= 0 then return d < 0 end
    end
    return false
  end)
  local classes = {
    [0] = {'PAST', 'text-pst'},
    {'PRESENT', 'text-prs'},
    {'FUTURE',  'text-ftr'},
    {'BEYOND',  'text-byd'},
    {'ETERNAL', 'text-etr'},
  }
  local function chart(cursor, o)
    local text, class = unpack(classes[o.ratingClass])
    cursor:tag 'dt':addClass(class):wikitext(text)
    for _, i in ipairs(o) do cursor:tag 'dd':wikitext(i) end
  end
  local function song(cursor, o)
    local ch = cursor:tag 'tr'
        :tag 'td':css('text-align', 'center'):wikitext('[[', o.link, ']]'):done()
        :tag 'td':tag 'dl':css('column-count', #o)
    for _, i in ipairs(o) do chart(ch, i) end
  end
  local pack
  local res = mw.html.create 'table':addClass 'wikitable mw-collapsible mw-collapsed'
      :tag 'tr'
      :tag 'th':wikitext '曲目':done()
      :tag 'th':wikitext '解锁条件':done()
      :done()
  for _, value in ipairs(uis) do
    if pack ~= value.name then
      pack = value.name
      res:tag 'tr':tag 'th':attr('colspan', '2'):wikitext(pack)
    end
    song(res, view[value.id])
  end
  return res
end
function p.detail()
  return tostring(createText(createView()))
end

return p