#organiser.py
#by VladVP

from commands import add, admin, name

on_by_default = True

@name('to')
@admin
def toggleorganisation(connection):
	global on_by_default
	if on_by_default is True:
		on_by_default = False
		connection.protocol.send_chat("Team organisation disabled!")
	else:
		on_by_default = True
		connection.protocol.send_chat("Team organisation enabled!")

add(toggleorganisation)

def apply_script(protocol, connection, config):
	class orgConnection(connection):
		def spawn(self):
			if on_by_default:
				protocol = self.protocol
				blue = config.get("team_preset", None).get("blue", None)
				if blue == None: return False
				green = config.get("team_preset", None).get("green", None)
				if green == None: return False
				if   self.name in blue:
					self.team = protocol.blue_team
				elif self.name in green:
					self.team = protocol.green_team
				else:
					self.team = protocol.spectator_team
			return connection.spawn(self)
	return protocol, orgConnection