85 lines
2.3 KiB
Python
85 lines
2.3 KiB
Python
"""
|
|
Module used to provide fixes for slixmpp functions not yet fixed
|
|
upstream.
|
|
|
|
TODO: Check that they are fixed and remove those hacks
|
|
"""
|
|
|
|
from slixmpp.stanza import Message
|
|
from slixmpp.xmlstream import ET
|
|
|
|
import logging
|
|
|
|
log = logging.getLogger(__name__)
|
|
|
|
|
|
def has_identity(xmpp, jid, identity, on_true=None, on_false=None):
|
|
def _cb(iq):
|
|
ident = lambda x: x[0]
|
|
res = identity in map(ident, iq['disco_info']['identities'])
|
|
if res and on_true is not None:
|
|
on_true()
|
|
if not res and on_false is not None:
|
|
on_false()
|
|
|
|
xmpp.plugin['xep_0030'].get_info(jid=jid, callback=_cb)
|
|
|
|
|
|
def get_room_form(xmpp, room, callback):
|
|
def _cb(result):
|
|
if result["type"] == "error":
|
|
return callback(None)
|
|
xform = result.xml.find(
|
|
'{http://jabber.org/protocol/muc#owner}query/{jabber:x:data}x')
|
|
if xform is None:
|
|
return callback(None)
|
|
form = xmpp.plugin['xep_0004'].build_form(xform)
|
|
return callback(form)
|
|
|
|
iq = xmpp.make_iq_get(ito=room)
|
|
query = ET.Element('{http://jabber.org/protocol/muc#owner}query')
|
|
iq.append(query)
|
|
iq.send(callback=_cb)
|
|
|
|
|
|
def _filter_add_receipt_request(self, stanza):
|
|
"""
|
|
Auto add receipt requests to outgoing messages, if:
|
|
|
|
- ``self.auto_request`` is set to ``True``
|
|
- The message is not for groupchat
|
|
- The message does not contain a receipt acknowledgment
|
|
- The recipient is a bare JID or, if a full JID, one
|
|
that has the ``urn:xmpp:receipts`` feature enabled
|
|
- The message has a body
|
|
|
|
The disco cache is checked if a full JID is specified in
|
|
the outgoing message, which may mean a round-trip disco#info
|
|
delay for the first message sent to the JID if entity caps
|
|
are not used.
|
|
"""
|
|
|
|
if not self.auto_request:
|
|
return stanza
|
|
|
|
if not isinstance(stanza, Message):
|
|
return stanza
|
|
|
|
if stanza['request_receipt']:
|
|
return stanza
|
|
|
|
if stanza['type'] not in self.ack_types:
|
|
return stanza
|
|
|
|
if stanza['receipt']:
|
|
return stanza
|
|
|
|
if not stanza['body']:
|
|
return stanza
|
|
|
|
# hack
|
|
if stanza['to'].resource and not hasattr(stanza, '_add_receipt'):
|
|
return stanza
|
|
|
|
stanza['request_receipt'] = True
|
|
return stanza
|