import logging
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__":
	import getopt
	import sys

	def usage():
		print """%s - megahal ding
Argumente:
  -h, --help		Diese Hilfe
  --host=HOST		(Standard: HOST=omegle.com)
  -d DBNAME  or
  --dbname=DBNAME	Database (Standard: DBNAME=omegle)
  --profile=NAME	Database select profile (Standard: NAME=standard)
			see sourcecode""" % (sys.argv[0])

	#parse args
	try:
		opts, args = getopt.getopt(sys.argv[1:], "hd:", ["help", "host=", "profile=","dbname="])
	except getopt.GetoptError,err:
		print str(err)
		usage()
		sys.exit(2)
	
	#defaults
	host = "omegle.com"
	sql = {"standard":
                   """SELECT message FROM omegle_messages 
                       WHERE LENGTH(message) < 35
                         AND NOT ( message ILIKE '%www.%'
                             OR message ILIKE '%http%'
                             OR message ILIKE '%@%.%')
                     ;""",
	       "offensive":
                   """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 '%@%.%');""",
	       }
	sqlKey = "standard"
	dbname = "omegle"

	for o, a in opts:
		if o in ('h', '--help'):
			usage()
			sys.exit(0)
		elif o in ('d', '--dbname'):
			dbname = a
		elif o == "--host":
			host = a
		elif o == "--profile":
			sqlKey = a

	mh_python.initbrain()

	conn = pyPgSQL.PgSQL.connect(database=dbname)
	cur = conn.cursor()

	cur.execute(sql[sqlKey])
	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()
	bot=MegahalBot(event,"A",host=host,color=31)

	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()