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)
  -l              or
  --local               Do local testing
  --profile=NAME	Database select profile (Standard: NAME=standard)
			see sourcecode""" % (sys.argv[0])

	#parse args
	try:
		opts, args = getopt.getopt(sys.argv[1:], "hd:l", ["help", "host=", "profile=","dbname=","local"])
	except getopt.GetoptError,err:
		print str(err)
		usage()
		sys.exit(2)
	
	#defaults
	local = False
	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 message ILIKE '%pants %'
                            OR message ILIKE '%penis%'
                            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 in ('-l', '--local'):
			local = True
		elif o == "--host":
			host = a
		elif o == "--profile":
			if sql.has_key(a):
				sqlKey = a
				print sql[a]
			else:
				print "Profile %s does not exist, create it!" % a
				sys.exit(0)

	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)
	if local:
		print ">>> ",
		while True:
			x=sys.stdin.readline().strip()
			print "<<< %s\n>>> " % mh_python.doreply(x),
	else:
		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()