Editing Module:Team

local function getGlobalDate()
	local globalDate = mw.ext.VariablesLua.var('date')
	if not(globalDate and globalDate ~= '') then
		globalDate = mw.ext.VariablesLua.var('tournament_edate')
		if not(globalDate and globalDate ~= '') then
			globalDate = mw.ext.VariablesLua.var('tournament_enddate')
			if not(globalDate and globalDate ~= '') then
				globalDate = mw.ext.VariablesLua.var('tournament_startdate')
				if not(globalDate and globalDate ~= '') then
					globalDate = mw.ext.VariablesLua.var('tournament_sdate')
					if not(globalDate and globalDate ~= '') then
						globalDate = os.date( '%F' )
					end
				end
			end
		end
	end
	return globalDate
end

local function getNullMessage(name)
	mw.log('Missing team: ' .. name)
	return '<small class="error">No team template exists for name "' .. name .. '".</small>' .. '[[Category:Pages with missing team templates]]'
end

local custom = {
	['team'] = function (name, img) return '<span data-highlightingclass="' .. name .. '" class="team-template-team-standard"><span class="team-template-image-icon">[[File:' .. img .. '|100x50px|link=' .. name .. ']]</span> <span class="team-template-text">[[' .. name .. ']]</span></span>' end,
	['team2'] = function (name, img) return '<span data-highlightingclass="' .. name .. '" class="team-template-team-standard"><span class="team-template-text">[[' .. name .. ']]</span> <span class="team-template-image-icon">[[File:' .. img .. '|100x50px|link=' .. name .. ']]</span></span>' end,
	['teamshort'] = function (name, img) return '<span data-highlightingclass="' .. name .. '" class="team-template-team-short"><span class="team-template-image-icon">[[File:' .. img .. '|100x50px|link=' .. name .. ']]</span> <span class="team-template-text">[[' .. name .. ']]</span></span>' end,
	['team2short'] = function (name, img) return '<span data-highlightingclass="' .. name .. '" class="team-template-team-short"><span class="team-template-text">[[' .. name .. ']]</span> <span class="team-template-image-icon">[[File:' .. img .. '|100x50px|link=' .. name .. ']]</span></span>' end,
	['teambracket'] = function (name, img) return '<span data-highlightingclass="' .. name .. '" class="team-template-team-bracket"><span class="team-template-image-icon">[[File:' .. img .. '|100x50px|link=|' .. name .. ']]</span> <span class="team-template-text">' .. name .. '</span></span>' end,
	['teamicon'] = function (name, img) return '<span data-highlightingclass="' .. name .. '" class="team-template-team-icon"><span class="team-template-image-icon">[[File:' .. img .. '|100x50px|link=' .. name .. ']]</span></span>' end,
	['teampart'] = function (name, img) return '<span class="team-template-team-part"><span class="team-template-image-icon">[[File:' .. img .. '|100x50px|link=' .. name .. ']]</span></span>' end,
	['iconfile'] = function (_, img) return img end,
	['imageFile'] = function (_, img) return img end,
	['imageFileDark'] = function (_, img) return img end,
	['teampage'] = function (name, _) return name end
}

local function getOverride(form, name, data)
	if name == nil then return nil end
	local overrides = mw.loadData('Module:Team/override')
	if overrides.templates[name:lower()] then
		return overrides.templates[name:lower()][form:lower()]
	elseif overrides.game[name:lower()] and data then
		return custom[form:lower()](data, overrides.games[mw.ext.VariablesLua.var('tournament_game', ''):lower()])
	elseif overrides.games[name:lower()] and data then
		return custom[form:lower()](data, overrides.games[name:lower()])
	elseif name:lower() == 'noteam' then
		return overrides.templates[''][form:lower()]
	else
		return nil
	end
end

local function getTemplate(form, name, date, skipOverride)
	if not(skipOverride) then
		local override = getOverride(form, name, date)
		if override then return override end
	end
	if mw.ext.TeamTemplate.teamexists(name) then return mw.ext.TeamTemplate[form](name, date or getGlobalDate())
	elseif mw.ext.TeamTemplate.teamexists(mw.text.trim(name)) then mw.log("Trimmed needed on team name: ".. name) return mw.ext.TeamTemplate[form](mw.text.trim(name), date or getGlobalDate()) .. '[[Category:Pages with trimmed team templates]]'
	elseif mw.ext.TeamTemplate.teamexists(name:gsub("_", " ")) then mw.log("Underscore in team name: ".. name) return mw.ext.TeamTemplate[form](name:gsub("_", " "), date or getGlobalDate()) .. '[[Category:Pages with underscore team templates]]'
	elseif mw.ext.TeamTemplate.teamexists(name:gsub(" ", "_")) then mw.log("Underscore in team name: ".. name) return mw.ext.TeamTemplate[form](name:gsub(" ", "_"), date or getGlobalDate()) .. '[[Category:Pages with underscore team templates]]'
	else return nil
	end
end

local p = {
	['team'] = function (_, name, date)
		return getTemplate('team', name, date) or getNullMessage(name)
	end,
	
	['team2'] = function (_, name, date)
		return getTemplate('team2', name, date) or getNullMessage(name)
	end,
	
	['short'] = function (_, name, date)
		return getTemplate('teamshort', name, date) or ('<div class="error">' .. name .. ' (missing)</div>' .. '[[Category:Pages with missing team templates]]')
	end,
	
	['short2'] = function (_, name, date)
		return getTemplate('team2short', name, date) or ('<span class="error">(missing) ' .. name .. '</span>' .. '[[Category:Pages with missing team templates]]')
	end,
	
	['bracket'] = function (_, name, date, skipOverride)
		return getTemplate('teambracket', name, date, skipOverride) or ('<span class="error">Missing: ' .. name .. '</span>' .. '[[Category:Pages with missing team templates]]')
	end,
	
	['bracketShort'] = function (_, name, date, skipOverride)
		if not(skipOverride) then
			local override = getOverride('teambracket', name, date)
			if override then return override end
		end
		local output
		if mw.ext.TeamTemplate.teamexists(name) then output = mw.ext.TeamTemplate.raw(name, date or getGlobalDate())
		elseif mw.ext.TeamTemplate.teamexists(mw.text.trim(name)) then output = mw.ext.TeamTemplate.raw(mw.text.trim(name), date or getGlobalDate())
		elseif mw.ext.TeamTemplate.teamexists(name:gsub("_", " ")) then output = mw.ext.TeamTemplate.raw(name:gsub("_", " "), date or getGlobalDate())
		elseif mw.ext.TeamTemplate.teamexists(name:gsub(" ", "_")) then output = mw.ext.TeamTemplate.raw(name:gsub(" ", "_"), date or getGlobalDate())
		else mw.log('Missing team: ' .. name) return '<span class="error">Missing: ' .. name .. '</span>' .. '[[Category:Pages with missing team templates]]'
		end
		if output.image ~= '' then
			return '<span data-highlightingclass="' .. output.page .. '" class="team-template-team-bracket"><span class="team-template-image-icon lightmode">[[File:' .. output.image .. '|100x50px|link=]]</span><span class="team-template-image-icon darkmode" style="display:none">[[File:' .. (output.imagedark ~= '' and output.imagedark or output.image) .. '|100x50px|link=]]</span> <span class="team-template-text">' .. output.shortname .. '</span></span>'
		else
			return '<span data-highlightingclass="' .. output.page .. '" class="team-template-team-bracket"><span class="team-template-image-legacy">[[File:' .. output.legacyimage .. '|link=]]</span> <span class="team-template-text">' .. output.shortname .. '</span></span>'
		end
	end,
	
	['icon'] = function (_, name, date)
		return getTemplate('teamicon', name, date) or getNullMessage(name)
	end,
	
	['iconFile'] = function (_, name, date)
		local output = getOverride('iconfile', name, date)
		if output then return output
		elseif mw.ext.TeamTemplate.teamexists(name) then output = mw.ext.TeamTemplate.raw(name, date or getGlobalDate())
		elseif mw.ext.TeamTemplate.teamexists(mw.text.trim(name)) then output = mw.ext.TeamTemplate.raw(mw.text.trim(name), date or getGlobalDate())
		elseif mw.ext.TeamTemplate.teamexists(name:gsub("_", " ")) then output = mw.ext.TeamTemplate.raw(name:gsub("_", " "), date or getGlobalDate())
		elseif mw.ext.TeamTemplate.teamexists(name:gsub(" ", "_")) then output = mw.ext.TeamTemplate.raw(name:gsub(" ", "_"), date or getGlobalDate())
		else mw.log('Missing team: ' .. name .. ' (icon)') return mw.loadData('Module:Team/override').games['']
		end
		return output.image ~= '' and output.image or output.legacyimage
	end,
	
	['imageFile'] = function (_, name, date)
		local output
		if mw.ext.TeamTemplate.teamexists(name) then output = mw.ext.TeamTemplate.raw(name, date or getGlobalDate())
		elseif mw.ext.TeamTemplate.teamexists(mw.text.trim(name)) then output = mw.ext.TeamTemplate.raw(mw.text.trim(name), date or getGlobalDate())
		elseif mw.ext.TeamTemplate.teamexists(name:gsub("_", " ")) then output = mw.ext.TeamTemplate.raw(name:gsub("_", " "), date or getGlobalDate())
		elseif mw.ext.TeamTemplate.teamexists(name:gsub(" ", "_")) then output = mw.ext.TeamTemplate.raw(name:gsub(" ", "_"), date or getGlobalDate())
		else mw.log('Missing team: ' .. name .. ' (icon)') return nil
		end
		return output.image ~= '' and output.image or nil
	end,
	
	['imageFileDark'] = function (_, name, date)
		local output
		if mw.ext.TeamTemplate.teamexists(name) then output = mw.ext.TeamTemplate.raw(name, date or getGlobalDate())
		elseif mw.ext.TeamTemplate.teamexists(mw.text.trim(name)) then output = mw.ext.TeamTemplate.raw(mw.text.trim(name), date or getGlobalDate())
		elseif mw.ext.TeamTemplate.teamexists(name:gsub("_", " ")) then output = mw.ext.TeamTemplate.raw(name:gsub("_", " "), date or getGlobalDate())
		elseif mw.ext.TeamTemplate.teamexists(name:gsub(" ", "_")) then output = mw.ext.TeamTemplate.raw(name:gsub(" ", "_"), date or getGlobalDate())
		else mw.log('Missing team: ' .. name .. ' (icon)') return nil
		end
		return (output.imagedark ~= '' and output.imagedark) or (output.image ~= '' and output.image) or nil
	end,
	
	['part'] = function (_, name, date)
		return getTemplate('teampart', name, date) or getNullMessage(name)
	end,
	
	['page'] = function (_, name, date)
		local override = getOverride('teampage', name, date)
		if override then return override
		elseif mw.ext.TeamTemplate.teamexists(name) then return mw.ext.TeamTemplate.teampage(name, date or getGlobalDate())
		elseif mw.ext.TeamTemplate.teamexists(mw.text.trim(name)) then return mw.ext.TeamTemplate.teampage(mw.text.trim(name), date or getGlobalDate())
		elseif mw.ext.TeamTemplate.teamexists(name:gsub("_", " ")) then return mw.ext.TeamTemplate.teampage(name:gsub("_", " "), date or getGlobalDate())
		elseif mw.ext.TeamTemplate.teamexists(name:gsub(" ", "_")) then return mw.ext.TeamTemplate.teampage(name:gsub(" ", "_"), date or getGlobalDate())
		else return name
		end
	end,
	
	['shortname'] = function (_, name, date)
		return (getTemplate('raw', name, date) or {}).shortname or getNullMessage(name)
	end,
	
	['name'] = function (_, name, date)
		return (getTemplate('raw', name, date) or {}).name or getNullMessage(name)
	end,
	
	['template'] = function (_, name, date)
		return (getTemplate('raw', name, date) or {}).templatename or getNullMessage(name)
	end,
	
	['bracketname'] = function (_, name, date, skipOverride)
		if not(skipOverride) then
			local override = getOverride('teambracket', name, date)
			if override then return override end
		end
		local output
		if mw.ext.TeamTemplate.teamexists(name) then output = mw.ext.TeamTemplate.raw(name, date or getGlobalDate())
		elseif mw.ext.TeamTemplate.teamexists(mw.text.trim(name)) then output = mw.ext.TeamTemplate.raw(mw.text.trim(name), date or getGlobalDate())
		elseif mw.ext.TeamTemplate.teamexists(name:gsub("_", " ")) then output = mw.ext.TeamTemplate.raw(name:gsub("_", " "), date or getGlobalDate())
		elseif mw.ext.TeamTemplate.teamexists(name:gsub(" ", "_")) then output = mw.ext.TeamTemplate.raw(name:gsub(" ", "_"), date or getGlobalDate())
		else mw.log('Missing team: ' .. name) return '<span class="error">Missing: ' .. name .. '</span>' .. '[[Category:Pages with missing team templates]]'
		end
		if output.image ~= '' then
			return '<span data-highlightingclass="' .. output.page .. '" class="team-template-team-short"><span class="team-template-image-icon">[[File:' .. output.image .. '|link=' .. name .. ']]</span> <span class="team-template-text">[[' .. output.bracketname .. ']]</span></span>'
		else
			return '<span data-highlightingclass="' .. output.page .. '" class="team-template-team-short"><span class="team-template-image-legacy">[[File:' .. output.legacyimage .. '|link=' .. name .. ']]</span> <span class="team-template-text">[[' .. output.bracketname .. ']]</span>'
		end
	end,
	
	['override'] = getOverride,
}

function p.get(frame)
	local args = {}
	for k, v in pairs(frame.args) do
		if v ~= '' then args[k] = mw.text.trim(v) end
	end
	if args[1] then
		return p[args[1]](frame, args[2] or '', args[3])
	else
		return '<span class="error">Invalid team template form specified.</span>' .. '[[Category:Pages with invalid team templates]]'
	end
end

function p.queryDB(type, name, date)
	if mw.ext.TeamTemplate.teamexists(name) then
		return mw.ext.TeamTemplate[type](name, date)
	else
		return nil
	end
end

function p.queryRaw(name, date)
	if mw.ext.TeamTemplate.teamexists(name) then
		return mw.ext.TeamTemplate.raw(name, date)
	else
		return nil
	end
end

function p.queryHistorical(name)
	return mw.ext.TeamTemplate.raw_historical(name)
end

local Set
function p.queryHistoricalNames(name)
	if not Set then Set = require('Module:Set') end
	if mw.ext.TeamTemplate.teamexists(name) then
		local index = mw.ext.TeamTemplate.raw_historical(name)
		if index then
			local set = Set{}
			for _, template in pairs(index) do
				set:add(mw.ext.TeamTemplate.teampage(template))
			end
			return set:toArray()
		else
			return {mw.ext.TeamTemplate.teampage(name)}
		end
	else
		return nil
	end
end

return p