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 style="background-color: #FFD700;">Gold</th><th style="background-color: C0C0C0;">Silver</th><th style="background-color: #CD7F32;">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
