libervia-backend: libervia/backend/plugins/plugin_xep

annotate libervia/backend/plugins/plugin_xep_0376.py @ 4321:2246eeeccc74

tests (unit): fix tests: - test_ap-gateway: fix missing implementation of `client.is_local` - test_plugin_xep_0215: fix missing return value of `has_feature`

author	Goffi <goffi@goffi.org>
date	Mon, 30 Sep 2024 14:15:47 +0200
parents	9658c534287e
children

rev	line source
3758 b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	1 #!/usr/bin/env python3
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	2
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	3 # SàT plugin for XEP-0376
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	4 # Copyright (C) 2009-2021 Jérôme Poisson (goffi@goffi.org)
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	5
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	6 # This program is free software: you can redistribute it and/or modify
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	7 # it under the terms of the GNU Affero General Public License as published by
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	8 # the Free Software Foundation, either version 3 of the License, or
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	9 # (at your option) any later version.
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	10
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	11 # This program is distributed in the hope that it will be useful,
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	12 # but WITHOUT ANY WARRANTY; without even the implied warranty of
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	13 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	14 # GNU Affero General Public License for more details.
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	15
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	16 # You should have received a copy of the GNU Affero General Public License
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	17 # along with this program. If not, see <http://www.gnu.org/licenses/>.
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	18
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	19 from typing import Dict, List, Tuple, Optional, Any
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	20 from zope.interface import implementer
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	21 from twisted.words.protocols.jabber import jid
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	22 from twisted.words.protocols.jabber.xmlstream import XMPPHandler
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	23 from wokkel import disco, iwokkel, pubsub, data_form
4071 4b842c1fb686 refactoring: renamed `sat` package to `libervia.backend` Goffi <goffi@goffi.org> parents: 4037 diff changeset	24 from libervia.backend.core.i18n import _
4b842c1fb686 refactoring: renamed `sat` package to `libervia.backend` Goffi <goffi@goffi.org> parents: 4037 diff changeset	25 from libervia.backend.core.constants import Const as C
4b842c1fb686 refactoring: renamed `sat` package to `libervia.backend` Goffi <goffi@goffi.org> parents: 4037 diff changeset	26 from libervia.backend.core import exceptions
4b842c1fb686 refactoring: renamed `sat` package to `libervia.backend` Goffi <goffi@goffi.org> parents: 4037 diff changeset	27 from libervia.backend.core.xmpp import SatXMPPEntity
4b842c1fb686 refactoring: renamed `sat` package to `libervia.backend` Goffi <goffi@goffi.org> parents: 4037 diff changeset	28 from libervia.backend.core.log import getLogger
3758 b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	29
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	30 log = getLogger(__name__)
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	31
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	32 PLUGIN_INFO = {
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	33 C.PI_NAME: "Pubsub Account Management",
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	34 C.PI_IMPORT_NAME: "XEP-0376",
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	35 C.PI_TYPE: C.PLUG_TYPE_XEP,
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	36 C.PI_MODES: C.PLUG_MODE_BOTH,
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	37 C.PI_PROTOCOLS: ["XEP-0376"],
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	38 C.PI_DEPENDENCIES: ["XEP-0060"],
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	39 C.PI_MAIN: "XEP_0376",
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	40 C.PI_HANDLER: "yes",
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	41 C.PI_DESCRIPTION: _("""Pubsub Account Management"""),
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	42 }
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	43
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	44 NS_PAM = "urn:xmpp:pam:0"
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	45
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	46
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	47 class XEP_0376:
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	48
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	49 def __init__(self, host):
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	50 log.info(_("Pubsub Account Management initialization"))
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	51 self.host = host
4037 524856bd7b19 massive refactoring to switch from camelCase to snake_case: Goffi <goffi@goffi.org> parents: 3758 diff changeset	52 host.register_namespace("pam", NS_PAM)
3758 b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	53 self._p = self.host.plugins["XEP-0060"]
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	54 host.trigger.add("XEP-0060_subscribe", self.subscribe)
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	55 host.trigger.add("XEP-0060_unsubscribe", self.unsubscribe)
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	56 host.trigger.add("XEP-0060_subscriptions", self.subscriptions)
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	57
4037 524856bd7b19 massive refactoring to switch from camelCase to snake_case: Goffi <goffi@goffi.org> parents: 3758 diff changeset	58 def get_handler(self, client):
3758 b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	59 return XEP_0376_Handler()
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	60
4037 524856bd7b19 massive refactoring to switch from camelCase to snake_case: Goffi <goffi@goffi.org> parents: 3758 diff changeset	61 async def profile_connected(self, client):
4320 9658c534287e plugin XEP-0215, XEP-0376: fix bad calls to `hasFeature`: Goffi <goffi@goffi.org> parents: 4270 diff changeset	62 if not await self.host.memory.disco.has_feature(client, NS_PAM):
3758 b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	63 log.warning(
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	64 "Your server doesn't support Pubsub Account Management, this is used to "
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	65 "track all your subscriptions. You may ask your server administrator to "
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	66 "install it."
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	67 )
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	68
4037 524856bd7b19 massive refactoring to switch from camelCase to snake_case: Goffi <goffi@goffi.org> parents: 3758 diff changeset	69 async def _sub_request(
3758 b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	70 self,
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	71 client: SatXMPPEntity,
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	72 service: jid.JID,
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	73 nodeIdentifier: str,
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	74 sub_jid: Optional[jid.JID],
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	75 options: Optional[dict],
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4071 diff changeset	76 subscribe: bool,
3758 b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	77 ) -> None:
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	78 if sub_jid is None:
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	79 sub_jid = client.jid.userhostJID()
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	80 iq_elt = client.IQ()
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	81 pam_elt = iq_elt.addElement((NS_PAM, "pam"))
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	82 pam_elt["jid"] = service.full()
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	83 subscribe_elt = pam_elt.addElement(
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	84 (pubsub.NS_PUBSUB, "subscribe" if subscribe else "unsubscribe")
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	85 )
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	86 subscribe_elt["node"] = nodeIdentifier
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	87 subscribe_elt["jid"] = sub_jid.full()
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	88 if options:
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	89 options_elt = pam_elt.addElement((pubsub.NS_PUBSUB, "options"))
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	90 options_elt["node"] = nodeIdentifier
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	91 options_elt["jid"] = sub_jid.full()
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	92 form = data_form.Form(
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4071 diff changeset	93 formType="submit", formNamespace=pubsub.NS_PUBSUB_SUBSCRIBE_OPTIONS
3758 b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	94 )
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	95 form.makeFields(options)
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	96 options_elt.addChild(form.toElement())
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	97
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	98 await iq_elt.send(client.server_jid.full())
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	99
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	100 async def subscribe(
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	101 self,
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	102 client: SatXMPPEntity,
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	103 service: jid.JID,
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	104 nodeIdentifier: str,
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	105 sub_jid: Optional[jid.JID] = None,
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4071 diff changeset	106 options: Optional[dict] = None,
3758 b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	107 ) -> Tuple[bool, Optional[pubsub.Subscription]]:
4320 9658c534287e plugin XEP-0215, XEP-0376: fix bad calls to `hasFeature`: Goffi <goffi@goffi.org> parents: 4270 diff changeset	108 if not await self.host.memory.disco.has_feature(client, NS_PAM) or client.is_component:
3758 b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	109 return True, None
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	110
4037 524856bd7b19 massive refactoring to switch from camelCase to snake_case: Goffi <goffi@goffi.org> parents: 3758 diff changeset	111 await self._sub_request(client, service, nodeIdentifier, sub_jid, options, True)
3758 b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	112
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	113 # TODO: actual result is sent with <message> stanza, we have to get and use them
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	114 # to known the actual result. XEP-0376 returns an empty <iq> result, thus we don't
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	115 # know here is the subscription actually succeeded
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	116
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	117 sub_id = None
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	118 sub = pubsub.Subscription(nodeIdentifier, sub_jid, "subscribed", options, sub_id)
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	119 return False, sub
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	120
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	121 async def unsubscribe(
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	122 self,
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	123 client: SatXMPPEntity,
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	124 service: jid.JID,
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	125 nodeIdentifier: str,
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	126 sub_jid: Optional[jid.JID],
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	127 subscriptionIdentifier: Optional[str],
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	128 sender: Optional[jid.JID] = None,
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	129 ) -> bool:
4320 9658c534287e plugin XEP-0215, XEP-0376: fix bad calls to `hasFeature`: Goffi <goffi@goffi.org> parents: 4270 diff changeset	130 if not await self.host.memory.disco.has_feature(client, NS_PAM) or client.is_component:
3758 b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	131 return True
4037 524856bd7b19 massive refactoring to switch from camelCase to snake_case: Goffi <goffi@goffi.org> parents: 3758 diff changeset	132 await self._sub_request(client, service, nodeIdentifier, sub_jid, None, False)
3758 b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	133 return False
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	134
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	135 async def subscriptions(
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	136 self,
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	137 client: SatXMPPEntity,
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	138 service: Optional[jid.JID],
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	139 node: str,
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	140 ) -> Tuple[bool, Optional[List[Dict[str, Any]]]]:
4320 9658c534287e plugin XEP-0215, XEP-0376: fix bad calls to `hasFeature`: Goffi <goffi@goffi.org> parents: 4270 diff changeset	141 if not await self.host.has_feature(client, NS_PAM):
3758 b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	142 return True, None
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	143 if service is not None or node is not None:
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	144 # if we have service and/or node subscriptions, it's a regular XEP-0060
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	145 # subscriptions request
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	146 return True, None
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	147
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	148 iq_elt = client.IQ("get")
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	149 subscriptions_elt = iq_elt.addElement((NS_PAM, "subscriptions"))
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	150 result_elt = await iq_elt.send()
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	151 try:
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	152 subscriptions_elt = next(result_elt.elements(NS_PAM, "subscriptions"))
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	153 except StopIteration:
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	154 raise ValueError(f"invalid PAM response: {result_elt.toXml()}")
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	155 subs = []
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	156 for subscription_elt in subscriptions_elt.elements(NS_PAM, "subscription"):
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	157 sub = {}
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	158 try:
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	159 for attr, key in (
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	160 ("service", "service"),
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	161 ("node", "node"),
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	162 ("jid", "subscriber"),
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4071 diff changeset	163 ("subscription", "state"),
3758 b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	164 ):
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	165 sub[key] = subscription_elt[attr]
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	166 except KeyError as e:
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	167 log.warning(
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	168 f"Invalid <subscription> element (missing {e.args[0]!r} attribute): "
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	169 f"{subscription_elt.toXml()}"
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	170 )
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	171 continue
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	172 sub_id = subscription_elt.getAttribute("subid")
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	173 if sub_id:
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	174 sub["id"] = sub_id
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	175 subs.append(sub)
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	176
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	177 return False, subs
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	178
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	179
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	180 @implementer(iwokkel.IDisco)
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	181 class XEP_0376_Handler(XMPPHandler):
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	182
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	183 def getDiscoInfo(self, requestor, service, nodeIdentifier=""):
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	184 return [disco.DiscoFeature(NS_PAM)]
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	185
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	186 def getDiscoItems(self, requestor, service, nodeIdentifier=""):
b7cef1b24f83 plugins XEP-0060, XEP-0376, XEP-0465, CLI: PAM + PSS implementation: Goffi <goffi@goffi.org> parents: diff changeset	187 return []

Mercurial > libervia-backend

annotate libervia/backend/plugins/plugin_xep_0376.py @ 4321:2246eeeccc74