diff sat/plugins/plugin_misc_attach.py @ 3181:5ff2cf7f0aba

plugin attach: first draft: this plugin handles the uploading of files attached in message data.
author Goffi <goffi@goffi.org>
date Sun, 23 Feb 2020 17:48:26 +0100
parents
children f2bb57348587
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/sat/plugins/plugin_misc_attach.py	Sun Feb 23 17:48:26 2020 +0100
@@ -0,0 +1,115 @@
+#!/usr/bin/env python3
+
+# SàT plugin for attaching files
+# Copyright (C) 2009-2020 Jérôme Poisson (goffi@goffi.org)
+
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Affero General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU Affero General Public License for more details.
+
+# You should have received a copy of the GNU Affero General Public License
+# along with this program.  If not, see <http://www.gnu.org/licenses/>.
+
+from functools import partial
+from pathlib import Path
+from twisted.internet import defer
+from sat.core.i18n import _
+from sat.core.constants import Const as C
+from sat.core.log import getLogger
+
+log = getLogger(__name__)
+
+
+PLUGIN_INFO = {
+    C.PI_NAME: "File Attach",
+    C.PI_IMPORT_NAME: "ATTACH",
+    C.PI_TYPE: C.PLUG_TYPE_MISC,
+    C.PI_DEPENDENCIES: ["UPLOAD"],
+    C.PI_MAIN: "AttachPlugin",
+    C.PI_HANDLER: "no",
+    C.PI_DESCRIPTION: _("""Attachments handler"""),
+}
+
+
+class AttachPlugin:
+
+    def __init__(self, host):
+        log.info(_("plugin Attach initialization"))
+        self.host = host
+        self._u = host.plugins["UPLOAD"]
+        host.trigger.add("sendMessage", self._sendMessageTrigger)
+
+    def _attachFiles(self, client, data):
+        # TODO: handle xhtml-im
+        body_elt = next(data["xml"].elements((C.NS_CLIENT, "body")))
+        for attachment in data["extra"][C.MESS_KEY_ATTACHMENTS]:
+            body_elt.addContent(f'\n{attachment["url"]}')
+        return data
+
+    async def uploadFiles(self, client, data):
+        uploads_d = []
+        to_delete = []
+        attachments = data["extra"]["attachments"]
+
+        for attachment in attachments:
+            try:
+                path = Path(attachment["path"])
+            except KeyError:
+                log.warning("no path in attachment: {attachment}")
+                to_delete.append(attachment)
+                continue
+
+            if "url" in attachment:
+                log.warning("invalid attachment: unknown URL")
+                to_delete.append(attachment)
+
+            try:
+                name = attachment["name"]
+            except KeyError:
+                name = attachment["name"] = path.name
+
+            options = {}
+            if client.encryption.isEncryptionRequested(data):
+                # FIXME: we should not use implementation specific value here
+                #   but for not it's the only file encryption method available with
+                #   with upload.
+                options['encryption'] = C.ENC_AES_GCM
+
+            __, upload_d = await self._u.upload(
+                client,
+                filepath=path,
+                filename=name,
+            )
+            uploads_d.append(upload_d)
+
+        for attachment in to_delete:
+            attachments.remove(attachment)
+
+        upload_results = await defer.DeferredList(uploads_d)
+        for idx, (success, ret) in enumerate(upload_results):
+            attachment = attachments[idx]
+
+            if not success:
+                # ret is a failure here
+                log.warning(f"error while uploading {attachment}: {ret}")
+                continue
+
+            attachment["url"] = ret
+
+        return data
+
+    def _uploadFiles(self, client, data):
+        return defer.ensureDeferred(self.uploadFiles(client, data))
+
+    def _sendMessageTrigger(
+        self, client, mess_data, pre_xml_treatments, post_xml_treatments):
+        if mess_data['extra'].get(C.MESS_KEY_ATTACHMENTS):
+            pre_xml_treatments.addCallback(partial(self._uploadFiles, client))
+            post_xml_treatments.addCallback(partial(self._attachFiles, client))
+        return True