Changeset 55:a8cda961ee7e for wokkel/generic.py

Timestamp:

Apr 22, 2009, 4:50:28 PM (13 years ago)

Author:

Ralph Meijer <ralphm@…>

Branch:

default

Convert:

svn:b33ecbfc-034c-dc11-8662-000475d9059e/trunk@167

Message:

Implement server-to-server dialback protocol and stream management.

Author: ralphm.
Fixes #33.

This also abstracts the concept of a client factory that fires a deferred.

File:

• wokkel/generic.py (modified) (2 diffs)

wokkel/generic.py

@@ -10 +10 @@
 from zope.interface import implements
 
-from twisted.internet import defer
-from twisted.words.protocols.jabber import error
+from twisted.internet import defer, protocol
+from twisted.words.protocols.jabber import error, xmlstream
 from twisted.words.protocols.jabber.xmlstream import toResponse
 from twisted.words.xish import domish, utility
+
+try:
+    from twisted.words.xish.xmlstream import BootstrapMixin
+except ImportError:
+    from wokkel.compat import BootstrapMixin
 
 from wokkel import disco
@@ -156 +161 @@
         self.source.send = lambda obj: self.sink.dispatch(obj)
         self.sink.send = lambda obj: self.source.dispatch(obj)
+
+
+class DeferredXmlStreamFactory(BootstrapMixin, protocol.ClientFactory):
+    protocol = xmlstream.XmlStream
+
+    def __init__(self, authenticator):
+        BootstrapMixin.__init__(self)
+
+        self.authenticator = authenticator
+
+        deferred = defer.Deferred()
+        self.deferred = deferred
+        self.addBootstrap(xmlstream.STREAM_AUTHD_EVENT, self.deferred.callback)
+        self.addBootstrap(xmlstream.INIT_FAILED_EVENT, deferred.errback)
+
+
+    def buildProtocol(self, addr):
+        """
+        Create an instance of XmlStream.
+
+        A new authenticator instance will be created and passed to the new
+        XmlStream. Registered bootstrap event observers are installed as well.
+        """
+        xs = self.protocol(self.authenticator)
+        xs.factory = self
+        self.installBootstraps(xs)
+        return xs
+
+
+    def clientConnectionFailed(self, connector, reason):
+        self.deferred.errback(reason)