XEP-0439: Add events

This commit is contained in:
mathieui 2020-12-07 20:58:16 +01:00
parent a61f2248fd
commit 91c3d64ca2

View file

@ -11,9 +11,12 @@ from typing import (
Tuple, Tuple,
) )
from slixmpp import JID from slixmpp import JID, Message
from slixmpp.plugins import BasePlugin from slixmpp.plugins import BasePlugin
from slixmpp.plugins.xep_0439 import stanza from slixmpp.plugins.xep_0439 import stanza
from slixmpp.xmlstream.matcher import StanzaPath
from slixmpp.xmlstream.handler import Callback
class XEP_0439(BasePlugin): class XEP_0439(BasePlugin):
@ -27,11 +30,40 @@ class XEP_0439(BasePlugin):
def plugin_init(self) -> None: def plugin_init(self) -> None:
stanza.register_plugins() stanza.register_plugins()
self.xmpp.register_handler(Callback(
'Action received',
StanzaPath('message/action'),
self._handle_action,
))
self.xmpp.register_handler(Callback(
'Response received',
StanzaPath('message/response'),
self._handle_response,
))
self.xmpp.register_handler(Callback(
'ActionSelected received',
StanzaPath('message/action_selected'),
self._handle_action_selected,
))
def ask_for_responses(self, mto: JID, body: str, def plugin_send(self):
responses: Iterable[Tuple[str, str]], self.xmpp.remove_handler('Action received')
mtype: str = 'chat', lang: Optional[str] = None, *, self.xmpp.remove_handler('Response received')
mfrom: Optional[JID] = None): self.xmpp.remove_handler('ActionSelected received')
def _handle_response(self, msg: Message):
self.xmpp.event('responses_received', msg)
def _handle_action(self, msg: Message):
self.xmpp.event('action_received', msg)
def _handle_action_selected(self, msg: Message):
self.xmpp.event('action_selected', msg)
def ask_for_response(self, mto: JID, body: str,
responses: Iterable[Tuple[str, str]],
mtype: str = 'chat', lang: Optional[str] = None, *,
mfrom: Optional[JID] = None):
""" """
Send a message with a set of responses. Send a message with a set of responses.