"""
Creating a SleekXMPP Plugin

This is a minimal implementation of XEP-0077 to serve
as a tutorial for creating SleekXMPP plugins.
"""

from sleekxmpp.plugins.base import base_plugin
from sleekxmpp.xmlstream.handler.callback import Callback
from sleekxmpp.xmlstream.matcher.xpath import MatchXPath
from sleekxmpp.xmlstream import ElementBase, ET, register_stanza_plugin
from sleekxmpp import Iq
from user import UserHandler
from config import config
from sleekxmpp.plugins.xep_0004 import Form

class Registration(ElementBase):
    namespace = 'jabber:iq:register'
    name = 'query'
    plugin_attrib = 'register'
    interfaces = set(('registered', 'remove', 'instructions', 'form'))
    sub_interfaces = interfaces
    subitem = (Form,)

    def getRegistered(self):
        present = self.xml.find('{%s}registered' % self.namespace)
        return present is not None

    def getRemove(self):
        present = self.xml.find('{%s}remove' % self.namespace)
        return present is not None

    def setRegistered(self, registered):
        if registered:
            self.addField('registered')
        else:
            del self['registered']

    def setRemove(self, remove):
        if remove:
            self.addField('remove')
        else:
            del self['remove']

    def addField(self, name):
        itemXML = ET.Element('{%s}%s' % (self.namespace, name))
        self.xml.append(itemXML)

    def addForm(self):
        aliasform = Form(None,self)
        aliasform.addField(ftype="hidden",var="FORM_TYPE",value="jabber:iq:register")
        aliasform.addField(var="pubkey",ftype="text-single", label="Public Key", required=True)
        aliasform.addField(var="privkey",ftype="text-single", label="Private Key", required=True)
        aliasform.addField(var="salt",ftype="text-single", label="Salt", required=True)

    def getForm(self):
        self.xml.findall('{jabber:x:data}field')

class xep_0077(base_plugin):
    """
    XEP-0077 In-Band Registration
    """

    def plugin_init(self):
        self.description = "In-Band Registration"
        self.xep = "0077"
        self.form_fields = ()
        self.form_instructions = "Please provide the following information to register\
                                  an alias account"
        self.backend = None

        self.xmpp.registerHandler(
            Callback('In-Band Registration',
                     MatchXPath('{%s}iq/{jabber:iq:register}query' % self.xmpp.default_ns),
                     self.__handleRegistration))
        register_stanza_plugin(Iq, Registration)

    def post_init(self):
        base_plugin.post_init(self)
        self.xmpp['xep_0030'].add_feature("jabber:iq:register")

    def __handleRegistration(self, iq):
        if iq['type'] == 'get':
            # Registration form requested
            self.sendRegistrationForm(iq, self.backend)
        elif iq['type'] == 'set':
            if iq['register']['remove']:
                # Remove an account
                self.backend.unregister(iq['from'].bare)
                #self.xmpp.event('unregistered_user', iq)
                iq.reply().send()
                return

            aliasforms = iq['register']['substanzas']
            for aliasform in aliasforms:
                aliasdict = aliasform.getValues()

            for field in self.form_fields:
                if not aliasdict[field]:
                    # Incomplete Registration
                    self._sendError(iq, '406', 'modify', 'not-acceptable',
                                    "Please fill in all fields.")
                    return

            #if self.backend.register(iq['from'].bare, iq['register']):
                # Successful registration
                #self.xmpp.event('registered_user', iq)
            iq.reply().setPayload(iq['register'].xml)
            iq.send()
            #else:
                # Conflicting registration
            #    self._sendError(iq, '409', 'cancel', 'conflict',
#                                "That username is already taken.")

    def setForm(self, *fields):
        self.form_fields = fields

    def setInstructions(self, instructions):
        self.form_instructions = instructions

    def sendRegistrationForm(self, iq, userData=None):
        reg = iq['register']
        if userData is None:
            userData = {}
        else:
            reg['registered'] = True

        if self.form_instructions:
            reg['instructions'] = self.form_instructions

        reg.addForm()

        iq.reply().setPayload(reg.xml)
        iq.send()

    def _sendError(self, iq, code, error_type, name, text=''):
        iq.reply().setPayload(iq['register'].xml)
        iq.error()
        iq['error']['code'] = code
        iq['error']['type'] = error_type
        iq['error']['condition'] = name
        iq['error']['text'] = text
        iq.send()