Mercurial > libervia-backend
view libervia/backend/plugins/plugin_exp_list_of_interest.py @ 4334:111dce64dcb5
plugins XEP-0300, XEP-0446, XEP-0447, XEP0448 and others: Refactoring to use Pydantic:
Pydantic models are used more and more in Libervia, for the bridge API, and also to
convert `domish.Element` to internal representation.
Type hints have also been added in many places.
rel 453
author | Goffi <goffi@goffi.org> |
---|---|
date | Tue, 03 Dec 2024 00:12:38 +0100 |
parents | 0d7bb4df2343 |
children |
line wrap: on
line source
#!/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 []