Module:GroupDraw

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

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

local function getTeamDisplay(teamName, tData, mData)
local lightLogo = "Shield_team.png"
local darkLogo = "Shield_team_dark.png"
local displayName = teamName

if tData then
if tData.display_name and tData.display_name ~= "" then
displayName = tData.display_name
end
if tData.image and tData.image ~= "" then
lightLogo = tData.image
darkLogo = (tData.image_dark and tData.image_dark ~= "") and tData.image_dark or lightLogo
end
end

if lightLogo == "Shield_team.png" and mData then
if mData.image and mData.image ~= "" then
lightLogo = mData.image
else
lightLogo = teamName .. ".png"
end
if mData.image_dark and mData.image_dark ~= "" then
darkLogo = mData.image_dark
else
darkLogo = lightLogo
end
end

local container = html.create('div'):addClass('team-cell-flex')
local logoWrap = container:tag('span'):addClass('team-logo-wrapper')
logoWrap:tag('span'):addClass('logo-lightmode'):wikitext('[[File:' .. lightLogo .. '|30px|link=|class=team-logo]]')
logoWrap:tag('span'):addClass('logo-darkmode'):wikitext('[[File:' .. darkLogo .. '|30px|link=|class=team-logo]]')
container:tag('span'):wikitext('[[' .. teamName .. '|' .. displayName .. ']]')

return tostring(container)
end

function p.main(frame)
local args = frame:getParent().args
local mainTitle = args.title
local tournamentName = args.tournament or mw.title.getCurrentTitle().text

local groupsData = {}
local allTeams = {}
local maxRows = 0

for i = 1, 20 do
local gTitle = args['g' .. i]
local gTeamsStr = args['g' .. i .. 'teams']

if gTitle and gTeamsStr then
local teamList = mw.text.split(gTeamsStr, ",")
local cleanedTeams = {}

for _, tName in ipairs(teamList) do
local trimmedTeam = mw.text.trim(tName)
if trimmedTeam ~= "" then
table.insert(cleanedTeams, trimmedTeam)
table.insert(allTeams, "'" .. trimmedTeam:gsub("'", "\\'") .. "'")
end
end

table.insert(groupsData, { heading = gTitle, squad = cleanedTeams })
if #cleanedTeams > maxRows then maxRows = #cleanedTeams end
end
end

if #groupsData == 0 then return "" end

local teamSql = table.concat(allTeams, ",")
local tourneyDb = {}
local masterDb = {}

if cargo and cargo.query and teamSql ~= "" then
local tResults = cargo.query("Tournament_Teams", "team, display_name, image, image_dark", { where = "tournament='" .. tournamentName:gsub("'", "\\'") .. "' AND team IN (" .. teamSql .. ")", limit = 500 })
if tResults then
for _, row in ipairs(tResults) do tourneyDb[row.team] = row end
end

local mResults = cargo.query("Teams", "name, image, image_dark", { where = "name IN (" .. teamSql .. ")", limit = 500 })
if mResults then
for _, row in ipairs(mResults) do masterDb[row.name] = row end
end
end

local wrapper = html.create('div'):addClass('modern-group-draw')
local drawTable = wrapper:tag('table'):addClass('wikitable sharp-table')

if mainTitle and mainTitle ~= "" then
local titleRow = drawTable:tag('tr')
titleRow:tag('th'):attr('colspan', tostring(#groupsData)):addClass('distinct-main-header'):wikitext(mainTitle)
end

local topHeader = drawTable:tag('tr')
for _, grp in ipairs(groupsData) do
topHeader:tag('th'):addClass('distinct-group-header'):wikitext(grp.heading)
end

for rIndex = 1, maxRows do
local tableRow = drawTable:tag('tr')
for _, grp in ipairs(groupsData) do
local cellContent = tableRow:tag('td')
local targetTeam = grp.squad[rIndex]

if targetTeam then
local tData = tourneyDb[targetTeam]
local mData = masterDb[targetTeam]
cellContent:wikitext(getTeamDisplay(targetTeam, tData, mData))
end
end
end

return tostring(wrapper)
end

return p