source: ralphm-patches/roster_server.patch @ 49:537d1413b661

wokkel/im.py

@@ -14 +14 @@
 
 from twisted.internet import defer
 from twisted.words.protocols.jabber.jid import JID, internJID
-from twisted.words.protocols.jabber import error
+from twisted.words.protocols.jabber import error, xmlstream
 from twisted.words.xish import domish
 
 from wokkel.compat import IQ
@@ -25 +25 @@
 NS_ROSTER = 'jabber:iq:roster'
 
 XPATH_ROSTER_SET = "/iq[@type='set']/query[@xmlns='%s']" % NS_ROSTER
+XPATH_ROSTER_GET = "/iq[@type='get']/query[@xmlns='%s']" % NS_ROSTER
 
 
 
@@ -579 +580 @@
         @param entity: The entity for which the roster item has been removed.
         @type entity: L{JID}
         """
+
+
+
+class RosterServerProtocol(XMPPHandler):
+    """
+    XMPP subprotocol handler for the roster, server side.
+    """
+
+    def connectionInitialized(self):
+        self.xmlstream.addObserver(XPATH_ROSTER_GET, self._onRosterGet)
+        self.xmlstream.addObserver(XPATH_ROSTER_SET, self._onRosterSet)
+
+
+    def _toRosterReply(self, roster, iq):
+        response = xmlstream.toResponse(iq, 'result')
+        response.addElement((NS_ROSTER, 'query'))
+
+        for item in roster:
+            response.query.addChild(item.toElement())
+
+        return response
+
+
+    def _onRosterGet(self, iq):
+        iq.handled = True
+
+        d = self.getRoster(internJID(iq["from"]))
+        d.addCallback(self._toRosterReply, iq)
+        d.addErrback(lambda _: error.ErrorStanza('internal-error').toResponse(iq))
+        d.addBoth(self.send)
+
+
+    def _onRosterSet(self, iq):
+        iq.handled = True
+        response = error.StanzaError('bad-request').toResponse(iq)
+        self.send(response)
+
+
+    def getRoster(self, entity):
+        raise NotImplemented