模块:Db
本模块为自动表核心功能程序，根据输入的信息自动筛选数据。
数据库储存在Module:db_song，通过Template:db可以调用数据库中的信息，也可以通过Template:songlist生成自动表格。
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
-- 格式化单个值的私有函数 (基础格式化，不应用BPM或难度样式)
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
return tostring(value)
end
end
-- 仅在表格中应用的BPM样式
local function formatBPMValue(rawValue)
local strValue = tostring(rawValue)
if string.find(strValue, '-') then
return '<span style="color:red;">' .. strValue .. '</span>'
end
return rawValue
end
-- 获取难度模式 (使用哈希表检查，使用 sub 提取模式)
local function getDifficultyDetails(field)
local checkField = field and string.lower(mw.text.trim(field)) or ''
-- 哈希表包含所有难度字段，排除 _notes
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')
local mode = checkField:sub(-2)
local btn = checkField:match("^(%d+)b")
return btn, mode
end
return nil, nil
end
-- **核心功能：根据难度和数值应用内联样式**
local function formatDifficultyValue(rawValue, btn, mode)
local numValue = tonumber(rawValue)
if numValue == nil then
return rawValue -- 如果不是数字则返回原始值
end
local style = ''
if mode == 'sc' then
style = 'font-weight: bold;'
if numValue >= 1 and numValue <= 5 then
style = style .. ' color: #E00075;'
elseif numValue >= 6 and numValue <= 10 then
style = style .. ' color: #C604E3;'
elseif numValue >= 11 and numValue <= 12 then
-- 11 and 12 颜色为 #3D66FF
style = style .. ' color: #3D66FF;'
elseif numValue == 13 then
-- NEW FIX: 13 颜色为白色
style = style .. ' color: white;'
elseif numValue == 14 then
-- 14: 白色，#3D66FF 光晕
style = style .. ' color: white; text-shadow: 0 0 5px #3D66FF, 0 0 8px #3D66FF;'
elseif numValue == 15 then
-- 15: 黄色，#3D66FF 光晕
style = style .. ' color: yellow; text-shadow: 0 0 5px #3D66FF, 0 0 8px #3D66FF;'
end
elseif (mode == 'nm' or mode == 'hd' or mode == 'mx') then
if numValue == 14 then
style = 'font-weight: bold; color: orange;'
elseif numValue == 15 then
style = 'font-weight: bold; color: red;'
end
end
if style ~= '' then
return '<span style="' .. style .. '">' .. rawValue .. '</span>'
end
return rawValue -- 返回原始值（如果没有样式应用，例如 NM/HD/MX 不是 14/15）
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)
-- 在 p.value 中，只进行基础格式化，不应用任何颜色或加粗样式
local formattedValue = formatSingleValue(value, argType)
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 = {}
-- 表格样式: 宽度 100%, 居中, 内容居中
table.insert(output, '{| class="wikitable sortable" style="width: 100%; margin: 0 auto; text-align: center;"')
-- 构造表格头部 (不包含 ID)
table.insert(output, '|-')
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)
-- **应用 BPM 样式** (仅在表格中)
if field == 'bpm' then
formattedValue = formatBPMValue(formattedValue)
end
-- **应用难度样式** (仅在表格中)
local btn, mode = getDifficultyDetails(field)
if btn and mode then
formattedValue = formatDifficultyValue(formattedValue, btn, mode)
end
-- FIX: 确保每个单元格都在自己的行上，这是最可靠的 wikitext 结构
table.insert(output, '|' .. formattedValue)
end
end
table.insert(output, '|}')
-- FIX: 使用 \n 连接所有行，确保 MediaWiki 解析器能识别所有表格元素
return table.concat(output, '\n')
end
return p