Mercurial > libervia-backend
view libervia/backend/plugins/plugin_misc_file.py @ 4232:0fbe5c605eb6
tests (unit/webrtc,XEP-0176, XEP-0234): Fix tests and add webrtc file transfer tests:
fix 441
author | Goffi <goffi@goffi.org> |
---|---|
date | Sat, 06 Apr 2024 12:59:50 +0200 (8 months ago) |
parents | e11b13418ba6 |
children | 0d7bb4df2343 |
line wrap: on
line source
#!/usr/bin/env python3 # SAT plugin for file tansfer # Copyright (C) 2009-2021 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 import os import os.path from pathlib import Path from twisted.internet import defer from twisted.words.protocols.jabber import jid from libervia.backend.core import exceptions from libervia.backend.core.constants import Const as C from libervia.backend.core.core_types import SatXMPPEntity from libervia.backend.core.i18n import D_, _ from libervia.backend.core.log import getLogger from libervia.backend.tools import xml_tools from libervia.backend.tools import stream from libervia.backend.tools import utils from libervia.backend.tools.common import data_format, utils as common_utils log = getLogger(__name__) PLUGIN_INFO = { C.PI_NAME: "File Tansfer", C.PI_IMPORT_NAME: "FILE", C.PI_TYPE: C.PLUG_TYPE_MISC, C.PI_MODES: C.PLUG_MODE_BOTH, C.PI_MAIN: "FilePlugin", C.PI_HANDLER: "no", C.PI_DESCRIPTION: _( """File Tansfer Management: This plugin manage the various ways of sending a file, and choose the best one.""" ), } SENDING = D_("Please select a file to send to {peer}") SENDING_TITLE = D_("File sending") CONFIRM = D_( '{peer} wants to send the file "{name}" to you:\n{desc}\n\nThe file has a size of ' '{size_human}\n\nDo you accept ?' ) CONFIRM_TITLE = D_("Confirm file transfer") CONFIRM_OVERWRITE = D_("File {} already exists, are you sure you want to overwrite ?") CONFIRM_OVERWRITE_TITLE = D_("File exists") SECURITY_LIMIT = 30 PROGRESS_ID_KEY = "progress_id" class FilePlugin: File = stream.SatFile def __init__(self, host): log.info(_("plugin File initialization")) self.host = host host.bridge.add_method( "file_send", ".plugin", in_sign="ssssss", out_sign="s", method=self._file_send, async_=True, ) self._file_managers = [] host.import_menu( (D_("Action"), D_("send file")), self._file_send_menu, security_limit=10, help_string=D_("Send a file"), type_=C.MENU_SINGLE, ) def _file_send( self, peer_jid_s: str, filepath: str, name: str, file_desc: str, extra_s: str, profile: str = C.PROF_KEY_NONE ) -> defer.Deferred: client = self.host.get_client(profile) d = defer.ensureDeferred(self.file_send( client, jid.JID(peer_jid_s), filepath, name or None, file_desc or None, data_format.deserialise(extra_s) )) d.addCallback(data_format.serialise) return d async def file_send( self, client: SatXMPPEntity, peer_jid: jid.JID, filepath: str|Path, filename: str|None=None, file_desc: str|None=None, extra: dict|None=None ) -> dict: """Send a file using best available method @param peer_jid: jid of the destinee @param filepath: absolute path to the file @param filename: name to use, or None to find it from filepath @param file_desc: description of the file @param profile: %(doc_profile)s @return: action dictionary, with progress id in case of success, else xmlui message """ filepath = Path(filepath) if not filepath.is_file(): raise exceptions.DataError("The given path doesn't link to a file") if not filename: filename = filepath.name for manager, __ in self._file_managers: if await utils.as_deferred(manager.can_handle_file_send, client, peer_jid, str(filepath)): try: method_name = manager.name except AttributeError: method_name = manager.__class__.__name__ log.info( _("{name} method will be used to send the file").format( name=method_name ) ) try: file_data= await utils.as_deferred( manager.file_send, client, peer_jid, str(filepath), filename, file_desc, extra ) except Exception as e: log.warning( _("Can't send {filepath} to {peer_jid} with {method_name}: " "{reason}").format( filepath=filepath, peer_jid=peer_jid, method_name=method_name, reason=e ) ) continue if "progress" not in file_data: raise exceptions.InternalError( '"progress" should be set in "file_send" returned file data dict.' ) return file_data msg = "Can't find any method to send file to {jid}".format(jid=peer_jid.full()) log.warning(msg) return { "xmlui": xml_tools.note( "Can't transfer file", msg, C.XMLUI_DATA_LVL_WARNING ).toXml() } def _on_file_choosed(self, peer_jid, data, profile): client = self.host.get_client(profile) cancelled = C.bool(data.get("cancelled", C.BOOL_FALSE)) if cancelled: return path = data["path"] return self.file_send(client, peer_jid, path) def _file_send_menu(self, data, profile): """ XMLUI activated by menu: return file sending UI @param profile: %(doc_profile)s """ try: jid_ = jid.JID(data["jid"]) except RuntimeError: raise exceptions.DataError(_("Invalid JID")) file_choosed_id = self.host.register_callback( partial(self._on_file_choosed, jid_), with_data=True, one_shot=True, ) xml_ui = xml_tools.XMLUI( C.XMLUI_DIALOG, dialog_opt={ C.XMLUI_DATA_TYPE: C.XMLUI_DIALOG_FILE, C.XMLUI_DATA_MESS: _(SENDING).format(peer=jid_.full()), }, title=_(SENDING_TITLE), submit_id=file_choosed_id, ) return {"xmlui": xml_ui.toXml()} def register(self, manager, priority: int = 0) -> None: """Register a fileSending manager @param manager: object implementing can_handle_file_send, and file_send methods @param priority: pririoty of this manager, the higher available will be used """ m_data = (manager, priority) if m_data in self._file_managers: raise exceptions.ConflictError( f"Manager {manager} is already registered" ) if not hasattr(manager, "can_handle_file_send") or not hasattr(manager, "file_send"): raise ValueError( f'{manager} must have both "can_handle_file_send" and "file_send" methods to ' 'be registered') self._file_managers.append(m_data) self._file_managers.sort(key=lambda m: m[1], reverse=True) def unregister(self, manager): for idx, data in enumerate(self._file_managers): if data[0] == manager: break else: raise exceptions.NotFound("The file manager {manager} is not registered") del self._file_managers[idx] # Dialogs with user # the overwrite check is done here def open_file_write(self, client, file_path, transfer_data, file_data, stream_object): """create SatFile or FileStremaObject for the requested file and fill suitable data """ if stream_object: assert "stream_object" not in transfer_data transfer_data["stream_object"] = stream.FileStreamObject( self.host, client, file_path, mode="wb", uid=file_data[PROGRESS_ID_KEY], size=file_data["size"], data_cb=file_data.get("data_cb"), ) else: assert "file_obj" not in transfer_data transfer_data["file_obj"] = stream.SatFile( self.host, client, file_path, mode="wb", uid=file_data[PROGRESS_ID_KEY], size=file_data["size"], data_cb=file_data.get("data_cb"), ) async def _got_confirmation( self, client, data, peer_jid, transfer_data, file_data, stream_object ): """Called when the permission and dest path have been received @param peer_jid(jid.JID): jid of the file sender @param transfer_data(dict): same as for [self.get_dest_dir] @param file_data(dict): same as for [self.get_dest_dir] @param stream_object(bool): same as for [self.get_dest_dir] return (bool): True if copy is wanted and OK False if user wants to cancel if file exists ask confirmation and call again self._getDestDir if needed """ if data.get("cancelled", False): return False path = data["path"] file_data["file_path"] = file_path = os.path.join(path, file_data["name"]) log.debug("destination file path set to {}".format(file_path)) # we manage case where file already exists if os.path.exists(file_path): overwrite = await xml_tools.defer_confirm( self.host, _(CONFIRM_OVERWRITE).format(file_path), _(CONFIRM_OVERWRITE_TITLE), action_extra={ "from_jid": peer_jid.full(), "type": C.META_TYPE_OVERWRITE, "progress_id": file_data[PROGRESS_ID_KEY], }, security_limit=SECURITY_LIMIT, profile=client.profile, ) if not overwrite: return await self.get_dest_dir(client, peer_jid, transfer_data, file_data) self.open_file_write(client, file_path, transfer_data, file_data, stream_object) return True async def get_dest_dir( self, client, peer_jid, transfer_data, file_data, stream_object=False ): """Request confirmation and destination dir to user Overwrite confirmation is managed. if transfer is confirmed, 'file_obj' is added to transfer_data @param peer_jid(jid.JID): jid of the file sender @param filename(unicode): name of the file @param transfer_data(dict): data of the transfer session, it will be only used to store the file_obj. "file_obj" (or "stream_object") key *MUST NOT* exist before using get_dest_dir @param file_data(dict): information about the file to be transfered It MUST contain the following keys: - peer_jid (jid.JID): other peer jid - name (unicode): name of the file to trasnsfer the name must not be empty or contain a "/" character - size (int): size of the file - desc (unicode): description of the file - progress_id (unicode): id to use for progression It *MUST NOT* contain the "peer" key It may contain: - data_cb (callable): method called on each data read/write "file_path" will be added to this dict once destination selected "size_human" will also be added with human readable file size @param stream_object(bool): if True, a stream_object will be used instead of file_obj a stream.FileStreamObject will be used return: True if transfer is accepted """ cont, ret_value = await self.host.trigger.async_return_point( "FILE_getDestDir", client, peer_jid, transfer_data, file_data, stream_object ) if not cont: return ret_value filename = file_data["name"] assert filename and not "/" in filename assert PROGRESS_ID_KEY in file_data # human readable size file_data["size_human"] = common_utils.get_human_size(file_data["size"]) resp_data = await xml_tools.defer_dialog( self.host, _(CONFIRM).format(peer=peer_jid.full(), **file_data), _(CONFIRM_TITLE), type_=C.XMLUI_DIALOG_FILE, options={C.XMLUI_DATA_FILETYPE: C.XMLUI_DATA_FILETYPE_DIR}, action_extra={ "from_jid": peer_jid.full(), "type": C.META_TYPE_FILE, "progress_id": file_data[PROGRESS_ID_KEY], }, security_limit=SECURITY_LIMIT, profile=client.profile, ) accepted = await self._got_confirmation( client, resp_data, peer_jid, transfer_data, file_data, stream_object, ) return accepted