Module:Statistics

Documentation for this module may be created at Module:Statistics/doc

local p = {}
local cargo = mw.ext.cargo
local html = mw.html
local text = mw.text

-- ============================================================
-- HELPER: Argument Fetcher
-- ============================================================
local function getArgs(frame)
    local args = {}
    for k, v in pairs(frame.args) do args[k] = v end
    if frame:getParent() then
        for k, v in pairs(frame:getParent().args) do args[k] = v end
    end
    return args
end

-- ============================================================
-- CONFIGURATION
-- ============================================================
-- New Softer Blue (Sky Blue): RGB 14, 165, 233
local HEATMAP_R, HEATMAP_G, HEATMAP_B = 14, 165, 233

local HEADERS = {
    rank = "#", team = "Team", player = "Player", matches_played = "Matches",
    finishes = "Finishes", fpm = "FPM", knocks = "Knocks", damage = "Damage",
    headshots = "Headshots", longest = "Longest", assists = "Assists",
    grenade_kills = "Grenade Kills", vehicle_kills = "Vehicle Kills", contribution = "Contrib %", 
    survival = "Surv. Time", healings = "Heals", revives = "Revives", damage_taken = "Dmg Recv",
    grenades_used = "Nades Used", smokes_used = "Smokes Used", utility_used = "Util Used",
    dist_drive = "Drive Dist", dist_walk = "Walk Dist", dist_total = "Total Dist",
    bluezone = "Bluezone Time", air_drops = "Air Drops",
    total_pts = "Total Pts", place_pts = "Place Pts", elims = "Elims",
    avg_place = "Avg Place", avg_place_pts = "Avg Place Pts", avg_elims = "Avg Elims", avg_total = "Avg Pts",
    wwcd = "🥇", place_2 = "🥈", place_3 = "🥉", top_5 = "Top 5", top_8 = "Top 8", place_low = "> 8th",
    g_0 = "0", g_1_5 = "1–5", g_6_10 = "6–10", g_11_15 = "11–15", g_16_20 = "16–20", g_20_plus = "20+"
}

local NO_HEATMAP = { rank=true, team=true, player=true, matches_played=true }

-- ============================================================
-- HELPER FUNCTIONS
-- ============================================================

-- 1. Soft Heatmap (Max Opacity reduced to 0.25)
local function getHeatmapStyle(value, maxVal)
    if not tonumber(value) or not tonumber(maxVal) or maxVal == 0 then return "" end
    local v = tonumber(value)
    local ratio = v / maxVal
    
    -- New Range: 0.05 (5%) to 0.25 (25%)
    local alpha = 0.05 + (ratio * 0.20) 
    return string.format("background-color: rgba(%d, %d, %d, %.2f);", HEATMAP_R, HEATMAP_G, HEATMAP_B, alpha)
end

-- 2. Smart Rounding
local function formatNumber(val)
    if not val then return "" end
    local n = tonumber(val)
    if not n then return val end -- Return text as is
    
    -- If it's an integer (150.0), return 150
    if n == math.floor(n) then
        return math.floor(n)
    else
        -- If it has decimals (5.3333), round to 2 places
        return string.format("%.2f", n)
    end
end

local function getTeamLogo(teamName)
    if not teamName then return "" end
    local cleanName = teamName:gsub("'", "")
    local lightFile = cleanName .. '.png'
    local darkFile = cleanName .. '_dark.png'
    local hasLight = mw.title.new('File:' .. lightFile).exists
    local hasDark = mw.title.new('File:' .. darkFile).exists
    
    local html = ""
    if hasLight then html = html .. '[[File:' .. lightFile .. '|25px|link=' .. teamName .. '|class=logo-lightmode]]'
    else html = html .. '[[File:Shield_team.png|25px|link=' .. teamName .. '|class=logo-lightmode]]' end
    if hasDark then html = html .. '[[File:' .. darkFile .. '|25px|link=' .. teamName .. '|class=logo-darkmode]]'
    elseif hasLight then html = html .. '[[File:' .. lightFile .. '|25px|link=' .. teamName .. '|class=logo-darkmode]]'
    else html = html .. '[[File:Shield_team_dark.png|25px|link=' .. teamName .. '|class=logo-darkmode]]' end
    return html .. " "
end

-- ============================================================
-- MAIN GENERATOR
-- ============================================================
function p.main(frame)
    local args = getArgs(frame)
    local type = args.type or "player"
    local tournament = args.tournament 
    if not tournament or tournament == "" then
        tournament = mw.title.getCurrentTitle().text
    end

    local map = args.map or "All"
    local stage = args.stage
    local group = args.group
    
    local colsInput = args.columns or ""
    local colKeys = {}
    if colsInput == "" then
        if type == "player" then colKeys = {"rank", "player", "team", "matches_played", "finishes", "damage"}
        else colKeys = {"rank", "team", "matches_played", "total_pts", "wwcd"} end
    else
        colKeys = text.split(colsInput, ",")
    end
    
    local queryFields = {}
    for _, k in ipairs(colKeys) do
        local cleanK = k:match("^%s*(.-)%s*$")
        if cleanK ~= "rank" then table.insert(queryFields, cleanK) end
    end
    
    local whereParts = {}
    table.insert(whereParts, string.format("tournament='%s'", tournament:gsub("'", "\\'")))
    if map ~= "Any" then table.insert(whereParts, string.format("map='%s'", map:gsub("'", "\\'"))) end
    if stage and stage ~= "" then table.insert(whereParts, string.format("stage='%s'", stage:gsub("'", "\\'"))) end
    if group and group ~= "" then table.insert(whereParts, string.format("groupname='%s'", group:gsub("'", "\\'"))) end
    
    local table_name = (type == "player") and "Player_Stats" or "Team_Stats"
    local results = cargo.query(table_name, table.concat(queryFields, ",") .. ", team", {
        where = table.concat(whereParts, " AND "),
        orderBy = (type == "player" and "finishes DESC" or "total_pts DESC"),
        limit = 100
    })
    
    if not results or #results == 0 then
        return '<div style="padding:20px; color:var(--text-muted); font-style:italic;">No statistics found for: ' .. table.concat(whereParts, " | ") .. '</div>'
    end
    
    local maxValues = {}
    for _, row in ipairs(results) do
        for _, key in ipairs(queryFields) do
            local val = tonumber(row[key]) or 0
            if val > (maxValues[key] or 0) then maxValues[key] = val end
        end
    end
    
    -- SWITCHED STYLE: modern-history-table (Clean, Dark Header)
    local root = html.create('div'):addClass('stats-table-wrapper')
    local tbl = root:tag('table'):addClass('modern-history-table sortable')
    
    local trHead = tbl:tag('thead'):tag('tr')
    for _, key in ipairs(colKeys) do
        local label = HEADERS[key:match("^%s*(.-)%s*$")] or key:upper()
        trHead:tag('th'):wikitext(label)
    end
    
    local tbody = tbl:tag('tbody')
    for i, row in ipairs(results) do
        local tr = tbody:tag('tr')
        for _, key in ipairs(colKeys) do
            local cleanKey = key:match("^%s*(.-)%s*$")
            local cell = tr:tag('td')
            
            if cleanKey == "rank" then
                cell:wikitext(i .. '.'):css("font-weight", "bold"):css("text-align", "center")
            elseif cleanKey == "team" then
                local teamName = row.team or ""
                cell:wikitext(getTeamLogo(teamName) .. '[[' .. teamName .. ']]'):css("text-align", "left"):css("white-space", "nowrap")
            elseif cleanKey == "player" then
                cell:wikitext('[[' .. (row.player or "") .. ']]'):css("text-align", "left"):css("font-weight", "bold")
            else
                -- Apply Smart Rounding
                local rawVal = row[cleanKey]
                local formattedVal = formatNumber(rawVal)
                cell:wikitext(formattedVal):css("text-align", "center")
                
                if not NO_HEATMAP[cleanKey] then
                    local style = getHeatmapStyle(rawVal, maxValues[cleanKey])
                    if style ~= "" then cell:attr("style", style .. " text-align:center;") end
                end
            end
        end
    end
    
    return tostring(root)
end

return p