diff libervia/backend/plugins/plugin_xep_0059.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_xep_0059.py@524856bd7b19
children 0d7bb4df2343
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/libervia/backend/plugins/plugin_xep_0059.py	Fri Jun 02 11:49:51 2023 +0200
@@ -0,0 +1,159 @@
+#!/usr/bin/env python3
+
+# Result Set Management (XEP-0059)
+# Copyright (C) 2009-2021 Jérôme Poisson (goffi@goffi.org)
+# Copyright (C) 2013-2016 Adrien Cossa (souliane@mailoo.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 typing import Optional
+from zope.interface import implementer
+from twisted.words.protocols.jabber import xmlstream
+from wokkel import disco
+from wokkel import iwokkel
+from wokkel import rsm
+from libervia.backend.core.i18n import _
+from libervia.backend.core.constants import Const as C
+from libervia.backend.core.log import getLogger
+
+
+log = getLogger(__name__)
+
+
+PLUGIN_INFO = {
+    C.PI_NAME: "Result Set Management",
+    C.PI_IMPORT_NAME: "XEP-0059",
+    C.PI_TYPE: "XEP",
+    C.PI_MODES: C.PLUG_MODE_BOTH,
+    C.PI_PROTOCOLS: ["XEP-0059"],
+    C.PI_MAIN: "XEP_0059",
+    C.PI_HANDLER: "yes",
+    C.PI_DESCRIPTION: _("""Implementation of Result Set Management"""),
+}
+
+RSM_PREFIX = "rsm_"
+
+
+class XEP_0059(object):
+    # XXX: RSM management is done directly in Wokkel.
+
+    def __init__(self, host):
+        log.info(_("Result Set Management plugin initialization"))
+
+    def get_handler(self, client):
+        return XEP_0059_handler()
+
+    def parse_extra(self, extra):
+        """Parse extra dictionnary to retrieve RSM arguments
+
+        @param extra(dict): data for parse
+        @return (rsm.RSMRequest, None): request with parsed arguments
+            or None if no RSM arguments have been found
+        """
+        if int(extra.get(RSM_PREFIX + 'max', 0)) < 0:
+            raise ValueError(_("rsm_max can't be negative"))
+
+        rsm_args = {}
+        for arg in ("max", "after", "before", "index"):
+            try:
+                argname = "max_" if arg == "max" else arg
+                rsm_args[argname] = extra.pop(RSM_PREFIX + arg)
+            except KeyError:
+                continue
+
+        if rsm_args:
+            return rsm.RSMRequest(**rsm_args)
+        else:
+            return None
+
+    def response2dict(self, rsm_response, data=None):
+        """Return a dict with RSM response
+
+        Key set in data can be:
+            - rsm_first: first item id in the page
+            - rsm_last: last item id in the page
+            - rsm_index: position of the first item in the full set (may be approximate)
+            - rsm_count: total number of items in the full set (may be approximage)
+        If a value doesn't exists, it's not set.
+        All values are set as strings.
+        @param rsm_response(rsm.RSMResponse): response to parse
+        @param data(dict, None): dict to update with rsm_* data.
+            If None, a new dict is created
+        @return (dict): data dict
+        """
+        if data is None:
+            data = {}
+        if rsm_response.first is not None:
+            data["first"] = rsm_response.first
+        if rsm_response.last is not None:
+            data["last"] = rsm_response.last
+        if rsm_response.index is not None:
+            data["index"] = rsm_response.index
+        return data
+
+    def get_next_request(
+        self,
+        rsm_request: rsm.RSMRequest,
+        rsm_response: rsm.RSMResponse,
+        log_progress: bool = True,
+    ) -> Optional[rsm.RSMRequest]:
+        """Generate next request to paginate through all items
+
+        Page will be retrieved forward
+        @param rsm_request: last request used
+        @param rsm_response: response from the last request
+        @return: request to retrive next page, or None if we are at the end
+            or if pagination is not possible
+        """
+        if rsm_request.max == 0:
+            log.warning("Can't do pagination if max is 0")
+            return None
+        if rsm_response is None:
+            # may happen if result set it empty, or we are at the end
+            return None
+        if (
+            rsm_response.count is not None
+            and rsm_response.index is not None
+        ):
+            next_index = rsm_response.index + rsm_request.max
+            if next_index >= rsm_response.count:
+                # we have reached the last page
+                return None
+
+            if log_progress:
+                log.debug(
+                    f"retrieving items {next_index} to "
+                    f"{min(next_index+rsm_request.max, rsm_response.count)} on "
+                    f"{rsm_response.count} ({next_index/rsm_response.count*100:.2f}%)"
+                )
+
+        if rsm_response.last is None:
+            if rsm_response.count:
+                log.warning("Can't do pagination, no \"last\" received")
+            return None
+
+        return rsm.RSMRequest(
+            max_=rsm_request.max,
+            after=rsm_response.last
+        )
+
+
+@implementer(iwokkel.IDisco)
+class XEP_0059_handler(xmlstream.XMPPHandler):
+
+    def getDiscoInfo(self, requestor, target, nodeIdentifier=""):
+        return [disco.DiscoFeature(rsm.NS_RSM)]
+
+    def getDiscoItems(self, requestor, target, nodeIdentifier=""):
+        return []