diff options
author | Maxime “pep” Buquet <pep@bouah.net> | 2018-06-09 15:55:12 +0100 |
---|---|---|
committer | Maxime “pep” Buquet <pep@bouah.net> | 2018-07-01 01:29:31 +0100 |
commit | 037706552c68ec900475633af0474a7cdac29814 (patch) | |
tree | 5beb821c162e652f20f86cbe53f1b58fcfda1026 /slixmpp/plugins/xep_0030 | |
parent | b881c6729b79e37aba5ab2d99c3ba0fd79bc1b76 (diff) | |
download | slixmpp-037706552c68ec900475633af0474a7cdac29814.tar.gz slixmpp-037706552c68ec900475633af0474a7cdac29814.tar.bz2 slixmpp-037706552c68ec900475633af0474a7cdac29814.tar.xz slixmpp-037706552c68ec900475633af0474a7cdac29814.zip |
Factor out fetching of identities in xep_0363 to xep_0030
Signed-off-by: Maxime “pep” Buquet <pep@bouah.net>
Diffstat (limited to 'slixmpp/plugins/xep_0030')
-rw-r--r-- | slixmpp/plugins/xep_0030/disco.py | 27 |
1 files changed, 27 insertions, 0 deletions
diff --git a/slixmpp/plugins/xep_0030/disco.py b/slixmpp/plugins/xep_0030/disco.py index b1c261a9..daeee879 100644 --- a/slixmpp/plugins/xep_0030/disco.py +++ b/slixmpp/plugins/xep_0030/disco.py @@ -124,6 +124,8 @@ class XEP_0030(BasePlugin): for op in self._disco_ops: self.api.register(getattr(self.static, op), op, default=True) + self.domain_infos = {} + def session_bind(self, jid): self.add_feature('http://jabber.org/protocol/disco#info') @@ -296,6 +298,31 @@ class XEP_0030(BasePlugin): 'cached': cached} return self.api['has_identity'](jid, node, ifrom, data) + async def find_identities(category, type_, domain=None, timeout=None): + if domain is None: + domain = self.xmpp.boundjid.domain + + if domain not in self.domain_infos: + infos = [self.xmpp['xep_0030'].get_info( + domain, timeout=timeout)] + iq_items = await self.xmpp['xep_0030'].get_items( + domain, timeout=timeout) + items = iq_items['disco_items']['items'] + infos += [ + self.xmpp['xep_0030'].get_info(item[0], timeout=timeout) + for item in items] + info_futures, _ = await asyncio.wait(infos, timeout=timeout) + + self.domain_infos[domain] = [ + future.result() for future in info_futures] + + results = [] + for info in self.domain_infos[domain]: + for identity in info['disco_info']['identities']: + if identity[0] == category and identity[1] == type_: + results.append(info) + return results + @future_wrapper def get_info(self, jid=None, node=None, local=None, cached=None, **kwargs): |