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