Editing Module:Bracket

local getArgs = require('Module:Arguments').getArgs
local html = require('Module:MatchHTML')
local Bracket = {}

local VariablesLua = mw.ext.VariablesLua

function Bracket.main(frame)
	local args = getArgs(frame)
	if args.teams == nil then return 'Have to enter a "teams" value as a Bracket template paramater' end
	-- if best-of set to bracket, then set VariableLua for it for entire bracket to use
	if args.bestof then VariablesLua.vardefine('bestof', args.bestof) end
	if args.solo then VariablesLua.vardefine('solo', 1) end
    if args.twitch then VariablesLua.vardefine('twitch', args.twitch) end
	if args.youtube then VariablesLua.vardefine('youtube', args.youtube) end
	-- set const values
	local roundW = args.roundW or 180
	-- create all the html parent elements
	local container = mw.html.create('div'):addClass('bracket')
	if args.lteams then container:addClass('de') end
	local wrapper = mw.html.create('div'):addClass('bracket-wrapper')
	local ubheader = mw.html.create('div'):addClass('bracket-header')
	local ubrounds = mw.html.create('div'):addClass('bracket-rounds')
	local lbheader = mw.html.create('div'):addClass('bracket-header')
	local lbrounds = mw.html.create('div'):addClass('bracket-rounds')
	container:node(wrapper)
	-- get max amount of matches for the first round
	local maxUbroundAmount = args.rounds or math.ceil(math.log(args.teams) / math.log(2))
	local ubroundAmount = 0
	local lbroundAmount = 0

	local upper = args.upper ~= nil and mw.text.jsonDecode(args.upper) or {}
	local lower = args.lower ~= nil and mw.text.jsonDecode(args.lower) or {}
	local lteams = tonumber(args.lteams)

	local matchAmount = (2^maxUbroundAmount)
	local upperMatches = args.teams and args.teams / 2 or 0
	local lowerMatches = lteams and lteams / 2 or 0
	local dropping = upperMatches
	local upperRounds = {}
	local lowerRounds = {}

	-- Go through upper bracket to see how many rounds and matches it has
	while tonumber(matchAmount) > 1 do
		ubroundAmount = ubroundAmount + 1
	
		local round = upper['R' .. ubroundAmount] ~= nil and mw.text.jsonDecode(upper['R' .. ubroundAmount]) or nil
		local roundTitle = (round and round.title) or lteams and 'Upper round ' .. ubroundAmount or 'Round ' .. ubroundAmount 
		
		-- if round.matches is lower than previous round divided by 2
		if (round and round.matches) and round.matches * 2 < matchAmount then
			matchAmount = matchAmount / 2
		else 
			matchAmount = (round and round.matches) or matchAmount / 2
		end
		upperRounds[ubroundAmount] = {
			matches = tonumber(matchAmount),
			title = roundTitle
		}
	end
	
	-- If lower bracket exists then populate lowerRounds and adjust upperRounds
	if lteams ~= nil then
		local ubRound = 1
		if lteams > 0 then ubRound = 0 end
		local lbRound = 1
		local lower = args.lower ~= nil and mw.text.jsonDecode(args.lower) or {}
		local upperMatches = upperRounds[1].matches
		local lowerMatches = lteams > 0 and lteams / 2 or upperMatches / 2
		if args.ladvance then lowerMatches = lowerMatches + args.ladvance / 2 end
		-- Start loop
		while #upperRounds >= ubRound and lowerMatches >= 1 do
			upperMatches = (upperRounds[ubRound] and upperRounds[ubRound].matches) or 0
			local nextUpperMatches = upperRounds[ubRound + 1] and upperRounds[ubRound + 1].matches or 0
			local round = lower['R' .. lbRound] ~= nil and mw.text.jsonDecode(lower['R' .. lbRound]) or {}
			local roundTitle = round.title or upperMatches ~= 1 and 'LB Round ' .. lbRound or 'LB Final'
			-- If amount of matches is manually entered, then overwrite
			if round.matches then lowerMatches = round.matches end
			
			-- If lowerMatches is exactly 2 times lower than upperMatches
			if lowerMatches * 2 == upperMatches then 
				table.insert(lowerRounds, Bracket.newRound(lowerMatches, roundTitle))
				ubRound = ubRound + 1
			-- If lowerMatches is same as upperMatches
			elseif lowerMatches == upperMatches then
				table.insert(lowerRounds, Bracket.newRound(lowerMatches, roundTitle))
				if upperMatches ~= 1 then 
					table.insert(upperRounds, ubRound + 1, {matches = 0})
					nextUpperMatches = 0
				end
				ubRound = ubRound + 1
			-- Else if upperMatches is lower
			elseif upperMatches < lowerMatches then
				table.insert(lowerRounds, Bracket.newRound(lowerMatches, roundTitle))
			end
			
			lowerMatches = upperMatches == 0 and lowerMatches or nextUpperMatches / 2 + lowerMatches / 2
			if lbRound == 1 and args.ladvance then lowerMatches = lowerMatches / 2 end
			lbRound = lbRound + 1
			if upperMatches == 0 then ubRound = ubRound + 1 end
		end
	end

	-- If lower and upper bracket both start with same amount of teams, then shift upper bracket one column to the right
	if args.teams == args.lteams then
		table.insert(upperRounds, 1, {matches = 0})
	end
	-- If ladvance is set then add empty round to UB round 2
	if args.ladvance then table.insert(upperRounds, 2, {matches = 0}) end

	-- draw upper bracket
	local upperOffset = 0
	for i = 1, #upperRounds do
		local roundNode = mw.html.create('div'):addClass('bracket-round')
		local round = upperRounds[i]
		local argsRound = upper['R' .. i - upperOffset] ~= nil and mw.text.jsonDecode(upper['R' .. i - upperOffset]) or nil
		local matches = round.matches
		if round.matches == 0 then upperOffset = upperOffset + 1 end

		-- Add round header to container
		ubheader:node(mw.html.create('div'):wikitext(round.title))
		
		for j = 1, matches do
			local match = argsRound and argsRound['M' .. j] or nil
			-- if no existing match html, then create it from scratch
			if match == nil then
				roundNode:node(mw.html.create('div'):addClass('match'):node(html.team('')):node(html.team('')))
				-- roundNode:node(mw.html.create('div'):addClass('match'):node(team1):node(team2):node(html.team('')):node(html.team('')))
			else 
				roundNode:node(match)
			end
		end
		--see if next rounds matchAmount is equal to or greater than current rounds
		local nextRound = upperRounds[i + 1] or {} 
		if tonumber(matches) <= tonumber(nextRound.matches or 0) then
			roundNode:addClass('round-offset')
		else
			if tonumber(matches) ~= 1 or lteams ~= nil then
				roundNode:addClass('conn-r')
			end
		end

		-- If previous round was empty
		if upperRounds[i - 1] and upperRounds[i - 1].matches == 0 then 
			if i ~= 2 then
				roundNode:addClass('long') 
			else 
				roundNode:addClass('first') 
			end
		end
		ubrounds:node(roundNode)
	end
	
	-- draw lower bracket
	if args.lteams then
		for i = 1, #lowerRounds do
			local roundNode = mw.html.create('div'):addClass('bracket-round')
			local round = lowerRounds[i]
			local argsRound = lower['R' .. i] ~= nil and mw.text.jsonDecode(lower['R' .. i]) or nil
			local matches = round.matches
			-- Add round header to container
			lbheader:node(mw.html.create('div'):wikitext(round.title))
			for j = 1, matches do
				local match = argsRound and argsRound['M' .. j] or nil
				-- if no existing match html, then create it from scratch
				if match == nil then
					roundNode:node(mw.html.create('div'):addClass('match'):node(html.team('')):node(html.team('')))
				else 
					roundNode:node(match)
				end
			end
			--see if next rounds matchAmount is equal to or greater than current rounds
			local nextRound = lowerRounds[i + 1] or {} 
			if tonumber(matches) <= tonumber(nextRound.matches or 0) then
				roundNode:addClass('round-offset')
			else
				if tonumber(matches) ~= 1 or lteams ~= nil then
					roundNode:addClass('conn-r')
				end
			end
			
			-- Add adjusting classes for connectors if next round is round offset
			if lowerRounds[i + 1] and lowerRounds[i + 2] and lowerRounds[i + 1].matches == lowerRounds[i + 1].matches then
				roundNode:addClass('conn-adj')
			end
			
			-- If previous round was empty
			if lowerRounds[i - 1] and lowerRounds[i - 1].matches == 0 then roundNode:addClass('long') end
			if i == #lowerRounds then roundNode:addClass('last') end
			lbrounds:node(roundNode)
		end
		-- Add grand final
		local maxMatches = 0
	    for _, obj in ipairs(upperRounds) do
	        if obj.matches and obj.matches > maxMatches then
	            maxMatches = obj.matches
	        end
	    end

		local grandFinal = mw.html.create('div'):addClass('grand-final'):css('min-width', roundW .. 'px'):css('width', roundW .. 'px'):css('height', (maxMatches * 56 + (maxMatches - 1) * 10 + 26 + 10) .. 'px')
		grandFinal:node(mw.html.create('div'):addClass('bracket-header'):node(mw.html.create('div'):wikitext('Grand Final')))
		if args.final == nil then
			grandFinal:node(mw.html.create('div'):addClass('match'):node(html.team('')):node(html.team('')))
		else 
			grandFinal:node(mw.html.create('div'):addClass('grand-final-round'):node(args.final))
		end
		container:node(grandFinal)
	end
	
	local maxRounds = #upperRounds <= #lowerRounds and #lowerRounds or #upperRounds
	ubheader:css('grid-template-columns', 'repeat(' .. maxRounds .. ', ' .. roundW .. 'px)')
	ubrounds:css('grid-template-columns', 'repeat(' .. maxRounds .. ', ' .. roundW .. 'px)')
	lbheader:css('grid-template-columns', 'repeat(' .. maxRounds .. ', ' .. roundW .. 'px)')
	lbrounds:css('grid-template-columns', 'repeat(' .. maxRounds .. ', ' .. roundW .. 'px)')
	
	wrapper:node(ubheader):node(ubrounds)
	if args.lteams then wrapper:node(lbheader):node(lbrounds) end

    if args.bestof then VariablesLua.vardefine('bestof', 0) end
    if args.twitch then VariablesLua.vardefine('twitch', 0) end
	if args.youtube then VariablesLua.vardefine('youtube', 0) end

	return container
end

function Bracket.newRound(matches, title)
	return {
		matches = matches,
		title = title
	}
end

return Bracket