Mercurial > libervia-backend
diff libervia/backend/plugins/plugin_xep_0163.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_0163.py@524856bd7b19 |
children | 0d7bb4df2343 |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/libervia/backend/plugins/plugin_xep_0163.py Fri Jun 02 11:49:51 2023 +0200 @@ -0,0 +1,203 @@ +#!/usr/bin/env python3 + + +# SAT plugin for Personal Eventing Protocol (xep-0163) +# 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 typing import Optional, Callable +from libervia.backend.core.i18n import _ +from libervia.backend.core import exceptions +from libervia.backend.core.constants import Const as C +from libervia.backend.core.log import getLogger + +from twisted.words.xish import domish + +from wokkel import disco, pubsub +from wokkel.formats import Mood +from libervia.backend.tools.common import data_format + + +log = getLogger(__name__) + +NS_USER_MOOD = "http://jabber.org/protocol/mood" + +PLUGIN_INFO = { + C.PI_NAME: "Personal Eventing Protocol Plugin", + C.PI_IMPORT_NAME: "XEP-0163", + C.PI_TYPE: "XEP", + C.PI_MODES: C.PLUG_MODE_BOTH, + C.PI_PROTOCOLS: ["XEP-0163", "XEP-0107"], + C.PI_DEPENDENCIES: ["XEP-0060"], + C.PI_MAIN: "XEP_0163", + C.PI_HANDLER: "no", + C.PI_DESCRIPTION: _("""Implementation of Personal Eventing Protocol"""), +} + + +class XEP_0163(object): + def __init__(self, host): + log.info(_("PEP plugin initialization")) + self.host = host + self.pep_events = set() + self.pep_out_cb = {} + host.trigger.add("PubSub Disco Info", self.diso_info_trigger) + host.bridge.add_method( + "pep_send", + ".plugin", + in_sign="sa{ss}s", + out_sign="", + method=self.pep_send, + async_=True, + ) # args: type(MOOD, TUNE, etc), data, profile_key; + self.add_pep_event("MOOD", NS_USER_MOOD, self.user_mood_cb, self.send_mood) + + def diso_info_trigger(self, disco_info, profile): + """Add info from managed PEP + + @param disco_info: list of disco feature as returned by PubSub, + will be filled with PEP features + @param profile: profile we are handling + """ + disco_info.extend(list(map(disco.DiscoFeature, self.pep_events))) + return True + + def add_pep_event( + self, + event_type: Optional[str], + node: str, + in_callback: Callable, + out_callback: Optional[Callable] = None, + notify: bool = True + ) -> None: + """Add a Personal Eventing Protocol event manager + + @param event_type: type of the event (stored uppercase), + only used when out_callback is set. + Can be MOOD, TUNE, etc. + @param node: namespace of the node (e.g. http://jabber.org/protocol/mood + for User Mood) + @param in_callback: method to call when this event occur + the callable will be called with (itemsEvent, profile) as arguments + @param out_callback: method to call when we want to publish this + event (must return a deferred) + the callable will be called when send_pep_event is called + @param notify: add autosubscribe (+notify) if True + """ + if event_type and out_callback: + event_type = event_type.upper() + if event_type in self.pep_out_cb: + raise exceptions.ConflictError( + f"event_type {event_type!r} already exists" + ) + self.pep_out_cb[event_type] = out_callback + self.pep_events.add(node) + if notify: + self.pep_events.add(node + "+notify") + + def filter_pep_event(client, itemsEvent): + """Ignore messages which are not coming from PEP (i.e. a bare jid) + + @param itemsEvent(pubsub.ItemsEvent): pubsub event + """ + if not itemsEvent.sender.user or itemsEvent.sender.resource: + log.debug( + "ignoring non PEP event from {} (profile={})".format( + itemsEvent.sender.full(), client.profile + ) + ) + return + in_callback(itemsEvent, client.profile) + + self.host.plugins["XEP-0060"].add_managed_node(node, items_cb=filter_pep_event) + + def send_pep_event(self, node, data, profile): + """Publish the event data + + @param node(unicode): node namespace + @param data: domish.Element to use as payload + @param profile: profile which send the data + """ + client = self.host.get_client(profile) + item = pubsub.Item(payload=data) + return self.host.plugins["XEP-0060"].publish(client, None, node, [item]) + + def pep_send(self, event_type, data, profile_key=C.PROF_KEY_NONE): + """Send personal event after checking the data is alright + + @param event_type: type of event (eg: MOOD, TUNE), + must be in self.pep_out_cb.keys() + @param data: dict of {string:string} of event_type dependant data + @param profile_key: profile who send the event + """ + profile = self.host.memory.get_profile_name(profile_key) + if not profile: + log.error( + _("Trying to send personal event with an unknown profile key [%s]") + % profile_key + ) + raise exceptions.ProfileUnknownError + if not event_type in list(self.pep_out_cb.keys()): + log.error(_("Trying to send personal event for an unknown type")) + raise exceptions.DataError("Type unknown") + return self.pep_out_cb[event_type](data, profile) + + def user_mood_cb(self, itemsEvent, profile): + if not itemsEvent.items: + log.debug(_("No item found")) + return + try: + mood_elt = [ + child for child in itemsEvent.items[0].elements() if child.name == "mood" + ][0] + except IndexError: + log.error(_("Can't find mood element in mood event")) + return + mood = Mood.fromXml(mood_elt) + if not mood: + log.debug(_("No mood found")) + return + self.host.bridge.ps_event( + C.PS_PEP, + itemsEvent.sender.full(), + itemsEvent.nodeIdentifier, + "MOOD", + data_format.serialise({"mood": mood.value or "", "text": mood.text or ""}), + profile, + ) + + def send_mood(self, data, profile): + """Send XEP-0107's User Mood + + @param data: must include mood and text + @param profile: profile which send the mood""" + try: + value = data["mood"].lower() + text = data["text"] if "text" in data else "" + except KeyError: + raise exceptions.DataError("Mood data must contain at least 'mood' key") + mood = UserMood(value, text) + return self.send_pep_event(NS_USER_MOOD, mood, profile) + + +class UserMood(Mood, domish.Element): + """Improved wokkel Mood which is also a domish.Element""" + + def __init__(self, value, text=None): + Mood.__init__(self, value, text) + domish.Element.__init__(self, (NS_USER_MOOD, "mood")) + self.addElement(value) + if text: + self.addElement("text", content=text)