summaryrefslogtreecommitdiff
path: root/plugins/upload.py
diff options
context:
space:
mode:
Diffstat (limited to 'plugins/upload.py')
-rw-r--r--plugins/upload.py59
1 files changed, 59 insertions, 0 deletions
diff --git a/plugins/upload.py b/plugins/upload.py
new file mode 100644
index 00000000..98379c33
--- /dev/null
+++ b/plugins/upload.py
@@ -0,0 +1,59 @@
+"""
+Upload a file and auto-complete the input with its URL.
+
+Usage
+-----
+
+This plugin adds a command to the chat tabs.
+
+.. glossary::
+
+ /upload
+ **Usage:** ``/upload <filename>``
+
+ Uploads the <filename> file to the preferred HTTP File Upload
+ service (see XEP-0363) and fill the input with its URL.
+
+
+"""
+import asyncio
+from os.path import expanduser
+from glob import glob
+
+from poezio.plugin import BasePlugin
+from poezio.core.structs import Completion
+from poezio.decorators import command_args_parser
+from poezio import tabs
+
+class Plugin(BasePlugin):
+
+ def init(self):
+ for _class in (tabs.PrivateTab, tabs.ConversationTab, tabs.MucTab):
+ self.api.add_tab_command(_class, 'upload', self.command_upload,
+ usage='<filename>',
+ help='Upload a file and auto-complete the input with its URL.',
+ short='Upload a file',
+ completion=self.completion_filename)
+
+ @asyncio.coroutine
+ def async_upload(self, filename):
+ try:
+ url = yield from self.core.xmpp['xep_0363'].upload_file(filename)
+ except Exception as e:
+ self.api.information('Failed to upload the file: %s(%s)' % (type(e), e), 'Error')
+ self.core.insert_input_text(url)
+
+ @command_args_parser.quoted(1)
+ def command_upload(self, args):
+ if args is None:
+ self.core.command.help('upload')
+ return
+ filename, = args
+ filename = expanduser(filename)
+ asyncio.ensure_future(self.async_upload(filename))
+
+ @staticmethod
+ def completion_filename(the_input):
+ txt = expanduser(the_input.get_text()[8:])
+ files = glob(txt + '*')
+ return Completion(the_input.auto_completion, files, quotify=False)