--- a/plugins/plugin_xep_0077.py	Mon Dec 14 02:11:05 2009 +1100
+++ b/plugins/plugin_xep_0077.py	Tue Dec 15 01:27:32 2009 +1100
@@ -40,16 +40,21 @@
 }
 
 class XEP_0077():
-
+ 
     def __init__(self, host):
         info("Plugin XEP_0077 initialization")
         self.host = host
+        self.triggers = {}  #used by other protocol (e.g. XEP-0100) to finish registration. key = target_jid
         host.bridge.addMethod("in_band_register", ".communication", in_sign='s', out_sign='s', method=self.in_band_register)
+        host.bridge.addMethod("in_band_submit", ".request", in_sign='sa(ss)', out_sign='s', method=self.in_band_submit)
    
+    def addTrigger(self, target, cb):
+        """Add a callback which is called when registration to target is successful"""
+        self.triggers[target] = cb
+    
     def reg_ok(self, answer):
         """Called after the first get IQ"""
         form = data_form.Form.fromElement(answer.firstChildElement().firstChildElement())
-        pdb.set_trace()
         xml_data = XMLTools.dataForm2xml(form)
         self.host.bridge.actionResult("FORM", answer['id'], {"target":answer["from"], "type":"registration", "xml":xml_data})
 
@@ -62,6 +67,53 @@
         answer_type = "ERROR"
         self.host.bridge.actionResult(answer_type, failure.value.stanza['id'], answer_data)
    
+    def unregistrationAnswer(self, answer):
+        debug ("registration answer: %s" % answer.toXml())
+        answer_type = "SUCCESS"
+        answer_data={"message":"Your are now unregistred"}
+        self.host.bridge.actionResult(answer_type, answer['id'], answer_data)
+        
+    def unregistrationFailure(self, failure):
+        info ("Unregistration failure: %s" % str(failure.value))
+        answer_type = "ERROR"
+        answer_data = {}
+        answer_data['reason'] = 'unknown'
+        answer_data={"message":"Unregistration failed: %s" % failure.value.condition}
+        self.host.bridge.actionResult(answer_type, failure.value.stanza['id'], answer_data)
+    
+    def registrationAnswer(self, answer):
+        debug ("registration answer: %s" % answer.toXml())
+        answer_type = "SUCCESS"
+        answer_data={"message":"Registration successfull"}
+        self.host.bridge.actionResult(answer_type, answer['id'], answer_data)
+        if self.triggers.has_key(answer["from"]):
+            self.triggers[answer["from"]](answer["from"])
+            del self.triggers[answer["from"]]
+        
+    def registrationFailure(self, failure):
+        info ("Registration failure: %s" % str(failure.value))
+        print failure.value.stanza.toXml()
+        answer_type = "ERROR"
+        answer_data = {}
+        if failure.value.condition == 'conflict':
+            answer_data['reason'] = 'conflict'
+            answer_data={"message":"Username already exists, please choose an other one"}
+        else:
+            answer_data['reason'] = 'unknown'
+            answer_data={"message":"Registration failed"}
+        self.host.bridge.actionResult(answer_type, failure.value.stanza['id'], answer_data)
+        if self.triggers.has_key(answer["from"]):
+            del self.triggers[answer["from"]]
+
+    def in_band_submit(self, action, target, fields):
+        """Submit a form for registration, using data_form"""
+        id, deferred = self.host.submitForm(action, target, fields)
+        if action == 'CANCEL':
+            deferred.addCallbacks(self.unregistrationAnswer, self.unregistrationFailure)
+        else:    
+            deferred.addCallbacks(self.registrationAnswer, self.registrationFailure)
+        return id
+    
     def in_band_register(self, target):
         """register to a target JID"""
         to_jid = jid.JID(target)