from xmpp.protocol import Message, Presence
import config
import libomegle
import xmpp
import xmppUtils
from threading import Thread

class OmegleClient:
    nick = None
    isPublic = True
    targetHandle = None
    muc = None
    omegleClient = None
    xmppClient = None
    connected = False
    thread = None

    def __init__(self, muc, nick='Stranger'):
        jid = xmpp.JID(config.get("id", "omegle"))
        user = jid.getNode()
        server = jid.getDomain()
        password = config.get("password")
        self.nick = nick
        self.muc = muc
        self.xmppClient = xmpp.Client(server, debug=[]);
        connResult = self.xmppClient.connect()
        if not connResult: print "Error connecting to server: " + server
        connResult = self.xmppClient.auth(user, password, nick)
        if not connResult: print "Error authenticating user: " + user

        self.xmppClient.sendInitPresence()
        self.xmppClient.RegisterHandler('message', self.xmppMessageHandler)
        self.xmppClient.RegisterHandler('presence', self.xmppPresenceHandler)

        self.omegleClient = libomegle.Omegle()
        self.omegleClient.registerHook('join', self.omegleJoin)
        self.omegleClient.registerHook('leave', self.omegleLeave)
        self.omegleClient.registerHook('message', self.omegleMessage)
        self.omegleClient.registerHook('captchaRequired', self.omegleCaptchaRequired)
        self.omegleClient.registerHook('captchaRejected', self.omegleCaptchaRejected)

        self.thread = Thread(target=self.loop)

    def start(self, isPublic=True):
        self.isPublic = isPublic

        if self.omegleClient.isConnected():
            self.sendToXMPP('Why are you starting me again?')
            return

        self.omegleClient.start()

        self.thread.start()

    def loop(self):
        while self.xmppClient.Process(0.1):
            self.omegleClient.step()

        if self.omegleClient.isConnected():
            self.omegleClient.disconnect()
        self.xmppClient.disconnect()

    def setScope(self, isPublic):
        self.isPublic = isPublic

    def stop(self):
        if self.omegleClient.isConnected():
            self.omegleClient.disconnect()
        self.xmppClient.sendPresence(typ='unavailable')
        self.connected = False

    def kick(self):
        self.sendToOmegle("** You have been kicked out of the groupchat for some reason or another.")
        self.stop()

    def sendCaptcha(self, captchaText):
        self.omegleClient.confirmCaptcha(unicode(captchaText).encode('utf-8'))

    def sendToOmegle(self, msg):
        print '[o:' + self.nick + '] tx: ' + msg
        self.omegleClient.sendMessage(unicode(msg).encode('utf-8'))

    def sendToXMPP(self, msg):
        self.xmppClient.send(Message(self.muc, unicode(msg), 'groupchat'))

    def xmppMessageHandler(self, client, msg):
        nick = msg.getFrom().getResource()
        if not msg or not msg.getBody() or not nick or nick == self.nick:
            return
        text = msg.getBody().strip()
        if text[:1] == '.' or text[:1] == '!':
            return



        if self.omegleClient.isConnected():
            s = u'<' + nick + u'> ' + unicode(text);
            self.sendToOmegle(s)

    def xmppPresenceHandler(self, client, pres):
        nick = pres.getFrom().getResource()
        if pres.getType() == 'unavailable':
            if nick == self.nick:
                self.kick()

    def omegleJoin(self):
        presence = Presence(to=self.muc + '/' + self.nick)
        presence.setTag('x', namespace=xmpp.NS_MUC).setTagData('password', '')
        presence.getTag('x').addChild('history', {'maxchars': '0', 'maxstanzas': '0'});
        self.xmppClient.send(presence)
        self.connected = True
        print '[o:' + self.nick + '] joined'
        self.sendToXMPP('. ** A stranger has taken the bait!')
        self.sendToOmegle('**PLEASE READ** Welcome to the Supernova groupchat! We request that you be kind and respect the rules, which are as follows- No flooding, no disrespect, and no sexchat which means no ASL and no M/F. You do NOT need a name, you have one already. Ask for it if you\'re curious, and don\'t be shy to ask us other information regarding the chat. We are neither on Omegle nor affiliated with it. So go ahead, and have fun!')

    def omegleMessage(self, msg):
        print '[o:' + self.nick + '] rx: ' + msg
        if 'asl' in msg:
            self.sendToOmegle('No, that was lazy. Try again.')
        else:
            self.sendToXMPP(msg)

    def omegleLeave(self):
        self.sendToOmegle('**We\'re sorry you didn\'t enjoy your stay.')
        self.sendToXMPP('.** The stranger ran away!')
        print '[o:' + self.nick + '] left'
        self.stop()

    def omegleCaptchaRequired(self, url):
        if not self.connected:
            presence = Presence(to=self.muc + '/' + self.nick)
            presence.setTag('x', namespace=xmpp.NS_MUC).setTagData('password', '')
            presence.getTag('x').addChild('history', {'maxchars': '0', 'maxstanzas': '0'});
            self.xmppClient.send(presence)
        print '[o:' + self.nick + '] captcha: ' + url
        self.sendToXMPP('.* CAPTCHA: ' + url)

    def omegleCaptchaRejected(self):
        print '[o:' + self.nick + '] captcha rejected'
        self.sendToXMPP('CAPTCHA rejected, disconnecting...')
        self.stop()