1 files changed, 147 insertions, 0 deletions
diff --git a/slixmpp/plugins/xep_0054/vcard_temp.py b/slixmpp/plugins/xep_0054/vcard_temp.py
new file mode 100644
index 00000000..f0173386
--- /dev/null
+++ b/slixmpp/plugins/xep_0054/vcard_temp.py
@@ -0,0 +1,147 @@
+"""
+    Slixmpp: The Slick XMPP Library
+    Copyright (C) 2012 Nathanael C. Fritz, Lance J.T. Stout
+    This file is part of Slixmpp.
+
+    See the file LICENSE for copying permission.
+"""
+
+import logging
+
+from slixmpp import JID, Iq
+from slixmpp.exceptions import XMPPError
+from slixmpp.xmlstream import register_stanza_plugin
+from slixmpp.xmlstream.handler import Callback
+from slixmpp.xmlstream.matcher import StanzaPath
+from slixmpp.plugins import BasePlugin
+from slixmpp.plugins.xep_0054 import VCardTemp, stanza
+from slixmpp import future_wrapper
+
+
+log = logging.getLogger(__name__)
+
+
+class XEP_0054(BasePlugin):
+
+    """
+    XEP-0054: vcard-temp
+    """
+
+    name = 'xep_0054'
+    description = 'XEP-0054: vcard-temp'
+    dependencies = set(['xep_0030', 'xep_0082'])
+    stanza = stanza
+
+    def plugin_init(self):
+        """
+        Start the XEP-0054 plugin.
+        """
+        register_stanza_plugin(Iq, VCardTemp)
+
+
+        self.api.register(self._set_vcard, 'set_vcard', default=True)
+        self.api.register(self._get_vcard, 'get_vcard', default=True)
+        self.api.register(self._del_vcard, 'del_vcard', default=True)
+
+        self._vcard_cache = {}
+
+        self.xmpp.register_handler(
+                Callback('VCardTemp',
+                    StanzaPath('iq/vcard_temp'),
+                    self._handle_get_vcard))
+
+    def plugin_end(self):
+        self.xmpp.remove_handler('VCardTemp')
+        self.xmpp['xep_0030'].del_feature(feature='vcard-temp')
+
+    def session_bind(self, jid):
+        self.xmpp['xep_0030'].add_feature('vcard-temp')
+
+    def make_vcard(self):
+        return VCardTemp()
+
+    @future_wrapper
+    def get_vcard(self, jid=None, ifrom=None, local=None, cached=False,
+                  callback=None, timeout=None, timeout_callback=None):
+        if local is None:
+            if jid is not None and not isinstance(jid, JID):
+                jid = JID(jid)
+                if self.xmpp.is_component:
+                    if jid.domain == self.xmpp.boundjid.domain:
+                        local = True
+                else:
+                    if str(jid) == str(self.xmpp.boundjid):
+                        local = True
+                jid = jid.full
+            elif jid in (None, ''):
+                local = True
+
+        if local:
+            vcard = self.api['get_vcard'](jid, None, ifrom)
+            if not isinstance(vcard, Iq):
+                iq = self.xmpp.Iq()
+                if vcard is None:
+                    vcard = VCardTemp()
+                iq.append(vcard)
+                return iq
+            return vcard
+
+        if cached:
+            vcard = self.api['get_vcard'](jid, None, ifrom)
+            if vcard is not None:
+                if not isinstance(vcard, Iq):
+                    iq = self.xmpp.Iq()
+                    iq.append(vcard)
+                    return iq
+                return vcard
+
+        iq = self.xmpp.Iq()
+        iq['to'] = jid
+        iq['from'] = ifrom
+        iq['type'] = 'get'
+        iq.enable('vcard_temp')
+
+        return iq.send(callback=callback, timeout=timeout,
+                       timeout_callback=timeout_callback)
+
+    @future_wrapper
+    def publish_vcard(self, vcard=None, jid=None, ifrom=None,
+                      callback=None, timeout=None, timeout_callback=None):
+        self.api['set_vcard'](jid, None, ifrom, vcard)
+        if self.xmpp.is_component:
+            return
+
+        iq = self.xmpp.Iq()
+        iq['to'] = jid
+        iq['from'] = ifrom
+        iq['type'] = 'set'
+        iq.append(vcard)
+        return iq.send(callback=callback, timeout=timeout,
+                       timeout_callback=timeout_callback)
+
+    def _handle_get_vcard(self, iq):
+        if iq['type'] == 'result':
+            self.api['set_vcard'](jid=iq['from'], args=iq['vcard_temp'])
+            return
+        elif iq['type'] == 'get':
+            vcard = self.api['get_vcard'](iq['from'].bare)
+            if isinstance(vcard, Iq):
+                vcard.send()
+            else:
+                iq = iq.reply()
+                iq.append(vcard)
+                iq.send()
+        elif iq['type'] == 'set':
+            raise XMPPError('service-unavailable')
+
+    # =================================================================
+
+    def _set_vcard(self, jid, node, ifrom, vcard):
+        self._vcard_cache[jid.bare] = vcard
+
+    def _get_vcard(self, jid, node, ifrom, vcard):
+        return self._vcard_cache.get(jid.bare, None)
+
+    def _del_vcard(self, jid, node, ifrom, vcard):
+        if jid.bare in self._vcard_cache:
+            del self._vcard_cache[jid.bare]