source: ralphm-patches/roster_server.patch @ 34:e46c5701df9e

wokkel/xmppim.py

@@ -12 +12 @@
 All of it should eventually move to Twisted.
 """
 
+from twisted.words.protocols.jabber import error, xmlstream
 from twisted.words.protocols.jabber.jid import JID
 from twisted.words.xish import domish
 
@@ -22 +23 @@
 NS_XML = 'http://www.w3.org/XML/1998/namespace'
 NS_ROSTER = 'jabber:iq:roster'
 
+XPATH_ROSTER_GET = "//iq[@type='get']/query[@xmlns='%s']" % NS_ROSTER
+XPATH_ROSTER_SET = "//iq[@type='set']/query[@xmlns='%s']" % NS_ROSTER
+
 class Presence(domish.Element):
     def __init__(self, to=None, type=None):
         domish.Element.__init__(self, (None, "presence"))
@@ -605 +609 @@
         self.groups = set()
 
 
+
 class RosterClientProtocol(XMPPHandler):
     """
     Client side XMPP roster protocol.
@@ -614 +619 @@
         ROSTER_SET = "/iq[@type='set']/query[@xmlns='%s']" % NS_ROSTER
         self.xmlstream.addObserver(ROSTER_SET, self._onRosterSet)
 
+
     def _parseRosterItem(self, element):
         jid = JID(element['jid'])
         item = RosterItem(jid)
@@ -628 +634 @@
 
         return item
 
+
     def getRoster(self):
         """
         Retrieve contact list.
@@ -683 +690 @@
             item = self._parseRosterItem(iq.query.item)
             self.onRosterSet(item)
 
+
     def onRosterSet(self, item):
         """
         Called when a roster push for a new or update item was received.
@@ -691 +699 @@
         @type item: L{RosterItem}
         """
 
+
     def onRosterRemove(self, entity):
         """
         Called when a roster push for the removal of an item was received.
@@ -699 +708 @@
         @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, -1)
+        self.xmlstream.addObserver(XPATH_ROSTER_SET, self._onRosterSet, -1)
+
+
+    def _toRosterReply(self, items, iq):
+        response = xmlstream.toResponse(iq, 'result')
+        response.addElement((NS_ROSTER, 'query'))
+
+        for item in items:
+            element = response.query.addElement('item')
+            element['jid'] = item['jid']
+            if item['name']:
+                element['name'] = item['name']
+            element['subscription'] = item['subscription']
+            if item['ask']:
+                element['ask'] = item['ask']
+
+            for group in item['groups']:
+                element.addElement('group', content=group)
+
+        return response
+
+
+    def _onRosterGet(self, iq):
+        iq.handled = True
+
+        d = self.getRoster(JID(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
+
+
+
 class MessageProtocol(XMPPHandler):
     """
     Generic XMPP subprotocol handler for incoming message stanzas.