poezio/plugins/disco.py
2018-05-30 21:04:02 +02:00

53 lines
1.8 KiB
Python

"""
Do a disco#info query on a JID
Usage
-----
.. glossary::
/disco
**Usage:** ``/disco <JID>``
This command queries a JID for its disco#info.
There is no cache, as this is generally used for debug more than
anything user-related.
"""
from poezio.plugin import BasePlugin
from slixmpp.jid import InvalidJID
class Plugin(BasePlugin):
def init(self):
self.api.add_command('disco', self.command_disco,
usage='<JID>',
short='Get the disco#info of a JID',
help='Get the disco#info of a JID')
def on_disco(self, iq):
info = iq['disco_info']
identities = (str(identity) for identity in info['identities'])
self.api.information('\n'.join(identities), 'Identities')
features = sorted(str(feature) for feature in info['features'])
self.api.information('\n'.join(features), 'Features')
title = 'Server Info'
server_info = []
for field in info['form']:
var = field['var']
if field['type'] == 'hidden' and var == 'FORM_TYPE':
title = field['value'][0]
continue
sep = '\n ' + len(var) * ' '
field_value = field.get_value(convert=False)
value = sep.join(field_value) if isinstance(field_value, list) else field_value
server_info.append('%s: %s' % (var, value))
if server_info:
self.api.information('\n'.join(server_info), title)
def command_disco(self, jid):
try:
self.core.xmpp.plugin['xep_0030'].get_info(jid=jid, cached=False,
callback=self.on_disco)
except InvalidJID as e:
self.api.information('Invalid JID “%s”: %s' % (jid, e), 'Error')