libervia-backend: sat/plugins/plugin_xep

annotate sat/plugins/plugin_xep_0329.py @ 3413:f4d417593e77

jp (pubsub/node/info, pubsub/get): better exit code: the correct exit code is now returned in case of NOT_FOUND or bridge exception. If the exception is not a bridge exception, an internal error exit code is used.

author	Goffi <goffi@goffi.org>
date	Thu, 12 Nov 2020 14:53:16 +0100
parents	08a3e34aead1
children	be6d91572633

rev	line source
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2937 diff changeset	1 #!/usr/bin/env python3
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	2
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	3 # SAT plugin for File Information Sharing (XEP-0329)
3136 9d0df638c8b4 dates update Goffi <goffi@goffi.org> parents: 3118 diff changeset	4 # Copyright (C) 2009-2020 Jérôme Poisson (goffi@goffi.org)
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	5
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	6 # This program is free software: you can redistribute it and/or modify
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	7 # it under the terms of the GNU Affero General Public License as published by
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	8 # the Free Software Foundation, either version 3 of the License, or
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	9 # (at your option) any later version.
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	10
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	11 # This program is distributed in the hope that it will be useful,
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	12 # but WITHOUT ANY WARRANTY; without even the implied warranty of
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	13 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	14 # GNU Affero General Public License for more details.
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	15
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	16 # You should have received a copy of the GNU Affero General Public License
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	17 # along with this program. If not, see <http://www.gnu.org/licenses/>.
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	18
3320 bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	19 import mimetypes
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	20 import json
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	21 import os
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	22 from pathlib import Path
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	23 from typing import Optional, Dict
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2937 diff changeset	24 from zope.interface import implementer
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	25 from twisted.words.protocols.jabber import xmlstream
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	26 from twisted.words.protocols.jabber import jid
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	27 from twisted.words.protocols.jabber import error as jabber_error
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	28 from twisted.internet import defer
3320 bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	29 from wokkel import disco, iwokkel, data_form
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	30 from sat.core.i18n import _
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	31 from sat.core.xmpp import SatXMPPEntity
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	32 from sat.core import exceptions
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	33 from sat.core.constants import Const as C
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	34 from sat.core.log import getLogger
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	35 from sat.tools import stream
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	36 from sat.tools.common import regex
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	37
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	38
3320 bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	39 log = getLogger(__name__)
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	40
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	41 PLUGIN_INFO = {
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	42 C.PI_NAME: "File Information Sharing",
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	43 C.PI_IMPORT_NAME: "XEP-0329",
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	44 C.PI_TYPE: "XEP",
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	45 C.PI_MODES: C.PLUG_MODE_BOTH,
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	46 C.PI_PROTOCOLS: ["XEP-0329"],
3333 ac9342f359e9 plugin XEP-0329: download thumbnails: Goffi <goffi@goffi.org> parents: 3321 diff changeset	47 C.PI_DEPENDENCIES: ["XEP-0231", "XEP-0234", "XEP-0300", "XEP-0106"],
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	48 C.PI_MAIN: "XEP_0329",
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	49 C.PI_HANDLER: "yes",
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2937 diff changeset	50 C.PI_DESCRIPTION: _("""Implementation of File Information Sharing"""),
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	51 }
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	52
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	53 NS_FIS = "urn:xmpp:fis:0"
3320 bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	54 NS_FIS_AFFILIATION = "org.salut-a-toi.fis-affiliation"
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	55 NS_FIS_CONFIGURATION = "org.salut-a-toi.fis-configuration"
3359 000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	56 NS_FIS_CREATE = "org.salut-a-toi.fis-create"
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	57
3320 bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	58 IQ_FIS_REQUEST = f'{C.IQ_GET}/query[@xmlns="{NS_FIS}"]'
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	59 # not in the standard, but needed, and it's handy to keep it here
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	60 IQ_FIS_AFFILIATION_GET = f'{C.IQ_GET}/affiliations[@xmlns="{NS_FIS_AFFILIATION}"]'
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	61 IQ_FIS_AFFILIATION_SET = f'{C.IQ_SET}/affiliations[@xmlns="{NS_FIS_AFFILIATION}"]'
3321 8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	62 IQ_FIS_CONFIGURATION_GET = f'{C.IQ_GET}/configuration[@xmlns="{NS_FIS_CONFIGURATION}"]'
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	63 IQ_FIS_CONFIGURATION_SET = f'{C.IQ_SET}/configuration[@xmlns="{NS_FIS_CONFIGURATION}"]'
3359 000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	64 IQ_FIS_CREATE_DIR = f'{C.IQ_SET}/dir[@xmlns="{NS_FIS_CREATE}"]'
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2937 diff changeset	65 SINGLE_FILES_DIR = "files"
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2937 diff changeset	66 TYPE_VIRTUAL = "virtual"
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2937 diff changeset	67 TYPE_PATH = "path"
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	68 SHARE_TYPES = (TYPE_PATH, TYPE_VIRTUAL)
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2937 diff changeset	69 KEY_TYPE = "type"
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	70
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	71
3320 bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	72 class RootPathException(Exception):
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	73 """Root path is requested"""
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	74
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	75
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	76 class ShareNode(object):
2909 90146552cde5 core (memory), plugin XEP-0329, plugin invitation: minor style improvments Goffi <goffi@goffi.org> parents: 2771 diff changeset	77 """Node containing directory or files to share, virtual or real"""
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	78
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	79 host = None
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	80
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	81 def __init__(self, name, parent, type_, access, path=None):
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	82 assert type_ in SHARE_TYPES
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	83 if name is not None:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2937 diff changeset	84 if name == ".." or "/" in name or "\\" in name:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	85 log.warning(
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2937 diff changeset	86 _("path change chars found in name [{name}], hack attempt?").format(
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	87 name=name
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	88 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	89 )
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2937 diff changeset	90 if name == "..":
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2937 diff changeset	91 name = "--"
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	92 else:
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	93 name = regex.pathEscape(name)
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	94 self.name = name
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	95 self.children = {}
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	96 self.type = type_
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	97 self.access = {} if access is None else access
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	98 assert isinstance(self.access, dict)
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	99 self.parent = None
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	100 if parent is not None:
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	101 assert name
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	102 parent.addChild(self)
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	103 else:
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	104 assert name is None
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	105 if path is not None:
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	106 if type_ != TYPE_PATH:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2937 diff changeset	107 raise exceptions.InternalError(_("path can only be set on path nodes"))
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	108 self._path = path
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	109
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	110 @property
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	111 def path(self):
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	112 return self._path
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	113
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	114 def __getitem__(self, key):
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	115 return self.children[key]
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	116
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	117 def __contains__(self, item):
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	118 return self.children.__contains__(item)
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	119
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	120 def __iter__(self):
2589 282d1314d574 plugin XEP-0329: new methods/signals to handle shares: Goffi <goffi@goffi.org> parents: 2562 diff changeset	121 return self.children.__iter__()
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	122
3032 95e2fd14a761 plugin XEP-0329: fixed ShareNode item() and values after Python 3 port Goffi <goffi@goffi.org> parents: 3028 diff changeset	123 def items(self):
95e2fd14a761 plugin XEP-0329: fixed ShareNode item() and values after Python 3 port Goffi <goffi@goffi.org> parents: 3028 diff changeset	124 return self.children.items()
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	125
3032 95e2fd14a761 plugin XEP-0329: fixed ShareNode item() and values after Python 3 port Goffi <goffi@goffi.org> parents: 3028 diff changeset	126 def values(self):
95e2fd14a761 plugin XEP-0329: fixed ShareNode item() and values after Python 3 port Goffi <goffi@goffi.org> parents: 3028 diff changeset	127 return self.children.values()
2589 282d1314d574 plugin XEP-0329: new methods/signals to handle shares: Goffi <goffi@goffi.org> parents: 2562 diff changeset	128
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	129 def getOrCreate(self, name, type_=TYPE_VIRTUAL, access=None):
2909 90146552cde5 core (memory), plugin XEP-0329, plugin invitation: minor style improvments Goffi <goffi@goffi.org> parents: 2771 diff changeset	130 """Get a node or create a virtual node and return it"""
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	131 if access is None:
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	132 access = {C.ACCESS_PERM_READ: {KEY_TYPE: C.ACCESS_TYPE_PUBLIC}}
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	133 try:
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	134 return self.children[name]
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	135 except KeyError:
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	136 node = ShareNode(name, self, type_=type_, access=access)
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	137 return node
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	138
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	139 def addChild(self, node):
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	140 if node.parent is not None:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2937 diff changeset	141 raise exceptions.ConflictError(_("a node can't have several parents"))
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	142 node.parent = self
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	143 self.children[node.name] = node
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	144
2589 282d1314d574 plugin XEP-0329: new methods/signals to handle shares: Goffi <goffi@goffi.org> parents: 2562 diff changeset	145 def removeFromParent(self):
282d1314d574 plugin XEP-0329: new methods/signals to handle shares: Goffi <goffi@goffi.org> parents: 2562 diff changeset	146 try:
282d1314d574 plugin XEP-0329: new methods/signals to handle shares: Goffi <goffi@goffi.org> parents: 2562 diff changeset	147 del self.parent.children[self.name]
282d1314d574 plugin XEP-0329: new methods/signals to handle shares: Goffi <goffi@goffi.org> parents: 2562 diff changeset	148 except TypeError:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	149 raise exceptions.InternalError(
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2937 diff changeset	150 "trying to remove a node from inexisting parent"
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	151 )
2589 282d1314d574 plugin XEP-0329: new methods/signals to handle shares: Goffi <goffi@goffi.org> parents: 2562 diff changeset	152 except KeyError:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2937 diff changeset	153 raise exceptions.InternalError("node not found in parent's children")
2589 282d1314d574 plugin XEP-0329: new methods/signals to handle shares: Goffi <goffi@goffi.org> parents: 2562 diff changeset	154 self.parent = None
282d1314d574 plugin XEP-0329: new methods/signals to handle shares: Goffi <goffi@goffi.org> parents: 2562 diff changeset	155
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	156 def _checkNodePermission(self, client, node, perms, peer_jid):
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	157 """Check access to this node for peer_jid
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	158
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	159 @param node(SharedNode): node to check access
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	160 @param perms(unicode): permissions to check, iterable of C.ACCESS_PERM_*
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	161 @param peer_jid(jid.JID): entity which try to access the node
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	162 @return (bool): True if entity can access
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	163 """
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2937 diff changeset	164 file_data = {"access": self.access, "owner": client.jid.userhostJID()}
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	165 try:
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	166 self.host.memory.checkFilePermission(file_data, peer_jid, perms)
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	167 except exceptions.PermissionError:
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	168 return False
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	169 else:
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	170 return True
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	171
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	172 def checkPermissions(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	173 self, client, peer_jid, perms=(C.ACCESS_PERM_READ,), check_parents=True
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	174 ):
2909 90146552cde5 core (memory), plugin XEP-0329, plugin invitation: minor style improvments Goffi <goffi@goffi.org> parents: 2771 diff changeset	175 """Check that peer_jid can access this node and all its parents
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	176
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	177 @param peer_jid(jid.JID): entrity trying to access the node
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	178 @param perms(unicode): permissions to check, iterable of C.ACCESS_PERM_*
2909 90146552cde5 core (memory), plugin XEP-0329, plugin invitation: minor style improvments Goffi <goffi@goffi.org> parents: 2771 diff changeset	179 @param check_parents(bool): if True, access of all parents of this node will be
90146552cde5 core (memory), plugin XEP-0329, plugin invitation: minor style improvments Goffi <goffi@goffi.org> parents: 2771 diff changeset	180 checked too
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	181 @return (bool): True if entity can access this node
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	182 """
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	183 peer_jid = peer_jid.userhostJID()
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	184 if peer_jid == client.jid.userhostJID():
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	185 return True
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	186
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	187 parent = self
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	188 while parent != None:
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	189 if not self._checkNodePermission(client, parent, perms, peer_jid):
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	190 return False
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	191 parent = parent.parent
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	192
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	193 return True
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	194
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	195 @staticmethod
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	196 def find(client, path, peer_jid, perms=(C.ACCESS_PERM_READ,)):
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	197 """find node corresponding to a path
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	198
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	199 @param path(unicode): path to the requested file or directory
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	200 @param peer_jid(jid.JID): entity trying to find the node
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	201 used to check permission
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	202 @return (dict, unicode): shared data, remaining path
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	203 @raise exceptions.PermissionError: user can't access this file
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	204 @raise exceptions.DataError: path is invalid
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	205 @raise NotFound: path lead to a non existing file/directory
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	206 """
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2937 diff changeset	207 path_elts = [_f for _f in path.split("/") if _f]
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	208
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2937 diff changeset	209 if ".." in path_elts:
2909 90146552cde5 core (memory), plugin XEP-0329, plugin invitation: minor style improvments Goffi <goffi@goffi.org> parents: 2771 diff changeset	210 log.warning(_(
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2937 diff changeset	211 'parent dir ("..") found in path, hack attempt? path is {path} '
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2937 diff changeset	212 '[{profile}]').format(path=path, profile=client.profile))
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2937 diff changeset	213 raise exceptions.PermissionError("illegal path elements")
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	214
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	215 if not path_elts:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2937 diff changeset	216 raise exceptions.DataError(_("path is invalid: {path}").format(path=path))
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	217
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	218 node = client._XEP_0329_root_node
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	219
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	220 while path_elts:
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	221 if node.type == TYPE_VIRTUAL:
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	222 try:
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	223 node = node[path_elts.pop(0)]
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	224 except KeyError:
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	225 raise exceptions.NotFound
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	226 elif node.type == TYPE_PATH:
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	227 break
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	228
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	229 if not node.checkPermissions(client, peer_jid, perms=perms):
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2937 diff changeset	230 raise exceptions.PermissionError("permission denied")
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	231
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2937 diff changeset	232 return node, "/".join(path_elts)
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	233
2589 282d1314d574 plugin XEP-0329: new methods/signals to handle shares: Goffi <goffi@goffi.org> parents: 2562 diff changeset	234 def findByLocalPath(self, path):
282d1314d574 plugin XEP-0329: new methods/signals to handle shares: Goffi <goffi@goffi.org> parents: 2562 diff changeset	235 """retrieve nodes linking to local path
282d1314d574 plugin XEP-0329: new methods/signals to handle shares: Goffi <goffi@goffi.org> parents: 2562 diff changeset	236
282d1314d574 plugin XEP-0329: new methods/signals to handle shares: Goffi <goffi@goffi.org> parents: 2562 diff changeset	237 @return (list[ShareNode]): found nodes associated to path
282d1314d574 plugin XEP-0329: new methods/signals to handle shares: Goffi <goffi@goffi.org> parents: 2562 diff changeset	238 @raise exceptions.NotFound: no node has been found with this path
282d1314d574 plugin XEP-0329: new methods/signals to handle shares: Goffi <goffi@goffi.org> parents: 2562 diff changeset	239 """
282d1314d574 plugin XEP-0329: new methods/signals to handle shares: Goffi <goffi@goffi.org> parents: 2562 diff changeset	240 shared_paths = self.getSharedPaths()
282d1314d574 plugin XEP-0329: new methods/signals to handle shares: Goffi <goffi@goffi.org> parents: 2562 diff changeset	241 try:
282d1314d574 plugin XEP-0329: new methods/signals to handle shares: Goffi <goffi@goffi.org> parents: 2562 diff changeset	242 return shared_paths[path]
282d1314d574 plugin XEP-0329: new methods/signals to handle shares: Goffi <goffi@goffi.org> parents: 2562 diff changeset	243 except KeyError:
282d1314d574 plugin XEP-0329: new methods/signals to handle shares: Goffi <goffi@goffi.org> parents: 2562 diff changeset	244 raise exceptions.NotFound
282d1314d574 plugin XEP-0329: new methods/signals to handle shares: Goffi <goffi@goffi.org> parents: 2562 diff changeset	245
282d1314d574 plugin XEP-0329: new methods/signals to handle shares: Goffi <goffi@goffi.org> parents: 2562 diff changeset	246 def _getSharedPaths(self, node, paths):
282d1314d574 plugin XEP-0329: new methods/signals to handle shares: Goffi <goffi@goffi.org> parents: 2562 diff changeset	247 if node.type == TYPE_VIRTUAL:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2937 diff changeset	248 for node in node.values():
2589 282d1314d574 plugin XEP-0329: new methods/signals to handle shares: Goffi <goffi@goffi.org> parents: 2562 diff changeset	249 self._getSharedPaths(node, paths)
282d1314d574 plugin XEP-0329: new methods/signals to handle shares: Goffi <goffi@goffi.org> parents: 2562 diff changeset	250 elif node.type == TYPE_PATH:
282d1314d574 plugin XEP-0329: new methods/signals to handle shares: Goffi <goffi@goffi.org> parents: 2562 diff changeset	251 paths.setdefault(node.path, []).append(node)
282d1314d574 plugin XEP-0329: new methods/signals to handle shares: Goffi <goffi@goffi.org> parents: 2562 diff changeset	252 else:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	253 raise exceptions.InternalError(
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2937 diff changeset	254 "unknown node type: {type}".format(type=node.type)
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	255 )
2589 282d1314d574 plugin XEP-0329: new methods/signals to handle shares: Goffi <goffi@goffi.org> parents: 2562 diff changeset	256
282d1314d574 plugin XEP-0329: new methods/signals to handle shares: Goffi <goffi@goffi.org> parents: 2562 diff changeset	257 def getSharedPaths(self):
282d1314d574 plugin XEP-0329: new methods/signals to handle shares: Goffi <goffi@goffi.org> parents: 2562 diff changeset	258 """retrieve nodes by shared path
282d1314d574 plugin XEP-0329: new methods/signals to handle shares: Goffi <goffi@goffi.org> parents: 2562 diff changeset	259
282d1314d574 plugin XEP-0329: new methods/signals to handle shares: Goffi <goffi@goffi.org> parents: 2562 diff changeset	260 this method will retrieve recursively shared path in children of this node
282d1314d574 plugin XEP-0329: new methods/signals to handle shares: Goffi <goffi@goffi.org> parents: 2562 diff changeset	261 @return (dict): map from shared path to list of nodes
282d1314d574 plugin XEP-0329: new methods/signals to handle shares: Goffi <goffi@goffi.org> parents: 2562 diff changeset	262 """
282d1314d574 plugin XEP-0329: new methods/signals to handle shares: Goffi <goffi@goffi.org> parents: 2562 diff changeset	263 if self.type == TYPE_PATH:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	264 raise exceptions.InternalError(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	265 "getSharedPaths must be used on a virtual node"
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	266 )
2589 282d1314d574 plugin XEP-0329: new methods/signals to handle shares: Goffi <goffi@goffi.org> parents: 2562 diff changeset	267 paths = {}
282d1314d574 plugin XEP-0329: new methods/signals to handle shares: Goffi <goffi@goffi.org> parents: 2562 diff changeset	268 self._getSharedPaths(self, paths)
282d1314d574 plugin XEP-0329: new methods/signals to handle shares: Goffi <goffi@goffi.org> parents: 2562 diff changeset	269 return paths
282d1314d574 plugin XEP-0329: new methods/signals to handle shares: Goffi <goffi@goffi.org> parents: 2562 diff changeset	270
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	271
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	272 class XEP_0329(object):
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	273 def __init__(self, host):
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	274 log.info(_("File Information Sharing initialization"))
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	275 self.host = host
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	276 ShareNode.host = host
3333 ac9342f359e9 plugin XEP-0329: download thumbnails: Goffi <goffi@goffi.org> parents: 3321 diff changeset	277 self._b = host.plugins["XEP-0231"]
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	278 self._h = host.plugins["XEP-0300"]
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	279 self._jf = host.plugins["XEP-0234"]
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	280 host.bridge.addMethod(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	281 "FISList",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	282 ".plugin",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	283 in_sign="ssa{ss}s",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	284 out_sign="aa{ss}",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	285 method=self._listFiles,
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2937 diff changeset	286 async_=True,
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	287 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	288 host.bridge.addMethod(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	289 "FISLocalSharesGet",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	290 ".plugin",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	291 in_sign="s",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	292 out_sign="as",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	293 method=self._localSharesGet,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	294 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	295 host.bridge.addMethod(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	296 "FISSharePath",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	297 ".plugin",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	298 in_sign="ssss",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	299 out_sign="s",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	300 method=self._sharePath,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	301 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	302 host.bridge.addMethod(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	303 "FISUnsharePath",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	304 ".plugin",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	305 in_sign="ss",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	306 out_sign="",
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	307 method=self._unsharePath,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	308 )
3320 bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	309 host.bridge.addMethod(
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	310 "FISAffiliationsGet",
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	311 ".plugin",
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	312 in_sign="ssss",
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	313 out_sign="a{ss}",
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	314 method=self._affiliationsGet,
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	315 async_=True,
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	316 )
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	317 host.bridge.addMethod(
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	318 "FISAffiliationsSet",
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	319 ".plugin",
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	320 in_sign="sssa{ss}s",
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	321 out_sign="",
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	322 method=self._affiliationsSet,
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	323 async_=True,
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	324 )
3321 8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	325 host.bridge.addMethod(
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	326 "FISConfigurationGet",
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	327 ".plugin",
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	328 in_sign="ssss",
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	329 out_sign="a{ss}",
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	330 method=self._configurationGet,
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	331 async_=True,
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	332 )
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	333 host.bridge.addMethod(
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	334 "FISConfigurationSet",
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	335 ".plugin",
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	336 in_sign="sssa{ss}s",
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	337 out_sign="",
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	338 method=self._configurationSet,
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	339 async_=True,
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	340 )
3359 000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	341 host.bridge.addMethod(
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	342 "FISCreateDir",
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	343 ".plugin",
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	344 in_sign="sssa{ss}s",
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	345 out_sign="",
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	346 method=self._createDir,
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	347 async_=True,
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	348 )
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	349 host.bridge.addSignal("FISSharedPathNew", ".plugin", signature="sss")
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	350 host.bridge.addSignal("FISSharedPathRemoved", ".plugin", signature="ss")
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	351 host.trigger.add("XEP-0234_fileSendingRequest", self._fileSendingRequestTrigger)
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	352 host.registerNamespace("fis", NS_FIS)
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	353
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	354 def getHandler(self, client):
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	355 return XEP_0329_handler(self)
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	356
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	357 def profileConnected(self, client):
3359 000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	358 if client.is_component:
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	359 client._file_sharing_allowed_hosts = self.host.memory.getConfig(
3402 08a3e34aead1 plugin XEP-0329: minor reformatting Goffi <goffi@goffi.org> parents: 3366 diff changeset	360 'component file_sharing',
08a3e34aead1 plugin XEP-0329: minor reformatting Goffi <goffi@goffi.org> parents: 3366 diff changeset	361 'http_upload_allowed_hosts_list') or [client.host]
3359 000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	362 else:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	363 client._XEP_0329_root_node = ShareNode(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	364 None,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	365 None,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	366 TYPE_VIRTUAL,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	367 {C.ACCESS_PERM_READ: {KEY_TYPE: C.ACCESS_TYPE_PUBLIC}},
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	368 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	369 client._XEP_0329_names_data = {} # name to share map
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	370
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	371 def _fileSendingRequestTrigger(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	372 self, client, session, content_data, content_name, file_data, file_elt
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	373 ):
2909 90146552cde5 core (memory), plugin XEP-0329, plugin invitation: minor style improvments Goffi <goffi@goffi.org> parents: 2771 diff changeset	374 """This trigger check that a requested file is available, and fill suitable data
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	375
2909 90146552cde5 core (memory), plugin XEP-0329, plugin invitation: minor style improvments Goffi <goffi@goffi.org> parents: 2771 diff changeset	376 Path and name are used to retrieve the file. If path is missing, we try our luck
90146552cde5 core (memory), plugin XEP-0329, plugin invitation: minor style improvments Goffi <goffi@goffi.org> parents: 2771 diff changeset	377 with known names
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	378 """
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	379 if client.is_component:
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	380 return True, None
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	381
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	382 try:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2937 diff changeset	383 name = file_data["name"]
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	384 except KeyError:
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	385 return True, None
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2937 diff changeset	386 assert "/" not in name
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	387
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2937 diff changeset	388 path = file_data.get("path")
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	389 if path is not None:
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	390 # we have a path, we can follow it to find node
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	391 try:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2937 diff changeset	392 node, rem_path = ShareNode.find(client, path, session["peer_jid"])
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	393 except (exceptions.PermissionError, exceptions.NotFound):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	394 # no file, or file not allowed, we continue normal workflow
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	395 return True, None
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	396 except exceptions.DataError:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2937 diff changeset	397 log.warning(_("invalid path: {path}").format(path=path))
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	398 return True, None
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	399
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	400 if node.type == TYPE_VIRTUAL:
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	401 # we have a virtual node, so name must link to a path node
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	402 try:
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	403 path = node[name].path
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	404 except KeyError:
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	405 return True, None
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	406 elif node.type == TYPE_PATH:
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	407 # we have a path node, so we can retrieve the full path now
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	408 path = os.path.join(node.path, rem_path, name)
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	409 else:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	410 raise exceptions.InternalError(
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2937 diff changeset	411 "unknown type: {type}".format(type=node.type)
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	412 )
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	413 if not os.path.exists(path):
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	414 return True, None
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	415 size = os.path.getsize(path)
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	416 else:
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	417 # we don't have the path, we try to find the file by its name
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	418 try:
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	419 name_data = client._XEP_0329_names_data[name]
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	420 except KeyError:
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	421 return True, None
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	422
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2937 diff changeset	423 for path, shared_file in name_data.items():
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	424 if True: # FIXME: filters are here
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	425 break
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	426 else:
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	427 return True, None
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2937 diff changeset	428 parent_node = shared_file["parent"]
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2937 diff changeset	429 if not parent_node.checkPermissions(client, session["peer_jid"]):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	430 log.warning(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	431 _(
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2937 diff changeset	432 "{peer_jid} requested a file (s)he can't access [{profile}]"
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2937 diff changeset	433 ).format(peer_jid=session["peer_jid"], profile=client.profile)
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	434 )
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	435 return True, None
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2937 diff changeset	436 size = shared_file["size"]
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	437
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2937 diff changeset	438 file_data["size"] = size
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2937 diff changeset	439 file_elt.addElement("size", content=str(size))
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2937 diff changeset	440 hash_algo = file_data["hash_algo"] = self._h.getDefaultAlgo()
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2937 diff changeset	441 hasher = file_data["hash_hasher"] = self._h.getHasher(hash_algo)
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	442 file_elt.addChild(self._h.buildHashUsedElt(hash_algo))
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	443 content_data["stream_object"] = stream.FileStreamObject(
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	444 self.host,
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	445 client,
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	446 path,
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	447 uid=self._jf.getProgressId(session, content_name),
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	448 size=size,
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	449 data_cb=lambda data: hasher.update(data),
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	450 )
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	451 return False, True
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	452
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	453 # common methods
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	454
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	455 def _requestHandler(self, client, iq_elt, root_nodes_cb, files_from_node_cb):
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	456 iq_elt.handled = True
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	457 node = iq_elt.query.getAttribute("node")
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	458 if not node:
2923 1fd3ecb3351a plugin XEP-0329: use local part of jid for components: Goffi <goffi@goffi.org> parents: 2909 diff changeset	459 d = defer.maybeDeferred(root_nodes_cb, client, iq_elt)
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	460 else:
2923 1fd3ecb3351a plugin XEP-0329: use local part of jid for components: Goffi <goffi@goffi.org> parents: 2909 diff changeset	461 d = defer.maybeDeferred(files_from_node_cb, client, iq_elt, node)
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	462 d.addErrback(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	463 lambda failure_: log.error(
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2937 diff changeset	464 _("error while retrieving files: {msg}").format(msg=failure_)
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	465 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	466 )
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	467
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	468 def _iqError(self, client, iq_elt, condition="item-not-found"):
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	469 error_elt = jabber_error.StanzaError(condition).toResponse(iq_elt)
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	470 client.send(error_elt)
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	471
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	472 # client
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	473
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	474 def _addPathData(self, client, query_elt, path, parent_node):
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	475 """Fill query_elt with files/directories found in path"""
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	476 name = os.path.basename(path)
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	477 if os.path.isfile(path):
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	478 size = os.path.getsize(path)
2512 025afb04c10b plugin XEP-0234: some cleaning + added triggers to allow plugins to change parsing/generation of <file> element Goffi <goffi@goffi.org> parents: 2503 diff changeset	479 mime_type = mimetypes.guess_type(path, strict=False)[0]
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	480 file_elt = self._jf.buildFileElement(
3314 5887fb414758 component file sharing: add/parse affiliation when possible Goffi <goffi@goffi.org> parents: 3136 diff changeset	481 client=client, name=name, size=size, mime_type=mime_type,
5887fb414758 component file sharing: add/parse affiliation when possible Goffi <goffi@goffi.org> parents: 3136 diff changeset	482 modified=os.path.getmtime(path)
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	483 )
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	484
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	485 query_elt.addChild(file_elt)
2909 90146552cde5 core (memory), plugin XEP-0329, plugin invitation: minor style improvments Goffi <goffi@goffi.org> parents: 2771 diff changeset	486 # we don't specify hash as it would be too resource intensive to calculate
90146552cde5 core (memory), plugin XEP-0329, plugin invitation: minor style improvments Goffi <goffi@goffi.org> parents: 2771 diff changeset	487 # it for all files.
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	488 # we add file to name_data, so users can request it later
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	489 name_data = client._XEP_0329_names_data.setdefault(name, {})
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	490 if path not in name_data:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	491 name_data[path] = {
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	492 "size": size,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	493 "mime_type": mime_type,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	494 "parent": parent_node,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	495 }
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	496 else:
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	497 # we have a directory
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	498 directory_elt = query_elt.addElement("directory")
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	499 directory_elt["name"] = name
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	500
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	501 def _pathNodeHandler(self, client, iq_elt, query_elt, node, path):
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	502 """Fill query_elt for path nodes, i.e. physical directories"""
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	503 path = os.path.join(node.path, path)
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	504
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	505 if not os.path.exists(path):
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	506 # path may have been moved since it has been shared
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	507 return self._iqError(client, iq_elt)
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	508 elif os.path.isfile(path):
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	509 self._addPathData(client, query_elt, path, node)
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	510 else:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	511 for name in sorted(os.listdir(path.encode("utf-8")), key=lambda n: n.lower()):
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	512 try:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	513 name = name.decode("utf-8", "strict")
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	514 except UnicodeDecodeError as e:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	515 log.warning(
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2937 diff changeset	516 _("ignoring invalid unicode name ({name}): {msg}").format(
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	517 name=name.decode("utf-8", "replace"), msg=e
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	518 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	519 )
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	520 continue
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	521 full_path = os.path.join(path, name)
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	522 self._addPathData(client, query_elt, full_path, node)
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	523
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	524 def _virtualNodeHandler(self, client, peer_jid, iq_elt, query_elt, node):
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	525 """Fill query_elt for virtual nodes"""
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2937 diff changeset	526 for name, child_node in node.items():
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	527 if not child_node.checkPermissions(client, peer_jid, check_parents=False):
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	528 continue
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	529 node_type = child_node.type
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	530 if node_type == TYPE_VIRTUAL:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	531 directory_elt = query_elt.addElement("directory")
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	532 directory_elt["name"] = name
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	533 elif node_type == TYPE_PATH:
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	534 self._addPathData(client, query_elt, child_node.path, child_node)
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	535 else:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	536 raise exceptions.InternalError(
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2937 diff changeset	537 _("unexpected type: {type}").format(type=node_type)
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	538 )
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	539
2923 1fd3ecb3351a plugin XEP-0329: use local part of jid for components: Goffi <goffi@goffi.org> parents: 2909 diff changeset	540 def _getRootNodesCb(self, client, iq_elt):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	541 peer_jid = jid.JID(iq_elt["from"])
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	542 iq_result_elt = xmlstream.toResponse(iq_elt, "result")
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	543 query_elt = iq_result_elt.addElement((NS_FIS, "query"))
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2937 diff changeset	544 for name, node in client._XEP_0329_root_node.items():
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	545 if not node.checkPermissions(client, peer_jid, check_parents=False):
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	546 continue
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	547 directory_elt = query_elt.addElement("directory")
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	548 directory_elt["name"] = name
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	549 client.send(iq_result_elt)
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	550
2923 1fd3ecb3351a plugin XEP-0329: use local part of jid for components: Goffi <goffi@goffi.org> parents: 2909 diff changeset	551 def _getFilesFromNodeCb(self, client, iq_elt, node_path):
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	552 """Main method to retrieve files/directories from a node_path"""
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2937 diff changeset	553 peer_jid = jid.JID(iq_elt["from"])
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	554 try:
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	555 node, path = ShareNode.find(client, node_path, peer_jid)
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	556 except (exceptions.PermissionError, exceptions.NotFound):
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	557 return self._iqError(client, iq_elt)
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	558 except exceptions.DataError:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	559 return self._iqError(client, iq_elt, condition="not-acceptable")
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	560
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	561 node_type = node.type
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	562 peer_jid = jid.JID(iq_elt["from"])
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	563 iq_result_elt = xmlstream.toResponse(iq_elt, "result")
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	564 query_elt = iq_result_elt.addElement((NS_FIS, "query"))
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2937 diff changeset	565 query_elt["node"] = node_path
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	566
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	567 # we now fill query_elt according to node_type
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	568 if node_type == TYPE_PATH:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	569 # it's a physical path
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	570 self._pathNodeHandler(client, iq_elt, query_elt, node, path)
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	571 elif node_type == TYPE_VIRTUAL:
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	572 assert not path
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	573 self._virtualNodeHandler(client, peer_jid, iq_elt, query_elt, node)
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	574 else:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	575 raise exceptions.InternalError(
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2937 diff changeset	576 _("unknown node type: {type}").format(type=node_type)
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	577 )
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	578
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	579 client.send(iq_result_elt)
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	580
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	581 def onRequest(self, iq_elt, client):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	582 return self._requestHandler(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	583 client, iq_elt, self._getRootNodesCb, self._getFilesFromNodeCb
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	584 )
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	585
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	586 # Component
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	587
2923 1fd3ecb3351a plugin XEP-0329: use local part of jid for components: Goffi <goffi@goffi.org> parents: 2909 diff changeset	588 def _compParseJids(self, client, iq_elt):
1fd3ecb3351a plugin XEP-0329: use local part of jid for components: Goffi <goffi@goffi.org> parents: 2909 diff changeset	589 """Retrieve peer_jid and owner to use from IQ stanza
1fd3ecb3351a plugin XEP-0329: use local part of jid for components: Goffi <goffi@goffi.org> parents: 2909 diff changeset	590
1fd3ecb3351a plugin XEP-0329: use local part of jid for components: Goffi <goffi@goffi.org> parents: 2909 diff changeset	591 @param iq_elt(domish.Element): IQ stanza of the FIS request
1fd3ecb3351a plugin XEP-0329: use local part of jid for components: Goffi <goffi@goffi.org> parents: 2909 diff changeset	592 @return (tuple[jid.JID, jid.JID]): peer_jid and owner
1fd3ecb3351a plugin XEP-0329: use local part of jid for components: Goffi <goffi@goffi.org> parents: 2909 diff changeset	593 """
1fd3ecb3351a plugin XEP-0329: use local part of jid for components: Goffi <goffi@goffi.org> parents: 2909 diff changeset	594
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	595 @defer.inlineCallbacks
2923 1fd3ecb3351a plugin XEP-0329: use local part of jid for components: Goffi <goffi@goffi.org> parents: 2909 diff changeset	596 def _compGetRootNodesCb(self, client, iq_elt):
3366 e09cb08166a3 plugin XEP-0329, core(xmpp): moved `_compParseJids` to `SatXMPPComponent`: Goffi <goffi@goffi.org> parents: 3365 diff changeset	597 peer_jid, owner = client.getOwnerAndPeer(iq_elt)
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	598 files_data = yield self.host.memory.getFiles(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	599 client,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	600 peer_jid=peer_jid,
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2937 diff changeset	601 parent="",
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	602 type_=C.FILE_TYPE_DIRECTORY,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	603 owner=owner,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	604 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	605 iq_result_elt = xmlstream.toResponse(iq_elt, "result")
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	606 query_elt = iq_result_elt.addElement((NS_FIS, "query"))
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	607 for file_data in files_data:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2937 diff changeset	608 name = file_data["name"]
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2937 diff changeset	609 directory_elt = query_elt.addElement("directory")
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2937 diff changeset	610 directory_elt["name"] = name
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	611 client.send(iq_result_elt)
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	612
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	613 @defer.inlineCallbacks
2923 1fd3ecb3351a plugin XEP-0329: use local part of jid for components: Goffi <goffi@goffi.org> parents: 2909 diff changeset	614 def _compGetFilesFromNodeCb(self, client, iq_elt, node_path):
1fd3ecb3351a plugin XEP-0329: use local part of jid for components: Goffi <goffi@goffi.org> parents: 2909 diff changeset	615 """Retrieve files from local files repository according to permissions
2528 65e278997715 component file sharing: comments metadata: Goffi <goffi@goffi.org> parents: 2527 diff changeset	616
65e278997715 component file sharing: comments metadata: Goffi <goffi@goffi.org> parents: 2527 diff changeset	617 result stanza is then built and sent to requestor
2909 90146552cde5 core (memory), plugin XEP-0329, plugin invitation: minor style improvments Goffi <goffi@goffi.org> parents: 2771 diff changeset	618 @trigger XEP-0329_compGetFilesFromNode(client, iq_elt, owner, node_path,
90146552cde5 core (memory), plugin XEP-0329, plugin invitation: minor style improvments Goffi <goffi@goffi.org> parents: 2771 diff changeset	619 files_data):
90146552cde5 core (memory), plugin XEP-0329, plugin invitation: minor style improvments Goffi <goffi@goffi.org> parents: 2771 diff changeset	620 can be used to add data/elements
2528 65e278997715 component file sharing: comments metadata: Goffi <goffi@goffi.org> parents: 2527 diff changeset	621 """
3366 e09cb08166a3 plugin XEP-0329, core(xmpp): moved `_compParseJids` to `SatXMPPComponent`: Goffi <goffi@goffi.org> parents: 3365 diff changeset	622 peer_jid, owner = client.getOwnerAndPeer(iq_elt)
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	623 try:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	624 files_data = yield self.host.memory.getFiles(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	625 client, peer_jid=peer_jid, path=node_path, owner=owner
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	626 )
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	627 except exceptions.NotFound:
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	628 self._iqError(client, iq_elt)
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	629 return
2937 db0890c9c7db plugin XEP-0329: correctly handle error if something is wrong when getting file from component. Goffi <goffi@goffi.org> parents: 2923 diff changeset	630 except exceptions.PermissionError:
db0890c9c7db plugin XEP-0329: correctly handle error if something is wrong when getting file from component. Goffi <goffi@goffi.org> parents: 2923 diff changeset	631 self._iqError(client, iq_elt, condition='not-allowed')
db0890c9c7db plugin XEP-0329: correctly handle error if something is wrong when getting file from component. Goffi <goffi@goffi.org> parents: 2923 diff changeset	632 return
db0890c9c7db plugin XEP-0329: correctly handle error if something is wrong when getting file from component. Goffi <goffi@goffi.org> parents: 2923 diff changeset	633 except Exception as e:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2937 diff changeset	634 log.error("internal server error: {e}".format(e=e))
2937 db0890c9c7db plugin XEP-0329: correctly handle error if something is wrong when getting file from component. Goffi <goffi@goffi.org> parents: 2923 diff changeset	635 self._iqError(client, iq_elt, condition='internal-server-error')
db0890c9c7db plugin XEP-0329: correctly handle error if something is wrong when getting file from component. Goffi <goffi@goffi.org> parents: 2923 diff changeset	636 return
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	637 iq_result_elt = xmlstream.toResponse(iq_elt, "result")
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	638 query_elt = iq_result_elt.addElement((NS_FIS, "query"))
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2937 diff changeset	639 query_elt["node"] = node_path
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	640 if not self.host.trigger.point(
3314 5887fb414758 component file sharing: add/parse affiliation when possible Goffi <goffi@goffi.org> parents: 3136 diff changeset	641 "XEP-0329_compGetFilesFromNode",
5887fb414758 component file sharing: add/parse affiliation when possible Goffi <goffi@goffi.org> parents: 3136 diff changeset	642 client,
5887fb414758 component file sharing: add/parse affiliation when possible Goffi <goffi@goffi.org> parents: 3136 diff changeset	643 iq_elt,
5887fb414758 component file sharing: add/parse affiliation when possible Goffi <goffi@goffi.org> parents: 3136 diff changeset	644 iq_result_elt,
5887fb414758 component file sharing: add/parse affiliation when possible Goffi <goffi@goffi.org> parents: 3136 diff changeset	645 owner,
5887fb414758 component file sharing: add/parse affiliation when possible Goffi <goffi@goffi.org> parents: 3136 diff changeset	646 node_path,
5887fb414758 component file sharing: add/parse affiliation when possible Goffi <goffi@goffi.org> parents: 3136 diff changeset	647 files_data
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	648 ):
2528 65e278997715 component file sharing: comments metadata: Goffi <goffi@goffi.org> parents: 2527 diff changeset	649 return
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	650 for file_data in files_data:
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3032 diff changeset	651 if file_data['type'] == C.FILE_TYPE_DIRECTORY:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3032 diff changeset	652 directory_elt = query_elt.addElement("directory")
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3032 diff changeset	653 directory_elt['name'] = file_data['name']
3314 5887fb414758 component file sharing: add/parse affiliation when possible Goffi <goffi@goffi.org> parents: 3136 diff changeset	654 self.host.trigger.point(
5887fb414758 component file sharing: add/parse affiliation when possible Goffi <goffi@goffi.org> parents: 3136 diff changeset	655 "XEP-0329_compGetFilesFromNode_build_directory",
5887fb414758 component file sharing: add/parse affiliation when possible Goffi <goffi@goffi.org> parents: 3136 diff changeset	656 client,
5887fb414758 component file sharing: add/parse affiliation when possible Goffi <goffi@goffi.org> parents: 3136 diff changeset	657 file_data,
5887fb414758 component file sharing: add/parse affiliation when possible Goffi <goffi@goffi.org> parents: 3136 diff changeset	658 directory_elt,
5887fb414758 component file sharing: add/parse affiliation when possible Goffi <goffi@goffi.org> parents: 3136 diff changeset	659 owner,
5887fb414758 component file sharing: add/parse affiliation when possible Goffi <goffi@goffi.org> parents: 3136 diff changeset	660 node_path,
5887fb414758 component file sharing: add/parse affiliation when possible Goffi <goffi@goffi.org> parents: 3136 diff changeset	661 )
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3032 diff changeset	662 else:
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3032 diff changeset	663 file_elt = self._jf.buildFileElementFromDict(
3314 5887fb414758 component file sharing: add/parse affiliation when possible Goffi <goffi@goffi.org> parents: 3136 diff changeset	664 client,
5887fb414758 component file sharing: add/parse affiliation when possible Goffi <goffi@goffi.org> parents: 3136 diff changeset	665 file_data,
5887fb414758 component file sharing: add/parse affiliation when possible Goffi <goffi@goffi.org> parents: 3136 diff changeset	666 modified=file_data.get("modified", file_data["created"])
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3032 diff changeset	667 )
fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3032 diff changeset	668 query_elt.addChild(file_elt)
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	669 client.send(iq_result_elt)
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	670
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	671 def onComponentRequest(self, iq_elt, client):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	672 return self._requestHandler(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	673 client, iq_elt, self._compGetRootNodesCb, self._compGetFilesFromNodeCb
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	674 )
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	675
3333 ac9342f359e9 plugin XEP-0329: download thumbnails: Goffi <goffi@goffi.org> parents: 3321 diff changeset	676 async def _parseResult(self, client, peer_jid, iq_elt):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	677 query_elt = next(iq_elt.elements(NS_FIS, "query"))
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	678 files = []
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	679
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	680 for elt in query_elt.elements():
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	681 if elt.name == "file":
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	682 # we have a file
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	683 try:
3333 ac9342f359e9 plugin XEP-0329: download thumbnails: Goffi <goffi@goffi.org> parents: 3321 diff changeset	684 file_data = await self._jf.parseFileElement(client, elt)
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	685 except exceptions.DataError:
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	686 continue
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2937 diff changeset	687 file_data["type"] = C.FILE_TYPE_FILE
3333 ac9342f359e9 plugin XEP-0329: download thumbnails: Goffi <goffi@goffi.org> parents: 3321 diff changeset	688 try:
ac9342f359e9 plugin XEP-0329: download thumbnails: Goffi <goffi@goffi.org> parents: 3321 diff changeset	689 thumbs = file_data['extra'][C.KEY_THUMBNAILS]
ac9342f359e9 plugin XEP-0329: download thumbnails: Goffi <goffi@goffi.org> parents: 3321 diff changeset	690 except KeyError:
ac9342f359e9 plugin XEP-0329: download thumbnails: Goffi <goffi@goffi.org> parents: 3321 diff changeset	691 log.debug(f"No thumbnail found for {file_data}")
ac9342f359e9 plugin XEP-0329: download thumbnails: Goffi <goffi@goffi.org> parents: 3321 diff changeset	692 else:
ac9342f359e9 plugin XEP-0329: download thumbnails: Goffi <goffi@goffi.org> parents: 3321 diff changeset	693 for thumb in thumbs:
ac9342f359e9 plugin XEP-0329: download thumbnails: Goffi <goffi@goffi.org> parents: 3321 diff changeset	694 if 'url' not in thumb and "id" in thumb:
ac9342f359e9 plugin XEP-0329: download thumbnails: Goffi <goffi@goffi.org> parents: 3321 diff changeset	695 try:
ac9342f359e9 plugin XEP-0329: download thumbnails: Goffi <goffi@goffi.org> parents: 3321 diff changeset	696 file_path = await self._b.getFile(client, peer_jid, thumb['id'])
ac9342f359e9 plugin XEP-0329: download thumbnails: Goffi <goffi@goffi.org> parents: 3321 diff changeset	697 except Exception as e:
ac9342f359e9 plugin XEP-0329: download thumbnails: Goffi <goffi@goffi.org> parents: 3321 diff changeset	698 log.warning(f"Can't get thumbnail {thumb['id']!r} for {file_data}: {e}")
ac9342f359e9 plugin XEP-0329: download thumbnails: Goffi <goffi@goffi.org> parents: 3321 diff changeset	699 else:
ac9342f359e9 plugin XEP-0329: download thumbnails: Goffi <goffi@goffi.org> parents: 3321 diff changeset	700 thumb['filename'] = file_path.name
ac9342f359e9 plugin XEP-0329: download thumbnails: Goffi <goffi@goffi.org> parents: 3321 diff changeset	701
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	702 elif elt.name == "directory" and elt.uri == NS_FIS:
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	703 # we have a directory
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	704
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	705 file_data = {"name": elt["name"], "type": C.FILE_TYPE_DIRECTORY}
3314 5887fb414758 component file sharing: add/parse affiliation when possible Goffi <goffi@goffi.org> parents: 3136 diff changeset	706 self.host.trigger.point(
5887fb414758 component file sharing: add/parse affiliation when possible Goffi <goffi@goffi.org> parents: 3136 diff changeset	707 "XEP-0329_parseResult_directory",
5887fb414758 component file sharing: add/parse affiliation when possible Goffi <goffi@goffi.org> parents: 3136 diff changeset	708 client,
5887fb414758 component file sharing: add/parse affiliation when possible Goffi <goffi@goffi.org> parents: 3136 diff changeset	709 elt,
5887fb414758 component file sharing: add/parse affiliation when possible Goffi <goffi@goffi.org> parents: 3136 diff changeset	710 file_data,
5887fb414758 component file sharing: add/parse affiliation when possible Goffi <goffi@goffi.org> parents: 3136 diff changeset	711 )
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	712 else:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	713 log.warning(
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3032 diff changeset	714 _(f"unexpected element, ignoring: {elt.toXml()}")
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	715 )
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	716 continue
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	717 files.append(file_data)
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	718 return files
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	719
3320 bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	720 # affiliations #
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	721
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	722 async def _parseElement(self, client, iq_elt, element, namespace):
3366 e09cb08166a3 plugin XEP-0329, core(xmpp): moved `_compParseJids` to `SatXMPPComponent`: Goffi <goffi@goffi.org> parents: 3365 diff changeset	723 peer_jid, owner = client.getOwnerAndPeer(iq_elt)
3320 bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	724 elt = next(iq_elt.elements(namespace, element))
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	725 path = Path("/", elt['path'])
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	726 if len(path.parts) < 2:
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	727 raise RootPathException
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	728 namespace = elt.getAttribute('namespace')
3360 57afccb91961 plugin XEP-0329: typo Goffi <goffi@goffi.org> parents: 3359 diff changeset	729 files_data = await self.host.memory.getFiles(
3320 bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	730 client,
3365 626046010a2d plugin XEP-0329: fix filtering by owner on components Goffi <goffi@goffi.org> parents: 3362 diff changeset	731 peer_jid=peer_jid,
3320 bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	732 path=str(path.parent),
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	733 name=path.name,
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	734 namespace=namespace,
3365 626046010a2d plugin XEP-0329: fix filtering by owner on components Goffi <goffi@goffi.org> parents: 3362 diff changeset	735 owner=owner,
3320 bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	736 )
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	737 if len(files_data) != 1:
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	738 client.sendError(iq_elt, 'item-not-found')
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	739 raise exceptions.CancelError
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	740 file_data = files_data[0]
3365 626046010a2d plugin XEP-0329: fix filtering by owner on components Goffi <goffi@goffi.org> parents: 3362 diff changeset	741 return peer_jid, elt, path, namespace, file_data
3320 bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	742
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	743 def _affiliationsGet(self, service_jid_s, namespace, path, profile):
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	744 client = self.host.getClient(profile)
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	745 service = jid.JID(service_jid_s)
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	746 d = defer.ensureDeferred(self.affiliationsGet(
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	747 client, service, namespace or None, path))
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	748 d.addCallback(
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	749 lambda affiliations: {
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	750 str(entity): affiliation for entity, affiliation in affiliations.items()
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	751 }
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	752 )
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	753 return d
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	754
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	755 async def affiliationsGet(
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	756 self,
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	757 client: SatXMPPEntity,
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	758 service: jid.JID,
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	759 namespace: Optional[str],
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	760 path: str
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	761 ) -> Dict[jid.JID, str]:
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	762 if not path:
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	763 raise ValueError(f"invalid path: {path!r}")
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	764 iq_elt = client.IQ("get")
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	765 iq_elt['to'] = service.full()
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	766 affiliations_elt = iq_elt.addElement((NS_FIS_AFFILIATION, "affiliations"))
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	767 if namespace:
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	768 affiliations_elt["namespace"] = namespace
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	769 affiliations_elt["path"] = path
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	770 iq_result_elt = await iq_elt.send()
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	771 try:
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	772 affiliations_elt = next(iq_result_elt.elements(NS_FIS_AFFILIATION, "affiliations"))
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	773 except StopIteration:
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	774 raise exceptions.DataError(f"Invalid result to affiliations request: {iq_result_elt.toXml()}")
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	775
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	776 affiliations = {}
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	777 for affiliation_elt in affiliations_elt.elements(NS_FIS_AFFILIATION, 'affiliation'):
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	778 try:
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	779 affiliations[jid.JID(affiliation_elt['jid'])] = affiliation_elt['affiliation']
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	780 except (KeyError, RuntimeError):
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	781 raise exceptions.DataError(
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	782 f"invalid affiliation element: {affiliation_elt.toXml()}")
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	783
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	784 return affiliations
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	785
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	786 def _affiliationsSet(self, service_jid_s, namespace, path, affiliations, profile):
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	787 client = self.host.getClient(profile)
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	788 service = jid.JID(service_jid_s)
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	789 affiliations = {jid.JID(e): a for e, a in affiliations.items()}
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	790 return defer.ensureDeferred(self.affiliationsSet(
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	791 client, service, namespace or None, path, affiliations))
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	792
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	793 async def affiliationsSet(
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	794 self,
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	795 client: SatXMPPEntity,
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	796 service: jid.JID,
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	797 namespace: Optional[str],
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	798 path: str,
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	799 affiliations: Dict[jid.JID, str],
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	800 ):
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	801 if not path:
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	802 raise ValueError(f"invalid path: {path!r}")
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	803 iq_elt = client.IQ("set")
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	804 iq_elt['to'] = service.full()
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	805 affiliations_elt = iq_elt.addElement((NS_FIS_AFFILIATION, "affiliations"))
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	806 if namespace:
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	807 affiliations_elt["namespace"] = namespace
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	808 affiliations_elt["path"] = path
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	809 for entity_jid, affiliation in affiliations.items():
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	810 affiliation_elt = affiliations_elt.addElement('affiliation')
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	811 affiliation_elt['jid'] = entity_jid.full()
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	812 affiliation_elt['affiliation'] = affiliation
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	813 await iq_elt.send()
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	814
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	815 def _onComponentAffiliationsGet(self, iq_elt, client):
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	816 iq_elt.handled = True
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	817 defer.ensureDeferred(self.onComponentAffiliationsGet(client, iq_elt))
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	818
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	819 async def onComponentAffiliationsGet(self, client, iq_elt):
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	820 try:
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	821 (
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	822 from_jid, affiliations_elt, path, namespace, file_data
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	823 ) = await self._parseElement(client, iq_elt, "affiliations", NS_FIS_AFFILIATION)
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	824 except exceptions.CancelError:
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	825 return
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	826 except RootPathException:
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	827 # if root path is requested, we only get owner affiliation
3366 e09cb08166a3 plugin XEP-0329, core(xmpp): moved `_compParseJids` to `SatXMPPComponent`: Goffi <goffi@goffi.org> parents: 3365 diff changeset	828 peer_jid, owner = client.getOwnerAndPeer(iq_elt)
3320 bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	829 is_owner = peer_jid.userhostJID() == owner
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	830 affiliations = {owner: 'owner'}
3355 33d9b38b5890 plugin XEP-0329: better error handling in `onComponentAffiliationsGet` Goffi <goffi@goffi.org> parents: 3333 diff changeset	831 except exceptions.NotFound:
33d9b38b5890 plugin XEP-0329: better error handling in `onComponentAffiliationsGet` Goffi <goffi@goffi.org> parents: 3333 diff changeset	832 client.sendError(iq_elt, "item-not-found")
33d9b38b5890 plugin XEP-0329: better error handling in `onComponentAffiliationsGet` Goffi <goffi@goffi.org> parents: 3333 diff changeset	833 return
33d9b38b5890 plugin XEP-0329: better error handling in `onComponentAffiliationsGet` Goffi <goffi@goffi.org> parents: 3333 diff changeset	834 except Exception as e:
33d9b38b5890 plugin XEP-0329: better error handling in `onComponentAffiliationsGet` Goffi <goffi@goffi.org> parents: 3333 diff changeset	835 client.sendError(iq_elt, "internal-server-error", str(e))
33d9b38b5890 plugin XEP-0329: better error handling in `onComponentAffiliationsGet` Goffi <goffi@goffi.org> parents: 3333 diff changeset	836 return
3320 bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	837 else:
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	838 from_jid_bare = from_jid.userhostJID()
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	839 is_owner = from_jid_bare == file_data.get('owner')
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	840 affiliations = self.host.memory.getFileAffiliations(file_data)
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	841 iq_result_elt = xmlstream.toResponse(iq_elt, "result")
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	842 affiliations_elt = iq_result_elt.addElement((NS_FIS_AFFILIATION, 'affiliations'))
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	843 for entity_jid, affiliation in affiliations.items():
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	844 if not is_owner and entity_jid.userhostJID() != from_jid_bare:
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	845 # only onwer can get all affiliations
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	846 continue
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	847 affiliation_elt = affiliations_elt.addElement('affiliation')
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	848 affiliation_elt['jid'] = entity_jid.userhost()
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	849 affiliation_elt['affiliation'] = affiliation
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	850 client.send(iq_result_elt)
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	851
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	852 def _onComponentAffiliationsSet(self, iq_elt, client):
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	853 iq_elt.handled = True
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	854 defer.ensureDeferred(self.onComponentAffiliationsSet(client, iq_elt))
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	855
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	856 async def onComponentAffiliationsSet(self, client, iq_elt):
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	857 try:
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	858 (
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	859 from_jid, affiliations_elt, path, namespace, file_data
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	860 ) = await self._parseElement(client, iq_elt, "affiliations", NS_FIS_AFFILIATION)
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	861 except exceptions.CancelError:
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	862 return
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	863 except RootPathException:
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	864 client.sendError(iq_elt, 'bad-request', "Root path can't be used")
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	865 return
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	866
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	867 if from_jid.userhostJID() != file_data['owner']:
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	868 log.warning(
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	869 f"{from_jid} tried to modify {path} affiliations while the owner is "
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	870 f"{file_data['owner']}"
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	871 )
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	872 client.sendError(iq_elt, 'forbidden')
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	873 return
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	874
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	875 try:
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	876 affiliations = {
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	877 jid.JID(e['jid']): e['affiliation']
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	878 for e in affiliations_elt.elements(NS_FIS_AFFILIATION, 'affiliation')
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	879 }
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	880 except (KeyError, RuntimeError):
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	881 log.warning(
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	882 f"invalid affiliation element: {affiliations_elt.toXml()}"
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	883 )
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	884 client.sendError(iq_elt, 'bad-request', "invalid affiliation element")
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	885 return
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	886 except Exception as e:
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	887 log.error(
3359 000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	888 f"unexepected exception while setting affiliation element: {e}\n"
3320 bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	889 f"{affiliations_elt.toXml()}"
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	890 )
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	891 client.sendError(iq_elt, 'internal-server-error', f"{e}")
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	892 return
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	893
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	894 await self.host.memory.setFileAffiliations(client, file_data, affiliations)
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	895
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	896 iq_result_elt = xmlstream.toResponse(iq_elt, "result")
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	897 client.send(iq_result_elt)
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	898
3321 8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	899 # configuration
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	900
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	901 def _configurationGet(self, service_jid_s, namespace, path, profile):
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	902 client = self.host.getClient(profile)
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	903 service = jid.JID(service_jid_s)
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	904 d = defer.ensureDeferred(self.configurationGet(
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	905 client, service, namespace or None, path))
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	906 d.addCallback(
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	907 lambda configuration: {
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	908 str(entity): affiliation for entity, affiliation in configuration.items()
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	909 }
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	910 )
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	911 return d
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	912
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	913 async def configurationGet(
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	914 self,
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	915 client: SatXMPPEntity,
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	916 service: jid.JID,
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	917 namespace: Optional[str],
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	918 path: str
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	919 ) -> Dict[str, str]:
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	920 if not path:
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	921 raise ValueError(f"invalid path: {path!r}")
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	922 iq_elt = client.IQ("get")
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	923 iq_elt['to'] = service.full()
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	924 configuration_elt = iq_elt.addElement((NS_FIS_CONFIGURATION, "configuration"))
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	925 if namespace:
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	926 configuration_elt["namespace"] = namespace
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	927 configuration_elt["path"] = path
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	928 iq_result_elt = await iq_elt.send()
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	929 try:
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	930 configuration_elt = next(iq_result_elt.elements(NS_FIS_CONFIGURATION, "configuration"))
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	931 except StopIteration:
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	932 raise exceptions.DataError(f"Invalid result to configuration request: {iq_result_elt.toXml()}")
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	933
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	934 form = data_form.findForm(configuration_elt, NS_FIS_CONFIGURATION)
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	935 configuration = {f.var: f.value for f in form.fields.values()}
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	936
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	937 return configuration
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	938
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	939 def _configurationSet(self, service_jid_s, namespace, path, configuration, profile):
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	940 client = self.host.getClient(profile)
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	941 service = jid.JID(service_jid_s)
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	942 return defer.ensureDeferred(self.configurationSet(
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	943 client, service, namespace or None, path, configuration))
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	944
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	945 async def configurationSet(
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	946 self,
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	947 client: SatXMPPEntity,
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	948 service: jid.JID,
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	949 namespace: Optional[str],
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	950 path: str,
3359 000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	951 configuration: Dict[str, str],
3321 8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	952 ):
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	953 if not path:
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	954 raise ValueError(f"invalid path: {path!r}")
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	955 iq_elt = client.IQ("set")
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	956 iq_elt['to'] = service.full()
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	957 configuration_elt = iq_elt.addElement((NS_FIS_CONFIGURATION, "configuration"))
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	958 if namespace:
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	959 configuration_elt["namespace"] = namespace
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	960 configuration_elt["path"] = path
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	961 form = data_form.Form(formType="submit", formNamespace=NS_FIS_CONFIGURATION)
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	962 form.makeFields(configuration)
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	963 configuration_elt.addChild(form.toElement())
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	964 await iq_elt.send()
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	965
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	966 def _onComponentConfigurationGet(self, iq_elt, client):
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	967 iq_elt.handled = True
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	968 defer.ensureDeferred(self.onComponentConfigurationGet(client, iq_elt))
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	969
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	970 async def onComponentConfigurationGet(self, client, iq_elt):
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	971 try:
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	972 (
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	973 from_jid, configuration_elt, path, namespace, file_data
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	974 ) = await self._parseElement(client, iq_elt, "configuration", NS_FIS_CONFIGURATION)
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	975 except exceptions.CancelError:
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	976 return
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	977 except RootPathException:
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	978 client.sendError(iq_elt, 'bad-request', "Root path can't be used")
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	979 return
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	980 try:
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	981 access_type = file_data['access'][C.ACCESS_PERM_READ]['type']
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	982 except KeyError:
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	983 access_model = 'whitelist'
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	984 else:
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	985 access_model = 'open' if access_type == C.ACCESS_TYPE_PUBLIC else 'whitelist'
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	986
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	987 iq_result_elt = xmlstream.toResponse(iq_elt, "result")
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	988 configuration_elt = iq_result_elt.addElement((NS_FIS_CONFIGURATION, 'configuration'))
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	989 form = data_form.Form(formType="form", formNamespace=NS_FIS_CONFIGURATION)
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	990 form.makeFields({'access_model': access_model})
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	991 configuration_elt.addChild(form.toElement())
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	992 client.send(iq_result_elt)
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	993
3359 000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	994 async def _setConfiguration(self, client, configuration_elt, file_data):
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	995 form = data_form.findForm(configuration_elt, NS_FIS_CONFIGURATION)
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	996 for name, value in form.items():
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	997 if name == 'access_model':
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	998 await self.host.memory.setFileAccessModel(client, file_data, value)
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	999 else:
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1000 # TODO: send a IQ error?
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1001 log.warning(
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1002 f"Trying to set a not implemented configuration option: {name}")
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1003
3321 8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	1004 def _onComponentConfigurationSet(self, iq_elt, client):
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	1005 iq_elt.handled = True
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	1006 defer.ensureDeferred(self.onComponentConfigurationSet(client, iq_elt))
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	1007
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	1008 async def onComponentConfigurationSet(self, client, iq_elt):
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	1009 try:
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	1010 (
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	1011 from_jid, configuration_elt, path, namespace, file_data
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	1012 ) = await self._parseElement(client, iq_elt, "configuration", NS_FIS_CONFIGURATION)
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	1013 except exceptions.CancelError:
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	1014 return
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	1015 except RootPathException:
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	1016 client.sendError(iq_elt, 'bad-request', "Root path can't be used")
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	1017 return
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	1018
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	1019 from_jid_bare = from_jid.userhostJID()
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	1020 is_owner = from_jid_bare == file_data.get('owner')
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	1021 if not is_owner:
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	1022 log.warning(
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	1023 f"{from_jid} tried to modify {path} configuration while the owner is "
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	1024 f"{file_data['owner']}"
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	1025 )
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	1026 client.sendError(iq_elt, 'forbidden')
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	1027 return
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	1028
3359 000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1029 await self._setConfiguration(client, configuration_elt, file_data)
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1030
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1031 iq_result_elt = xmlstream.toResponse(iq_elt, "result")
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1032 client.send(iq_result_elt)
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1033
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1034 # directory creation
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1035
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1036 def _createDir(self, service_jid_s, namespace, path, configuration, profile):
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1037 client = self.host.getClient(profile)
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1038 service = jid.JID(service_jid_s)
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1039 return defer.ensureDeferred(self.createDir(
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1040 client, service, namespace or None, path, configuration or None))
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1041
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1042 async def createDir(
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1043 self,
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1044 client: SatXMPPEntity,
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1045 service: jid.JID,
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1046 namespace: Optional[str],
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1047 path: str,
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1048 configuration: Optional[Dict[str, str]],
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1049 ):
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1050 if not path:
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1051 raise ValueError(f"invalid path: {path!r}")
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1052 iq_elt = client.IQ("set")
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1053 iq_elt['to'] = service.full()
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1054 create_dir_elt = iq_elt.addElement((NS_FIS_CREATE, "dir"))
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1055 if namespace:
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1056 create_dir_elt["namespace"] = namespace
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1057 create_dir_elt["path"] = path
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1058 if configuration:
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1059 configuration_elt = create_dir_elt.addElement((NS_FIS_CONFIGURATION, "configuration"))
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1060 form = data_form.Form(formType="submit", formNamespace=NS_FIS_CONFIGURATION)
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1061 form.makeFields(configuration)
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1062 configuration_elt.addChild(form.toElement())
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1063 await iq_elt.send()
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1064
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1065 def _onComponentCreateDir(self, iq_elt, client):
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1066 iq_elt.handled = True
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1067 defer.ensureDeferred(self.onComponentCreateDir(client, iq_elt))
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1068
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1069 async def onComponentCreateDir(self, client, iq_elt):
3366 e09cb08166a3 plugin XEP-0329, core(xmpp): moved `_compParseJids` to `SatXMPPComponent`: Goffi <goffi@goffi.org> parents: 3365 diff changeset	1070 peer_jid, owner = client.getOwnerAndPeer(iq_elt)
3359 000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1071 if peer_jid.host not in client._file_sharing_allowed_hosts:
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1072 client.sendError(iq_elt, 'forbidden')
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1073 return
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1074 create_dir_elt = next(iq_elt.elements(NS_FIS_CREATE, "dir"))
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1075 namespace = create_dir_elt.getAttribute('namespace')
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1076 path = Path("/", create_dir_elt['path'])
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1077 if len(path.parts) < 2:
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1078 client.sendError(iq_elt, 'bad-request', "Root path can't be used")
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1079 return
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1080 # for root directories, we check permission here
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1081 if len(path.parts) == 2 and owner != peer_jid.userhostJID():
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1082 log.warning(
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1083 f"{peer_jid} is trying to create a dir at {owner}'s repository:\n"
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1084 f"path: {path}\nnamespace: {namespace!r}"
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1085 )
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1086 client.sendError(iq_elt, 'forbidden', "You can't create a directory there")
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1087 return
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1088 # when going further into the path, the permissions will be checked by getFiles
3362 02583a401e51 plugin XEP-0329: fix conflict check + directory creation in `onComponentCreateDir` Goffi <goffi@goffi.org> parents: 3360 diff changeset	1089 if len(path.parts) > 2:
02583a401e51 plugin XEP-0329: fix conflict check + directory creation in `onComponentCreateDir` Goffi <goffi@goffi.org> parents: 3360 diff changeset	1090 files_data = await self.host.memory.getFiles(
3359 000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1091 client,
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1092 peer_jid=peer_jid,
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1093 path=path.parent,
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1094 namespace=namespace,
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1095 owner=owner,
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1096 )
3362 02583a401e51 plugin XEP-0329: fix conflict check + directory creation in `onComponentCreateDir` Goffi <goffi@goffi.org> parents: 3360 diff changeset	1097 if path.name in [d['name'] for d in files_data]:
02583a401e51 plugin XEP-0329: fix conflict check + directory creation in `onComponentCreateDir` Goffi <goffi@goffi.org> parents: 3360 diff changeset	1098 log.warning(
02583a401e51 plugin XEP-0329: fix conflict check + directory creation in `onComponentCreateDir` Goffi <goffi@goffi.org> parents: 3360 diff changeset	1099 f"Conflict when trying to create a directory (from: {peer_jid} "
02583a401e51 plugin XEP-0329: fix conflict check + directory creation in `onComponentCreateDir` Goffi <goffi@goffi.org> parents: 3360 diff changeset	1100 f"namespace: {namespace!r} path: {path!r})"
02583a401e51 plugin XEP-0329: fix conflict check + directory creation in `onComponentCreateDir` Goffi <goffi@goffi.org> parents: 3360 diff changeset	1101 )
02583a401e51 plugin XEP-0329: fix conflict check + directory creation in `onComponentCreateDir` Goffi <goffi@goffi.org> parents: 3360 diff changeset	1102 client.sendError(
02583a401e51 plugin XEP-0329: fix conflict check + directory creation in `onComponentCreateDir` Goffi <goffi@goffi.org> parents: 3360 diff changeset	1103 iq_elt, 'conflict', "there is already a file or dir at this path")
02583a401e51 plugin XEP-0329: fix conflict check + directory creation in `onComponentCreateDir` Goffi <goffi@goffi.org> parents: 3360 diff changeset	1104 return
3359 000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1105
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1106 try:
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1107 configuration_elt = next(
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1108 create_dir_elt.elements(NS_FIS_CONFIGURATION, 'configuration'))
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1109 except StopIteration:
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1110 configuration_elt = None
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1111
3362 02583a401e51 plugin XEP-0329: fix conflict check + directory creation in `onComponentCreateDir` Goffi <goffi@goffi.org> parents: 3360 diff changeset	1112 await self.host.memory.setFile(
02583a401e51 plugin XEP-0329: fix conflict check + directory creation in `onComponentCreateDir` Goffi <goffi@goffi.org> parents: 3360 diff changeset	1113 client,
02583a401e51 plugin XEP-0329: fix conflict check + directory creation in `onComponentCreateDir` Goffi <goffi@goffi.org> parents: 3360 diff changeset	1114 path.name,
02583a401e51 plugin XEP-0329: fix conflict check + directory creation in `onComponentCreateDir` Goffi <goffi@goffi.org> parents: 3360 diff changeset	1115 path=path.parent,
02583a401e51 plugin XEP-0329: fix conflict check + directory creation in `onComponentCreateDir` Goffi <goffi@goffi.org> parents: 3360 diff changeset	1116 type_=C.FILE_TYPE_DIRECTORY,
02583a401e51 plugin XEP-0329: fix conflict check + directory creation in `onComponentCreateDir` Goffi <goffi@goffi.org> parents: 3360 diff changeset	1117 namespace=namespace,
02583a401e51 plugin XEP-0329: fix conflict check + directory creation in `onComponentCreateDir` Goffi <goffi@goffi.org> parents: 3360 diff changeset	1118 owner=owner,
02583a401e51 plugin XEP-0329: fix conflict check + directory creation in `onComponentCreateDir` Goffi <goffi@goffi.org> parents: 3360 diff changeset	1119 peer_jid=peer_jid
02583a401e51 plugin XEP-0329: fix conflict check + directory creation in `onComponentCreateDir` Goffi <goffi@goffi.org> parents: 3360 diff changeset	1120 )
3359 000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1121
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1122 if configuration_elt is not None:
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1123 file_data = (await self.host.memory.getFiles(
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1124 client,
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1125 peer_jid=peer_jid,
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1126 path=path.parent,
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1127 name=path.name,
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1128 namespace=namespace,
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1129 owner=owner,
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1130 ))[0]
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1131
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1132 await self._setConfiguration(client, configuration_elt, file_data)
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1133
3321 8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	1134 iq_result_elt = xmlstream.toResponse(iq_elt, "result")
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	1135 client.send(iq_result_elt)
8bbd2ed924e8 plugin XEP-0329: added way to change `access_model` using PubSub-like configuration: Goffi <goffi@goffi.org> parents: 3320 diff changeset	1136
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	1137 # file methods #
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	1138
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	1139 def _serializeData(self, files_data):
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	1140 for file_data in files_data:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2937 diff changeset	1141 for key, value in file_data.items():
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	1142 file_data[key] = (
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2937 diff changeset	1143 json.dumps(value) if key in ("extra",) else str(value)
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	1144 )
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	1145 return files_data
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	1146
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	1147 def _listFiles(self, target_jid, path, extra, profile):
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	1148 client = self.host.getClient(profile)
3040 fee60f17ebac jp: jp asyncio port: Goffi <goffi@goffi.org> parents: 3032 diff changeset	1149 target_jid = client.jid if not target_jid else jid.JID(target_jid)
3333 ac9342f359e9 plugin XEP-0329: download thumbnails: Goffi <goffi@goffi.org> parents: 3321 diff changeset	1150 d = defer.ensureDeferred(self.listFiles(client, target_jid, path or None))
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	1151 d.addCallback(self._serializeData)
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	1152 return d
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	1153
3333 ac9342f359e9 plugin XEP-0329: download thumbnails: Goffi <goffi@goffi.org> parents: 3321 diff changeset	1154 async def listFiles(self, client, peer_jid, path=None, extra=None):
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	1155 """List file shared by an entity
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	1156
3333 ac9342f359e9 plugin XEP-0329: download thumbnails: Goffi <goffi@goffi.org> parents: 3321 diff changeset	1157 @param peer_jid(jid.JID): jid of the sharing entity
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	1158 @param path(unicode, None): path to the directory containing shared files
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	1159 None to get root directories
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	1160 @param extra(dict, None): extra data
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	1161 @return list(dict): shared files
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	1162 """
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	1163 iq_elt = client.IQ("get")
3333 ac9342f359e9 plugin XEP-0329: download thumbnails: Goffi <goffi@goffi.org> parents: 3321 diff changeset	1164 iq_elt["to"] = peer_jid.full()
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	1165 query_elt = iq_elt.addElement((NS_FIS, "query"))
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	1166 if path:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	1167 query_elt["node"] = path
3333 ac9342f359e9 plugin XEP-0329: download thumbnails: Goffi <goffi@goffi.org> parents: 3321 diff changeset	1168 iq_result_elt = await iq_elt.send()
ac9342f359e9 plugin XEP-0329: download thumbnails: Goffi <goffi@goffi.org> parents: 3321 diff changeset	1169 return await self._parseResult(client, peer_jid, iq_result_elt)
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	1170
2589 282d1314d574 plugin XEP-0329: new methods/signals to handle shares: Goffi <goffi@goffi.org> parents: 2562 diff changeset	1171 def _localSharesGet(self, profile):
282d1314d574 plugin XEP-0329: new methods/signals to handle shares: Goffi <goffi@goffi.org> parents: 2562 diff changeset	1172 client = self.host.getClient(profile)
282d1314d574 plugin XEP-0329: new methods/signals to handle shares: Goffi <goffi@goffi.org> parents: 2562 diff changeset	1173 return self.localSharesGet(client)
282d1314d574 plugin XEP-0329: new methods/signals to handle shares: Goffi <goffi@goffi.org> parents: 2562 diff changeset	1174
282d1314d574 plugin XEP-0329: new methods/signals to handle shares: Goffi <goffi@goffi.org> parents: 2562 diff changeset	1175 def localSharesGet(self, client):
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2937 diff changeset	1176 return list(client._XEP_0329_root_node.getSharedPaths().keys())
2589 282d1314d574 plugin XEP-0329: new methods/signals to handle shares: Goffi <goffi@goffi.org> parents: 2562 diff changeset	1177
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	1178 def _sharePath(self, name, path, access, profile):
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	1179 client = self.host.getClient(profile)
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	1180 access = json.loads(access)
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	1181 return self.sharePath(client, name or None, path, access)
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	1182
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	1183 def sharePath(self, client, name, path, access):
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	1184 if client.is_component:
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	1185 raise exceptions.ClientTypeError
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	1186 if not os.path.exists(path):
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2937 diff changeset	1187 raise ValueError(_("This path doesn't exist!"))
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2937 diff changeset	1188 if not path or not path.strip(" /"):
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2937 diff changeset	1189 raise ValueError(_("A path need to be specified"))
2589 282d1314d574 plugin XEP-0329: new methods/signals to handle shares: Goffi <goffi@goffi.org> parents: 2562 diff changeset	1190 if not isinstance(access, dict):
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2937 diff changeset	1191 raise ValueError(_("access must be a dict"))
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	1192
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	1193 node = client._XEP_0329_root_node
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	1194 node_type = TYPE_PATH
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	1195 if os.path.isfile(path):
2909 90146552cde5 core (memory), plugin XEP-0329, plugin invitation: minor style improvments Goffi <goffi@goffi.org> parents: 2771 diff changeset	1196 # we have a single file, the workflow is diferrent as we store all single
90146552cde5 core (memory), plugin XEP-0329, plugin invitation: minor style improvments Goffi <goffi@goffi.org> parents: 2771 diff changeset	1197 # files in the same dir
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	1198 node = node.getOrCreate(SINGLE_FILES_DIR)
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	1199
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	1200 if not name:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2937 diff changeset	1201 name = os.path.basename(path.rstrip(" /"))
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	1202 if not name:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2937 diff changeset	1203 raise exceptions.InternalError(_("Can't find a proper name"))
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	1204
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	1205 if name in node or name == SINGLE_FILES_DIR:
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	1206 idx = 1
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2937 diff changeset	1207 new_name = name + "_" + str(idx)
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	1208 while new_name in node:
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	1209 idx += 1
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2937 diff changeset	1210 new_name = name + "_" + str(idx)
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	1211 name = new_name
2909 90146552cde5 core (memory), plugin XEP-0329, plugin invitation: minor style improvments Goffi <goffi@goffi.org> parents: 2771 diff changeset	1212 log.info(_(
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2937 diff changeset	1213 "A directory with this name is already shared, renamed to {new_name} "
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2937 diff changeset	1214 "[{profile}]".format( new_name=new_name, profile=client.profile)))
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	1215
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	1216 ShareNode(name=name, parent=node, type_=node_type, access=access, path=path)
2589 282d1314d574 plugin XEP-0329: new methods/signals to handle shares: Goffi <goffi@goffi.org> parents: 2562 diff changeset	1217 self.host.bridge.FISSharedPathNew(path, name, client.profile)
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	1218 return name
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	1219
2589 282d1314d574 plugin XEP-0329: new methods/signals to handle shares: Goffi <goffi@goffi.org> parents: 2562 diff changeset	1220 def _unsharePath(self, path, profile):
282d1314d574 plugin XEP-0329: new methods/signals to handle shares: Goffi <goffi@goffi.org> parents: 2562 diff changeset	1221 client = self.host.getClient(profile)
282d1314d574 plugin XEP-0329: new methods/signals to handle shares: Goffi <goffi@goffi.org> parents: 2562 diff changeset	1222 return self.unsharePath(client, path)
282d1314d574 plugin XEP-0329: new methods/signals to handle shares: Goffi <goffi@goffi.org> parents: 2562 diff changeset	1223
282d1314d574 plugin XEP-0329: new methods/signals to handle shares: Goffi <goffi@goffi.org> parents: 2562 diff changeset	1224 def unsharePath(self, client, path):
282d1314d574 plugin XEP-0329: new methods/signals to handle shares: Goffi <goffi@goffi.org> parents: 2562 diff changeset	1225 nodes = client._XEP_0329_root_node.findByLocalPath(path)
282d1314d574 plugin XEP-0329: new methods/signals to handle shares: Goffi <goffi@goffi.org> parents: 2562 diff changeset	1226 for node in nodes:
282d1314d574 plugin XEP-0329: new methods/signals to handle shares: Goffi <goffi@goffi.org> parents: 2562 diff changeset	1227 node.removeFromParent()
282d1314d574 plugin XEP-0329: new methods/signals to handle shares: Goffi <goffi@goffi.org> parents: 2562 diff changeset	1228 self.host.bridge.FISSharedPathRemoved(path, client.profile)
282d1314d574 plugin XEP-0329: new methods/signals to handle shares: Goffi <goffi@goffi.org> parents: 2562 diff changeset	1229
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	1230
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2937 diff changeset	1231 @implementer(iwokkel.IDisco)
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	1232 class XEP_0329_handler(xmlstream.XMPPHandler):
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	1233
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	1234 def __init__(self, plugin_parent):
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	1235 self.plugin_parent = plugin_parent
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	1236 self.host = plugin_parent.host
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	1237
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	1238 def connectionInitialized(self):
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	1239 if self.parent.is_component:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	1240 self.xmlstream.addObserver(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	1241 IQ_FIS_REQUEST, self.plugin_parent.onComponentRequest, client=self.parent
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	1242 )
3320 bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	1243 self.xmlstream.addObserver(
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	1244 IQ_FIS_AFFILIATION_GET,
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	1245 self.plugin_parent._onComponentAffiliationsGet,
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	1246 client=self.parent
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	1247 )
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	1248 self.xmlstream.addObserver(
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	1249 IQ_FIS_AFFILIATION_SET,
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	1250 self.plugin_parent._onComponentAffiliationsSet,
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	1251 client=self.parent
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	1252 )
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	1253 self.xmlstream.addObserver(
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	1254 IQ_FIS_CONFIGURATION_GET,
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	1255 self.plugin_parent._onComponentConfigurationGet,
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	1256 client=self.parent
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	1257 )
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	1258 self.xmlstream.addObserver(
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	1259 IQ_FIS_CONFIGURATION_SET,
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	1260 self.plugin_parent._onComponentConfigurationSet,
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	1261 client=self.parent
bb92085720c8 plugin XEP-0329: implemented ways to get/set affiliations: Goffi <goffi@goffi.org> parents: 3314 diff changeset	1262 )
3359 000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1263 self.xmlstream.addObserver(
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1264 IQ_FIS_CREATE_DIR,
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1265 self.plugin_parent._onComponentCreateDir,
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1266 client=self.parent
000b6722bd35 plugin XEP-0329: added `FISCreateDir` method: Goffi <goffi@goffi.org> parents: 3355 diff changeset	1267 )
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	1268 else:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	1269 self.xmlstream.addObserver(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	1270 IQ_FIS_REQUEST, self.plugin_parent.onRequest, client=self.parent
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	1271 )
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	1272
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	1273 def getDiscoInfo(self, requestor, target, nodeIdentifier=""):
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	1274 return [disco.DiscoFeature(NS_FIS)]
c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	1275
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2589 diff changeset	1276 def getDiscoItems(self, requestor, target, nodeIdentifier=""):
2503 c0bec8bac2b5 XEP-0329: implementation of File Information Sharing: Goffi <goffi@goffi.org> parents: diff changeset	1277 return []

Mercurial > libervia-backend

annotate sat/plugins/plugin_xep_0329.py @ 3413:f4d417593e77