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

local Utils = require("Module:Utils")
local Items = require("Module:Items")

---@diagnostic disable-next-line: undefined-global
local mw = mw
local ObjectData = mw.loadData('Module:Object/data')

local p = {}

local ERROR_NOT_FOUND = "物品不存在。"

local nameIndex

local function normalizeId(value)
    if value == nil then return nil end
    local text = tostring(value):match("^%s*(.-)%s*$")
    if text == "" then return nil end
    return text
end

local function fetchItemById(rawId)
    local id = normalizeId(rawId)
    if not id then return nil, nil end
    return ObjectData[id], id
end

local function ensureNameIndex()
    if nameIndex then return nameIndex end
    nameIndex = {}
    for id, data in pairs(ObjectData) do
        local name = data and data.Name
        if type(name) == "string" then
            local normalized = Utils.normalizeKey(name)
            if normalized and normalized ~= "" and not nameIndex[normalized] then
                nameIndex[normalized] = id
            end
        end
    end
    return nameIndex
end

local function matchField(fields, requested)
    local key = Utils.normalizeKey(requested)
    if not key then return nil end
    for fieldName, value in pairs(fields) do
        if Utils.normalizeKey(fieldName) == key then return value end
    end
    return nil
end

-- mw.logObject(p.getFieldsById{ args = { "137"} })
function p.getFieldsById(input)
    local args = Utils.resolveArgs(input)
    local id = args[1] or args["1"]
    local item = fetchItemById(id)
    if not item then return ERROR_NOT_FOUND end

    local fields = {
        Name = item.Name,
        Type = item.Type,
        Category = item.Category,
        Price = item.Price,
        ContextTags = item.ContextTags
    }
    local requestedField = args[2] or args["2"]
    if requestedField and requestedField ~= "" then
        return matchField(fields, requestedField) or ""
    end

    return fields
end

-- =p.getColorById{ args = { "137"} }
-- mw.logObject(p.getColorById{ args = { "137"} })
function p.getColorById(input)
    local id = Utils.getArg(input)
    local item = fetchItemById(id)
    local tags = item and item.ContextTags
    if type(tags) ~= "table" then return "" end

    for _, tag in ipairs(tags) do
        if type(tag) == "string" and tag:match("^color_") then
            local formatted = tag:gsub("color_", ""):gsub("_", " ")
            formatted = formatted:gsub("(%a)(%w*)", function(first, rest)
                return first:upper() .. rest:lower()
            end)
            return formatted
        end
    end
    return ""
end

-- mw.logObject(p.getPriceById{ args = { "137"} })
function p.getPriceById(input)
    local id = Utils.getArg(input)
    local item = fetchItemById(id)
    if not item then return "" end
    return item.Price or ""
end

-- mw.logObject(p.getPriceByName{ args = { "Smallmouth Bass"} })
function p.getPriceByName(input)
	local name = Utils.normalizeKey(Utils.getArg(input))
	if not name then return "" end
    local id = Items.getId(name):gsub("%(O%)", "")
    if not id or id == "" then return "" end
    return p.getPriceById(id)
end

return p