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

local common = require('Module:Common')

local p = {}

local data_cache
local by_id
local mapping

local gender_map = {
    male = '男',
    female = '女',
}

local season_map = {
    ['1'] = 'Spring',
    ['2'] = 'Summer',
    ['3'] = 'Autumn',
    ['4'] = 'Winter',
}

local function normalize_key(value)
    return common.normalizeKey(value)
end

local function pick_first(...)
    local values = { ... }
    for _, value in ipairs(values) do
        value = common.trim(value)
        if value ~= '' then
            return value
        end
    end
    return ''
end

local function format_birthday(record)
    if type(record) ~= 'table' then
        return ''
    end

    local month = common.trim(record.birthday_month)
    local day = common.trim(record.birthday_day)
    local season = season_map[month]
    if season ~= nil and day ~= '' then
        return mw.getCurrentFrame():expandTemplate{
            title = 'Season',
            args = { season, day },
        }
    end
    return ''
end

local function localize_gender(value)
    local normalized = normalize_key(value)
    return gender_map[normalized] or common.trim(value)
end

local function display_image(record)
    if type(record) ~= 'table' then
        return ''
    end
    return pick_first(record.default_image, record.upgrade_icon, record.map_image)
end

local function load_data()
    if data_cache then
        return
    end

    data_cache = common.readJsonPage('数据:Character/character_index.json') or {}
    by_id = data_cache
    mapping = common.readJsonPage('数据:Character/character_mapping.json') or {
        name_to_id = {},
        id_to_name = {},
        aliases = {},
        overrides = {
            name_to_id = {},
            aliases = {},
        },
    }
end

local function find_record(key)
    load_data()

    local resolved = common.trim(key)
    if resolved == '' then
        resolved = common.getCurrentTitleText()
    end

    local normalized = normalize_key(resolved)
    if by_id[normalized] then
        return by_id[normalized]
    end

    local override_id = mapping.overrides and mapping.overrides.name_to_id and mapping.overrides.name_to_id[normalized]
    if override_id and by_id[normalize_key(override_id)] then
        return by_id[normalize_key(override_id)]
    end

    local alias_id = mapping.overrides and mapping.overrides.aliases and mapping.overrides.aliases[normalized]
    if alias_id and by_id[normalize_key(alias_id)] then
        return by_id[normalize_key(alias_id)]
    end

    local mapped_id = mapping.name_to_id and mapping.name_to_id[normalized]
    if mapped_id and by_id[normalize_key(mapped_id)] then
        return by_id[normalize_key(mapped_id)]
    end

    local mapped_alias = mapping.aliases and mapping.aliases[normalized]
    if mapped_alias and by_id[normalize_key(mapped_alias)] then
        return by_id[normalize_key(mapped_alias)]
    end

    return nil
end

local function get_display_field(record, field)
    if type(record) ~= 'table' then
        return ''
    end

    if field == 'birthday_display' or field == 'birthday' then
        return format_birthday(record)
    end
    if field == 'gender_display' or field == 'gender_zh' then
        return localize_gender(record.gender)
    end
    if field == 'image' or field == 'image_fallback' then
        return display_image(record)
    end
    if field == 'display_name' then
        return pick_first(record.name, record.title, record.id)
    end
    if field == 'display_title' then
        return pick_first(record.title, record.name)
    end

    return record[field]
end

function p.getField(frame)
    local key = common.getArg(frame, 1, '')
    local field = common.getArg(frame, 2, '')
    if field == '' then
        return ''
    end

    local record = find_record(key)
    if not record then
        return ''
    end

    return common.toText(get_display_field(record, field))
end

function p.getId(frame)
    local key = common.getArg(frame, 1, '')
    local record = find_record(key)
    if not record then
        return ''
    end

    return common.toText(record.id)
end

function p.getImage(frame)
    local key = common.getArg(frame, 1, '')
    local record = find_record(key)
    if not record then
        return ''
    end

    return common.toText(display_image(record))
end

function p.getBirthday(frame)
    local key = common.getArg(frame, 1, '')
    local record = find_record(key)
    if not record then
        return ''
    end

    return format_birthday(record)
end

function p.getGender(frame)
    local key = common.getArg(frame, 1, '')
    local record = find_record(key)
    if not record then
        return ''
    end

    return localize_gender(record.gender)
end

return p