import logging
import sys
from threading import Event
import random

import mh_python
import time
import pyPgSQL.PgSQL
from omegle import OmegleChat



class MegahalBot(OmegleChat):
	def __init__(self,disconnect_event,name,host="omegle.com",color=0):
		self.disconnect_event = disconnect_event
		self.name=name
		self.color = color
		OmegleChat.__init__(self,name=" Megahal",host=host)
		self.idlecount = 0
		

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


	def respond(self,message,prepend=""):
		r=mh_python.doreply(message.encode("utf-8"))
		print self.c("%s%s [MSG] %s: %s" %(prepend,time.strftime("%H:%M:%S"), self.name, r),bold=True)
		try:
			self.send(r) #.decode("utf-8"))
		except Exception,e:
			print e
			self.disconnect_event.set()

	def on_idle(self):
		self.idlecount += 1
		if self.idlecount > 8:
			self.logger.info("Idle > 8. disconnect()")
			self.disconnect()
		else:
			self.logger.info("Idle count %s/8" % self.idlecount)
			self.respond("hey?") #restart the conv.

	def on_message(self,message):
		print self.c("%s%s [MSG] %s: %s" %(
				"\033[1A\033[2K\033[30D",
				time.strftime("%H:%M:%S"), "Stranger", message),bold=True,color=32)
		self.idlecount = 0
		time.sleep(0.05)
		typestr = "%s [EVT] %s: typing" %(time.strftime("%H:%M:%S"),self.name)
		print self.c(typestr,bold=True,color=32)
		try:
			self.typing()
		except Exception,e:
			try:
				self.disconnect()
			except:
				self.disconnect_event.set()
			return
		time.sleep(random.randint(1,3))
		self.respond(message,prepend="\033[1A\033[2K\033[%sD"%(len(typestr))) #move up 1; delete line; move left %s

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


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



if __name__ == "__main__":
	mh_python.initbrain()

	conn = pyPgSQL.PgSQL.connect(database="omegle")
	cur = conn.cursor()
	#normal
#	sql = """SELECT message FROM omegle_messages 
 #                      WHERE LENGTH(message) < 35
  #                       AND NOT ( message ILIKE '%www.%'
   #                          OR message ILIKE '%http%'
    #                         OR message ILIKE '%@%.%')
     #                ;
#	"""

	#Offensive
	sql = """SELECT message FROM omegle_messages 
                       WHERE ( message ILIKE '%sex%' 
                            OR message ILIKE '%boob%' 
                            OR message ILIKE '%horny %'
                            OR message ILIKE '%pussy %'
                            OR message ILIKE '%ass %'
                            OR message ILIKE '%naked%'
                            OR message ILIKE '%fuck%'
                            OR LENGTH(message) < 7 AND CAST(RANDOM()*10 as integer)=1)
                        AND LENGTH(message) < 35
                        AND NOT ( message ILIKE '%www.%'
                            OR message ILIKE '%http%'
                            OR message ILIKE '%@%.%')
                     ;
	"""

	cur.execute(sql)
	messages = cur.fetchall()
	c=0
	for message in messages:
		c+=1
		if c%20==0:
			print "%s%s/%s" % ("\033[2K\033[E",c,len(messages)),
		#XXX pass massage[0] to mh and end with a fucked up python
		mh_python.learn("%s " % (message[0]))
		mh_python.learn("%s " % (message[0]))

	print "%s%s/%s" % ("\033[2K\033[E",len(messages),len(messages))
	cur.close()
	conn.close()
	del messages

	logging.basicConfig(level=logging.DEBUG)
	print "press ctrl-c to abort"
	
	event = Event()
	if len(sys.argv) == 2:
		bot=MegahalBot(event,"A",host=sys.argv[1],color=31)
	else: 
		bot=MegahalBot(event,"A",host="omegle.com",color=31)#red

	bot.start()

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

	if bot.is_connected: bot.disconnect()