pep/slixmpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

XEP-0115: add a broadcast parameter to update_caps

Browse source 
and do not send a presence after updating if it is false
This commit is contained in:
mathieui 2021-04-19 23:17:10 +02:00
parent 0d52344a31
commit 8828a5b99d
1 changed files with 15 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
slixmpp/plugins/xep_0115/caps.py
View file

@ -8,6 +8,7 @@ import hashlib
import base64 import base64
from asyncio import Future from asyncio import Future
from typing import Optional
from slixmpp import __version__ from slixmpp import __version__
from slixmpp.stanza import StreamFeatures, Presence, Iq from slixmpp.stanza import StreamFeatures, Presence, Iq
@ -15,10 +16,10 @@ from slixmpp.xmlstream import register_stanza_plugin, JID
from slixmpp.xmlstream.handler import Callback from slixmpp.xmlstream.handler import Callback
from slixmpp.xmlstream.matcher import StanzaPath from slixmpp.xmlstream.matcher import StanzaPath
from slixmpp.util import MemoryCache from slixmpp.util import MemoryCache
from slixmpp import asyncio from slixmpp.exceptions import XMPPError
from slixmpp.exceptions import XMPPError, IqError, IqTimeout
from slixmpp.plugins import BasePlugin from slixmpp.plugins import BasePlugin
from slixmpp.plugins.xep_0115 import stanza, StaticCaps from slixmpp.plugins.xep_0115 import stanza, StaticCaps
from slixmpp.types import OptJidStr
log = logging.getLogger(__name__) log = logging.getLogger(__name__)
@ -285,7 +286,17 @@ class XEP_0115(BasePlugin):
binary = hash(S.encode('utf8')).digest() binary = hash(S.encode('utf8')).digest()
return base64.b64encode(binary).decode('utf-8') return base64.b64encode(binary).decode('utf-8')
async def update_caps(self, jid=None, node=None, preserve=False): async def update_caps(self, jid: OptJidStr = None,
node: Optional[str] = None,
preserve: bool = False,
broadcast: bool = True):
"""Update caps for a local JID based on current data.
:param jid: JID whose info to update
:param node: Node to fetch info from
:param broadcast: Send a presence after updating.
:param preserve: Send presence only to contacts found in the roster.
"""
try: try:
info = await self.xmpp['xep_0030'].get_info(jid, node, local=True) info = await self.xmpp['xep_0030'].get_info(jid, node, local=True)
if isinstance(info, Iq): if isinstance(info, Iq):
@ -299,7 +310,7 @@ class XEP_0115(BasePlugin):
await self.cache_caps(ver, info) await self.cache_caps(ver, info)
await self.assign_verstring(jid, ver) await self.assign_verstring(jid, ver)
if self.xmpp.sessionstarted and self.broadcast: if broadcast and self.xmpp.sessionstarted and self.broadcast:
if self.xmpp.is_component or preserve: if self.xmpp.is_component or preserve:
for contact in self.xmpp.roster[jid]: for contact in self.xmpp.roster[jid]:
self.xmpp.roster[jid][contact].send_last_presence() self.xmpp.roster[jid][contact].send_last_presence()