查看“Module:Arcaea Data”的源代码

local getArgs = require('Module:Arguments').getArgs
local loadJson = require("Module:LoadJson")

local p = {}

function makeInvokeFunc(funcName)
    return function(frame)
        local args = getArgs(frame)
        return p[funcName](args)
    end
end

-- p.pack = makeInvokeFunc('_pack')

local packlist

function p.packId(packId)
    -- 将曲包ID转换为曲包名称。
    local covList = {
        ['chunithm_append_1'] = 'CHUNITHM Collaboration - Collaboration Chapter 2',
        ['dividedheart'] = 'Divided Heart',
        ['groovecoaster'] = 'Groove Coaster Collaboration',
        ['lanota'] = 'Lanota Collaboration',
        ['lanota_append_1'] = 'Lanota Collaboration - Collaboration Chapter 2',
        ['tonesphere'] = 'Tone Sphere Collaboration',
        ['chunithm'] = 'CHUNITHM Collaboration',
        ['dynamix'] = 'Dynamix Collaboration',
        ['maimai'] = 'maimai Collaboration',
        ['ongeki'] = 'O.N.G.E.K.I. Collaboration',
        ['musedash'] = 'Muse Dash Collaboration',
        ['single'] = 'Memory Archive'
    }
    if covList[args[1]] then return covList[args[1]] end

    if packlist == nil then
        packlist = mw.text.jsonDecode(mw.getCurrentFrame():expandTemplate{ title = 'Packlist' })
    end

    for i, k in ipairs(packlist) do
        if k['id'] == packId then return k['name_localized']['en'] end
    end
end

function p.unlock(songId)
    local json = loadJson.unlocks()['unlocks']
    local result = {}
    for i, k in ipairs(json) do
        if k['songId'] == songId then
            result[tonumber(k['ratingClass'] + 1)] = k['conditions']
        end
    end
    return result
end

p.song = makeInvokeFunc('_song')

function p._song(args)
    local json
    if args['type'] or args[3] == 'ns' then
        json = loadJson.Songlist()
    else
        json = loadJson.SonglistNS()
    end

    return getSingle(json, args[1], args['index'])(args[2])
end

function getSingle(json, index, genre)
    -- 传入JSON文件，索引值和索引值类型（索引是ID还是曲名）。分析JSON文件以获得曲目信息。仅获取单一曲目的信息。table中的索引名请查看下方switch表。

    -- Songlist是以数字为索引值，因此遍历Songlist，直到发现需要找的曲目，将信息存入info变量中。
    local s, info = "", {}
    -- 如果模式是ID则遍历songlist直至发现id值为索引值为止，并获取内容。曲名同理。
    if genre == "id" then
        for i, k in ipairs(json['songs']) do
            if k['id'] == index then list[index] = k end
        end
    else
        for i, k in ipairs(json['songs']) do
            if k['title_localized']['en'] == index then info = k end
        end
    end

    -- 此处返回一个函数用于查询内容。

    return function(key)

        -- 如果info无值即无法在songlist中查找到索引值，则返回nil。
        
        if info == nil 
        then 
            mw.log( '无法在Songlist中发现目标,索引值为:'..index )
            return nil 
        end

        -- 此处定义一个switch table以达到switch函数的效用。如果需要查找的参数在switch中不存在，返回nil。
        local switch = {
            ["id"] = function() return info["id"] end,
            ["title"] = function() return info['title_localized']['en'] end,
            ["artist"] = function() return info["artist"] end,
            ["bpm"] = function() return info["bpm"] end,
            ["set"] = function() return info["set"] end,
            ["side"] = function() return info["side"] end,
            ["date"] = function() return info["date"] end,
            ["version"] = function() return info["version"] end,
            ["bg"] = function()
                if info["bg"] ~= "" then
                    return info["bg"]
                else
                    if info["side"] == 0 then
                        return 'base_light'
                    else
                        if info['side'] ==1 then
                            return 'base_conflict'
                        else
                            return 'undefined'
                        end
                    end
                end
            end,
            ["pstChartDesigner"] = function()
                return info["difficulties"][1]['chartDesigner']
            end,
            ["pstJacketDesigner"] = function()
                return info["difficulties"][1]['jacketDesigner']
            end,
            ["pstRating"] = function()
                return info["difficulties"][1]['rating']
            end,
            ["prsChartDesigner"] = function()
                return info["difficulties"][2]['chartDesigner']
            end,
            ["prsJacketDesigner"] = function()
                return info["difficulties"][2]['jacketDesigner']
            end,
            ["prsRating"] = function()
                return info["difficulties"][2]['rating']
            end,
            ["ftrChartDesigner"] = function()
                return info["difficulties"][3]['chartDesigner']
            end,
            ["ftrJacketDesigner"] = function()
                return info["difficulties"][3]['jacketDesigner']
            end,
            ["ftrRating"] = function()
                if info["difficulties"][3]['ratingPlus'] then
                    return info["difficulties"][3]['rating'] .. '+'
                else
                    return info["difficulties"][3]['rating']
                end
            end,
            ["bydChartDesigner"] = function()
                if info["difficulties"][4] then
                    return info["difficulties"][4]['chartDesigner']
                end
            end,
            ["bydJacketDesigner"] = function()
                if info["difficulties"][4] then
                    return info["difficulties"][4]['jacketDesigner']
                end
            end,
            ["bydRating"] = function()
                if info["difficulties"][4] then
                    return info["difficulties"][4]['rating']
                end
            end,
            ["pstChange"] = function()
                if info["difficulties"][1] then
                    return info["difficulties"][1]['has_controller_alt_chart']
                end
            end,
            ["prsChange"] = function()
                if info["difficulties"][2] then
                    return info["difficulties"][2]['has_controller_alt_chart']
                end
            end,
            ["ftrChange"] = function()
                if info["difficulties"][3] then
                    return info["difficulties"][3]['has_controller_alt_chart']
                end
            end,
            ["bydAudioChange"] = function()
                if info["difficulties"][4] then
                    return info["difficulties"][4]['audioOverride']
                end
            end,
            ["bydBg"] = function()
                if info["difficulties"][4] then
                    return info["difficulties"][4]['bg']
                end
            end,
            ["bydBPM"] = function()
                if info["difficulties"][4] then
                    return info["difficulties"][4]['bpm']
                end
            end,
            ["bydArtist"] = function()
                if info["difficulties"][4] then
                    return info["difficulties"][4]['artist']
                end
            end
        }

        if switch[key] == nil 
        then 
            mw.log( '未定义的索引类型，请检查是否拼写错误。' )
            return nil 
        end

        return switch[key]()
    end
end

function p.getMobile(name, id)
    -- 传入曲名或ID，获得一个查阅信息的函数（移动版信息）。直接在下一个模块使用。
    if id then
        return getSingle(mw.text.jsonDecode(mw.getCurrentFrame():expandTemplate{ title = 'Songlist.json' } ), id, 'id')
    else
        return getSingle(mw.text.jsonDecode(mw.getCurrentFrame():expandTemplate{ title = 'Songlist.json' } ), name, 'name')
    end
end

function p.getNS(name, id)
    -- 传入曲名或ID，获得一个查阅信息的函数（NS版信息）。直接在下一个模块使用。
    if id then
        return getSingle(mw.text.jsonDecode(mw.getCurrentFrame():expandTemplate{ title = 'Songlist NS.json' } ), id, 'id')
    else
        return getSingle(mw.text.jsonDecode(mw.getCurrentFrame():expandTemplate{ title = 'Songlist NS.json' } ), name, 'name')
    end
end

return p