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
This commit is contained in:
parent
f2e11805b2
commit
610e457bec
1 changed files with 246 additions and 0 deletions
246
sat/plugins/plugin_xep_0292.py
Normal file
246
sat/plugins/plugin_xep_0292.py
Normal file
|
|
@ -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 []
|
||||
Loading…
Reference in a new issue