查看“︁模块:GiftsByItem”︁的源代码

local Utils = require("Module:Utils")
local ID = require("Module:Expanded/ID")
local Vanilla = require("Module:GiftsByItem")

local p = {}

-- SVE 新增人物
local newVillagers = {
    ["Lance"] = "兰斯", ["Magnus"] = "马格努斯", ["Marlon"] = "马龙",
    ["Martin"] = "马丁", ["Morgan"] = "摩根", ["Morris"] = "莫里斯",
    ["Olivia"] = "奥利维亚", ["Andy"] = "安迪", ["Scarlett"] = "斯嘉丽",
    ["Sophia"] = "索菲娅", ["Susan"] = "苏珊", ["Victor"] = "维克多",
    ["Henchman"] = "仆从", ["Claire"] = "克莱尔", ["Gunther"] = "冈瑟",
    ["Apples"] = "苹果"
}

local orderAdditions = {
    "索菲娅", "维克多", "克莱尔", "马丁", "安迪", "苏珊", "奥利维亚", "兰斯", "斯嘉丽", "莫里斯", "马龙", "冈瑟", "仆从", "马格努斯", "摩根", "苹果"
}

-- 缓存合并后的数据
local translationMap
local orderList
local originalMap
local orderIndexMap

-- 初始化函数
local function initializeData()
    if translationMap then return end
    
    translationMap = {}
    for k, v in pairs(Vanilla.translation) do
        translationMap[k] = v
    end
    for k, v in pairs(newVillagers) do
        translationMap[k] = v
    end
    
    orderList = {}
    for _, name in ipairs(Vanilla.predefinedOrder) do
        table.insert(orderList, name)
    end
    for _, name in ipairs(orderAdditions) do
        table.insert(orderList, name)
    end
    
    originalMap = {}
    for en, zh in pairs(translationMap) do
        originalMap[zh] = en
    end
    
    orderIndexMap = {}
    for i, name in ipairs(orderList) do
        orderIndexMap[name] = i
    end
end

-- SVE 的主函数
function p.tssve(frame)
    initializeData()
    local villagerlist = frame.args[1]
    return Vanilla.processVillagers(villagerlist, translationMap, orderIndexMap, originalMap, "SVE")
end

-- 重写 getCurrentProcessor 函数，返回 SVE 的处理函数
function p.getCurrentProcessor()
    return p.tssve
end

-- 重写 getCurrentIDFunction 函数，返回 SVE 的 ID 函数
function p.getCurrentIDFunction()
    return ID.getId
end

-- 重写 getCurrentDataSource 函数，返回 SVE 的数据源
function p.getCurrentDataSource()
    return Utils.LazyLoad('Module:Expanded/GiftsByItem/data')
end

-- SVE 的 generateGiftTable 函数
function p.generateGiftTable(frame)
    return Vanilla.generateGiftTable(frame, p.getCurrentProcessor(), p.getCurrentIDFunction(), p.getCurrentDataSource())
end

-- 导出一些有用的数据和函数
p.newVillagers = newVillagers
p.orderAdditions = orderAdditions
p.base = Vanilla

function p.getTranslationMap()
    initializeData()
    return translationMap
end

function p.getOrderList()
    initializeData()
    return orderList
end

return p