import sys
import time
import logging
from threading import Event
from omegle import OmegleChat


__all__ = ['OmegleProxyChat']


colors = {"default":0, "black":30, "red":31, "green":32, "yellow":33,
          "blue":34,"magenta":35, "cyan":36, "white":37, "black":38, "black":39}




class OmegleProxyChat(OmegleChat):
	def __init__(self,disconnect_event,name,color=0):
		OmegleChat.__init__(self,name="(" + name + ")")
		self.disconnect_event = disconnect_event
		self.name = name
		self.color = color

	def c(self,str,bold=False):
		if bold:
			return "\033[1m\033[%sm%s\033[0m\033[0m" %(self.color, str)
		else:
			return "\033[%sm%s\033[0m" %(self.color, str)

	def set_partner(self,partner):
		self.partner = partner

	def on_message(self,message):
		print self.c("%s [MSG] %s: %s" %(time.strftime("%H:%M:%S"), self.name, message),bold=True)
		if self.partner.is_connected:
			self.partner.send(message)

	def on_connect(self):
		print self.c("%s [EVT] %s Connection confirmed" % (time.strftime("%H:%M:%S"), self.name))

	def on_typing(self):
		print self.c("%s [EVT] %s is typing" % (time.strftime("%H:%M:%S"), self.name))
		if self.partner.is_connected:
			self.partner.typing()

	def on_stopped_typing(self):
		print self.c("%s [EVT] %s stopped typing" % (time.strftime("%H:%M:%S"), self.name))
		if self.partner.is_connected:
			self.partner.stopped_typing()

	def on_disconnect(self):
		print self.c("%s [EVT] %s disconnect" % (time.strftime("%H:%M:%S"), self.name))
		if self.partner.is_connected:
			self.partner.disconnect()
		self.disconnect_event.set()



if __name__ == "__main__":
	logging.basicConfig(level=logging.DEBUG)
	print "press ctrl-c to abort"
	
	event = Event()
	A=OmegleProxyChat(event,"A",colors['red'])
	B=OmegleProxyChat(event,"B",colors['blue'])

	A.set_partner(B)
	B.set_partner(A)

	A.start()
	B.start()

	try:
		while True:
			event.wait(0.5)
			if event.isSet():
				print "One part disconnected, exiting"
				break
	except KeyboardInterrupt:
		print "CTRL-C pressed, exiting"

	if A.is_connected: A.disconnect()
	if B.is_connected: B.disconnect()