模組:CCHPaward

require("strict")

local getArgs = require("Module:Arguments").getArgs

-- 数据
local RECORD_MODULE = "Module:CCHPaward/list"

local function load_record()
    return mw.loadData(RECORD_MODULE)
end

-- 获取用户名（页面名称或指定参数）
local function fetch_username(user)
    if user == nil or user == "" then
        return mw.title.getCurrentTitle().baseText
    end
    return user
end

-- 计算等级
local function calculate_level(article, image, review)
    local total_score = article * 1 + image * 0.2 + review * 0.4
    return math.floor(total_score / 10)
end

local p = {}

-- 获取用户等级
function p.level(frame)
    local args = getArgs(frame)
    local user = args.user
    local record = load_record()
    local username = fetch_username(user)
    
    -- 获取得分
    local user_data = record[username] or {article = 0, image = 0, review = 0}
    
    -- 计算并返回等级
    return tostring(calculate_level(user_data.article, user_data.image, user_data.review))
end

-- 生成等级排行榜
function p._ranking()
    local record = load_record()
    local user_list = {}
    
    -- 整理数据
    for username, scores in pairs(record) do
        local level = calculate_level(scores.article, scores.image, scores.review)
        table.insert(user_list, {
            username = username,
            level = level,
            article = scores.article,
            image = scores.image,
            review = scores.review
        })
    end
    
    -- 排序
    table.sort(user_list, function(a, b)
        if a.level == b.level then
            return a.username < b.username
        end
        return a.level > b.level
    end)
    
    -- 生成排行榜
    local lines = {}
    local current_level_group = math.huge
    local rank_counter = 1
    
    for _, user in ipairs(user_list) do
        -- 每10级一个分组标题
        local level_group = math.floor(user.level / 10) * 10
        if level_group < current_level_group then
            current_level_group = level_group
            if #lines > 0 then
                table.insert(lines, "</ol>") -- 关闭前一个列表
            end
            table.insert(lines, "")
            table.insert(lines, "=== " .. current_level_group .. "级及以上 ===")
            table.insert(lines, "<ol start='" .. rank_counter .. "'>")
        end
        
        -- 添加用户行
        local user_line = string.format(
            "[[User:%s|%s]]：%d级（%d条目分，%d图片分，%d评审分）",
            user.username,
            user.username,
            user.level,
            user.article,
            user.image,
            user.review
        )
        table.insert(lines, "<li>" .. user_line .. "</li>")
        rank_counter = rank_counter + 1
    end
    
    -- 结束列表
    if #lines > 0 then
        table.insert(lines, "</ol>")
    end
    
    return table.concat(lines, "\n")
end

function p.ranking(frame)
    return p._ranking()
end

return p