-- These are two ComputerCraft Lua programs that I've made for a Capture the Flag game in Tekkit.
-- This is a rough program and not considered complete. I'm posting it here for a RFC.
-- Check out my reddit thread at http://www.reddit.com/r/tekkit/comments/131787/work_in_progress_any_ideas_for_a_ctf_map/

-- Referee
-- The referee is a ComputerCraft computer that's intended to be in an observation area. It's programmed
-- to read signals from three bundled cables, and then provide a simple display on a monitor.
-- The bundled cable off the right is for the red team, the left is for the blue team, and the back is
-- reserved for a transmitter that signals the start of the game.

function checkError()
	-- this will check to see if there's an error for either team's computer and return a bool
	if (redstone.testBundledInput("left", colors.orange)) or (redstone.testBundledInput("right", colors.orange)) then
		return true
	else
		return false
	end
end

function checkReady()
	-- this will check to see that both teams are ready to start the game
	if (redstone.testBundledInput("left", colors.lime)) and (redstone.testBundledInput("right", colors.lime)) then
		return true
	else
		return false
	end
end

function setStart()
	redstone.setBundledOutput("back", colors.white)
end


while true do
	
	print("Please wait five seconds while I make sure there are no errors.\n")
	sleep(5)
	if (checkError()) then
		print("There has been an error and I am unable to continue.")
		print("Make sure there is a disk in the left disk drive of each team's computer.")
		break
	end
	
	print("Ready to start.")
	print("Waiting for each team to signal that they are ready.\n")
	
	while true do
		os.pullEvent("redstone")
		if checkReady() then
			print("Both teams ready!")
			print("Starting game!")
			setStart()
			break
		end
	end
	
	local win = false
	
	while true do
	
		monitor = peripheral.wrap("top")
		monitor.setTextScale(1.5)
		term.redirect(monitor)
		
		os.pullEvent("redstone")
		
		term.clear()
		term.setCursorPos(10,1)
		
		term.write("RED")
		
		term.setCursorPos(20,1)
		term.write("BLUE")
		
		term.setCursorPos(1,2)
		term.write("FLAG")
		
		-- stuff for flag here
		if (redstone.testBundledInput("right", colors.magenta)) then
			term.setCursorPos(10,2)
			term.write("STOLEN")
		end
		
		if (redstone.testBundledInput("left", colors.magenta)) then
			term.setCursorPos(20,2)
			term.write("STOLEN")
		end
		
		if (redstone.testBundledInput("left", colors.lightBlue)) then
			-- red team has captured blue flag
			term.setCursorPos(10,2)
			term.write("CAPTURED!")
		end
		
		if (redstone.testBundledInput("right", colors.lightBlue)) then
			term.setCursorPos(20,2)
			term.write("CAPTURED!")
		end
		
		if (redstone.testBundledInput("left", colors.yellow)) then
			term.clear()
			term.setCursorPos(20,4)
			term.write("GAME OVER!")
			term.setCursorPos(20,5)
			term.write("BLUE TEAM WINS!")
			win = true
			break
		end
		
		if (redstone.testBundledInput("right", colors.yellow)) then
			term.clear()
			term.setCursorPos(20,4)
			term.write("GAME OVER!")
			term.setCursorPos(17,5)
			term.write("RED TEAM WINS!")
			win = true
			break
		end		
		
	end
	
	if win then
		term.setCursorPos(1,18)
		term.restore()
		break
	end
	
end


-- RedTeamBase
-- Each base has a ComputerCraft setup with two disk drives with bundled cable off the back, and
-- transmitters/receivers. The "flags" for the game are labeled floppy disks for ComputerCraft. The
-- program for each base is identical except where the red/blue things have been transposed.

function isStolen()
	if disk.hasData("left") then
		label = disk.getLabel("left")
		if label == "Red Flag" then
			return false
		end
	end
	if disk.hasData("right") then
		label = disk.getLabel("right")
		if label == "Red Flag" then
			return false
		end
	end
	return true
end

function isCaptured()
	if disk.hasData("left") then
		label = disk.getLabel("left")
		if label == "Blue Flag" then
			return true
		end
	end
	if disk.hasData("right") then
		label = disk.getLabel("right")
		if label == "Blue Flag" then
			return true
		end
	end
	return false
end

function amReady()
	if disk.hasData("left") then
		label = disk.getLabel("left")
		if label == "Red Flag" then
			return true
		else
			disk.setLabel("left", "Red Flag")
			return true
		end
	else
		return false
	end
end

while true do
	redstone.setBundledOutput("back",0)
	if amReady() then
		print("Ready to start!")
		print("Press any key to signal the referee that your team is ready to start.")
		local event, param1 = os.pullEvent("key")
		print("Regeree signaled. Game will start once other team is ready.")
		curOut = redstone.getBundledOutput("back")
		newOut = colors.combine(curOut, colors.lime)
		redstone.setBundledOutput("back",newOut)
		while true do
			if redstone.testBundledInput("back",colors.white) then -- start signal is on
				print("Game started!")
				while true do
					
					if (isStolen()) then
						-- turn on "stolen" signal
						curOut = redstone.getBundledOutput("back")
						newOut = colors.combine(curOut, colors.magenta)
						redstone.setBundledOutput("back",newOut)
						print("Flag stolen!")
					else
						-- turn off the "stolen" signal
						curOut = redstone.getBundledOutput("back")
						newOut = colors.subtract(curOut, colors.magenta)
						redstone.setBundledOutput("back",newOut)
						if isCaptured() then
							-- turn on "captured" signal
							curOut = redstone.getBundledOutput("back")
							newOut = colors.combine(curOut, colors.lightBlue)
							redstone.setBundledOutput("back",newOut)
							print("Enemy flag captured! You must now wait 30 seconds before claiming victory.")
							sleep(30) -- the team must hold onto the flag for a while before claiming victory
							if not (isStolen()) and (isCaptured()) then
								-- turn on win signal
								curOut = redstone.getBundledOutput("back")
								newOut = colors.combine(curOut, colors.yellow)
								redstone.setBundledOutput("back",newOut)
								print("Win conditions met. Check referee for final result.")
								break
							else
								-- turn off captured signal
								curOut = redstone.getBundledOutput("back")
								newOut = colors.subtract(curOut, colors.lightBlue)
								redstone.setBundledOutput("back",newOut)
								print("Enemy flag retaken! Game continues!")
							end
						end
					end
					sleep(1)
					
				end
				
				break
				
			else
				sleep(1)
			end
			
		end
		
		break
		
	else
		-- turn on the error signal
		curOut = redstone.getBundledOutput("back")
		newOut = colors.combine(curOut, colors.orange)
		redstone.setBundledOutput("back",newOut)
		print("ERROR: There must be a disk in the left drive before I can start.")
		print("Insert a disk and restart the computer.")
		break
	end
end