From 3b00c56bdc85db189029bd30e14e0d53e15ff497 Mon Sep 17 00:00:00 2001 From: mathieui Date: Sat, 12 Dec 2020 14:43:18 +0100 Subject: XEP-0424: Add an event for message retraction + update the integration test --- slixmpp/plugins/xep_0424/retraction.py | 11 +++++++++++ 1 file changed, 11 insertions(+) (limited to 'slixmpp/plugins/xep_0424/retraction.py') diff --git a/slixmpp/plugins/xep_0424/retraction.py b/slixmpp/plugins/xep_0424/retraction.py index d3461d66..09e3adee 100644 --- a/slixmpp/plugins/xep_0424/retraction.py +++ b/slixmpp/plugins/xep_0424/retraction.py @@ -9,6 +9,8 @@ from typing import Optional from slixmpp import JID, Message from slixmpp.exceptions import IqError, IqTimeout +from slixmpp.xmlstream.matcher import StanzaPath +from slixmpp.xmlstream.handler import Callback from slixmpp.plugins import BasePlugin from slixmpp.plugins.xep_0424 import stanza @@ -30,6 +32,11 @@ class XEP_0424(BasePlugin): def plugin_init(self) -> None: stanza.register_plugins() + self.xmpp.register_handler(Callback( + "Message Retracted", + StanzaPath("message/apply_to/retract"), + self._handle_retract_message, + )) def session_bind(self, jid): self.xmpp.plugin['xep_0030'].add_feature(feature=stanza.NS) @@ -37,6 +44,10 @@ class XEP_0424(BasePlugin): def plugin_end(self): self.xmpp.plugin['xep_0030'].del_feature(feature=stanza.NS) + def _handle_retract_message(self, message: Message): + if message['type'] != 'groupchat': + self.xmpp.event('message_retract', message) + def send_retraction(self, mto: JID, id: str, mtype: str = 'chat', include_fallback: bool = True, fallback_text: Optional[str] = None, *, -- cgit v1.2.3