Module:Medal tally

local p = {}

function p.render(frame)
    local args = frame:getParent().args
    local header = args.header or "Medal Tally"
    local class = args.class or ""

    local result = {}
    table.insert(result, string.format('<table class="wikitable %s" style="text-align:center;">', class))
    table.insert(result, string.format('<caption>%s</caption>', header))
    table.insert(result, '<tr><th>Rank</th><th>Team</th><th>Gold</th><th>Silver</th><th>Bronze</th><th>Total</th></tr>')

    local medals = {}
    local totalGold, totalSilver, totalBronze = 0, 0, 0

    for i = 1, 8 do
        local team = args['team' .. i]
        local gold = tonumber(args['gold' .. i]) or 0
        local silver = tonumber(args['silver' .. i]) or 0
        local bronze = tonumber(args['bronze' .. i]) or 0
        local total = gold + silver + bronze

        if team then
            table.insert(medals, {rank = i, team = team, gold = gold, silver = silver, bronze = bronze, total = total})
            totalGold = totalGold + gold
            totalSilver = totalSilver + silver
            totalBronze = totalBronze + bronze
        end
    end

    table.sort(medals, function(a, b)
        if a.gold ~= b.gold then return a.gold > b.gold end
        if a.silver ~= b.silver then return a.silver > b.silver end
        return a.bronze > b.bronze
    end)

    for index, entry in ipairs(medals) do
        table.insert(result, string.format('<tr><td>%d</td><td>%s</td><td>%d</td><td>%d</td><td>%d</td><td>%d</td></tr>',
            index, entry.team, entry.gold, entry.silver, entry.bronze, entry.total))
    end

    local totalMedals = totalGold + totalSilver + totalBronze
    table.insert(result, string.format('<tr><th colspan="2">Total</th><td>%d</td><td>%d</td><td>%d</td><td>%d</td></tr>',
        totalGold, totalSilver, totalBronze, totalMedals))
    table.insert(result, '</table>')

    return table.concat(result, "\n")
end

return p