Mercurial > libervia-backend
diff libervia/backend/plugins/plugin_misc_lists.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_misc_lists.py@524856bd7b19 |
children | 0d7bb4df2343 |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/libervia/backend/plugins/plugin_misc_lists.py Fri Jun 02 11:49:51 2023 +0200 @@ -0,0 +1,519 @@ +#!/usr/bin/env python3 + +# 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/>. + +import shortuuid +from typing import List, Tuple, Optional +from twisted.internet import defer +from twisted.words.xish import domish +from twisted.words.protocols.jabber import jid +from libervia.backend.core.i18n import _, D_ +from libervia.backend.core.xmpp import SatXMPPEntity +from libervia.backend.core.constants import Const as C +from libervia.backend.tools import xml_tools +from libervia.backend.tools.common import uri +from libervia.backend.tools.common import data_format +from libervia.backend.core.log import getLogger + +log = getLogger(__name__) + +# XXX: this plugin was formely named "tickets", thus the namespace keeps this +# name +APP_NS_TICKETS = "org.salut-a-toi.tickets:0" +NS_TICKETS_TYPE = "org.salut-a-toi.tickets#type:0" + +PLUGIN_INFO = { + C.PI_NAME: _("Pubsub Lists"), + C.PI_IMPORT_NAME: "LISTS", + C.PI_TYPE: "EXP", + C.PI_PROTOCOLS: [], + C.PI_DEPENDENCIES: ["XEP-0060", "XEP-0346", "XEP-0277", "IDENTITY", + "PUBSUB_INVITATION"], + C.PI_MAIN: "PubsubLists", + C.PI_HANDLER: "no", + C.PI_DESCRIPTION: _("""Pubsub lists management plugin"""), +} + +TEMPLATES = { + "todo": { + "name": D_("TODO List"), + "icon": "check", + "fields": [ + {"name": "title"}, + {"name": "author"}, + {"name": "created"}, + {"name": "updated"}, + {"name": "time_limit"}, + {"name": "labels", "type": "text-multi"}, + { + "name": "status", + "label": D_("status"), + "type": "list-single", + "options": [ + { + "label": D_("to do"), + "value": "todo" + }, + { + "label": D_("in progress"), + "value": "in_progress" + }, + { + "label": D_("done"), + "value": "done" + }, + ], + "value": "todo" + }, + { + "name": "priority", + "label": D_("priority"), + "type": "list-single", + "options": [ + { + "label": D_("major"), + "value": "major" + }, + { + "label": D_("normal"), + "value": "normal" + }, + { + "label": D_("minor"), + "value": "minor" + }, + ], + "value": "normal" + }, + {"name": "body", "type": "xhtml"}, + {"name": "comments_uri"}, + ] + }, + "grocery": { + "name": D_("Grocery List"), + "icon": "basket", + "fields": [ + {"name": "name", "label": D_("name")}, + {"name": "quantity", "label": D_("quantity")}, + { + "name": "status", + "label": D_("status"), + "type": "list-single", + "options": [ + { + "label": D_("to buy"), + "value": "to_buy" + }, + { + "label": D_("bought"), + "value": "bought" + }, + ], + "value": "to_buy" + }, + ] + }, + "tickets": { + "name": D_("Tickets"), + "icon": "clipboard", + "fields": [ + {"name": "title"}, + {"name": "author"}, + {"name": "created"}, + {"name": "updated"}, + {"name": "labels", "type": "text-multi"}, + { + "name": "type", + "label": D_("type"), + "type": "list-single", + "options": [ + { + "label": D_("bug"), + "value": "bug" + }, + { + "label": D_("feature request"), + "value": "feature" + }, + ], + "value": "bug" + }, + { + "name": "status", + "label": D_("status"), + "type": "list-single", + "options": [ + { + "label": D_("queued"), + "value": "queued" + }, + { + "label": D_("started"), + "value": "started" + }, + { + "label": D_("review"), + "value": "review" + }, + { + "label": D_("closed"), + "value": "closed" + }, + ], + "value": "queued" + }, + { + "name": "priority", + "label": D_("priority"), + "type": "list-single", + "options": [ + { + "label": D_("major"), + "value": "major" + }, + { + "label": D_("normal"), + "value": "normal" + }, + { + "label": D_("minor"), + "value": "minor" + }, + ], + "value": "normal" + }, + {"name": "body", "type": "xhtml"}, + {"name": "comments_uri"}, + ] + } +} + + +class PubsubLists: + + def __init__(self, host): + log.info(_("Pubsub lists plugin initialization")) + self.host = host + self._s = self.host.plugins["XEP-0346"] + self.namespace = self._s.get_submitted_ns(APP_NS_TICKETS) + host.register_namespace("tickets", APP_NS_TICKETS) + host.register_namespace("tickets_type", NS_TICKETS_TYPE) + self.host.plugins["PUBSUB_INVITATION"].register( + APP_NS_TICKETS, self + ) + self._p = self.host.plugins["XEP-0060"] + self._m = self.host.plugins["XEP-0277"] + host.bridge.add_method( + "list_get", + ".plugin", + in_sign="ssiassss", + out_sign="s", + method=lambda service, node, max_items, items_ids, sub_id, extra, profile_key: + self._s._get( + service, + node, + max_items, + items_ids, + sub_id, + extra, + default_node=self.namespace, + form_ns=APP_NS_TICKETS, + filters={ + "author": self._s.value_or_publisher_filter, + "created": self._s.date_filter, + "updated": self._s.date_filter, + "time_limit": self._s.date_filter, + }, + profile_key=profile_key), + async_=True, + ) + host.bridge.add_method( + "list_set", + ".plugin", + in_sign="ssa{sas}ssss", + out_sign="s", + method=self._set, + async_=True, + ) + host.bridge.add_method( + "list_delete_item", + ".plugin", + in_sign="sssbs", + out_sign="", + method=self._delete, + async_=True, + ) + host.bridge.add_method( + "list_schema_get", + ".plugin", + in_sign="sss", + out_sign="s", + method=lambda service, nodeIdentifier, profile_key: self._s._get_ui_schema( + service, nodeIdentifier, default_node=self.namespace, + profile_key=profile_key), + async_=True, + ) + host.bridge.add_method( + "lists_list", + ".plugin", + in_sign="sss", + out_sign="s", + method=self._lists_list, + async_=True, + ) + host.bridge.add_method( + "list_templates_names_get", + ".plugin", + in_sign="ss", + out_sign="s", + method=self._get_templates_names, + ) + host.bridge.add_method( + "list_template_get", + ".plugin", + in_sign="sss", + out_sign="s", + method=self._get_template, + ) + host.bridge.add_method( + "list_template_create", + ".plugin", + in_sign="ssss", + out_sign="(ss)", + method=self._create_template, + async_=True, + ) + + async def on_invitation_preflight( + self, + client: SatXMPPEntity, + namespace: str, + name: str, + extra: dict, + service: jid.JID, + node: str, + item_id: Optional[str], + item_elt: domish.Element + ) -> None: + try: + schema = await self._s.get_schema_form(client, service, node) + except Exception as e: + log.warning(f"Can't retrive node schema as {node!r} [{service}]: {e}") + else: + try: + field_type = schema[NS_TICKETS_TYPE] + except KeyError: + log.debug("no type found in list schema") + else: + list_elt = extra["element"] = domish.Element((APP_NS_TICKETS, "list")) + list_elt["type"] = field_type + + def _set(self, service, node, values, schema=None, item_id=None, extra_s='', + profile_key=C.PROF_KEY_NONE): + client, service, node, schema, item_id, extra = self._s.prepare_bridge_set( + service, node, schema, item_id, extra_s, profile_key + ) + d = defer.ensureDeferred(self.set( + client, service, node, values, schema, item_id, extra, deserialise=True + )) + d.addCallback(lambda ret: ret or "") + return d + + async def set( + self, client, service, node, values, schema=None, item_id=None, extra=None, + deserialise=False, form_ns=APP_NS_TICKETS + ): + """Publish a tickets + + @param node(unicode, None): Pubsub node to use + None to use default tickets node + @param values(dict[key(unicode), [iterable[object]|object]]): values of the ticket + + if value is not iterable, it will be put in a list + 'created' and 'updated' will be forced to current time: + - 'created' is set if item_id is None, i.e. if it's a new ticket + - 'updated' is set everytime + @param extra(dict, None): same as for [XEP-0060.send_item] with additional keys: + - update(bool): if True, get previous item data to merge with current one + if True, item_id must be set + other arguments are same as for [self._s.send_data_form_item] + @return (unicode): id of the created item + """ + if not node: + node = self.namespace + + if not item_id: + comments_service = await self._m.get_comments_service(client, service) + + # we need to use uuid for comments node, because we don't know item id in + # advance (we don't want to set it ourselves to let the server choose, so we + # can have a nicer id if serial ids is activated) + comments_node = self._m.get_comments_node( + node + "_" + str(shortuuid.uuid()) + ) + options = { + self._p.OPT_ACCESS_MODEL: self._p.ACCESS_OPEN, + self._p.OPT_PERSIST_ITEMS: 1, + self._p.OPT_DELIVER_PAYLOADS: 1, + self._p.OPT_SEND_ITEM_SUBSCRIBE: 1, + self._p.OPT_PUBLISH_MODEL: self._p.ACCESS_OPEN, + } + await self._p.createNode(client, comments_service, comments_node, options) + values["comments_uri"] = uri.build_xmpp_uri( + "pubsub", + subtype="microblog", + path=comments_service.full(), + node=comments_node, + ) + + return await self._s.set( + client, service, node, values, schema, item_id, extra, deserialise, form_ns + ) + + def _delete( + self, service_s, nodeIdentifier, itemIdentifier, notify, profile_key + ): + client = self.host.get_client(profile_key) + return defer.ensureDeferred(self.delete( + client, + jid.JID(service_s) if service_s else None, + nodeIdentifier, + itemIdentifier, + notify + )) + + async def delete( + self, + client: SatXMPPEntity, + service: Optional[jid.JID], + node: Optional[str], + itemIdentifier: str, + notify: Optional[bool] = None + ) -> None: + if not node: + node = self.namespace + return await self._p.retract_items( + service, node, (itemIdentifier,), notify, client.profile + ) + + def _lists_list(self, service, node, profile): + service = jid.JID(service) if service else None + node = node or None + client = self.host.get_client(profile) + d = defer.ensureDeferred(self.lists_list(client, service, node)) + d.addCallback(data_format.serialise) + return d + + async def lists_list( + self, client, service: Optional[jid.JID], node: Optional[str]=None + ) -> List[dict]: + """Retrieve list of pubsub lists registered in personal interests + + @return list: list of lists metadata + """ + items, metadata = await self.host.plugins['LIST_INTEREST'].list_interests( + client, service, node, namespace=APP_NS_TICKETS) + lists = [] + for item in items: + interest_elt = item.interest + if interest_elt is None: + log.warning(f"invalid interest for {client.profile}: {item.toXml}") + continue + if interest_elt.getAttribute("namespace") != APP_NS_TICKETS: + continue + pubsub_elt = interest_elt.pubsub + list_data = { + "id": item["id"], + "name": interest_elt["name"], + "service": pubsub_elt["service"], + "node": pubsub_elt["node"], + "creator": C.bool(pubsub_elt.getAttribute("creator", C.BOOL_FALSE)), + } + try: + list_elt = next(pubsub_elt.elements(APP_NS_TICKETS, "list")) + except StopIteration: + pass + else: + list_type = list_data["type"] = list_elt["type"] + if list_type in TEMPLATES: + list_data["icon_name"] = TEMPLATES[list_type]["icon"] + lists.append(list_data) + + return lists + + def _get_templates_names(self, language, profile): + client = self.host.get_client(profile) + return data_format.serialise(self.get_templates_names(client, language)) + + def get_templates_names(self, client, language: str) -> list: + """Retrieve well known list templates""" + + templates = [{"id": tpl_id, "name": d["name"], "icon": d["icon"]} + for tpl_id, d in TEMPLATES.items()] + return templates + + def _get_template(self, name, language, profile): + client = self.host.get_client(profile) + return data_format.serialise(self.get_template(client, name, language)) + + def get_template(self, client, name: str, language: str) -> dict: + """Retrieve a well known template""" + return TEMPLATES[name] + + def _create_template(self, template_id, name, access_model, profile): + client = self.host.get_client(profile) + d = defer.ensureDeferred(self.create_template( + client, template_id, name, access_model + )) + d.addCallback(lambda node_data: (node_data[0].full(), node_data[1])) + return d + + async def create_template( + self, client, template_id: str, name: str, access_model: str + ) -> Tuple[jid.JID, str]: + """Create a list from a template""" + name = name.strip() + if not name: + name = shortuuid.uuid() + fields = TEMPLATES[template_id]["fields"].copy() + fields.insert( + 0, + {"type": "hidden", "name": NS_TICKETS_TYPE, "value": template_id} + ) + schema = xml_tools.data_dict_2_data_form( + {"namespace": APP_NS_TICKETS, "fields": fields} + ).toElement() + + service = client.jid.userhostJID() + node = self._s.get_submitted_ns(f"{APP_NS_TICKETS}_{name}") + options = { + self._p.OPT_ACCESS_MODEL: access_model, + } + if template_id == "grocery": + # for grocery list, we want all publishers to be able to set all items + # XXX: should node options be in TEMPLATE? + options[self._p.OPT_OVERWRITE_POLICY] = self._p.OWPOL_ANY_PUB + await self._p.createNode(client, service, node, options) + await self._s.set_schema(client, service, node, schema) + list_elt = domish.Element((APP_NS_TICKETS, "list")) + list_elt["type"] = template_id + try: + await self.host.plugins['LIST_INTEREST'].register_pubsub( + client, APP_NS_TICKETS, service, node, creator=True, + name=name, element=list_elt) + except Exception as e: + log.warning(f"Can't add list to interests: {e}") + return service, node