diff libervia/backend/plugins/plugin_exp_list_of_interest.py @ 4071:4b842c1fb686

refactoring: renamed `sat` package to `libervia.backend`
author Goffi <goffi@goffi.org>
date Fri, 02 Jun 2023 11:49:51 +0200
parents sat/plugins/plugin_exp_list_of_interest.py@524856bd7b19
children 0d7bb4df2343
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/libervia/backend/plugins/plugin_exp_list_of_interest.py	Fri Jun 02 11:49:51 2023 +0200
@@ -0,0 +1,321 @@
+#!/usr/bin/env python3
+
+
+# SAT plugin to detect language (experimental)
+# Copyright (C) 2009-2021 Jérôme Poisson (goffi@goffi.org)
+
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Affero General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU Affero General Public License for more details.
+
+# You should have received a copy of the GNU Affero General Public License
+# along with this program.  If not, see <http://www.gnu.org/licenses/>.
+
+from libervia.backend.core.i18n import _
+from libervia.backend.core.constants import Const as C
+from libervia.backend.core.xmpp import SatXMPPEntity
+from libervia.backend.core import exceptions
+from libervia.backend.core.log import getLogger
+from libervia.backend.tools.common import data_format
+from libervia.backend.tools.common import uri
+from wokkel import disco, iwokkel, pubsub
+from zope.interface import implementer
+from twisted.internet import defer
+from twisted.words.protocols.jabber import error as jabber_error, jid
+from twisted.words.protocols.jabber.xmlstream import XMPPHandler
+from twisted.words.xish import domish
+
+log = getLogger(__name__)
+
+
+PLUGIN_INFO = {
+    C.PI_NAME: "List of Interest",
+    C.PI_IMPORT_NAME: "LIST_INTEREST",
+    C.PI_TYPE: "EXP",
+    C.PI_PROTOCOLS: [],
+    C.PI_DEPENDENCIES: ["XEP-0060", "XEP-0329", "XEP-0106"],
+    C.PI_RECOMMENDATIONS: [],
+    C.PI_MAIN: "ListInterest",
+    C.PI_HANDLER: "yes",
+    C.PI_DESCRIPTION: _("Experimental handling of interesting XMPP locations"),
+}
+
+NS_LIST_INTEREST = "https://salut-a-toi/protocol/list-interest:0"
+
+
+class ListInterest(object):
+    namespace = NS_LIST_INTEREST
+
+    def __init__(self, host):
+        log.info(_("List of Interest plugin initialization"))
+        self.host = host
+        self._p = self.host.plugins["XEP-0060"]
+        host.bridge.add_method(
+            "interests_list",
+            ".plugin",
+            in_sign="ssss",
+            out_sign="aa{ss}",
+            method=self._list_interests,
+            async_=True,
+        )
+        host.bridge.add_method(
+            "interests_file_sharing_register",
+            ".plugin",
+            in_sign="sssssss",
+            out_sign="",
+            method=self._register_file_sharing,
+            async_=True,
+        )
+        host.bridge.add_method(
+            "interest_retract",
+            ".plugin",
+            in_sign="sss",
+            out_sign="",
+            method=self._interest_retract,
+            async_=True,
+        )
+
+    def get_handler(self, client):
+        return ListInterestHandler(self)
+
+    @defer.inlineCallbacks
+    def createNode(self, client):
+        try:
+            # TODO: check auto-create, no need to create node first if available
+            options = {self._p.OPT_ACCESS_MODEL: self._p.ACCESS_WHITELIST}
+            yield self._p.createNode(
+                client,
+                client.jid.userhostJID(),
+                nodeIdentifier=NS_LIST_INTEREST,
+                options=options,
+            )
+        except jabber_error.StanzaError as e:
+            if e.condition == "conflict":
+                log.debug(_("requested node already exists"))
+
+    async def register_pubsub(self, client, namespace, service, node, item_id=None,
+                       creator=False, name=None, element=None, extra=None):
+        """Register an interesting element in personal list
+
+        @param namespace(unicode): namespace of the interest
+            this is used as a cache, to avoid the need to retrieve the item only to get
+            its namespace
+        @param service(jid.JID): target pubsub service
+        @param node(unicode): target pubsub node
+        @param item_id(unicode, None): target pubsub id
+        @param creator(bool): True if client's profile is the creator of the node
+            This is used a cache, to avoid the need to retrieve affiliations
+        @param name(unicode, None): name of the interest
+        @param element(domish.Element, None): element to attach
+            may be used to cache some extra data
+        @param extra(dict, None): extra data, key can be:
+            - thumb_url: http(s) URL of a thumbnail
+        """
+        if extra is None:
+            extra = {}
+        await self.createNode(client)
+        interest_elt = domish.Element((NS_LIST_INTEREST, "interest"))
+        interest_elt["namespace"] = namespace
+        if name is not None:
+            interest_elt['name'] = name
+        thumb_url = extra.get('thumb_url')
+        if thumb_url:
+            interest_elt['thumb_url'] = thumb_url
+        pubsub_elt = interest_elt.addElement("pubsub")
+        pubsub_elt["service"] = service.full()
+        pubsub_elt["node"] = node
+        if item_id is not None:
+            pubsub_elt["item"] = item_id
+        if creator:
+            pubsub_elt["creator"] = C.BOOL_TRUE
+        if element is not None:
+            pubsub_elt.addChild(element)
+        uri_kwargs = {
+            "path": service.full(),
+            "node": node
+        }
+        if item_id:
+            uri_kwargs['id'] = item_id
+        interest_uri = uri.build_xmpp_uri("pubsub", **uri_kwargs)
+        # we use URI of the interest as item id to avoid duplicates
+        item_elt = pubsub.Item(interest_uri, payload=interest_elt)
+        await self._p.publish(
+            client, client.jid.userhostJID(), NS_LIST_INTEREST, items=[item_elt]
+        )
+
+    def _register_file_sharing(
+        self, service, repos_type, namespace, path, name, extra_raw,
+        profile
+    ):
+        client = self.host.get_client(profile)
+        extra = data_format.deserialise(extra_raw)
+
+        return defer.ensureDeferred(self.register_file_sharing(
+            client, jid.JID(service), repos_type or None, namespace or None, path or None,
+            name or None, extra
+        ))
+
+    def normalise_file_sharing_service(self, client, service):
+        # FIXME: Q&D fix as the bare file sharing service JID will lead to user own
+        #   repository, which thus would not be the same for the host and the guest.
+        #   By specifying the user part, we for the use of the host repository.
+        #   A cleaner way should be implemented
+        if service.user is None:
+            service.user = self.host.plugins['XEP-0106'].escape(client.jid.user)
+
+    def get_file_sharing_id(self, service, namespace, path):
+        return f"{service}_{namespace or ''}_{path or ''}"
+
+    async def register_file_sharing(
+            self, client, service, repos_type=None, namespace=None, path=None, name=None,
+            extra=None):
+        """Register an interesting file repository in personal list
+
+        @param service(jid.JID): service of the file repository
+        @param repos_type(unicode): type of the repository
+        @param namespace(unicode, None): namespace of the repository
+        @param path(unicode, None): path of the repository
+        @param name(unicode, None): name of the repository
+        @param extra(dict, None): same as [register_pubsub]
+        """
+        if extra is None:
+            extra = {}
+        self.normalise_file_sharing_service(client, service)
+        await self.createNode(client)
+        item_id = self.get_file_sharing_id(service, namespace, path)
+        interest_elt = domish.Element((NS_LIST_INTEREST, "interest"))
+        interest_elt["namespace"] = self.host.get_namespace("fis")
+        if name is not None:
+            interest_elt['name'] = name
+        thumb_url = extra.get('thumb_url')
+        if thumb_url:
+            interest_elt['thumb_url'] = thumb_url
+
+        file_sharing_elt = interest_elt.addElement("file_sharing")
+        file_sharing_elt["service"] = service.full()
+        if repos_type is not None:
+            file_sharing_elt["type"] = repos_type
+        if namespace is not None:
+            file_sharing_elt["namespace"] = namespace
+        if path is not None:
+            file_sharing_elt["path"] = path
+        item_elt = pubsub.Item(item_id, payload=interest_elt)
+        await self._p.publish(
+            client, client.jid.userhostJID(), NS_LIST_INTEREST, items=[item_elt]
+        )
+
+    def _list_interests_serialise(self, interests_data):
+        interests = []
+        for item_elt in interests_data[0]:
+            interest_data = {"id": item_elt['id']}
+            interest_elt = item_elt.interest
+            if interest_elt.hasAttribute('namespace'):
+                interest_data['namespace'] = interest_elt.getAttribute('namespace')
+            if interest_elt.hasAttribute('name'):
+                interest_data['name'] = interest_elt.getAttribute('name')
+            if interest_elt.hasAttribute('thumb_url'):
+                interest_data['thumb_url'] = interest_elt.getAttribute('thumb_url')
+            elt = interest_elt.firstChildElement()
+            if elt.uri != NS_LIST_INTEREST:
+                log.warning("unexpected child element, ignoring: {xml}".format(
+                    xml = elt.toXml()))
+                continue
+            if elt.name == 'pubsub':
+                interest_data.update({
+                    "type": "pubsub",
+                    "service": elt['service'],
+                    "node": elt['node'],
+                })
+                for attr in ('item', 'creator'):
+                    if elt.hasAttribute(attr):
+                        interest_data[attr] = elt[attr]
+            elif elt.name == 'file_sharing':
+                interest_data.update({
+                    "type": "file_sharing",
+                    "service": elt['service'],
+                })
+                if elt.hasAttribute('type'):
+                    interest_data['subtype'] = elt['type']
+                for attr in ('files_namespace', 'path'):
+                    if elt.hasAttribute(attr):
+                        interest_data[attr] = elt[attr]
+            else:
+                log.warning("unknown element, ignoring: {xml}".format(xml=elt.toXml()))
+                continue
+            interests.append(interest_data)
+
+        return interests
+
+    def _list_interests(self, service, node, namespace, profile):
+        service = jid.JID(service) if service else None
+        node = node or None
+        namespace = namespace or None
+        client = self.host.get_client(profile)
+        d = defer.ensureDeferred(self.list_interests(client, service, node, namespace))
+        d.addCallback(self._list_interests_serialise)
+        return d
+
+    async def list_interests(self, client, service=None, node=None, namespace=None):
+        """Retrieve list of interests
+
+        @param service(jid.JID, None): service to use
+            None to use own PEP
+        @param node(unicode, None): node to use
+            None to use default node
+        @param namespace(unicode, None): filter interests of this namespace
+            None to retrieve all interests
+        @return: same as [XEP_0060.get_items]
+        """
+        # TODO: if a MAM filter were available, it would improve performances
+        if not node:
+            node = NS_LIST_INTEREST
+        items, metadata = await self._p.get_items(client, service, node)
+        if namespace is not None:
+            filtered_items = []
+            for item in items:
+                try:
+                    interest_elt = next(item.elements(NS_LIST_INTEREST, "interest"))
+                except StopIteration:
+                    log.warning(_("Missing interest element: {xml}").format(
+                        xml=item.toXml()))
+                    continue
+                if interest_elt.getAttribute("namespace") == namespace:
+                    filtered_items.append(item)
+            items = filtered_items
+
+        return (items, metadata)
+
+    def _interest_retract(self, service_s, item_id, profile_key):
+        d = self._p._retract_item(
+            service_s, NS_LIST_INTEREST, item_id, True, profile_key)
+        d.addCallback(lambda __: None)
+        return d
+
+    async def get(self, client: SatXMPPEntity, item_id: str) -> dict:
+        """Retrieve a specific interest in profile's list"""
+        items_data = await self._p.get_items(client, None, NS_LIST_INTEREST, item_ids=[item_id])
+        try:
+            return self._list_interests_serialise(items_data)[0]
+        except IndexError:
+            raise exceptions.NotFound
+
+
+@implementer(iwokkel.IDisco)
+class ListInterestHandler(XMPPHandler):
+
+    def __init__(self, plugin_parent):
+        self.plugin_parent = plugin_parent
+
+    def getDiscoInfo(self, requestor, target, nodeIdentifier=""):
+        return [
+            disco.DiscoFeature(NS_LIST_INTEREST),
+        ]
+
+    def getDiscoItems(self, requestor, target, nodeIdentifier=""):
+        return []