diff sat/plugins/plugin_xep_0292.py @ 3819:4f02e339d184

plugin XEP-0292: vCard4 Over XMPP implementation: implementation of vCard4, the dedicated <IQ/> protocol is not implemented because it's obsoleted (see comments). Fow now, only `fn`, `nickname` and `note` are handled. rel 368
author Goffi <goffi@goffi.org>
date Wed, 29 Jun 2022 11:58:00 +0200
parents
children 524856bd7b19
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/sat/plugins/plugin_xep_0292.py	Wed Jun 29 11:58:00 2022 +0200
@@ -0,0 +1,246 @@
+#!/usr/bin/env python3
+
+# Libervia plugin for XEP-0292
+# Copyright (C) 2009-2022 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 typing import List, Dict, Union, Any, Optional
+from functools import partial
+
+from twisted.words.protocols.jabber.xmlstream import XMPPHandler
+from twisted.words.protocols.jabber import jid
+from twisted.words.xish import domish
+from zope.interface import implementer
+from wokkel import disco, iwokkel
+
+from sat.core.constants import Const as C
+from sat.core.i18n import _
+from sat.core.log import getLogger
+from sat.core.core_types import SatXMPPEntity
+from sat.core import exceptions
+from sat.tools.common.async_utils import async_lru
+
+
+log = getLogger(__name__)
+
+IMPORT_NAME = "XEP-0292"
+
+PLUGIN_INFO = {
+    C.PI_NAME: "vCard4 Over XMPP",
+    C.PI_IMPORT_NAME: IMPORT_NAME,
+    C.PI_TYPE: C.PLUG_TYPE_XEP,
+    C.PI_MODES: C.PLUG_MODE_BOTH,
+    C.PI_PROTOCOLS: ["XEP-0292"],
+    C.PI_DEPENDENCIES: ["IDENTITY", "XEP-0060", "XEP-0163"],
+    C.PI_MAIN: "XEP_0292",
+    C.PI_HANDLER: "yes",
+    C.PI_DESCRIPTION: _("""XEP-0292 (vCard4 Over XMPP) implementation"""),
+}
+
+NS_VCARD4 = "urn:ietf:params:xml:ns:vcard-4.0"
+VCARD4_NODE = "urn:xmpp:vcard4"
+text_fields = {
+    "fn": "name",
+    "nickname": "nicknames",
+    "note": "description"
+}
+text_fields_inv = {v: k for k,v in text_fields.items()}
+
+
+class XEP_0292:
+    namespace = NS_VCARD4
+    node = VCARD4_NODE
+
+    def __init__(self, host):
+        # XXX: as of XEP-0292 v0.11, there is a dedicated <IQ/> protocol in this XEP which
+        # should be used according to the XEP. Hovewer it feels like an outdated behaviour
+        # and other clients don't seem to use it. After discussing it on xsf@ MUC, it
+        # seems that implemeting the dedicated <IQ/> protocol is a waste of time, and thus
+        # it is not done here. It is expected that this dedicated protocol will be removed
+        # from a future version of the XEP.
+        log.info(_("vCard4 Over XMPP initialization"))
+        host.registerNamespace("vcard4", NS_VCARD4)
+        self.host = host
+        self._p = host.plugins["XEP-0060"]
+        self._i = host.plugins['IDENTITY']
+        self._i.register(
+            IMPORT_NAME,
+            'nicknames',
+            partial(self.getValue, field="nicknames"),
+            partial(self.setValue, field="nicknames"),
+            priority=1000
+        )
+        self._i.register(
+            IMPORT_NAME,
+            'description',
+            partial(self.getValue, field="description"),
+            partial(self.setValue, field="description"),
+            priority=1000
+        )
+
+    def getHandler(self, client):
+        return XEP_0292_Handler()
+
+    def vcard2Dict(self, vcard_elt: domish.Element) -> Dict[str, Any]:
+        """Convert vcard element to equivalent identity metadata"""
+        vcard: Dict[str, Any] = {}
+
+        for metadata_elt in vcard_elt.elements():
+            # Text values
+            for source_field, dest_field in text_fields.items():
+                if metadata_elt.name == source_field:
+                    if metadata_elt.text is not None:
+                        dest_type = self._i.getFieldType(dest_field)
+                        value = str(metadata_elt.text)
+                        if dest_type is str:
+                            if dest_field in vcard:
+                                vcard[dest_field] +=  value
+                            else:
+                                vcard[dest_field] = value
+                        elif dest_type is list:
+                            vcard.setdefault(dest_field, []).append(value)
+                        else:
+                            raise exceptions.InternalError(
+                                f"unexpected dest_type: {dest_type!r}"
+                            )
+                    break
+            else:
+                log.debug(
+                    f"Following element is currently unmanaged: {metadata_elt.toXml()}"
+                )
+        return vcard
+
+    def dict2VCard(self, vcard: dict[str, Any]) -> domish.Element:
+        """Convert vcard metadata to vCard4 element"""
+        vcard_elt = domish.Element((NS_VCARD4, "vcard"))
+        for field, elt_name in text_fields_inv.items():
+            value = vcard.get(field)
+            if value:
+                if isinstance(value, str):
+                    value = [value]
+                if isinstance(value, list):
+                    for v in value:
+                        field_elt = vcard_elt.addElement(elt_name)
+                        field_elt.addElement("text", content=v)
+                else:
+                    log.warning(
+                        f"ignoring unexpected value: {value!r}"
+                    )
+
+        return vcard_elt
+
+    @async_lru(5)
+    async def getCard(self, client: SatXMPPEntity, entity: jid.JID) -> dict:
+        try:
+            items, metadata = await self._p.getItems(
+                client, entity, VCARD4_NODE, item_ids=["current"]
+            )
+        except exceptions.NotFound:
+            log.info(f"No vCard node found for {entity}")
+            return {}
+        item_elt = items[0]
+        try:
+            vcard_elt = next(item_elt.elements(NS_VCARD4, "vcard"))
+        except StopIteration:
+            log.info(f"vCard element is not present for {entity}")
+            return {}
+
+        return self.vcard2Dict(vcard_elt)
+
+    async def updateVCardElt(
+        self,
+        client: SatXMPPEntity,
+        vcard_elt: domish.Element,
+        entity: Optional[jid.JID] = None
+    ) -> None:
+        """Update VCard 4 of given entity, create node if doesn't already exist
+
+        @param vcard_elt: whole vCard element to update
+        @param entity: entity for which the vCard must be updated
+            None to update profile's own vCard
+        """
+        service = entity or client.jid.userhostJID()
+        node_options = {
+            self._p.OPT_ACCESS_MODEL: self._p.ACCESS_OPEN,
+            self._p.OPT_PUBLISH_MODEL: self._p.PUBLISH_MODEL_PUBLISHERS
+        }
+        await self._p.createIfNewNode(client, service, VCARD4_NODE, node_options)
+        await self._p.sendItem(
+            client, service, VCARD4_NODE, vcard_elt, item_id=self._p.ID_SINGLETON
+        )
+
+    async def updateVCard(
+        self,
+        client: SatXMPPEntity,
+        vcard: Dict[str, Any],
+        entity: Optional[jid.JID] = None,
+        update: bool = True,
+    ) -> None:
+        """Update VCard 4 of given entity, create node if doesn't already exist
+
+        @param vcard: identity metadata
+        @param entity: entity for which the vCard must be updated
+            None to update profile's own vCard
+        @param update: if True, current vCard will be retrieved and updated with given
+        vcard (thus if False, `vcard` data will fully replace previous one).
+        """
+        service = entity or client.jid.userhostJID()
+        if update:
+            current_vcard = await self.getCard(client, service)
+            current_vcard.update(vcard)
+            vcard = current_vcard
+        vcard_elt = self.dict2VCard(vcard)
+        await self.updateVCardElt(client, vcard_elt, service)
+
+    async def getValue(
+        self,
+        client: SatXMPPEntity,
+        entity: jid.JID,
+        field: str,
+    ) -> Optional[Union[str, List[str]]]:
+        """Return generic value
+
+        @param entity: entity from who the vCard comes
+        @param field: name of the field to get
+            This has to be a string field
+        @return request value
+        """
+        vcard_data = await self.getCard(client, entity)
+        return vcard_data.get(field)
+
+    async def setValue(
+        self,
+        client: SatXMPPEntity,
+        value: Union[str, List[str]],
+        entity: jid.JID,
+        field: str
+    ) -> None:
+        """Set generic value
+
+        @param entity: entity from who the vCard comes
+        @param field: name of the field to get
+            This has to be a string field
+        """
+        await self.updateVCard(client, {field: value}, entity)
+
+
+@implementer(iwokkel.IDisco)
+class XEP_0292_Handler(XMPPHandler):
+
+    def getDiscoInfo(self, requestor, service, nodeIdentifier=""):
+        return [disco.DiscoFeature(NS_VCARD4)]
+
+    def getDiscoItems(self, requestor, service, nodeIdentifier=""):
+        return []