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

local p = {}

-- [[------------------------------------------------------------------]]
-- [[ 1. 国际化和常量 ]]
-- [[------------------------------------------------------------------]]
local i18n = {
	['true'] = '是',      
	['false'] = '否',     
	['none'] = '无',
	['category_missing_value'] = '缺失%s',
	['error_no_type'] = '<span class="error">Db：调用时必须提供type参数。</span>',
    ['error_find_keys_args'] = '<span class="error">Db：findKeysByValue必须提供field和value参数。</span>', 
    ['error_generate_table_args'] = '<span class="error">Db：generateListTable调用缺少field, value或columns参数。</span>',
}

-- [[------------------------------------------------------------------]]
-- [[ 2. 表格标题映射 ]]
-- [[------------------------------------------------------------------]]
local displayNames = {
    -- 核心字段
    ['title'] = '歌曲标题',
    ['composer'] = '艺术家',
    ['bpm'] = 'BPM',
    ['singer'] = '歌手',
    ['genre'] = '流派',
    ['id'] = '歌曲ID',
    ['key'] = 'KEY音',
    ['bga'] = 'BGA作者',

    -- 保留常用字段
    ['duration'] = '时长',

    -- 节奏游戏字段
    ['4b_nm'] = '4B NM',
    ['4b_nm_notes'] = '4B NM按键数',
    ['4b_hd'] = '4B HD',
    ['4b_hd_notes'] = '4B HD按键数',
    ['4b_mx'] = '4B MX',
    ['4b_mx_notes'] = '4B MX按键数',
    ['4b_sc'] = '4B SC',
    ['4b_sc_notes'] = '4B SC按键数',
    ['5b_nm'] = '5B NM',
    ['5b_nm_notes'] = '5B NM按键数',
    ['5b_hd'] = '5B HD',
    ['5b_hd_notes'] = '5B HD按键数',
    ['5b_mx'] = '5B MX',
    ['5b_mx_notes'] = '5B MX按键数',
    ['5b_sc'] = '5B SC',
    ['5b_sc_notes'] = '5B SC按键数',
    ['6b_nm'] = '6B NM',
    ['6b_nm_notes'] = '6B NM按键数',
    ['6b_hd'] = '6B HD',
    ['6b_hd_notes'] = '6B HD按键数',
    ['6b_mx'] = '6B MX',
    ['6b_mx_notes'] = '6B MX按键数',
    ['6b_sc'] = '6B SC',
    ['6b_sc_notes'] = '6B SC按键数',
    ['8b_nm'] = '8B NM',
    ['8b_nm_notes'] = '8B NM按键数',
    ['8b_hd'] = '8B HD',
    ['8b_hd_notes'] = '8B HD按键数',
    ['8b_mx'] = '8B MX',
    ['8b_mx_notes'] = '8B MX按键数',
    ['8b_sc'] = '8B SC',
    ['8b_sc_notes'] = '8B SC按键数',
}

-- [[------------------------------------------------------------------]]
-- [[ 3. 本地辅助函数定义 ]]
-- [[------------------------------------------------------------------]]
local function GetValuesTable()
    return mw.loadData('Module:db_song')
end 

local function ValueFromValuesByKey(values, key)
	if values and key then
		return values[key]
	end
	return nil
end 

-- 格式化单个值的私有函数
local function formatSingleValue(value, fieldType)
    if value == nil then
        return i18n['none']
    elseif type(value) == 'boolean' then
        return value and i18n['true'] or i18n['false']
    elseif type(p.valueMappingMethod[fieldType]) == 'function' then
        return p.valueMappingMethod[fieldType](value)
    else
        local strValue = tostring(value)
        
        -- BPM 颜色逻辑
        if fieldType == 'bpm' and string.find(strValue, '-') then
            return '<span style="color:red;">' .. strValue .. '</span>'
        end
        
        return strValue
    end
end 

-- 获取难度模式 (使用哈希表检查，使用 sub 提取模式)
local function getDifficultyMode(field)
    local checkField = field and string.lower(mw.text.trim(field)) or ''
    
    -- 哈希表包含所有需要套用 {{df|}} 模板的精准字段
    local difficultyFields = {
        ['4b_nm'] = true, ['4b_hd'] = true, ['4b_mx'] = true, ['4b_sc'] = true,
        ['5b_nm'] = true, ['5b_hd'] = true, ['5b_mx'] = true, ['5b_sc'] = true,
        ['6b_nm'] = true, ['6b_hd'] = true, ['6b_mx'] = true, ['6b_sc'] = true,
        ['8b_nm'] = true, ['8b_hd'] = true, ['8b_mx'] = true, ['8b_sc'] = true,
    }

    if difficultyFields[checkField] == true then
        -- 匹配成功，安全提取模式 (最后两个字符: 'nm', 'hd', 'mx', or 'sc')
        return checkField:sub(-2) 
    end
    
    return nil
end


-- 4. 值格式化方法 (valueMappingMethod)
p.valueMappingMethod = {
    ['duration'] = (function (value)
        if type(value) == 'number' then
            local minutes = math.floor(value / 60)
            local seconds = value % 60
            return string.format('%d:%02d', minutes, seconds)
        end
        return value
    end),
}

-- [[------------------------------------------------------------------]]
-- [[ 5. 外部调用函数 (p.*) ]]
-- [[------------------------------------------------------------------]]

function p.value(f)
    local args = f
	local frame = mw.getCurrentFrame()
	if f == frame then
		args = require('Module:ProcessArgs').merge(true)
	end
    
	local argTargetName = mw.text.trim(args[1] or '') 
	local argTypeString = args.type                 
	local argNocat = args.nocat                     
    
    if not argTypeString then
        return i18n['error_no_type']
    end
    
    local values = GetValuesTable() 
    local songEntry = ValueFromValuesByKey(values, argTargetName)

    if not songEntry then
        if not argNocat then
            local title = mw.title.getCurrentTitle()
            if title.namespace == 0 and not title.isSubpage then
                return '?' .. '[[Category:' .. string.format(i18n['category_missing_value'], '歌曲ID ' .. argTargetName) .. ']]'
            end
        end
        return i18n['none']
    end
    
    local fieldNames = mw.text.split(argTypeString, ',')
    local resultTable = {}

    for _, argType in ipairs(fieldNames) do
        argType = mw.text.trim(argType)
        local value = ValueFromValuesByKey(songEntry, argType)
        local formattedValue = formatSingleValue(value, argType)
        
        -- 应用 {{df|}} 模板和 SC 前缀逻辑
        local mode = getDifficultyMode(argType)
        if mode then
            -- 如果是 SC 难度，将 'sc' 前缀加入参数中
            if mode == 'sc' then
                formattedValue = 'sc' .. formattedValue
            end
            formattedValue = '{{df|' .. formattedValue .. '}}'
        end
        
        table.insert(resultTable, formattedValue)
    end
    
	return table.concat(resultTable, ' / ')
end

function p.findKeysByValue(f)
    local args = f
    local frame = mw.getCurrentFrame()
    if f == frame then
        args = require('Module:ProcessArgs').merge(true)
    end
    
    local targetField = mw.text.trim(args.field or '')
    local targetValue = args.value 
    if type(targetValue) == 'string' then
        targetValue = mw.text.trim(targetValue)
    end
    
    if targetField == '' or targetValue == nil then
        return i18n['error_find_keys_args']
    end

    local songData = GetValuesTable()
    local matchingKeys = {}
    
    for songId, songEntry in pairs(songData) do
        local currentValue = ValueFromValuesByKey(songEntry, targetField)
        
        if currentValue == targetValue then
            table.insert(matchingKeys, songId)
        end
    end
    
    table.sort(matchingKeys)
    
    return table.concat(matchingKeys, ', ')
end


function p.generateListTable(f)
    local args = require('Module:ProcessArgs').merge(true) 
    
    local conditionField = mw.text.trim(args.field or '')   
    local conditionValue = args.value                       
    local displayFieldsString = mw.text.trim(args.columns or '') 
    
    -- 预处理 conditionValue 的类型
    if type(conditionValue) == 'string' then
        local trimmedValue = mw.text.trim(conditionValue)
        
        local lowerValue = string.lower(trimmedValue)
        if lowerValue == 'true' or trimmedValue == '是' then
            conditionValue = true
        elseif lowerValue == 'false' or trimmedValue == '否' then
            conditionValue = false
        else
            local num = tonumber(trimmedValue)
            if num ~= nil then
                conditionValue = num
            else
                conditionValue = trimmedValue
            end
        end
    end
    
    if conditionField == '' or conditionValue == nil or displayFieldsString == '' then
        return i18n['error_generate_table_args']
    end

    local songData = GetValuesTable()
    local displayFields = mw.text.split(displayFieldsString, ',')
    local matchingKeys = {}
    
    for songId, songEntry in pairs(songData) do
        local currentValue = ValueFromValuesByKey(songEntry, conditionField)
        if currentValue == conditionValue then
            table.insert(matchingKeys, songId)
        end
    end
    
    table.sort(matchingKeys)

    if #matchingKeys == 0 then
        return '未找到符合条件的歌曲。'
    end

    local output = {}
    
    -- 表格样式: 宽度 90%, 居中, 内容居中
    table.insert(output, '{| class="wikitable sortable" style="width: 90%; margin: 0 auto; text-align: center;"')
    table.insert(output, '|-')
    
    -- 构造表格头部 (不包含 ID)
    for _, field in ipairs(displayFields) do
        field = mw.text.trim(field)
        table.insert(output, '! ' .. (displayNames[field] or field))
    end
    
    -- 构造表格行
    for _, songId in ipairs(matchingKeys) do
        local songEntry = songData[songId]
        table.insert(output, '|-')
        
        -- 数据字段
        for _, field in ipairs(displayFields) do
            field = mw.text.trim(field)
            local rawValue = ValueFromValuesByKey(songEntry, field)
            local formattedValue = formatSingleValue(rawValue, field)
            
            -- 应用 {{df|}} 模板和 SC 前缀逻辑
            local mode = getDifficultyMode(field)
            if mode then
                -- 如果是 SC 难度，将 'sc' 前缀加入参数中
                if mode == 'sc' then
                    formattedValue = 'sc' .. formattedValue
                end
                formattedValue = '{{df|' .. formattedValue .. '}}'
            end
            
            -- FIX: 确保每个单元格都在新行上，前面有 |，以符合标准的 Wikitext 语法
            table.insert(output, '\n|' .. formattedValue)
        end
    end
    
    table.insert(output, '\n|}')
    
    return table.concat(output, '')
end

return p