From 5ab77c745270d7d5c016c1dc7ef2a82533a4b16e Mon Sep 17 00:00:00 2001 From: Florent Le Coz Date: Thu, 17 Jul 2014 14:19:04 +0200 Subject: Rename to slixmpp --- slixmpp/plugins/xep_0231/__init__.py | 16 ++++ slixmpp/plugins/xep_0231/bob.py | 140 +++++++++++++++++++++++++++++++++++ slixmpp/plugins/xep_0231/stanza.py | 36 +++++++++ 3 files changed, 192 insertions(+) create mode 100644 slixmpp/plugins/xep_0231/__init__.py create mode 100644 slixmpp/plugins/xep_0231/bob.py create mode 100644 slixmpp/plugins/xep_0231/stanza.py (limited to 'slixmpp/plugins/xep_0231') diff --git a/slixmpp/plugins/xep_0231/__init__.py b/slixmpp/plugins/xep_0231/__init__.py new file mode 100644 index 00000000..57d57846 --- /dev/null +++ b/slixmpp/plugins/xep_0231/__init__.py @@ -0,0 +1,16 @@ +""" + Slixmpp: The Slick XMPP Library + Copyright (C) 2012 Nathanael C. Fritz, + Emmanuel Gil Peyrot + This file is part of Slixmpp. + + See the file LICENSE for copying permission. +""" + +from slixmpp.plugins.base import register_plugin + +from slixmpp.plugins.xep_0231.stanza import BitsOfBinary +from slixmpp.plugins.xep_0231.bob import XEP_0231 + + +register_plugin(XEP_0231) diff --git a/slixmpp/plugins/xep_0231/bob.py b/slixmpp/plugins/xep_0231/bob.py new file mode 100644 index 00000000..d590c9dd --- /dev/null +++ b/slixmpp/plugins/xep_0231/bob.py @@ -0,0 +1,140 @@ +""" + Slixmpp: The Slick XMPP Library + Copyright (C) 2012 Nathanael C. Fritz, + Emmanuel Gil Peyrot + This file is part of Slixmpp. + + See the file LICENSE for copying permission. +""" + +import logging +import hashlib + +from slixmpp.stanza import Iq, Message, Presence +from slixmpp.exceptions import XMPPError +from slixmpp.xmlstream.handler import Callback +from slixmpp.xmlstream.matcher import StanzaPath +from slixmpp.xmlstream import register_stanza_plugin +from slixmpp.plugins.base import BasePlugin +from slixmpp.plugins.xep_0231 import stanza, BitsOfBinary + + +log = logging.getLogger(__name__) + + +class XEP_0231(BasePlugin): + + """ + XEP-0231 Bits of Binary + """ + + name = 'xep_0231' + description = 'XEP-0231: Bits of Binary' + dependencies = set(['xep_0030']) + + def plugin_init(self): + self._cids = {} + + register_stanza_plugin(Iq, BitsOfBinary) + register_stanza_plugin(Message, BitsOfBinary) + register_stanza_plugin(Presence, BitsOfBinary) + + self.xmpp.register_handler( + Callback('Bits of Binary - Iq', + StanzaPath('iq/bob'), + self._handle_bob_iq)) + + self.xmpp.register_handler( + Callback('Bits of Binary - Message', + StanzaPath('message/bob'), + self._handle_bob)) + + self.xmpp.register_handler( + Callback('Bits of Binary - Presence', + StanzaPath('presence/bob'), + self._handle_bob)) + + self.api.register(self._get_bob, 'get_bob', default=True) + self.api.register(self._set_bob, 'set_bob', default=True) + self.api.register(self._del_bob, 'del_bob', default=True) + + def plugin_end(self): + self.xmpp['xep_0030'].del_feature(feature='urn:xmpp:bob') + self.xmpp.remove_handler('Bits of Binary - Iq') + self.xmpp.remove_handler('Bits of Binary - Message') + self.xmpp.remove_handler('Bits of Binary - Presence') + + def session_bind(self, jid): + self.xmpp['xep_0030'].add_feature('urn:xmpp:bob') + + def set_bob(self, data, mtype, cid=None, max_age=None): + if cid is None: + cid = 'sha1+%s@bob.xmpp.org' % hashlib.sha1(data).hexdigest() + + bob = BitsOfBinary() + bob['data'] = data + bob['type'] = mtype + bob['cid'] = cid + bob['max_age'] = max_age + + self.api['set_bob'](args=bob) + + return cid + + def get_bob(self, jid=None, cid=None, cached=True, ifrom=None, + block=True, timeout=None, callback=None): + if cached: + data = self.api['get_bob'](None, None, ifrom, args=cid) + if data is not None: + if not isinstance(data, Iq): + iq = self.xmpp.Iq() + iq.append(data) + return iq + return data + + iq = self.xmpp.Iq() + iq['to'] = jid + iq['from'] = ifrom + iq['type'] = 'get' + iq['bob']['cid'] = cid + return iq.send(block=block, timeout=timeout, callback=callback) + + def del_bob(self, cid): + self.api['del_bob'](args=cid) + + def _handle_bob_iq(self, iq): + cid = iq['bob']['cid'] + + if iq['type'] == 'result': + self.api['set_bob'](iq['from'], None, iq['to'], args=iq['bob']) + self.xmpp.event('bob', iq) + elif iq['type'] == 'get': + data = self.api['get_bob'](iq['to'], None, iq['from'], args=cid) + if isinstance(data, Iq): + data['id'] = iq['id'] + data.send() + return + + iq.reply() + iq.append(data) + iq.send() + + def _handle_bob(self, stanza): + self.api['set_bob'](stanza['from'], None, + stanza['to'], args=stanza['bob']) + self.xmpp.event('bob', stanza) + + # ================================================================= + + def _set_bob(self, jid, node, ifrom, bob): + self._cids[bob['cid']] = bob + + def _get_bob(self, jid, node, ifrom, cid): + if cid in self._cids: + return self._cids[cid] + else: + raise XMPPError('item-not-found') + + def _del_bob(self, jid, node, ifrom, cid): + if cid in self._cids: + del self._cids[cid] diff --git a/slixmpp/plugins/xep_0231/stanza.py b/slixmpp/plugins/xep_0231/stanza.py new file mode 100644 index 00000000..f7e6a66c --- /dev/null +++ b/slixmpp/plugins/xep_0231/stanza.py @@ -0,0 +1,36 @@ +""" + Slixmpp: The Slick XMPP Library + Copyright (C) 2012 Nathanael C. Fritz, + Emmanuel Gil Peyrot + This file is part of Slixmpp. + + See the file LICENSE for copying permission. +""" + +import base64 + + +from slixmpp.util import bytes +from slixmpp.xmlstream import ElementBase + + +class BitsOfBinary(ElementBase): + name = 'data' + namespace = 'urn:xmpp:bob' + plugin_attrib = 'bob' + interfaces = set(('cid', 'max_age', 'type', 'data')) + + def get_max_age(self): + return self._get_attr('max-age') + + def set_max_age(self, value): + self._set_attr('max-age', value) + + def get_data(self): + return base64.b64decode(bytes(self.xml.text)) + + def set_data(self, value): + self.xml.text = bytes(base64.b64encode(value)).decode('utf-8') + + def del_data(self): + self.xml.text = '' -- cgit v1.2.3