Mercurial > libervia-backend
view sat/plugins/plugin_exp_events.py @ 3476:11003f9f92fb
core: moved `sat_tmp.wokkel` install `sat_plugin`:
This patch need to be installed early so it's available before anything imports
`wokkel.data_form`.
author | Goffi <goffi@goffi.org> |
---|---|
date | Thu, 18 Mar 2021 22:35:59 +0100 |
parents | 12dc234f698c |
children | be6d91572633 |
line wrap: on
line source
#!/usr/bin/env python3 # SAT plugin to detect language (experimental) # Copyright (C) 2009-2020 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 Optional import shortuuid from sat.core.i18n import _ from sat.core import exceptions from sat.core.constants import Const as C from sat.core.log import getLogger from sat.core.xmpp import SatXMPPEntity from sat.tools import utils from sat.tools.common import uri as xmpp_uri from sat.tools.common import date_utils from twisted.internet import defer from twisted.words.protocols.jabber import jid, error from twisted.words.xish import domish from wokkel import disco, iwokkel from zope.interface import implementer from twisted.words.protocols.jabber.xmlstream import XMPPHandler from wokkel import pubsub log = getLogger(__name__) PLUGIN_INFO = { C.PI_NAME: "Events", C.PI_IMPORT_NAME: "EVENTS", C.PI_TYPE: "EXP", C.PI_PROTOCOLS: [], C.PI_DEPENDENCIES: ["XEP-0060", "INVITATION", "PUBSUB_INVITATION", "LIST_INTEREST"], C.PI_RECOMMENDATIONS: ["XEP-0277", "EMAIL_INVITATION"], C.PI_MAIN: "Events", C.PI_HANDLER: "yes", C.PI_DESCRIPTION: _("""Experimental implementation of XMPP events management"""), } NS_EVENT = "org.salut-a-toi.event:0" class Events(object): """Q&D module to handle event attendance answer, experimentation only""" def __init__(self, host): log.info(_("Event plugin initialization")) self.host = host self._p = self.host.plugins["XEP-0060"] self._i = self.host.plugins.get("EMAIL_INVITATION") self._b = self.host.plugins.get("XEP-0277") self.host.registerNamespace("event", NS_EVENT) self.host.plugins["PUBSUB_INVITATION"].register(NS_EVENT, self) host.bridge.addMethod( "eventGet", ".plugin", in_sign="ssss", out_sign="(ia{ss})", method=self._eventGet, async_=True, ) host.bridge.addMethod( "eventCreate", ".plugin", in_sign="ia{ss}ssss", out_sign="s", method=self._eventCreate, async_=True, ) host.bridge.addMethod( "eventModify", ".plugin", in_sign="sssia{ss}s", out_sign="", method=self._eventModify, async_=True, ) host.bridge.addMethod( "eventsList", ".plugin", in_sign="sss", out_sign="aa{ss}", method=self._eventsList, async_=True, ) host.bridge.addMethod( "eventInviteeGet", ".plugin", in_sign="ssss", out_sign="a{ss}", method=self._eventInviteeGet, async_=True, ) host.bridge.addMethod( "eventInviteeSet", ".plugin", in_sign="ssa{ss}s", out_sign="", method=self._eventInviteeSet, async_=True, ) host.bridge.addMethod( "eventInviteesList", ".plugin", in_sign="sss", out_sign="a{sa{ss}}", method=self._eventInviteesList, async_=True, ), host.bridge.addMethod( "eventInvite", ".plugin", in_sign="sssss", out_sign="", method=self._invite, async_=True, ) host.bridge.addMethod( "eventInviteByEmail", ".plugin", in_sign="ssssassssssss", out_sign="", method=self._inviteByEmail, async_=True, ) def getHandler(self, client): return EventsHandler(self) def _parseEventElt(self, event_elt): """Helper method to parse event element @param (domish.Element): event_elt @return (tuple[int, dict[unicode, unicode]): timestamp, event_data """ try: timestamp = date_utils.date_parse(next(event_elt.elements(NS_EVENT, "date"))) except StopIteration: timestamp = -1 data = {} for key in ("name",): try: data[key] = event_elt[key] except KeyError: continue for elt_name in ("description",): try: elt = next(event_elt.elements(NS_EVENT, elt_name)) except StopIteration: continue else: data[elt_name] = str(elt) for elt_name in ("image", "background-image"): try: image_elt = next(event_elt.elements(NS_EVENT, elt_name)) data[elt_name] = image_elt["src"] except StopIteration: continue except KeyError: log.warning(_("no src found for image")) for uri_type in ("invitees", "blog"): try: elt = next(event_elt.elements(NS_EVENT, uri_type)) uri = data[uri_type + "_uri"] = elt["uri"] uri_data = xmpp_uri.parseXMPPUri(uri) if uri_data["type"] != "pubsub": raise ValueError except StopIteration: log.warning(_("no {uri_type} element found!").format(uri_type=uri_type)) except KeyError: log.warning(_("incomplete {uri_type} element").format(uri_type=uri_type)) except ValueError: log.warning(_("bad {uri_type} element").format(uri_type=uri_type)) else: data[uri_type + "_service"] = uri_data["path"] data[uri_type + "_node"] = uri_data["node"] for meta_elt in event_elt.elements(NS_EVENT, "meta"): key = meta_elt["name"] if key in data: log.warning( "Ignoring conflicting meta element: {xml}".format( xml=meta_elt.toXml() ) ) continue data[key] = str(meta_elt) if event_elt.link: link_elt = event_elt.link data["service"] = link_elt["service"] data["node"] = link_elt["node"] data["item"] = link_elt["item"] if event_elt.getAttribute("creator") == "true": data["creator"] = True return timestamp, data @defer.inlineCallbacks def getEventElement(self, client, service, node, id_): """Retrieve event element @param service(jid.JID): pubsub service @param node(unicode): pubsub node @param id_(unicode, None): event id @return (domish.Element): event element @raise exceptions.NotFound: no event element found """ if not id_: id_ = NS_EVENT items, metadata = yield self._p.getItems(client, service, node, item_ids=[id_]) try: event_elt = next(items[0].elements(NS_EVENT, "event")) except StopIteration: raise exceptions.NotFound(_("No event element has been found")) except IndexError: raise exceptions.NotFound(_("No event with this id has been found")) defer.returnValue(event_elt) def _eventGet(self, service, node, id_="", profile_key=C.PROF_KEY_NONE): service = jid.JID(service) if service else None node = node if node else NS_EVENT client = self.host.getClient(profile_key) return self.eventGet(client, service, node, id_) @defer.inlineCallbacks def eventGet(self, client, service, node, id_=NS_EVENT): """Retrieve event data @param service(unicode, None): PubSub service @param node(unicode): PubSub node of the event @param id_(unicode): id_ with even data @return (tuple[int, dict[unicode, unicode]): event data: - timestamp of the event - event metadata where key can be: location: location of the event image: URL of a picture to use to represent event background-image: URL of a picture to use in background """ event_elt = yield self.getEventElement(client, service, node, id_) defer.returnValue(self._parseEventElt(event_elt)) def _eventCreate( self, timestamp, data, service, node, id_="", profile_key=C.PROF_KEY_NONE ): service = jid.JID(service) if service else None node = node or None client = self.host.getClient(profile_key) data["register"] = C.bool(data.get("register", C.BOOL_FALSE)) return defer.ensureDeferred( self.eventCreate(client, timestamp, data, service, node, id_ or NS_EVENT) ) async def eventCreate(self, client, timestamp, data, service, node=None, event_id=NS_EVENT): """Create or replace an event @param service(jid.JID, None): PubSub service @param node(unicode, None): PubSub node of the event None will create instant node. @param event_id(unicode): ID of the item to create. @param timestamp(timestamp, None) @param data(dict[unicode, unicode]): data to update dict will be cleared, do a copy if data are still needed key can be: - name: name of the event - description: details - image: main picture of the event - background-image: image to use as background - register: bool, True if we want to register the event in our local list @return (unicode): created node """ if not event_id: raise ValueError(_("event_id must be set")) if not service: service = client.jid.userhostJID() if not node: node = NS_EVENT + "__" + shortuuid.uuid() event_elt = domish.Element((NS_EVENT, "event")) if timestamp is not None and timestamp != -1: formatted_date = utils.xmpp_date(timestamp) event_elt.addElement((NS_EVENT, "date"), content=formatted_date) register = data.pop("register", False) for key in ("name",): if key in data: event_elt[key] = data.pop(key) for key in ("description",): if key in data: event_elt.addElement((NS_EVENT, key), content=data.pop(key)) for key in ("image", "background-image"): if key in data: elt = event_elt.addElement((NS_EVENT, key)) elt["src"] = data.pop(key) # we first create the invitees and blog nodes (if not specified in data) for uri_type in ("invitees", "blog"): key = uri_type + "_uri" for to_delete in ("service", "node"): k = uri_type + "_" + to_delete if k in data: del data[k] if key not in data: # FIXME: affiliate invitees uri_node = await self._p.createNode(client, service) await self._p.setConfiguration( client, service, uri_node, {self._p.OPT_ACCESS_MODEL: self._p.ACCESS_WHITELIST}, ) uri_service = service else: uri = data.pop(key) uri_data = xmpp_uri.parseXMPPUri(uri) if uri_data["type"] != "pubsub": raise ValueError( _("The given URI is not valid: {uri}").format(uri=uri) ) uri_service = jid.JID(uri_data["path"]) uri_node = uri_data["node"] elt = event_elt.addElement((NS_EVENT, uri_type)) elt["uri"] = xmpp_uri.buildXMPPUri( "pubsub", path=uri_service.full(), node=uri_node ) # remaining data are put in <meta> elements for key in list(data.keys()): elt = event_elt.addElement((NS_EVENT, "meta"), content=data.pop(key)) elt["name"] = key item_elt = pubsub.Item(id=event_id, payload=event_elt) try: # TODO: check auto-create, no need to create node first if available node = await self._p.createNode(client, service, nodeIdentifier=node) except error.StanzaError as e: if e.condition == "conflict": log.debug(_("requested node already exists")) await self._p.publish(client, service, node, items=[item_elt]) if register: extra = {} self.onInvitationPreflight( client, "", extra, service, node, event_id, item_elt ) await self.host.plugins['LIST_INTEREST'].registerPubsub( client, NS_EVENT, service, node, event_id, True, extra.pop("name", ""), extra.pop("element"), extra ) return node def _eventModify(self, service, node, id_, timestamp_update, data_update, profile_key=C.PROF_KEY_NONE): service = jid.JID(service) if service else None if not node: raise ValueError(_("missing node")) client = self.host.getClient(profile_key) return defer.ensureDeferred( self.eventModify( client, service, node, id_ or NS_EVENT, timestamp_update or None, data_update ) ) async def eventModify( self, client, service, node, id_=NS_EVENT, timestamp_update=None, data_update=None ): """Update an event Similar as create instead that it update existing item instead of creating or replacing it. Params are the same as for [eventCreate]. """ event_timestamp, event_metadata = await self.eventGet(client, service, node, id_) new_timestamp = event_timestamp if timestamp_update is None else timestamp_update new_data = event_metadata if data_update: for k, v in data_update.items(): new_data[k] = v await self.eventCreate(client, new_timestamp, new_data, service, node, id_) def _eventsListSerialise(self, events): for timestamp, data in events: data["date"] = str(timestamp) data["creator"] = C.boolConst(data.get("creator", False)) return [e[1] for e in events] def _eventsList(self, service, node, profile): service = jid.JID(service) if service else None node = node or None client = self.host.getClient(profile) d = self.eventsList(client, service, node) d.addCallback(self._eventsListSerialise) return d @defer.inlineCallbacks def eventsList(self, client, service, node=None): """Retrieve list of registered events @return list(tuple(int, dict)): list of events (timestamp + metadata) """ items, metadata = yield self.host.plugins['LIST_INTEREST'].listInterests( client, service, node, namespace=NS_EVENT) events = [] for item in items: try: event_elt = next(item.interest.pubsub.elements(NS_EVENT, "event")) except StopIteration: log.warning( _("No event found in item {item_id}, ignoring").format( item_id=item["id"]) ) else: timestamp, data = self._parseEventElt(event_elt) events.append((timestamp, data)) defer.returnValue(events) def _eventInviteeGet(self, service, node, invitee_jid_s, profile_key): service = jid.JID(service) if service else None node = node if node else NS_EVENT client = self.host.getClient(profile_key) invitee_jid = jid.JID(invitee_jid_s) if invitee_jid_s else None return self.eventInviteeGet(client, service, node, invitee_jid) @defer.inlineCallbacks def eventInviteeGet(self, client, service, node, invitee_jid=None): """Retrieve attendance from event node @param service(unicode, None): PubSub service @param node(unicode): PubSub node of the event's invitees @param invitee_jid(jid.JId, None): jid of the invitee to retrieve (None to retrieve profile's invitation). The bare jid correspond to the PubSub item id. @return (dict): a dict with current attendance status, an empty dict is returned if nothing has been answered yed """ if invitee_jid is None: invitee_jid = client.jid try: items, metadata = yield self._p.getItems( client, service, node, item_ids=[invitee_jid.userhost()] ) event_elt = next(items[0].elements(NS_EVENT, "invitee")) except (exceptions.NotFound, IndexError): # no item found, event data are not set yet defer.returnValue({}) data = {} for key in ("attend", "guests"): try: data[key] = event_elt[key] except KeyError: continue defer.returnValue(data) def _eventInviteeSet(self, service, node, event_data, profile_key): service = jid.JID(service) if service else None node = node if node else NS_EVENT client = self.host.getClient(profile_key) return self.eventInviteeSet(client, service, node, event_data) def eventInviteeSet(self, client, service, node, data): """Set or update attendance data in event node @param service(unicode, None): PubSub service @param node(unicode): PubSub node of the event @param data(dict[unicode, unicode]): data to update key can be: attend: one of "yes", "no", "maybe" guests: an int """ event_elt = domish.Element((NS_EVENT, "invitee")) for key in ("attend", "guests"): try: event_elt[key] = data.pop(key) except KeyError: pass item_elt = pubsub.Item(id=client.jid.userhost(), payload=event_elt) return self._p.publish(client, service, node, items=[item_elt]) def _eventInviteesList(self, service, node, profile_key): service = jid.JID(service) if service else None node = node if node else NS_EVENT client = self.host.getClient(profile_key) return self.eventInviteesList(client, service, node) @defer.inlineCallbacks def eventInviteesList(self, client, service, node): """Retrieve attendance from event node @param service(unicode, None): PubSub service @param node(unicode): PubSub node of the event @return (dict): a dict with current attendance status, an empty dict is returned if nothing has been answered yed """ items, metadata = yield self._p.getItems(client, service, node) invitees = {} for item in items: try: event_elt = next(item.elements(NS_EVENT, "invitee")) except StopIteration: # no item found, event data are not set yet log.warning(_( "no data found for {item_id} (service: {service}, node: {node})" .format(item_id=item["id"], service=service, node=node))) else: data = {} for key in ("attend", "guests"): try: data[key] = event_elt[key] except KeyError: continue invitees[item["id"]] = data defer.returnValue(invitees) async def invitePreflight( self, client: SatXMPPEntity, invitee_jid: jid.JID, service: jid.JID, node: str, item_id: Optional[str] = None, name: str = '', extra: Optional[dict] = None, ) -> None: if self._b is None: raise exceptions.FeatureNotFound( _('"XEP-0277" (blog) plugin is needed for this feature') ) if item_id is None: item_id = extra["default_item_id"] = NS_EVENT __, event_data = await self.eventGet(client, service, node, item_id) log.debug(_("got event data")) invitees_service = jid.JID(event_data["invitees_service"]) invitees_node = event_data["invitees_node"] blog_service = jid.JID(event_data["blog_service"]) blog_node = event_data["blog_node"] await self._p.setNodeAffiliations( client, invitees_service, invitees_node, {invitee_jid: "publisher"} ) log.debug( f"affiliation set on invitee node (jid: {invitees_service}, " f"node: {invitees_node!r})" ) await self._p.setNodeAffiliations( client, blog_service, blog_node, {invitee_jid: "member"} ) blog_items, __ = await self._b.mbGet(client, blog_service, blog_node, None) for item in blog_items: try: comments_service = jid.JID(item["comments_service"]) comments_node = item["comments_node"] except KeyError: log.debug( "no comment service set for item {item_id}".format( item_id=item["id"] ) ) else: await self._p.setNodeAffiliations( client, comments_service, comments_node, {invitee_jid: "publisher"} ) log.debug(_("affiliation set on blog and comments nodes")) def _invite(self, invitee_jid, service, node, item_id, profile): return self.host.plugins["PUBSUB_INVITATION"]._sendPubsubInvitation( invitee_jid, service, node, item_id or NS_EVENT, profile_key=profile ) def _inviteByEmail(self, service, node, id_=NS_EVENT, email="", emails_extra=None, name="", host_name="", language="", url_template="", message_subject="", message_body="", profile_key=C.PROF_KEY_NONE): client = self.host.getClient(profile_key) kwargs = { "profile": client.profile, "emails_extra": [str(e) for e in emails_extra], } for key in ( "email", "name", "host_name", "language", "url_template", "message_subject", "message_body", ): value = locals()[key] kwargs[key] = str(value) return defer.ensureDeferred(self.inviteByEmail( client, jid.JID(service) if service else None, node, id_ or NS_EVENT, **kwargs )) async def inviteByEmail(self, client, service, node, id_=NS_EVENT, **kwargs): """High level method to create an email invitation to an event @param service(unicode, None): PubSub service @param node(unicode): PubSub node of the event @param id_(unicode): id_ with even data """ if self._i is None: raise exceptions.FeatureNotFound( _('"Invitations" plugin is needed for this feature') ) if self._b is None: raise exceptions.FeatureNotFound( _('"XEP-0277" (blog) plugin is needed for this feature') ) service = service or client.jid.userhostJID() event_uri = xmpp_uri.buildXMPPUri( "pubsub", path=service.full(), node=node, item=id_ ) kwargs["extra"] = {"event_uri": event_uri} invitation_data = await self._i.create(**kwargs) invitee_jid = invitation_data["jid"] log.debug(_("invitation created")) # now that we have a jid, we can send normal invitation await self.invite(client, invitee_jid, service, node, id_) def onInvitationPreflight( self, client: SatXMPPEntity, name: str, extra: dict, service: jid.JID, node: str, item_id: Optional[str], item_elt: domish.Element ) -> None: event_elt = item_elt.event link_elt = event_elt.addElement("link") link_elt["service"] = service.full() link_elt["node"] = node link_elt["item"] = item_id __, event_data = self._parseEventElt(event_elt) try: name = event_data["name"] except KeyError: pass else: extra["name"] = name if 'image' in event_data: extra["thumb_url"] = event_data['image'] extra["element"] = event_elt @implementer(iwokkel.IDisco) class EventsHandler(XMPPHandler): def __init__(self, plugin_parent): self.plugin_parent = plugin_parent def getDiscoInfo(self, requestor, target, nodeIdentifier=""): return [ disco.DiscoFeature(NS_EVENT), ] def getDiscoItems(self, requestor, target, nodeIdentifier=""): return []