summaryrefslogtreecommitdiff
path: root/plugins/upload.py
blob: 0f9ef736751f00cc4e88b867f0b0bfb15a6f6ef7 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
"""
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
import traceback
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):
        if not self.core.xmpp['xep_0363']:
            raise Exception('slixmpp XEP-0363 plugin failed to load')
        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:
            exception = traceback.format_exc()
            self.api.information('Failed to upload file: %s' % exception, 'Error')
            return
        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)