Add a plugin sending Bits of Binary (XEP-0231) images.

2 files changed, 76 insertions, 0 deletions

diff --git a/plugins/bob.py b/plugins/bob.py
new file mode 100644
index 00000000..cce02b03
--- /dev/null
+++ b/plugins/bob.py
@@ -0,0 +1,75 @@
+"""
+This plugin sends a small image to the recipient of your choice, using XHTML-IM and Bits of Binary.
+
+Usage
+-----
+
+/bob some/image.png
+
+Configuration options
+---------------------
+
+.. glossary::
+
+    max_size
+        **Default:** ``2048``
+
+        The maximum acceptable size of a file, over which you will get an error instead.
+
+    max_age
+        **Default:** ``86400``
+
+        The time during which the file should stay in cache on the receiving side.
+"""
+
+from plugin import BasePlugin
+import tabs
+
+from pathlib import Path
+from glob import glob
+from os.path import expanduser
+from mimetypes import guess_type
+
+
+class Plugin(BasePlugin):
+
+    default_config = {'bob': {'max_size': 2048,
+                              'max_age': 86400}}
+
+    def init(self):
+        for tab in tabs.ConversationTab, tabs.PrivateTab, tabs.MucTab:
+            self.api.add_tab_command(tab, 'bob', self.command_bob,
+                                     usage='<image>',
+                                     help='Send image <image> to the current discussion',
+                                     short='Send a short image',
+                                     completion=self.completion_bob)
+
+    def command_bob(self, filename):
+        path = Path(expanduser(filename))
+        try:
+            size = path.stat().st_size
+        except OSError as exc:
+            self.api.information('Error sending “%s”: %s' % (path.name, exc), 'Error')
+            return
+        mime_type = guess_type(path.as_posix())[0]
+        if mime_type is None or not mime_type.startswith('image/'):
+            self.api.information('Error sending “%s”, not an image file.' % path.name, 'Error')
+            return
+        if size > self.config.get('max_size'):
+            self.api.information('Error sending “%s”, file too big.' % path.name, 'Error')
+            return
+        with open(path.as_posix(), 'rb') as file:
+            data = file.read()
+        max_age = self.config.get('max_age')
+        cid = self.core.xmpp.plugin['xep_0231'].set_bob(data, mime_type, max_age=max_age)
+        self.api.run_command('/xhtml <img src="cid:%s" alt="%s"/>' % (cid, path.name))
+
+    @staticmethod
+    def completion_bob(the_input):
+        txt = expanduser(the_input.get_text()[5:])
+        images = []
+        for filename in glob(txt + '*'):
+            mime_type = guess_type(filename)[0]
+            if mime_type is not None and mime_type.startswith('image/'):
+                images.append(filename)
+        return the_input.auto_completion(images, quotify=False)
diff --git a/src/connection.py b/src/connection.py
index bfa9603e..23c07d59 100644
--- a/src/connection.py
+++ b/src/connection.py
@@ -149,6 +149,7 @@ class Connection(slixmpp.ClientXMPP):
         if config.get('send_time'):
             self.register_plugin('xep_0202')
         self.register_plugin('xep_0224')
+        self.register_plugin('xep_0231')
         self.register_plugin('xep_0249')
         self.register_plugin('xep_0257')
         self.register_plugin('xep_0280')