# Copyright (C) 2006 Tomasz Melcer # Copyright (C) 2006-2014 Yann Leboulanger # Copyright (C) 2007 Jean-Marie Traissard # Copyright (C) 2008 Stephan Erb # Copyright (C) 2018 Philipp Hörist # # This file is part of Gajim. # # Gajim is free software; you can redistribute it and/or modify # it under the terms of the GNU General Public License as published # by the Free Software Foundation; version 3 only. # # Gajim is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # GNU General Public License for more details. # # You should have received a copy of the GNU General Public License # along with Gajim. If not, see . # XEP-0060: Publish-Subscribe import logging import nbxmpp from gajim.common import app log = logging.getLogger('gajim.c.m.pubsub') class PubSub: def __init__(self, con): self._con = con self._account = con.name self.handlers = [] def send_pb_subscription_query(self, jid, cb, **kwargs): if not app.account_is_connected(self._account): return query = nbxmpp.Iq('get', to=jid) pb = query.addChild('pubsub', namespace=nbxmpp.NS_PUBSUB) pb.addChild('subscriptions') self._con.connection.SendAndCallForResponse(query, cb, kwargs) def send_pb_subscribe(self, jid, node, cb, **kwargs): if not app.account_is_connected(self._account): return our_jid = app.get_jid_from_account(self._account) query = nbxmpp.Iq('set', to=jid) pb = query.addChild('pubsub', namespace=nbxmpp.NS_PUBSUB) pb.addChild('subscribe', {'node': node, 'jid': our_jid}) self._con.connection.SendAndCallForResponse(query, cb, kwargs) def send_pb_unsubscribe(self, jid, node, cb, **kwargs): if not app.account_is_connected(self._account): return our_jid = app.get_jid_from_account(self._account) query = nbxmpp.Iq('set', to=jid) pb = query.addChild('pubsub', namespace=nbxmpp.NS_PUBSUB) pb.addChild('unsubscribe', {'node': node, 'jid': our_jid}) self._con.connection.SendAndCallForResponse(query, cb, kwargs) def send_pb_publish(self, jid, node, item, id_=None, options=None, cb=None, **kwargs): if not app.account_is_connected(self._account): return if cb is None: cb = self._default_callback query = nbxmpp.Iq('set', to=jid) e = query.addChild('pubsub', namespace=nbxmpp.NS_PUBSUB) p = e.addChild('publish', {'node': node}) attrs = {} if id_: attrs = {'id': id_} p.addChild('item', attrs, [item]) if options: p = e.addChild('publish-options') p.addChild(node=options) self._con.connection.SendAndCallForResponse(query, cb, kwargs) @staticmethod def get_pb_retrieve_iq(jid, node, item_id=None): """ Get IQ to query items from a node """ query = nbxmpp.Iq('get', to=jid) r = query.addChild('pubsub', namespace=nbxmpp.NS_PUBSUB) r = r.addChild('items', {'node': node}) if item_id is not None: r.addChild('item', {'id': item_id}) return query def send_pb_retrieve(self, jid, node, item_id=None, cb=None, **kwargs): """ Get items from a node """ if not app.account_is_connected(self._account): return if cb is None: cb = self._default_callback query = self.get_pb_retrieve_iq(jid, node, item_id) self._con.connection.SendAndCallForResponse(query, cb, kwargs) def send_pb_retract(self, jid, node, id_, cb=None, **kwargs): """ Delete item from a node """ if not app.account_is_connected(self._account): return if cb is None: cb = self._default_callback query = nbxmpp.Iq('set', to=jid) r = query.addChild('pubsub', namespace=nbxmpp.NS_PUBSUB) r = r.addChild('retract', {'node': node, 'notify': '1'}) r = r.addChild('item', {'id': id_}) self._con.connection.SendAndCallForResponse(query, cb, kwargs) def send_pb_purge(self, jid, node, cb=None, **kwargs): """ Purge node: Remove all items """ if not app.account_is_connected(self._account): return if cb is None: cb = self._default_callback query = nbxmpp.Iq('set', to=jid) d = query.addChild('pubsub', namespace=nbxmpp.NS_PUBSUB_OWNER) d = d.addChild('purge', {'node': node}) self._con.connection.SendAndCallForResponse(query, cb, kwargs) def send_pb_delete(self, jid, node, on_ok=None, on_fail=None): """ Delete node """ if not app.account_is_connected(self._account): return query = nbxmpp.Iq('set', to=jid) d = query.addChild('pubsub', namespace=nbxmpp.NS_PUBSUB_OWNER) d = d.addChild('delete', {'node': node}) def response(con, resp, jid, node): if resp.getType() == 'result' and on_ok: on_ok(jid, node) elif on_fail: msg = resp.getErrorMsg() on_fail(jid, node, msg) self._con.connection.SendAndCallForResponse( query, response, {'jid': jid, 'node': node}) def send_pb_create(self, jid, node, cb, configure=False, configure_form=None): """ Create a new node """ if not app.account_is_connected(self._account): return query = nbxmpp.Iq('set', to=jid) c = query.addChild('pubsub', namespace=nbxmpp.NS_PUBSUB) c = c.addChild('create', {'node': node}) if configure: conf = c.addChild('configure') if configure_form is not None: conf.addChild(node=configure_form) self._con.connection.SendAndCallForResponse(query, cb) def send_pb_configure(self, jid, node, form, cb=None): if not app.account_is_connected(self._account): return if cb is None: cb = self._default_callback query = nbxmpp.Iq('set', to=jid) c = query.addChild('pubsub', namespace=nbxmpp.NS_PUBSUB_OWNER) c = c.addChild('configure', {'node': node}) c.addChild(node=form) self._con.connection.SendAndCallForResponse(query, cb) def request_pb_configuration(self, jid, node, cb=None): if not app.account_is_connected(self._account): return if cb is None: cb = self._default_callback query = nbxmpp.Iq('get', to=jid) e = query.addChild('pubsub', namespace=nbxmpp.NS_PUBSUB_OWNER) e = e.addChild('configure', {'node': node}) self._con.connection.SendAndCallForResponse(query, cb) def _default_callback(self, conn, stanza, *args, **kwargs): if not nbxmpp.isResultNode(stanza): log.warning('Error: %s', stanza.getError())