Changeset 55:a8cda961ee7e

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.

Location:

wokkel

Files:

• client.py (modified) (3 diffs)
• generic.py (modified) (2 diffs)
• server.py (added)
• test/test_server.py (added)

wokkel/client.py

@@ -12 +12 @@
 
 from twisted.application import service
-from twisted.internet import defer, protocol, reactor
+from twisted.internet import reactor
 from twisted.names.srvconnect import SRVConnector
 from twisted.words.protocols.jabber import client, sasl, xmlstream
 
-try:
-    from twisted.words.xish.xmlstream import BootstrapMixin
-except ImportError:
-    from wokkel.compat import BootstrapMixin
-
-from wokkel.subprotocols import StreamManager, XMPPHandler
+from wokkel import generic
+from wokkel.subprotocols import StreamManager
 
 class CheckAuthInitializer(object):
@@ -128 +124 @@
 
 
-class DeferredClientFactory(BootstrapMixin, protocol.ClientFactory):
-    protocol = xmlstream.XmlStream
+class DeferredClientFactory(generic.DeferredXmlStreamFactory):
 
     def __init__(self, jid, password):
-        BootstrapMixin.__init__(self)
-
-        self.jid = jid
-        self.password = password
-
-        deferred = defer.Deferred()
-        self.deferred = deferred
-        self.addBootstrap(xmlstream.INIT_FAILED_EVENT, deferred.errback)
-
-        class ConnectionInitializedHandler(XMPPHandler):
-            def connectionInitialized(self):
-                deferred.callback(None)
-
-        self.streamManager = StreamManager(self)
-        self.addHandler(ConnectionInitializedHandler())
-
-
-    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.
-        """
-        self.authenticator = client.XMPPAuthenticator(self.jid, self.password)
-        xs = self.protocol(self.authenticator)
-        xs.factory = self
-        self.installBootstraps(xs)
-        return xs
-
-
-    def clientConnectionFailed(self, connector, reason):
-        self.deferred.errback(reason)
+        authenticator = client.XMPPAuthenticator(jid, password)
+        generic.DeferredXmlStreamFactory.__init__(self, authenticator)
 
 
@@ -181 +145 @@
 
 
+
 def clientCreator(factory):
-    domain = factory.jid.host
+    domain = factory.authenticator.jid.host
     c = SRVConnector(reactor, 'xmpp-client', domain, factory)
     c.connect()

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)