import asyncio
import slixmpp

class Connection(slixmpp.ClientXMPP):
	def __init__(self, jid, password):
		self.__username = jid.split('@')[0]
		self.__password = password
		self.__login_failed = False
		
		super().__init__(jid, password)
		
		self.register_plugin('xep_0077')
		self.add_event_handler('connected', self.on_connected)
		self.add_event_handler('disconnected', self.on_disconnected)
		self.add_event_handler('failed_auth', self.on_failed_auth)
		self.add_event_handler('session_start', self.on_session_start)
		self.add_event_handler('session_end', self.on_session_end)
	
	def exception(self, exception):
		print("exception", exception)
		raise exception
	
	def on_connected(self, event):
		print("connected", event)
	
	def on_disconnected(self, event):
		print("disconnected", event)
	
	def on_failed_auth(self, event):
		print("failed_auth", event)
		self.__login_failed = True
	
	async def on_session_start(self, event):
		print("session_start", event)
		if self.__login_failed:
			print(" login failed, registering")
			
			iq = await (self['xep_0077'].get_registration())
			print(" recv:", iq)
			registration_form = iq['register']
			
			registration_request = self['xep_0077'].stanza.Register()
			registration_request.set_fields(registration_form.get_fields())
			for field in registration_request.get_fields():
				if field == 'username':
					registration_request[field] = self.__username
				elif field == 'password':
					registration_request[field] = self.__password
				elif field == 'email':
					registration_request[field] = 'bobby@example.net'
				else:
					print(" unsupported field:", field)
			iq = self['xep_0077'].xmpp.make_iq_set(registration_request)
			print(" send:", iq)
			print(" recv:", (await iq.send()))
			self.disconnect() # disconnect
		
		else:
			self.send_presence()
			self.get_roster()
			
			print(" login successful, unregistering")
			
			iq = await (self['xep_0077'].get_registration())
			print(" recv:", iq)
			registration_form = iq['register']
			
			registration_request = self['xep_0077'].stanza.Register()
			registration_request.set_fields(['remove'])
			iq = self['xep_0077'].xmpp.make_iq_set(registration_request)
			print(" send:", iq)
			await iq.send()
			# the server will disconnect
	
	def on_session_end(self, event):
		print("session_end", event)
		quit()
	
if __debug__ and __name__ == '__main__':
	connection = Connection('bobby@tigase.example.net/x12345', 'abcdef') # use tigase address server here, change username
	
	connection.connect()
	try:
		connection.process()
	except KeyboardInterrupt:
		connection.disconnect()
		connection.process(timeout=5)

if (session.isAuthorized()) {

if (session.isAuthorized() && ! session.isAnonymous()) {