diff libervia/backend/plugins/plugin_xep_0280.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_0280.py@c23cad65ae99
children 9162d3480b9e
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/libervia/backend/plugins/plugin_xep_0280.py	Fri Jun 02 11:49:51 2023 +0200
@@ -0,0 +1,174 @@
+#!/usr/bin/env python3
+
+
+# SAT plugin for managing xep-0280
+# 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 _, D_
+from libervia.backend.core.log import getLogger
+
+log = getLogger(__name__)
+from libervia.backend.core import exceptions
+from libervia.backend.core.constants import Const as C
+from twisted.words.protocols.jabber.error import StanzaError
+from twisted.internet import defer
+from wokkel import disco, iwokkel
+from zope.interface import implementer
+
+try:
+    from twisted.words.protocols.xmlstream import XMPPHandler
+except ImportError:
+    from wokkel.subprotocols import XMPPHandler
+
+
+PARAM_CATEGORY = "Misc"
+PARAM_NAME = "carbon"
+PARAM_LABEL = D_("Message carbons")
+NS_CARBONS = "urn:xmpp:carbons:2"
+
+PLUGIN_INFO = {
+    C.PI_NAME: "XEP-0280 Plugin",
+    C.PI_IMPORT_NAME: "XEP-0280",
+    C.PI_TYPE: "XEP",
+    C.PI_PROTOCOLS: ["XEP-0280"],
+    C.PI_DEPENDENCIES: [],
+    C.PI_MAIN: "XEP_0280",
+    C.PI_HANDLER: "yes",
+    C.PI_DESCRIPTION: D_("""Implementation of Message Carbons"""),
+}
+
+
+class XEP_0280(object):
+    #  TODO: param is only checked at profile connection
+    #       activate carbons on param change even after profile connection
+    # TODO: chat state notifications are not handled yet (and potentially other XEPs?)
+
+    params = """
+    <params>
+    <individual>
+    <category name="{category_name}" label="{category_label}">
+        <param name="{param_name}" label="{param_label}" value="true" type="bool" security="0" />
+     </category>
+    </individual>
+    </params>
+    """.format(
+        category_name=PARAM_CATEGORY,
+        category_label=D_(PARAM_CATEGORY),
+        param_name=PARAM_NAME,
+        param_label=PARAM_LABEL,
+    )
+
+    def __init__(self, host):
+        log.info(_("Plugin XEP_0280 initialization"))
+        self.host = host
+        host.memory.update_params(self.params)
+        host.trigger.add("message_received", self.message_received_trigger, priority=200000)
+
+    def get_handler(self, client):
+        return XEP_0280_handler()
+
+    def set_private(self, message_elt):
+        """Add a <private/> element to a message
+
+        this method is intented to be called on final domish.Element by other plugins
+        (in particular end 2 end encryption plugins)
+        @param message_elt(domish.Element): <message> stanza
+        """
+        if message_elt.name != "message":
+            log.error("addPrivateElt must be used with <message> stanzas")
+            return
+        message_elt.addElement((NS_CARBONS, "private"))
+
+    @defer.inlineCallbacks
+    def profile_connected(self, client):
+        """activate message carbons on connection if possible and activated in config"""
+        activate = self.host.memory.param_get_a(
+            PARAM_NAME, PARAM_CATEGORY, profile_key=client.profile
+        )
+        if not activate:
+            log.info(_("Not activating message carbons as requested in params"))
+            return
+        try:
+            yield self.host.check_features(client, (NS_CARBONS,))
+        except exceptions.FeatureNotFound:
+            log.warning(_("server doesn't handle message carbons"))
+        else:
+            log.info(_("message carbons available, enabling it"))
+            iq_elt = client.IQ()
+            iq_elt.addElement((NS_CARBONS, "enable"))
+            try:
+                yield iq_elt.send()
+            except StanzaError as e:
+                log.warning("Can't activate message carbons: {}".format(e))
+            else:
+                log.info(_("message carbons activated"))
+
+    def message_received_trigger(self, client, message_elt, post_treat):
+        """get message and handle it if carbons namespace is present"""
+        carbons_elt = None
+        for e in message_elt.elements():
+            if e.uri == NS_CARBONS:
+                carbons_elt = e
+                break
+
+        if carbons_elt is None:
+            # this is not a message carbons,
+            # we continue normal behaviour
+            return True
+
+        if message_elt["from"] != client.jid.userhost():
+            log.warning(
+                "The message carbon received is not from our server, hack attempt?\n{xml}".format(
+                    xml=message_elt.toXml()
+                )
+            )
+            return
+        forwarded_elt = next(carbons_elt.elements(C.NS_FORWARD, "forwarded"))
+        cc_message_elt = next(forwarded_elt.elements(C.NS_CLIENT, "message"))
+
+        # we replace the wrapping message with the CCed one
+        # and continue the normal behaviour
+        if carbons_elt.name == "received":
+            message_elt["from"] = cc_message_elt["from"]
+        elif carbons_elt.name == "sent":
+            try:
+                message_elt["to"] = cc_message_elt["to"]
+            except KeyError:
+                # we may not have "to" in case of message from ourself (from an other
+                # device)
+                pass
+        else:
+            log.warning(
+                "invalid message carbons received:\n{xml}".format(
+                    xml=message_elt.toXml()
+                )
+            )
+            return False
+
+        del message_elt.children[:]
+        for c in cc_message_elt.children:
+            message_elt.addChild(c)
+
+        return True
+
+@implementer(iwokkel.IDisco)
+class XEP_0280_handler(XMPPHandler):
+
+    def getDiscoInfo(self, requestor, target, nodeIdentifier=""):
+        return [disco.DiscoFeature(NS_CARBONS)]
+
+    def getDiscoItems(self, requestor, target, nodeIdentifier=""):
+        return []