poezio/plugins/disco.py
2017-12-19 15:37:42 +01:00

57 lines
2 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')
for field in info['form']['fields']:
if field['type'] == 'hidden' and field['var'] == 'FORM_TYPE':
value = field['value']
if 'http://jabber.org/network/serverinfo' not in value:
self.api.error('Unknown form type “%s' % value, 'Form')
return
break
server_info = []
for field in info['form']:
var = field['var']
if field['type'] == 'hidden' and var == 'FORM_TYPE':
continue
sep = '\n ' + len(var) * ' '
value = sep.join(field.get_value(convert=False))
server_info.append('%s: %s' % (var, value))
if server_info:
self.api.information('\n'.join(server_info), 'Server Info')
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')