740 lines
29 KiB
Python
740 lines
29 KiB
Python
"""
|
|
SleekXMPP: The Sleek XMPP Library
|
|
Copyright (C) 2010 Nathanael C. Fritz, Lance J.T. Stout
|
|
This file is part of SleekXMPP.
|
|
|
|
See the file LICENSE for copying permission.
|
|
"""
|
|
|
|
import logging
|
|
|
|
from sleekxmpp import Iq
|
|
from sleekxmpp.plugins import BasePlugin
|
|
from sleekxmpp.xmlstream.handler import Callback
|
|
from sleekxmpp.xmlstream.matcher import StanzaPath
|
|
from sleekxmpp.xmlstream import register_stanza_plugin, JID
|
|
from sleekxmpp.plugins.xep_0030 import stanza, DiscoInfo, DiscoItems
|
|
from sleekxmpp.plugins.xep_0030 import StaticDisco
|
|
|
|
|
|
log = logging.getLogger(__name__)
|
|
|
|
|
|
class XEP_0030(BasePlugin):
|
|
|
|
"""
|
|
XEP-0030: Service Discovery
|
|
|
|
Service discovery in XMPP allows entities to discover information about
|
|
other agents in the network, such as the feature sets supported by a
|
|
client, or signposts to other, related entities.
|
|
|
|
Also see <http://www.xmpp.org/extensions/xep-0030.html>.
|
|
|
|
The XEP-0030 plugin works using a hierarchy of dynamic
|
|
node handlers, ranging from global handlers to specific
|
|
JID+node handlers. The default set of handlers operate
|
|
in a static manner, storing disco information in memory.
|
|
However, custom handlers may use any available backend
|
|
storage mechanism desired, such as SQLite or Redis.
|
|
|
|
Node handler hierarchy:
|
|
JID | Node | Level
|
|
---------------------
|
|
None | None | Global
|
|
Given | None | All nodes for the JID
|
|
None | Given | Node on self.xmpp.boundjid
|
|
Given | Given | A single node
|
|
|
|
Stream Handlers:
|
|
Disco Info -- Any Iq stanze that includes a query with the
|
|
namespace http://jabber.org/protocol/disco#info.
|
|
Disco Items -- Any Iq stanze that includes a query with the
|
|
namespace http://jabber.org/protocol/disco#items.
|
|
|
|
Events:
|
|
disco_info -- Received a disco#info Iq query result.
|
|
disco_items -- Received a disco#items Iq query result.
|
|
disco_info_query -- Received a disco#info Iq query request.
|
|
disco_items_query -- Received a disco#items Iq query request.
|
|
|
|
Attributes:
|
|
stanza -- A reference to the module containing the
|
|
stanza classes provided by this plugin.
|
|
static -- Object containing the default set of
|
|
static node handlers.
|
|
default_handlers -- A dictionary mapping operations to the default
|
|
global handler (by default, the static handlers).
|
|
xmpp -- The main SleekXMPP object.
|
|
|
|
Methods:
|
|
set_node_handler -- Assign a handler to a JID/node combination.
|
|
del_node_handler -- Remove a handler from a JID/node combination.
|
|
get_info -- Retrieve disco#info data, locally or remote.
|
|
get_items -- Retrieve disco#items data, locally or remote.
|
|
set_identities --
|
|
set_features --
|
|
set_items --
|
|
del_items --
|
|
del_identity --
|
|
del_feature --
|
|
del_item --
|
|
add_identity --
|
|
add_feature --
|
|
add_item --
|
|
"""
|
|
|
|
name = 'xep_0030'
|
|
description = 'XEP-0030: Service Discovery'
|
|
dependencies = set()
|
|
stanza = stanza
|
|
|
|
def plugin_init(self):
|
|
"""
|
|
Start the XEP-0030 plugin.
|
|
"""
|
|
self.xmpp.register_handler(
|
|
Callback('Disco Info',
|
|
StanzaPath('iq/disco_info'),
|
|
self._handle_disco_info))
|
|
|
|
self.xmpp.register_handler(
|
|
Callback('Disco Items',
|
|
StanzaPath('iq/disco_items'),
|
|
self._handle_disco_items))
|
|
|
|
register_stanza_plugin(Iq, DiscoInfo)
|
|
register_stanza_plugin(Iq, DiscoItems)
|
|
|
|
self.static = StaticDisco(self.xmpp, self)
|
|
|
|
self.use_cache = self.config.get('use_cache', True)
|
|
self.wrap_results = self.config.get('wrap_results', False)
|
|
|
|
self._disco_ops = [
|
|
'get_info', 'set_info', 'set_identities', 'set_features',
|
|
'get_items', 'set_items', 'del_items', 'add_identity',
|
|
'del_identity', 'add_feature', 'del_feature', 'add_item',
|
|
'del_item', 'del_identities', 'del_features', 'cache_info',
|
|
'get_cached_info', 'supports', 'has_identity']
|
|
|
|
for op in self._disco_ops:
|
|
self.api.register(getattr(self.static, op), op, default=True)
|
|
|
|
def _add_disco_op(self, op, default_handler):
|
|
self.api.register(default_handler, op)
|
|
self.api.register_default(default_handler, op)
|
|
|
|
def set_node_handler(self, htype, jid=None, node=None, handler=None):
|
|
"""
|
|
Add a node handler for the given hierarchy level and
|
|
handler type.
|
|
|
|
Node handlers are ordered in a hierarchy where the
|
|
most specific handler is executed. Thus, a fallback,
|
|
global handler can be used for the majority of cases
|
|
with a few node specific handler that override the
|
|
global behavior.
|
|
|
|
Node handler hierarchy:
|
|
JID | Node | Level
|
|
---------------------
|
|
None | None | Global
|
|
Given | None | All nodes for the JID
|
|
None | Given | Node on self.xmpp.boundjid
|
|
Given | Given | A single node
|
|
|
|
Handler types:
|
|
get_info
|
|
get_items
|
|
set_identities
|
|
set_features
|
|
set_items
|
|
del_items
|
|
del_identities
|
|
del_identity
|
|
del_feature
|
|
del_features
|
|
del_item
|
|
add_identity
|
|
add_feature
|
|
add_item
|
|
|
|
Arguments:
|
|
htype -- The operation provided by the handler.
|
|
jid -- The JID the handler applies to. May be narrowed
|
|
further if a node is given.
|
|
node -- The particular node the handler is for. If no JID
|
|
is given, then the self.xmpp.boundjid.full is
|
|
assumed.
|
|
handler -- The handler function to use.
|
|
"""
|
|
self.api.register(handler, htype, jid, node)
|
|
|
|
def del_node_handler(self, htype, jid, node):
|
|
"""
|
|
Remove a handler type for a JID and node combination.
|
|
|
|
The next handler in the hierarchy will be used if one
|
|
exists. If removing the global handler, make sure that
|
|
other handlers exist to process existing nodes.
|
|
|
|
Node handler hierarchy:
|
|
JID | Node | Level
|
|
---------------------
|
|
None | None | Global
|
|
Given | None | All nodes for the JID
|
|
None | Given | Node on self.xmpp.boundjid
|
|
Given | Given | A single node
|
|
|
|
Arguments:
|
|
htype -- The type of handler to remove.
|
|
jid -- The JID from which to remove the handler.
|
|
node -- The node from which to remove the handler.
|
|
"""
|
|
self.api.unregister(htype, jid, node)
|
|
|
|
def restore_defaults(self, jid=None, node=None, handlers=None):
|
|
"""
|
|
Change all or some of a node's handlers to the default
|
|
handlers. Useful for manually overriding the contents
|
|
of a node that would otherwise be handled by a JID level
|
|
or global level dynamic handler.
|
|
|
|
The default is to use the built-in static handlers, but that
|
|
may be changed by modifying self.default_handlers.
|
|
|
|
Arguments:
|
|
jid -- The JID owning the node to modify.
|
|
node -- The node to change to using static handlers.
|
|
handlers -- Optional list of handlers to change to the
|
|
default version. If provided, only these
|
|
handlers will be changed. Otherwise, all
|
|
handlers will use the default version.
|
|
"""
|
|
if handlers is None:
|
|
handlers = self._disco_ops
|
|
for op in handlers:
|
|
self.api.restore_default(op, jid, node)
|
|
|
|
def supports(self, jid=None, node=None, feature=None, local=False,
|
|
cached=True, ifrom=None):
|
|
"""
|
|
Check if a JID supports a given feature.
|
|
|
|
Return values:
|
|
True -- The feature is supported
|
|
False -- The feature is not listed as supported
|
|
None -- Nothing could be found due to a timeout
|
|
|
|
Arguments:
|
|
jid -- Request info from this JID.
|
|
node -- The particular node to query.
|
|
feature -- The name of the feature to check.
|
|
local -- If true, then the query is for a JID/node
|
|
combination handled by this Sleek instance and
|
|
no stanzas need to be sent.
|
|
Otherwise, a disco stanza must be sent to the
|
|
remove JID to retrieve the info.
|
|
cached -- If true, then look for the disco info data from
|
|
the local cache system. If no results are found,
|
|
send the query as usual. The self.use_cache
|
|
setting must be set to true for this option to
|
|
be useful. If set to false, then the cache will
|
|
be skipped, even if a result has already been
|
|
cached. Defaults to false.
|
|
ifrom -- Specifiy the sender's JID.
|
|
"""
|
|
data = {'feature': feature,
|
|
'local': local,
|
|
'cached': cached}
|
|
return self.api['supports'](jid, node, ifrom, data)
|
|
|
|
def has_identity(self, jid=None, node=None, category=None, itype=None,
|
|
lang=None, local=False, cached=True, ifrom=None):
|
|
"""
|
|
Check if a JID provides a given identity.
|
|
|
|
Return values:
|
|
True -- The identity is provided
|
|
False -- The identity is not listed
|
|
None -- Nothing could be found due to a timeout
|
|
|
|
Arguments:
|
|
jid -- Request info from this JID.
|
|
node -- The particular node to query.
|
|
category -- The category of the identity to check.
|
|
itype -- The type of the identity to check.
|
|
lang -- The language of the identity to check.
|
|
local -- If true, then the query is for a JID/node
|
|
combination handled by this Sleek instance and
|
|
no stanzas need to be sent.
|
|
Otherwise, a disco stanza must be sent to the
|
|
remove JID to retrieve the info.
|
|
cached -- If true, then look for the disco info data from
|
|
the local cache system. If no results are found,
|
|
send the query as usual. The self.use_cache
|
|
setting must be set to true for this option to
|
|
be useful. If set to false, then the cache will
|
|
be skipped, even if a result has already been
|
|
cached. Defaults to false.
|
|
ifrom -- Specifiy the sender's JID.
|
|
"""
|
|
data = {'category': category,
|
|
'itype': itype,
|
|
'lang': lang,
|
|
'local': local,
|
|
'cached': cached}
|
|
return self.api['has_identity'](jid, node, ifrom, data)
|
|
|
|
def get_info(self, jid=None, node=None, local=False,
|
|
cached=None, **kwargs):
|
|
"""
|
|
Retrieve the disco#info results from a given JID/node combination.
|
|
|
|
Info may be retrieved from both local resources and remote agents;
|
|
the local parameter indicates if the information should be gathered
|
|
by executing the local node handlers, or if a disco#info stanza
|
|
must be generated and sent.
|
|
|
|
If requesting items from a local JID/node, then only a DiscoInfo
|
|
stanza will be returned. Otherwise, an Iq stanza will be returned.
|
|
|
|
Arguments:
|
|
jid -- Request info from this JID.
|
|
node -- The particular node to query.
|
|
local -- If true, then the query is for a JID/node
|
|
combination handled by this Sleek instance and
|
|
no stanzas need to be sent.
|
|
Otherwise, a disco stanza must be sent to the
|
|
remove JID to retrieve the info.
|
|
cached -- If true, then look for the disco info data from
|
|
the local cache system. If no results are found,
|
|
send the query as usual. The self.use_cache
|
|
setting must be set to true for this option to
|
|
be useful. If set to false, then the cache will
|
|
be skipped, even if a result has already been
|
|
cached. Defaults to false.
|
|
ifrom -- Specifiy the sender's JID.
|
|
block -- If true, block and wait for the stanzas' reply.
|
|
timeout -- The time in seconds to block while waiting for
|
|
a reply. If None, then wait indefinitely. The
|
|
timeout value is only used when block=True.
|
|
callback -- Optional callback to execute when a reply is
|
|
received instead of blocking and waiting for
|
|
the reply.
|
|
"""
|
|
if jid is not None and not isinstance(jid, JID):
|
|
jid = JID(jid)
|
|
if self.xmpp.is_component:
|
|
if jid.domain == self.xmpp.boundjid.domain:
|
|
local = True
|
|
else:
|
|
if str(jid) == str(self.xmpp.boundjid):
|
|
local = True
|
|
jid = jid.full
|
|
elif jid in (None, ''):
|
|
local = True
|
|
|
|
if local:
|
|
log.debug("Looking up local disco#info data " + \
|
|
"for %s, node %s.", jid, node)
|
|
info = self.api['get_info'](jid, node,
|
|
kwargs.get('ifrom', None),
|
|
kwargs)
|
|
info = self._fix_default_info(info)
|
|
return self._wrap(kwargs.get('ifrom', None), jid, info)
|
|
|
|
if cached:
|
|
log.debug("Looking up cached disco#info data " + \
|
|
"for %s, node %s.", jid, node)
|
|
info = self.api['get_cached_info'](jid, node,
|
|
kwargs.get('ifrom', None),
|
|
kwargs)
|
|
if info is not None:
|
|
return self._wrap(kwargs.get('ifrom', None), jid, info)
|
|
|
|
iq = self.xmpp.Iq()
|
|
# Check dfrom parameter for backwards compatibility
|
|
iq['from'] = kwargs.get('ifrom', kwargs.get('dfrom', ''))
|
|
iq['to'] = jid
|
|
iq['type'] = 'get'
|
|
iq['disco_info']['node'] = node if node else ''
|
|
return iq.send(timeout=kwargs.get('timeout', None),
|
|
block=kwargs.get('block', True),
|
|
callback=kwargs.get('callback', None))
|
|
|
|
def set_info(self, jid=None, node=None, info=None):
|
|
"""
|
|
Set the disco#info data for a JID/node based on an existing
|
|
disco#info stanza.
|
|
"""
|
|
if isinstance(info, Iq):
|
|
info = info['disco_info']
|
|
self.api['set_info'](jid, node, None, info)
|
|
|
|
def get_items(self, jid=None, node=None, local=False, **kwargs):
|
|
"""
|
|
Retrieve the disco#items results from a given JID/node combination.
|
|
|
|
Items may be retrieved from both local resources and remote agents;
|
|
the local parameter indicates if the items should be gathered by
|
|
executing the local node handlers, or if a disco#items stanza must
|
|
be generated and sent.
|
|
|
|
If requesting items from a local JID/node, then only a DiscoItems
|
|
stanza will be returned. Otherwise, an Iq stanza will be returned.
|
|
|
|
Arguments:
|
|
jid -- Request info from this JID.
|
|
node -- The particular node to query.
|
|
local -- If true, then the query is for a JID/node
|
|
combination handled by this Sleek instance and
|
|
no stanzas need to be sent.
|
|
Otherwise, a disco stanza must be sent to the
|
|
remove JID to retrieve the items.
|
|
ifrom -- Specifiy the sender's JID.
|
|
block -- If true, block and wait for the stanzas' reply.
|
|
timeout -- The time in seconds to block while waiting for
|
|
a reply. If None, then wait indefinitely.
|
|
callback -- Optional callback to execute when a reply is
|
|
received instead of blocking and waiting for
|
|
the reply.
|
|
iterator -- If True, return a result set iterator using
|
|
the XEP-0059 plugin, if the plugin is loaded.
|
|
Otherwise the parameter is ignored.
|
|
"""
|
|
if local or jid is None:
|
|
items = self.api['get_items'](jid, node,
|
|
kwargs.get('ifrom', None),
|
|
kwargs)
|
|
return self._wrap(kwargs.get('ifrom', None), jid, items)
|
|
|
|
iq = self.xmpp.Iq()
|
|
# Check dfrom parameter for backwards compatibility
|
|
iq['from'] = kwargs.get('ifrom', kwargs.get('dfrom', ''))
|
|
iq['to'] = jid
|
|
iq['type'] = 'get'
|
|
iq['disco_items']['node'] = node if node else ''
|
|
if kwargs.get('iterator', False) and self.xmpp['xep_0059']:
|
|
return self.xmpp['xep_0059'].iterate(iq, 'disco_items')
|
|
else:
|
|
return iq.send(timeout=kwargs.get('timeout', None),
|
|
block=kwargs.get('block', True),
|
|
callback=kwargs.get('callback', None))
|
|
|
|
def set_items(self, jid=None, node=None, **kwargs):
|
|
"""
|
|
Set or replace all items for the specified JID/node combination.
|
|
|
|
The given items must be in a list or set where each item is a
|
|
tuple of the form: (jid, node, name).
|
|
|
|
Arguments:
|
|
jid -- The JID to modify.
|
|
node -- Optional node to modify.
|
|
items -- A series of items in tuple format.
|
|
"""
|
|
self.api['set_items'](jid, node, None, kwargs)
|
|
|
|
def del_items(self, jid=None, node=None, **kwargs):
|
|
"""
|
|
Remove all items from the given JID/node combination.
|
|
|
|
Arguments:
|
|
jid -- The JID to modify.
|
|
node -- Optional node to modify.
|
|
"""
|
|
self.api['del_items'](jid, node, None, kwargs)
|
|
|
|
def add_item(self, jid='', name='', node=None, subnode='', ijid=None):
|
|
"""
|
|
Add a new item element to the given JID/node combination.
|
|
|
|
Each item is required to have a JID, but may also specify
|
|
a node value to reference non-addressable entities.
|
|
|
|
Arguments:
|
|
jid -- The JID for the item.
|
|
name -- Optional name for the item.
|
|
node -- The node to modify.
|
|
subnode -- Optional node for the item.
|
|
ijid -- The JID to modify.
|
|
"""
|
|
if not jid:
|
|
jid = self.xmpp.boundjid.full
|
|
kwargs = {'ijid': jid,
|
|
'name': name,
|
|
'inode': subnode}
|
|
self.api['add_item'](ijid, node, None, kwargs)
|
|
|
|
def del_item(self, jid=None, node=None, **kwargs):
|
|
"""
|
|
Remove a single item from the given JID/node combination.
|
|
|
|
Arguments:
|
|
jid -- The JID to modify.
|
|
node -- The node to modify.
|
|
ijid -- The item's JID.
|
|
inode -- The item's node.
|
|
"""
|
|
self.api['del_item'](jid, node, None, kwargs)
|
|
|
|
def add_identity(self, category='', itype='', name='',
|
|
node=None, jid=None, lang=None):
|
|
"""
|
|
Add a new identity to the given JID/node combination.
|
|
|
|
Each identity must be unique in terms of all four identity
|
|
components: category, type, name, and language.
|
|
|
|
Multiple, identical category/type pairs are allowed only
|
|
if the xml:lang values are different. Likewise, multiple
|
|
category/type/xml:lang pairs are allowed so long as the
|
|
names are different. A category and type is always required.
|
|
|
|
Arguments:
|
|
category -- The identity's category.
|
|
itype -- The identity's type.
|
|
name -- Optional name for the identity.
|
|
lang -- Optional two-letter language code.
|
|
node -- The node to modify.
|
|
jid -- The JID to modify.
|
|
"""
|
|
kwargs = {'category': category,
|
|
'itype': itype,
|
|
'name': name,
|
|
'lang': lang}
|
|
self.api['add_identity'](jid, node, None, kwargs)
|
|
|
|
def add_feature(self, feature, node=None, jid=None):
|
|
"""
|
|
Add a feature to a JID/node combination.
|
|
|
|
Arguments:
|
|
feature -- The namespace of the supported feature.
|
|
node -- The node to modify.
|
|
jid -- The JID to modify.
|
|
"""
|
|
kwargs = {'feature': feature}
|
|
self.api['add_feature'](jid, node, None, kwargs)
|
|
|
|
def del_identity(self, jid=None, node=None, **kwargs):
|
|
"""
|
|
Remove an identity from the given JID/node combination.
|
|
|
|
Arguments:
|
|
jid -- The JID to modify.
|
|
node -- The node to modify.
|
|
category -- The identity's category.
|
|
itype -- The identity's type value.
|
|
name -- Optional, human readable name for the identity.
|
|
lang -- Optional, the identity's xml:lang value.
|
|
"""
|
|
self.api['del_identity'](jid, node, None, kwargs)
|
|
|
|
def del_feature(self, jid=None, node=None, **kwargs):
|
|
"""
|
|
Remove a feature from a given JID/node combination.
|
|
|
|
Arguments:
|
|
jid -- The JID to modify.
|
|
node -- The node to modify.
|
|
feature -- The feature's namespace.
|
|
"""
|
|
self.api['del_feature'](jid, node, None, kwargs)
|
|
|
|
def set_identities(self, jid=None, node=None, **kwargs):
|
|
"""
|
|
Add or replace all identities for the given JID/node combination.
|
|
|
|
The identities must be in a set where each identity is a tuple
|
|
of the form: (category, type, lang, name)
|
|
|
|
Arguments:
|
|
jid -- The JID to modify.
|
|
node -- The node to modify.
|
|
identities -- A set of identities in tuple form.
|
|
lang -- Optional, xml:lang value.
|
|
"""
|
|
self.api['set_identities'](jid, node, None, kwargs)
|
|
|
|
def del_identities(self, jid=None, node=None, **kwargs):
|
|
"""
|
|
Remove all identities for a JID/node combination.
|
|
|
|
If a language is specified, only identities using that
|
|
language will be removed.
|
|
|
|
Arguments:
|
|
jid -- The JID to modify.
|
|
node -- The node to modify.
|
|
lang -- Optional. If given, only remove identities
|
|
using this xml:lang value.
|
|
"""
|
|
self.api['del_identities'](jid, node, None, kwargs)
|
|
|
|
def set_features(self, jid=None, node=None, **kwargs):
|
|
"""
|
|
Add or replace the set of supported features
|
|
for a JID/node combination.
|
|
|
|
Arguments:
|
|
jid -- The JID to modify.
|
|
node -- The node to modify.
|
|
features -- The new set of supported features.
|
|
"""
|
|
self.api['set_features'](jid, node, None, kwargs)
|
|
|
|
def del_features(self, jid=None, node=None, **kwargs):
|
|
"""
|
|
Remove all features from a JID/node combination.
|
|
|
|
Arguments:
|
|
jid -- The JID to modify.
|
|
node -- The node to modify.
|
|
"""
|
|
self.api['del_features'](jid, node, None, kwargs)
|
|
|
|
def _run_node_handler(self, htype, jid, node=None, ifrom=None, data={}):
|
|
"""
|
|
Execute the most specific node handler for the given
|
|
JID/node combination.
|
|
|
|
Arguments:
|
|
htype -- The handler type to execute.
|
|
jid -- The JID requested.
|
|
node -- The node requested.
|
|
data -- Optional, custom data to pass to the handler.
|
|
"""
|
|
return self.api[htype](jid, node, ifrom, data)
|
|
|
|
def _handle_disco_info(self, iq):
|
|
"""
|
|
Process an incoming disco#info stanza. If it is a get
|
|
request, find and return the appropriate identities
|
|
and features. If it is an info result, fire the
|
|
disco_info event.
|
|
|
|
Arguments:
|
|
iq -- The incoming disco#items stanza.
|
|
"""
|
|
if iq['type'] == 'get':
|
|
log.debug("Received disco info query from " + \
|
|
"<%s> to <%s>.", iq['from'], iq['to'])
|
|
if self.xmpp.is_component:
|
|
jid = iq['to'].full
|
|
else:
|
|
jid = iq['to'].bare
|
|
info = self.api['get_info'](jid,
|
|
iq['disco_info']['node'],
|
|
iq['from'],
|
|
iq)
|
|
if isinstance(info, Iq):
|
|
info['id'] = iq['id']
|
|
info.send()
|
|
else:
|
|
iq.reply()
|
|
if info:
|
|
info = self._fix_default_info(info)
|
|
iq.set_payload(info.xml)
|
|
iq.send()
|
|
elif iq['type'] == 'result':
|
|
log.debug("Received disco info result from " + \
|
|
"<%s> to <%s>.", iq['from'], iq['to'])
|
|
if self.use_cache:
|
|
log.debug("Caching disco info result from " \
|
|
"<%s> to <%s>.", iq['from'], iq['to'])
|
|
if self.xmpp.is_component:
|
|
ito = iq['to'].full
|
|
else:
|
|
ito = None
|
|
self.api['cache_info'](iq['from'].full,
|
|
iq['disco_info']['node'],
|
|
ito,
|
|
iq)
|
|
self.xmpp.event('disco_info', iq)
|
|
|
|
def _handle_disco_items(self, iq):
|
|
"""
|
|
Process an incoming disco#items stanza. If it is a get
|
|
request, find and return the appropriate items. If it
|
|
is an items result, fire the disco_items event.
|
|
|
|
Arguments:
|
|
iq -- The incoming disco#items stanza.
|
|
"""
|
|
if iq['type'] == 'get':
|
|
log.debug("Received disco items query from " + \
|
|
"<%s> to <%s>.", iq['from'], iq['to'])
|
|
if self.xmpp.is_component:
|
|
jid = iq['to'].full
|
|
else:
|
|
jid = iq['to'].bare
|
|
items = self.api['get_items'](jid,
|
|
iq['disco_items']['node'],
|
|
iq['from'].full,
|
|
iq)
|
|
if isinstance(items, Iq):
|
|
items.send()
|
|
else:
|
|
iq.reply()
|
|
if items:
|
|
iq.set_payload(items.xml)
|
|
iq.send()
|
|
elif iq['type'] == 'result':
|
|
log.debug("Received disco items result from " + \
|
|
"%s to %s.", iq['from'], iq['to'])
|
|
self.xmpp.event('disco_items', iq)
|
|
|
|
def _fix_default_info(self, info):
|
|
"""
|
|
Disco#info results for a JID are required to include at least
|
|
one identity and feature. As a default, if no other identity is
|
|
provided, SleekXMPP will use either the generic component or the
|
|
bot client identity. A the standard disco#info feature will also be
|
|
added if no features are provided.
|
|
|
|
Arguments:
|
|
info -- The disco#info quest (not the full Iq stanza) to modify.
|
|
"""
|
|
result = info
|
|
if isinstance(info, Iq):
|
|
info = info['disco_info']
|
|
if not info['node']:
|
|
if not info['identities']:
|
|
if self.xmpp.is_component:
|
|
log.debug("No identity found for this entity. " + \
|
|
"Using default component identity.")
|
|
info.add_identity('component', 'generic')
|
|
else:
|
|
log.debug("No identity found for this entity. " + \
|
|
"Using default client identity.")
|
|
info.add_identity('client', 'bot')
|
|
if not info['features']:
|
|
log.debug("No features found for this entity. " + \
|
|
"Using default disco#info feature.")
|
|
info.add_feature(info.namespace)
|
|
return result
|
|
|
|
def _wrap(self, ito, ifrom, payload, force=False):
|
|
"""
|
|
Ensure that results are wrapped in an Iq stanza
|
|
if self.wrap_results has been set to True.
|
|
|
|
Arguments:
|
|
ito -- The JID to use as the 'to' value
|
|
ifrom -- The JID to use as the 'from' value
|
|
payload -- The disco data to wrap
|
|
force -- Force wrapping, regardless of self.wrap_results
|
|
"""
|
|
if (force or self.wrap_results) and not isinstance(payload, Iq):
|
|
iq = self.xmpp.Iq()
|
|
# Since we're simulating a result, we have to treat
|
|
# the 'from' and 'to' values opposite the normal way.
|
|
iq['to'] = self.xmpp.boundjid if ito is None else ito
|
|
iq['from'] = self.xmpp.boundjid if ifrom is None else ifrom
|
|
iq['type'] = 'result'
|
|
iq.append(payload)
|
|
return iq
|
|
return payload
|