108 lines
3.3 KiB
Python
108 lines
3.3 KiB
Python
import xmpp
|
|
import gajim
|
|
import connection_handlers
|
|
|
|
class ConnectionPubSub:
|
|
def __init__(self):
|
|
self.__callbacks={}
|
|
|
|
def send_pb_subscription_query(self, jid, cb, *args, **kwargs):
|
|
query = xmpp.Iq('get', to=jid)
|
|
pb = query.addChild('pubsub', {'xmlns': xmpp.NS_PUBSUB})
|
|
pb.addChild('subscriptions')
|
|
|
|
id = self.connection.send(query)
|
|
|
|
self.__callbacks[id]=(cb, args, kwargs)
|
|
|
|
def send_pb_subscribe(self, jid, node, cb, *args, **kwargs):
|
|
our_jid = gajim.get_jid_from_account(self.name)
|
|
query = xmpp.Iq('set', to=jid)
|
|
pb = query.addChild('pubsub', namespace=xmpp.NS_PUBSUB)
|
|
pb.addChild('subscribe', {'node': node, 'jid': our_jid})
|
|
|
|
id = self.connection.send(query)
|
|
|
|
self.__callbacks[id]=(cb, args, kwargs)
|
|
|
|
def send_pb_unsubscribe(self, jid, node, cb, *args, **kwargs):
|
|
our_jid = gajim.get_jid_from_account(self.name)
|
|
query = xmpp.Iq('set', to=jid)
|
|
pb = query.addChild('pubsub', namespace=xmpp.NS_PUBSUB)
|
|
pb.addChild('unsubscribe', {'node': node, 'jid': our_jid})
|
|
|
|
id = self.connection.send(query)
|
|
|
|
self.__callbacks[id]=(cb, args, kwargs)
|
|
|
|
def send_pb_publish(self, jid, node, item, id):
|
|
'''Publish item to a node.'''
|
|
query = xmpp.Iq('set', to=jid)
|
|
e = query.addChild('pubsub', namespace=xmpp.NS_PUBSUB)
|
|
e = e.addChild('publish', {'node': node})
|
|
e = e.addChild('item', {'id': id}, [item]) # TODO: we should generate id... or we shouldn't?
|
|
|
|
self.connection.send(query)
|
|
|
|
def send_pb_retract(self, jid, node, id):
|
|
'''Delete item from a node'''
|
|
query = xmpp.Iq('set', to=jid)
|
|
r = query.addChild('pubsub', namespace=xmpp.NS_PUBSUB)
|
|
r = r.addChild('retract', {'node': node, 'notify': '1'})
|
|
r = r.addChild('item', {'id': id})
|
|
|
|
self.connection.send(query)
|
|
|
|
def send_pb_delete(self, jid, node):
|
|
'''Deletes node.'''
|
|
query = xmpp.Iq('set', to=jid)
|
|
d = query.addChild('pubsub', namespace=xmpp.NS_PUBSUB)
|
|
d = d.addChild('delete', {'node': node})
|
|
|
|
def response(con, resp, jid, node):
|
|
if resp.getType() == 'result':
|
|
self.dispatch('PUBSUB_NODE_REMOVED', (jid, node))
|
|
else:
|
|
msg = resp.getErrorMsg()
|
|
self.dispatch('PUBSUB_NODE_NOT_REMOVED', (jid, node, msg))
|
|
|
|
self.connection.SendAndCallForResponse(query, response, {'jid': jid,
|
|
'node': node})
|
|
|
|
def send_pb_create(self, jid, node, configure = False, configure_form = None):
|
|
'''Creates new node.'''
|
|
query = xmpp.Iq('set', to=jid)
|
|
c = query.addChild('pubsub', namespace=xmpp.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.connection.send(query)
|
|
|
|
def send_pb_configure(self, jid, node, form):
|
|
query = xmpp.Iq('set', to=jid)
|
|
c = query.addChild('pubsub', namespace=xmpp.NS_PUBSUB_OWNER)
|
|
c = c.addChild('configure', {'node': node})
|
|
c.addChild(node=form)
|
|
|
|
self.connection.send(query)
|
|
|
|
def _PubSubCB(self, conn, stanza):
|
|
try:
|
|
cb, args, kwargs = self.__callbacks.pop(stanza.getID())
|
|
cb(conn, stanza, *args, **kwargs)
|
|
except:
|
|
pass
|
|
|
|
def request_pb_configuration(self, jid, node):
|
|
query = xmpp.Iq('get', to=jid)
|
|
e = query.addChild('pubsub', namespace=xmpp.NS_PUBSUB_OWNER)
|
|
e = e.addChild('configure', {'node': node})
|
|
id = self.connection.getAnID()
|
|
query.setID(id)
|
|
self.awaiting_answers[id] = (connection_handlers.PEP_CONFIG,)
|
|
self.connection.send(query)
|
|
|
|
# vim: se ts=3: |