from omegle import OmegleChat
import sys, logging

colors = {"default":0, "black":30, "red":31, "green":32, "yellow":33,
          "blue":34,"magenta":35, "cyan":36, "white":37, "black":38,
          "black":39} #33[%colors%m
#        color_string = "\033[%dm\033[1m" % colors[color]

class OmegleProxyChat(OmegleChat):
	def __init__(self,name,color=0):
		OmegleChat.__init__(self)
		self.name = name
		self.color = color
	def c(self,str):
		return "\033[%sm\033[1m%s\033[0m\033[0m" %(self.color, str)

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

	def on_message(self,message):
		print self.c("[MSG] %s: %s" %(self.name, message))
		self.partner.send(message)
	def on_connect(self):
		print self.c("[EVT] %s Connection confirmed" % self.name)
	def on_typing(self):
		print self.c("[EVT] %s is typing" % self.name)
		self.partner.typing()
	def on_stopped_typing(self):
		print self.c("[EVT] %s stopped typing" % self.name)
		self.partner.stopped_typing()
	def on_disconnect(self):
		print self.c("[EVT] %s disconnect" % self.name)
		self.partner.disconnect()

logging.basicConfig(level=logging.DEBUG)

A=OmegleProxyChat("A",colors['red'])
B=OmegleProxyChat("B",colors['blue'])

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

A.start()
B.start()

print "press enter to abort"
sys.stdin.readline()

A.disconnect()
B.disconnect()