libervia-pubsub: sat_pubsub/delegation.py annotate

annotate sat_pubsub/delegation.py @ 318:d13526c0eb32

RSM improvments/refactoring: - a warning message is displayed if maxItems == 0 in getItems, and an empty list is returned in this case - use the new container.ItemData instead of doing tuple (un)packing - the list of ItemData => list of domish.Element conversion is done at the end of the workflow - rsm request is checked in self._items_rsm directly - better handling of Response.index in _items_rsm - itemsIdentifiers can't be used with RSM (the later will be ignored if this happen) - don't do approximative unpacking anymore in _items_rsm - countItems and getIndex have been refactored and renamed getItemsCount and getItemsIndex, don't use query duplications anymore - cleaned query handling in getItems - /!\ mam module is temporarly broken

author	Goffi <goffi@goffi.org>
date	Sun, 03 Jan 2016 18:33:22 +0100
parents	5d7c3787672e
children	54d90c73b8b5

rev	line source
242 a6170637690d remote roster partial support Goffi <goffi@goffi.org> parents: diff changeset	1 #!/usr/bin/python
a6170637690d remote roster partial support Goffi <goffi@goffi.org> parents: diff changeset	2 #-- coding: utf-8 --
a6170637690d remote roster partial support Goffi <goffi@goffi.org> parents: diff changeset	3 #
312 5d7c3787672e fixed copyright put in docstring instead of comments Goffi <goffi@goffi.org> parents: 310 diff changeset	4 # Copyright (c) 2015 Jérôme Poisson
242 a6170637690d remote roster partial support Goffi <goffi@goffi.org> parents: diff changeset	5
a6170637690d remote roster partial support Goffi <goffi@goffi.org> parents: diff changeset	6
312 5d7c3787672e fixed copyright put in docstring instead of comments Goffi <goffi@goffi.org> parents: 310 diff changeset	7 # This program is free software: you can redistribute it and/or modify
5d7c3787672e fixed copyright put in docstring instead of comments Goffi <goffi@goffi.org> parents: 310 diff changeset	8 # it under the terms of the GNU Affero General Public License as published by
5d7c3787672e fixed copyright put in docstring instead of comments Goffi <goffi@goffi.org> parents: 310 diff changeset	9 # the Free Software Foundation, either version 3 of the License, or
5d7c3787672e fixed copyright put in docstring instead of comments Goffi <goffi@goffi.org> parents: 310 diff changeset	10 # (at your option) any later version.
242 a6170637690d remote roster partial support Goffi <goffi@goffi.org> parents: diff changeset	11
312 5d7c3787672e fixed copyright put in docstring instead of comments Goffi <goffi@goffi.org> parents: 310 diff changeset	12 # This program is distributed in the hope that it will be useful,
5d7c3787672e fixed copyright put in docstring instead of comments Goffi <goffi@goffi.org> parents: 310 diff changeset	13 # but WITHOUT ANY WARRANTY; without even the implied warranty of
5d7c3787672e fixed copyright put in docstring instead of comments Goffi <goffi@goffi.org> parents: 310 diff changeset	14 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
5d7c3787672e fixed copyright put in docstring instead of comments Goffi <goffi@goffi.org> parents: 310 diff changeset	15 # GNU Affero General Public License for more details.
242 a6170637690d remote roster partial support Goffi <goffi@goffi.org> parents: diff changeset	16
312 5d7c3787672e fixed copyright put in docstring instead of comments Goffi <goffi@goffi.org> parents: 310 diff changeset	17 # You should have received a copy of the GNU Affero General Public License
5d7c3787672e fixed copyright put in docstring instead of comments Goffi <goffi@goffi.org> parents: 310 diff changeset	18 # along with this program. If not, see <http://www.gnu.org/licenses/>.
242 a6170637690d remote roster partial support Goffi <goffi@goffi.org> parents: diff changeset	19
312 5d7c3787672e fixed copyright put in docstring instead of comments Goffi <goffi@goffi.org> parents: 310 diff changeset	20 # ---
242 a6170637690d remote roster partial support Goffi <goffi@goffi.org> parents: diff changeset	21
312 5d7c3787672e fixed copyright put in docstring instead of comments Goffi <goffi@goffi.org> parents: 310 diff changeset	22 # This module implements XEP-0355 (Namespace delegation) to use SàT Pubsub as PEP service
242 a6170637690d remote roster partial support Goffi <goffi@goffi.org> parents: diff changeset	23
285 a87c155d0fd5 replaced former roster dirty hack by a XEP-0356 first draft implementation, only roster get is implemented so far Goffi <goffi@goffi.org> parents: 283 diff changeset	24 from wokkel.subprotocols import XMPPHandler
287 61f92273fb69 implementation of XEP-0355 (Namespace Delegation) to use SàT Pubsub as PEP service, first draft Goffi <goffi@goffi.org> parents: 286 diff changeset	25 from wokkel import pubsub
289 f08f8536cab8 mod delegation: extensions management (XEP-0128) Goffi <goffi@goffi.org> parents: 288 diff changeset	26 from wokkel import data_form
288 073161f6f143 namespace delegation: disco nesting management Goffi <goffi@goffi.org> parents: 287 diff changeset	27 from wokkel import disco, iwokkel
292 6918a0dad359 delegation: delegated stanza are tracked Goffi <goffi@goffi.org> parents: 291 diff changeset	28 from wokkel.iwokkel import IPubSubService
285 a87c155d0fd5 replaced former roster dirty hack by a XEP-0356 first draft implementation, only roster get is implemented so far Goffi <goffi@goffi.org> parents: 283 diff changeset	29 from twisted.python import log
291 61fb4817b77f delegation: iq forwarded management: Goffi <goffi@goffi.org> parents: 289 diff changeset	30 from twisted.words.protocols.jabber import jid, error
61fb4817b77f delegation: iq forwarded management: Goffi <goffi@goffi.org> parents: 289 diff changeset	31 from twisted.words.protocols.jabber.xmlstream import toResponse
61fb4817b77f delegation: iq forwarded management: Goffi <goffi@goffi.org> parents: 289 diff changeset	32 from twisted.words.xish import domish
288 073161f6f143 namespace delegation: disco nesting management Goffi <goffi@goffi.org> parents: 287 diff changeset	33 from zope.interface import implements
242 a6170637690d remote roster partial support Goffi <goffi@goffi.org> parents: diff changeset	34
287 61f92273fb69 implementation of XEP-0355 (Namespace Delegation) to use SàT Pubsub as PEP service, first draft Goffi <goffi@goffi.org> parents: 286 diff changeset	35 DELEGATION_NS = 'urn:xmpp:delegation:1'
61f92273fb69 implementation of XEP-0355 (Namespace Delegation) to use SàT Pubsub as PEP service, first draft Goffi <goffi@goffi.org> parents: 286 diff changeset	36 FORWARDED_NS = 'urn:xmpp:forward:0'
61f92273fb69 implementation of XEP-0355 (Namespace Delegation) to use SàT Pubsub as PEP service, first draft Goffi <goffi@goffi.org> parents: 286 diff changeset	37 DELEGATION_ADV_XPATH = '/message/delegation[@xmlns="{}"]'.format(DELEGATION_NS)
291 61fb4817b77f delegation: iq forwarded management: Goffi <goffi@goffi.org> parents: 289 diff changeset	38 DELEGATION_FWD_XPATH = '/iq[@type="set"]/delegation[@xmlns="{}"]/forwarded[@xmlns="{}"]'.format(DELEGATION_NS, FORWARDED_NS)
285 a87c155d0fd5 replaced former roster dirty hack by a XEP-0356 first draft implementation, only roster get is implemented so far Goffi <goffi@goffi.org> parents: 283 diff changeset	39
288 073161f6f143 namespace delegation: disco nesting management Goffi <goffi@goffi.org> parents: 287 diff changeset	40 DELEGATION_MAIN_SEP = "::"
073161f6f143 namespace delegation: disco nesting management Goffi <goffi@goffi.org> parents: 287 diff changeset	41 DELEGATION_BARE_SEP = ":bare:"
073161f6f143 namespace delegation: disco nesting management Goffi <goffi@goffi.org> parents: 287 diff changeset	42
285 a87c155d0fd5 replaced former roster dirty hack by a XEP-0356 first draft implementation, only roster get is implemented so far Goffi <goffi@goffi.org> parents: 283 diff changeset	43 class InvalidStanza(Exception):
a87c155d0fd5 replaced former roster dirty hack by a XEP-0356 first draft implementation, only roster get is implemented so far Goffi <goffi@goffi.org> parents: 283 diff changeset	44 pass
a87c155d0fd5 replaced former roster dirty hack by a XEP-0356 first draft implementation, only roster get is implemented so far Goffi <goffi@goffi.org> parents: 283 diff changeset	45
292 6918a0dad359 delegation: delegated stanza are tracked Goffi <goffi@goffi.org> parents: 291 diff changeset	46
6918a0dad359 delegation: delegated stanza are tracked Goffi <goffi@goffi.org> parents: 291 diff changeset	47
287 61f92273fb69 implementation of XEP-0355 (Namespace Delegation) to use SàT Pubsub as PEP service, first draft Goffi <goffi@goffi.org> parents: 286 diff changeset	48 class DelegationsHandler(XMPPHandler):
288 073161f6f143 namespace delegation: disco nesting management Goffi <goffi@goffi.org> parents: 287 diff changeset	49 implements(iwokkel.IDisco)
292 6918a0dad359 delegation: delegated stanza are tracked Goffi <goffi@goffi.org> parents: 291 diff changeset	50 _service_hacked = False
242 a6170637690d remote roster partial support Goffi <goffi@goffi.org> parents: diff changeset	51
285 a87c155d0fd5 replaced former roster dirty hack by a XEP-0356 first draft implementation, only roster get is implemented so far Goffi <goffi@goffi.org> parents: 283 diff changeset	52 def __init__(self):
287 61f92273fb69 implementation of XEP-0355 (Namespace Delegation) to use SàT Pubsub as PEP service, first draft Goffi <goffi@goffi.org> parents: 286 diff changeset	53 super(DelegationsHandler, self).__init__()
285 a87c155d0fd5 replaced former roster dirty hack by a XEP-0356 first draft implementation, only roster get is implemented so far Goffi <goffi@goffi.org> parents: 283 diff changeset	54
292 6918a0dad359 delegation: delegated stanza are tracked Goffi <goffi@goffi.org> parents: 291 diff changeset	55 def _service_hack(self):
6918a0dad359 delegation: delegated stanza are tracked Goffi <goffi@goffi.org> parents: 291 diff changeset	56 """Patch the PubSubService to track delegated stanzas"""
6918a0dad359 delegation: delegated stanza are tracked Goffi <goffi@goffi.org> parents: 291 diff changeset	57 # XXX: we need to monkey patch to track origin of the stanza in PubSubRequest.
6918a0dad359 delegation: delegated stanza are tracked Goffi <goffi@goffi.org> parents: 291 diff changeset	58 # As PubSubRequest from sat.tmp.wokkel.pubsub use _request_class while
6918a0dad359 delegation: delegated stanza are tracked Goffi <goffi@goffi.org> parents: 291 diff changeset	59 # original wokkel.pubsub use directly pubsub.PubSubRequest, we need to
6918a0dad359 delegation: delegated stanza are tracked Goffi <goffi@goffi.org> parents: 291 diff changeset	60 # check which version is used before monkeypatching
6918a0dad359 delegation: delegated stanza are tracked Goffi <goffi@goffi.org> parents: 291 diff changeset	61 for handler in self.parent.handlers:
6918a0dad359 delegation: delegated stanza are tracked Goffi <goffi@goffi.org> parents: 291 diff changeset	62 if IPubSubService.providedBy(handler):
6918a0dad359 delegation: delegated stanza are tracked Goffi <goffi@goffi.org> parents: 291 diff changeset	63 if hasattr(handler, '_request_class'):
6918a0dad359 delegation: delegated stanza are tracked Goffi <goffi@goffi.org> parents: 291 diff changeset	64 request_base_class = handler._request_class
6918a0dad359 delegation: delegated stanza are tracked Goffi <goffi@goffi.org> parents: 291 diff changeset	65 else:
6918a0dad359 delegation: delegated stanza are tracked Goffi <goffi@goffi.org> parents: 291 diff changeset	66 request_base_class = pubsub.PubSubRequest
6918a0dad359 delegation: delegated stanza are tracked Goffi <goffi@goffi.org> parents: 291 diff changeset	67
6918a0dad359 delegation: delegated stanza are tracked Goffi <goffi@goffi.org> parents: 291 diff changeset	68 class PubSubRequestWithDelegation(request_base_class):
6918a0dad359 delegation: delegated stanza are tracked Goffi <goffi@goffi.org> parents: 291 diff changeset	69 """A PubSubReques which put an indicator if the stanza comme from delegation"""
6918a0dad359 delegation: delegated stanza are tracked Goffi <goffi@goffi.org> parents: 291 diff changeset	70
6918a0dad359 delegation: delegated stanza are tracked Goffi <goffi@goffi.org> parents: 291 diff changeset	71 @classmethod
6918a0dad359 delegation: delegated stanza are tracked Goffi <goffi@goffi.org> parents: 291 diff changeset	72 def fromElement(cls, element):
6918a0dad359 delegation: delegated stanza are tracked Goffi <goffi@goffi.org> parents: 291 diff changeset	73 """Check if element comme from delegation, and set a delegated flags
6918a0dad359 delegation: delegated stanza are tracked Goffi <goffi@goffi.org> parents: 291 diff changeset	74
6918a0dad359 delegation: delegated stanza are tracked Goffi <goffi@goffi.org> parents: 291 diff changeset	75 delegated flaf is either False, or it's a jid of the delegating server
6918a0dad359 delegation: delegated stanza are tracked Goffi <goffi@goffi.org> parents: 291 diff changeset	76 the delegated flag must be set on element before use
6918a0dad359 delegation: delegated stanza are tracked Goffi <goffi@goffi.org> parents: 291 diff changeset	77 """
6918a0dad359 delegation: delegated stanza are tracked Goffi <goffi@goffi.org> parents: 291 diff changeset	78 try:
6918a0dad359 delegation: delegated stanza are tracked Goffi <goffi@goffi.org> parents: 291 diff changeset	79 # __getattr__ is overriden in domish.Element, so we use __getattribute__
6918a0dad359 delegation: delegated stanza are tracked Goffi <goffi@goffi.org> parents: 291 diff changeset	80 delegated = element.__getattribute__('delegated')
6918a0dad359 delegation: delegated stanza are tracked Goffi <goffi@goffi.org> parents: 291 diff changeset	81 except AttributeError:
6918a0dad359 delegation: delegated stanza are tracked Goffi <goffi@goffi.org> parents: 291 diff changeset	82 delegated = False
6918a0dad359 delegation: delegated stanza are tracked Goffi <goffi@goffi.org> parents: 291 diff changeset	83 instance = cls.__base__.fromElement(element)
6918a0dad359 delegation: delegated stanza are tracked Goffi <goffi@goffi.org> parents: 291 diff changeset	84 instance.delegated = delegated
6918a0dad359 delegation: delegated stanza are tracked Goffi <goffi@goffi.org> parents: 291 diff changeset	85 return instance
6918a0dad359 delegation: delegated stanza are tracked Goffi <goffi@goffi.org> parents: 291 diff changeset	86
6918a0dad359 delegation: delegated stanza are tracked Goffi <goffi@goffi.org> parents: 291 diff changeset	87 if hasattr(handler, '_request_class'):
6918a0dad359 delegation: delegated stanza are tracked Goffi <goffi@goffi.org> parents: 291 diff changeset	88 handler._request_class = PubSubRequestWithDelegation
6918a0dad359 delegation: delegated stanza are tracked Goffi <goffi@goffi.org> parents: 291 diff changeset	89 else:
6918a0dad359 delegation: delegated stanza are tracked Goffi <goffi@goffi.org> parents: 291 diff changeset	90 pubsub.PubSubRequest = PubSubRequestWithDelegation
6918a0dad359 delegation: delegated stanza are tracked Goffi <goffi@goffi.org> parents: 291 diff changeset	91 DelegationsHandler._service_hacked = True
6918a0dad359 delegation: delegated stanza are tracked Goffi <goffi@goffi.org> parents: 291 diff changeset	92
285 a87c155d0fd5 replaced former roster dirty hack by a XEP-0356 first draft implementation, only roster get is implemented so far Goffi <goffi@goffi.org> parents: 283 diff changeset	93 def connectionInitialized(self):
292 6918a0dad359 delegation: delegated stanza are tracked Goffi <goffi@goffi.org> parents: 291 diff changeset	94 if not self._service_hacked:
6918a0dad359 delegation: delegated stanza are tracked Goffi <goffi@goffi.org> parents: 291 diff changeset	95 self._service_hack()
287 61f92273fb69 implementation of XEP-0355 (Namespace Delegation) to use SàT Pubsub as PEP service, first draft Goffi <goffi@goffi.org> parents: 286 diff changeset	96 self.xmlstream.addObserver(DELEGATION_ADV_XPATH, self.onAdvertise)
291 61fb4817b77f delegation: iq forwarded management: Goffi <goffi@goffi.org> parents: 289 diff changeset	97 self.xmlstream.addObserver(DELEGATION_FWD_XPATH, self._obsWrapper, 0, self.onForward)
61fb4817b77f delegation: iq forwarded management: Goffi <goffi@goffi.org> parents: 289 diff changeset	98 self._current_iqs = {} # dict of iq being handler by delegation
61fb4817b77f delegation: iq forwarded management: Goffi <goffi@goffi.org> parents: 289 diff changeset	99 self._xs_send = self.xmlstream.send
61fb4817b77f delegation: iq forwarded management: Goffi <goffi@goffi.org> parents: 289 diff changeset	100 self.xmlstream.send = self._sendHack
61fb4817b77f delegation: iq forwarded management: Goffi <goffi@goffi.org> parents: 289 diff changeset	101
61fb4817b77f delegation: iq forwarded management: Goffi <goffi@goffi.org> parents: 289 diff changeset	102 def _sendHack(self, elt):
61fb4817b77f delegation: iq forwarded management: Goffi <goffi@goffi.org> parents: 289 diff changeset	103 """This method is called instead of xmlstream to control sending
61fb4817b77f delegation: iq forwarded management: Goffi <goffi@goffi.org> parents: 289 diff changeset	104
61fb4817b77f delegation: iq forwarded management: Goffi <goffi@goffi.org> parents: 289 diff changeset	105 @param obj(domsish.Element, unicode, str): obj sent to real xmlstream
61fb4817b77f delegation: iq forwarded management: Goffi <goffi@goffi.org> parents: 289 diff changeset	106 """
61fb4817b77f delegation: iq forwarded management: Goffi <goffi@goffi.org> parents: 289 diff changeset	107 if isinstance(elt, domish.Element) and elt.name=='iq':
61fb4817b77f delegation: iq forwarded management: Goffi <goffi@goffi.org> parents: 289 diff changeset	108 try:
292 6918a0dad359 delegation: delegated stanza are tracked Goffi <goffi@goffi.org> parents: 291 diff changeset	109 id_ = elt.getAttribute('id')
6918a0dad359 delegation: delegated stanza are tracked Goffi <goffi@goffi.org> parents: 291 diff changeset	110 ori_iq, managed_entity = self._current_iqs[id_]
291 61fb4817b77f delegation: iq forwarded management: Goffi <goffi@goffi.org> parents: 289 diff changeset	111 if jid.JID(elt['to']) != managed_entity:
292 6918a0dad359 delegation: delegated stanza are tracked Goffi <goffi@goffi.org> parents: 291 diff changeset	112 log.msg("IQ id conflict: the managed entity doesn't match (got {got} was expecting {expected})"
6918a0dad359 delegation: delegated stanza are tracked Goffi <goffi@goffi.org> parents: 291 diff changeset	113 .format(got=jid.JID(elt['to']), expected=managed_entity))
291 61fb4817b77f delegation: iq forwarded management: Goffi <goffi@goffi.org> parents: 289 diff changeset	114 raise KeyError
61fb4817b77f delegation: iq forwarded management: Goffi <goffi@goffi.org> parents: 289 diff changeset	115 except KeyError:
61fb4817b77f delegation: iq forwarded management: Goffi <goffi@goffi.org> parents: 289 diff changeset	116 # the iq is not a delegated one
61fb4817b77f delegation: iq forwarded management: Goffi <goffi@goffi.org> parents: 289 diff changeset	117 self._xs_send(elt)
61fb4817b77f delegation: iq forwarded management: Goffi <goffi@goffi.org> parents: 289 diff changeset	118 else:
292 6918a0dad359 delegation: delegated stanza are tracked Goffi <goffi@goffi.org> parents: 291 diff changeset	119 del self._current_iqs[id_]
291 61fb4817b77f delegation: iq forwarded management: Goffi <goffi@goffi.org> parents: 289 diff changeset	120 iq_result_elt = toResponse(ori_iq, 'result')
61fb4817b77f delegation: iq forwarded management: Goffi <goffi@goffi.org> parents: 289 diff changeset	121 fwd_elt = iq_result_elt.addElement('delegation', DELEGATION_NS).addElement('forwarded', FORWARDED_NS)
61fb4817b77f delegation: iq forwarded management: Goffi <goffi@goffi.org> parents: 289 diff changeset	122 fwd_elt.addChild(elt)
292 6918a0dad359 delegation: delegated stanza are tracked Goffi <goffi@goffi.org> parents: 291 diff changeset	123 elt.uri = elt.defaultUri = 'jabber:client'
291 61fb4817b77f delegation: iq forwarded management: Goffi <goffi@goffi.org> parents: 289 diff changeset	124 self._xs_send(iq_result_elt)
61fb4817b77f delegation: iq forwarded management: Goffi <goffi@goffi.org> parents: 289 diff changeset	125 else:
61fb4817b77f delegation: iq forwarded management: Goffi <goffi@goffi.org> parents: 289 diff changeset	126 self._xs_send(elt)
61fb4817b77f delegation: iq forwarded management: Goffi <goffi@goffi.org> parents: 289 diff changeset	127
61fb4817b77f delegation: iq forwarded management: Goffi <goffi@goffi.org> parents: 289 diff changeset	128 def _obsWrapper(self, observer, stanza):
61fb4817b77f delegation: iq forwarded management: Goffi <goffi@goffi.org> parents: 289 diff changeset	129 """Wrapper to observer which catch StanzaError
61fb4817b77f delegation: iq forwarded management: Goffi <goffi@goffi.org> parents: 289 diff changeset	130
61fb4817b77f delegation: iq forwarded management: Goffi <goffi@goffi.org> parents: 289 diff changeset	131 @param observer(callable): method to wrap
61fb4817b77f delegation: iq forwarded management: Goffi <goffi@goffi.org> parents: 289 diff changeset	132 """
61fb4817b77f delegation: iq forwarded management: Goffi <goffi@goffi.org> parents: 289 diff changeset	133 try:
61fb4817b77f delegation: iq forwarded management: Goffi <goffi@goffi.org> parents: 289 diff changeset	134 observer(stanza)
61fb4817b77f delegation: iq forwarded management: Goffi <goffi@goffi.org> parents: 289 diff changeset	135 except error.StanzaError as e:
61fb4817b77f delegation: iq forwarded management: Goffi <goffi@goffi.org> parents: 289 diff changeset	136 error_elt = e.toResponse(stanza)
61fb4817b77f delegation: iq forwarded management: Goffi <goffi@goffi.org> parents: 289 diff changeset	137 self._xs_send(error_elt)
61fb4817b77f delegation: iq forwarded management: Goffi <goffi@goffi.org> parents: 289 diff changeset	138 stanza.handled = True
285 a87c155d0fd5 replaced former roster dirty hack by a XEP-0356 first draft implementation, only roster get is implemented so far Goffi <goffi@goffi.org> parents: 283 diff changeset	139
a87c155d0fd5 replaced former roster dirty hack by a XEP-0356 first draft implementation, only roster get is implemented so far Goffi <goffi@goffi.org> parents: 283 diff changeset	140 def onAdvertise(self, message):
291 61fb4817b77f delegation: iq forwarded management: Goffi <goffi@goffi.org> parents: 289 diff changeset	141 """Manage the <message/> advertising delegations"""
287 61f92273fb69 implementation of XEP-0355 (Namespace Delegation) to use SàT Pubsub as PEP service, first draft Goffi <goffi@goffi.org> parents: 286 diff changeset	142 delegation_elt = message.elements(DELEGATION_NS, 'delegation').next()
61f92273fb69 implementation of XEP-0355 (Namespace Delegation) to use SàT Pubsub as PEP service, first draft Goffi <goffi@goffi.org> parents: 286 diff changeset	143 delegated = {}
61f92273fb69 implementation of XEP-0355 (Namespace Delegation) to use SàT Pubsub as PEP service, first draft Goffi <goffi@goffi.org> parents: 286 diff changeset	144 for delegated_elt in delegation_elt.elements(DELEGATION_NS):
285 a87c155d0fd5 replaced former roster dirty hack by a XEP-0356 first draft implementation, only roster get is implemented so far Goffi <goffi@goffi.org> parents: 283 diff changeset	145 try:
287 61f92273fb69 implementation of XEP-0355 (Namespace Delegation) to use SàT Pubsub as PEP service, first draft Goffi <goffi@goffi.org> parents: 286 diff changeset	146 if delegated_elt.name != 'delegated':
61f92273fb69 implementation of XEP-0355 (Namespace Delegation) to use SàT Pubsub as PEP service, first draft Goffi <goffi@goffi.org> parents: 286 diff changeset	147 raise InvalidStanza(u'unexpected element {}'.format(delegated_elt.name))
285 a87c155d0fd5 replaced former roster dirty hack by a XEP-0356 first draft implementation, only roster get is implemented so far Goffi <goffi@goffi.org> parents: 283 diff changeset	148 try:
287 61f92273fb69 implementation of XEP-0355 (Namespace Delegation) to use SàT Pubsub as PEP service, first draft Goffi <goffi@goffi.org> parents: 286 diff changeset	149 namespace = delegated_elt['namespace']
285 a87c155d0fd5 replaced former roster dirty hack by a XEP-0356 first draft implementation, only roster get is implemented so far Goffi <goffi@goffi.org> parents: 283 diff changeset	150 except KeyError:
287 61f92273fb69 implementation of XEP-0355 (Namespace Delegation) to use SàT Pubsub as PEP service, first draft Goffi <goffi@goffi.org> parents: 286 diff changeset	151 raise InvalidStanza(u'was expecting a "namespace" attribute in delegated element')
61f92273fb69 implementation of XEP-0355 (Namespace Delegation) to use SàT Pubsub as PEP service, first draft Goffi <goffi@goffi.org> parents: 286 diff changeset	152 delegated[namespace] = []
61f92273fb69 implementation of XEP-0355 (Namespace Delegation) to use SàT Pubsub as PEP service, first draft Goffi <goffi@goffi.org> parents: 286 diff changeset	153 for attribute_elt in delegated_elt.elements(DELEGATION_NS, 'attribute'):
61f92273fb69 implementation of XEP-0355 (Namespace Delegation) to use SàT Pubsub as PEP service, first draft Goffi <goffi@goffi.org> parents: 286 diff changeset	154 try:
61f92273fb69 implementation of XEP-0355 (Namespace Delegation) to use SàT Pubsub as PEP service, first draft Goffi <goffi@goffi.org> parents: 286 diff changeset	155 delegated[namespace].append(attribute_elt["name"])
61f92273fb69 implementation of XEP-0355 (Namespace Delegation) to use SàT Pubsub as PEP service, first draft Goffi <goffi@goffi.org> parents: 286 diff changeset	156 except KeyError:
61f92273fb69 implementation of XEP-0355 (Namespace Delegation) to use SàT Pubsub as PEP service, first draft Goffi <goffi@goffi.org> parents: 286 diff changeset	157 raise InvalidStanza(u'was expecting a "name" attribute in attribute element')
285 a87c155d0fd5 replaced former roster dirty hack by a XEP-0356 first draft implementation, only roster get is implemented so far Goffi <goffi@goffi.org> parents: 283 diff changeset	158 except InvalidStanza as e:
287 61f92273fb69 implementation of XEP-0355 (Namespace Delegation) to use SàT Pubsub as PEP service, first draft Goffi <goffi@goffi.org> parents: 286 diff changeset	159 log.msg("Invalid stanza received ({})".format(e))
285 a87c155d0fd5 replaced former roster dirty hack by a XEP-0356 first draft implementation, only roster get is implemented so far Goffi <goffi@goffi.org> parents: 283 diff changeset	160
287 61f92273fb69 implementation of XEP-0355 (Namespace Delegation) to use SàT Pubsub as PEP service, first draft Goffi <goffi@goffi.org> parents: 286 diff changeset	161 log.msg(u'delegations updated:\n{}'.format(
61f92273fb69 implementation of XEP-0355 (Namespace Delegation) to use SàT Pubsub as PEP service, first draft Goffi <goffi@goffi.org> parents: 286 diff changeset	162 u'\n'.join([u" - namespace {}{}".format(ns,
61f92273fb69 implementation of XEP-0355 (Namespace Delegation) to use SàT Pubsub as PEP service, first draft Goffi <goffi@goffi.org> parents: 286 diff changeset	163 u"" if not attributes else u" with filtering on {} attribute(s)".format(
61f92273fb69 implementation of XEP-0355 (Namespace Delegation) to use SàT Pubsub as PEP service, first draft Goffi <goffi@goffi.org> parents: 286 diff changeset	164 u", ".join(attributes))) for ns, attributes in delegated.items()])))
242 a6170637690d remote roster partial support Goffi <goffi@goffi.org> parents: diff changeset	165
287 61f92273fb69 implementation of XEP-0355 (Namespace Delegation) to use SàT Pubsub as PEP service, first draft Goffi <goffi@goffi.org> parents: 286 diff changeset	166 if not pubsub.NS_PUBSUB in delegated:
61f92273fb69 implementation of XEP-0355 (Namespace Delegation) to use SàT Pubsub as PEP service, first draft Goffi <goffi@goffi.org> parents: 286 diff changeset	167 log.msg(u"Didn't got pubsub delegation from server, can't act as a PEP service")
242 a6170637690d remote roster partial support Goffi <goffi@goffi.org> parents: diff changeset	168
291 61fb4817b77f delegation: iq forwarded management: Goffi <goffi@goffi.org> parents: 289 diff changeset	169 def onForward(self, iq):
61fb4817b77f delegation: iq forwarded management: Goffi <goffi@goffi.org> parents: 289 diff changeset	170 """Manage forwarded iq
61fb4817b77f delegation: iq forwarded management: Goffi <goffi@goffi.org> parents: 289 diff changeset	171
61fb4817b77f delegation: iq forwarded management: Goffi <goffi@goffi.org> parents: 289 diff changeset	172 @param iq(domish.Element): full delegation stanza
61fb4817b77f delegation: iq forwarded management: Goffi <goffi@goffi.org> parents: 289 diff changeset	173 """
310 e6a9a3c93314 delegation: fixed bad security check which was rejecting all delegations from external servers: Goffi <goffi@goffi.org> parents: 292 diff changeset	174
e6a9a3c93314 delegation: fixed bad security check which was rejecting all delegations from external servers: Goffi <goffi@goffi.org> parents: 292 diff changeset	175 # FIXME: we use a hack supposing that our delegation come from hostname
e6a9a3c93314 delegation: fixed bad security check which was rejecting all delegations from external servers: Goffi <goffi@goffi.org> parents: 292 diff changeset	176 # and we are a component named [name].hostname
e6a9a3c93314 delegation: fixed bad security check which was rejecting all delegations from external servers: Goffi <goffi@goffi.org> parents: 292 diff changeset	177 # but we need to manage properly allowed servers
e6a9a3c93314 delegation: fixed bad security check which was rejecting all delegations from external servers: Goffi <goffi@goffi.org> parents: 292 diff changeset	178 # TODO: do proper origin security check
e6a9a3c93314 delegation: fixed bad security check which was rejecting all delegations from external servers: Goffi <goffi@goffi.org> parents: 292 diff changeset	179 _, allowed = iq['to'].split('.', 1)
e6a9a3c93314 delegation: fixed bad security check which was rejecting all delegations from external servers: Goffi <goffi@goffi.org> parents: 292 diff changeset	180 if jid.JID(iq['from']) != jid.JID(allowed):
e6a9a3c93314 delegation: fixed bad security check which was rejecting all delegations from external servers: Goffi <goffi@goffi.org> parents: 292 diff changeset	181 log.msg((u"SECURITY WARNING: forwarded stanza doesn't come from our server: {}"
e6a9a3c93314 delegation: fixed bad security check which was rejecting all delegations from external servers: Goffi <goffi@goffi.org> parents: 292 diff changeset	182 .format(iq.toXml())).encode('utf-8'))
e6a9a3c93314 delegation: fixed bad security check which was rejecting all delegations from external servers: Goffi <goffi@goffi.org> parents: 292 diff changeset	183 raise error.StanzaError('not-allowed')
e6a9a3c93314 delegation: fixed bad security check which was rejecting all delegations from external servers: Goffi <goffi@goffi.org> parents: 292 diff changeset	184
291 61fb4817b77f delegation: iq forwarded management: Goffi <goffi@goffi.org> parents: 289 diff changeset	185 try:
61fb4817b77f delegation: iq forwarded management: Goffi <goffi@goffi.org> parents: 289 diff changeset	186 fwd_iq = (iq.elements(DELEGATION_NS, 'delegation').next()
61fb4817b77f delegation: iq forwarded management: Goffi <goffi@goffi.org> parents: 289 diff changeset	187 .elements(FORWARDED_NS, 'forwarded').next()
61fb4817b77f delegation: iq forwarded management: Goffi <goffi@goffi.org> parents: 289 diff changeset	188 .elements('jabber:client', 'iq').next())
61fb4817b77f delegation: iq forwarded management: Goffi <goffi@goffi.org> parents: 289 diff changeset	189 except StopIteration:
61fb4817b77f delegation: iq forwarded management: Goffi <goffi@goffi.org> parents: 289 diff changeset	190 raise error.StanzaError('not-acceptable')
61fb4817b77f delegation: iq forwarded management: Goffi <goffi@goffi.org> parents: 289 diff changeset	191
292 6918a0dad359 delegation: delegated stanza are tracked Goffi <goffi@goffi.org> parents: 291 diff changeset	192 managed_entity = jid.JID(fwd_iq['from'])
291 61fb4817b77f delegation: iq forwarded management: Goffi <goffi@goffi.org> parents: 289 diff changeset	193
61fb4817b77f delegation: iq forwarded management: Goffi <goffi@goffi.org> parents: 289 diff changeset	194 self._current_iqs[fwd_iq['id']] = (iq, managed_entity)
292 6918a0dad359 delegation: delegated stanza are tracked Goffi <goffi@goffi.org> parents: 291 diff changeset	195 fwd_iq.delegated = True
6918a0dad359 delegation: delegated stanza are tracked Goffi <goffi@goffi.org> parents: 291 diff changeset	196
6918a0dad359 delegation: delegated stanza are tracked Goffi <goffi@goffi.org> parents: 291 diff changeset	197 # we need a recipient in pubsub request for PEP
6918a0dad359 delegation: delegated stanza are tracked Goffi <goffi@goffi.org> parents: 291 diff changeset	198 # so we set "to" attribute if it doesn't exist
6918a0dad359 delegation: delegated stanza are tracked Goffi <goffi@goffi.org> parents: 291 diff changeset	199 if not fwd_iq.hasAttribute('to'):
6918a0dad359 delegation: delegated stanza are tracked Goffi <goffi@goffi.org> parents: 291 diff changeset	200 fwd_iq["to"] = jid.JID(fwd_iq["from"]).userhost()
6918a0dad359 delegation: delegated stanza are tracked Goffi <goffi@goffi.org> parents: 291 diff changeset	201
291 61fb4817b77f delegation: iq forwarded management: Goffi <goffi@goffi.org> parents: 289 diff changeset	202 # we now inject the element in the stream
61fb4817b77f delegation: iq forwarded management: Goffi <goffi@goffi.org> parents: 289 diff changeset	203 self.xmlstream.dispatch(fwd_iq)
61fb4817b77f delegation: iq forwarded management: Goffi <goffi@goffi.org> parents: 289 diff changeset	204
288 073161f6f143 namespace delegation: disco nesting management Goffi <goffi@goffi.org> parents: 287 diff changeset	205 def getDiscoInfo(self, requestor, target, nodeIdentifier=''):
073161f6f143 namespace delegation: disco nesting management Goffi <goffi@goffi.org> parents: 287 diff changeset	206 """Manage disco nesting
073161f6f143 namespace delegation: disco nesting management Goffi <goffi@goffi.org> parents: 287 diff changeset	207
073161f6f143 namespace delegation: disco nesting management Goffi <goffi@goffi.org> parents: 287 diff changeset	208 This method looks for DiscoHandler in sibling handlers and use it to
073161f6f143 namespace delegation: disco nesting management Goffi <goffi@goffi.org> parents: 287 diff changeset	209 collect main disco infos. It then filters by delegated namespace and return it.
073161f6f143 namespace delegation: disco nesting management Goffi <goffi@goffi.org> parents: 287 diff changeset	210 An identity is added for PEP if pubsub namespace is requested.
073161f6f143 namespace delegation: disco nesting management Goffi <goffi@goffi.org> parents: 287 diff changeset	211
073161f6f143 namespace delegation: disco nesting management Goffi <goffi@goffi.org> parents: 287 diff changeset	212 The same features/identities are returned for main and bare nodes
073161f6f143 namespace delegation: disco nesting management Goffi <goffi@goffi.org> parents: 287 diff changeset	213 """
073161f6f143 namespace delegation: disco nesting management Goffi <goffi@goffi.org> parents: 287 diff changeset	214 if not nodeIdentifier.startswith(DELEGATION_NS):
073161f6f143 namespace delegation: disco nesting management Goffi <goffi@goffi.org> parents: 287 diff changeset	215 return []
073161f6f143 namespace delegation: disco nesting management Goffi <goffi@goffi.org> parents: 287 diff changeset	216
073161f6f143 namespace delegation: disco nesting management Goffi <goffi@goffi.org> parents: 287 diff changeset	217 try:
073161f6f143 namespace delegation: disco nesting management Goffi <goffi@goffi.org> parents: 287 diff changeset	218 _, namespace = nodeIdentifier.split(DELEGATION_MAIN_SEP, 1)
073161f6f143 namespace delegation: disco nesting management Goffi <goffi@goffi.org> parents: 287 diff changeset	219 except ValueError:
073161f6f143 namespace delegation: disco nesting management Goffi <goffi@goffi.org> parents: 287 diff changeset	220 try:
073161f6f143 namespace delegation: disco nesting management Goffi <goffi@goffi.org> parents: 287 diff changeset	221 _, namespace = nodeIdentifier.split(DELEGATION_BARE_SEP, 1)
073161f6f143 namespace delegation: disco nesting management Goffi <goffi@goffi.org> parents: 287 diff changeset	222 except ValueError:
073161f6f143 namespace delegation: disco nesting management Goffi <goffi@goffi.org> parents: 287 diff changeset	223 log.msg("Unexpected disco node: {}".format(nodeIdentifier))
073161f6f143 namespace delegation: disco nesting management Goffi <goffi@goffi.org> parents: 287 diff changeset	224 raise error.StanzaError('not-acceptable')
073161f6f143 namespace delegation: disco nesting management Goffi <goffi@goffi.org> parents: 287 diff changeset	225
073161f6f143 namespace delegation: disco nesting management Goffi <goffi@goffi.org> parents: 287 diff changeset	226 if not namespace:
073161f6f143 namespace delegation: disco nesting management Goffi <goffi@goffi.org> parents: 287 diff changeset	227 log.msg("No namespace found in node {}".format(nodeIdentifier))
073161f6f143 namespace delegation: disco nesting management Goffi <goffi@goffi.org> parents: 287 diff changeset	228 return []
073161f6f143 namespace delegation: disco nesting management Goffi <goffi@goffi.org> parents: 287 diff changeset	229
073161f6f143 namespace delegation: disco nesting management Goffi <goffi@goffi.org> parents: 287 diff changeset	230 def gotInfos(infos):
073161f6f143 namespace delegation: disco nesting management Goffi <goffi@goffi.org> parents: 287 diff changeset	231 ns_features = []
073161f6f143 namespace delegation: disco nesting management Goffi <goffi@goffi.org> parents: 287 diff changeset	232 for info in infos:
073161f6f143 namespace delegation: disco nesting management Goffi <goffi@goffi.org> parents: 287 diff changeset	233 if isinstance(info, disco.DiscoFeature) and info.startswith(namespace):
073161f6f143 namespace delegation: disco nesting management Goffi <goffi@goffi.org> parents: 287 diff changeset	234 ns_features.append(info)
289 f08f8536cab8 mod delegation: extensions management (XEP-0128) Goffi <goffi@goffi.org> parents: 288 diff changeset	235 elif (isinstance(info, data_form.Form) and info.formNamespace
f08f8536cab8 mod delegation: extensions management (XEP-0128) Goffi <goffi@goffi.org> parents: 288 diff changeset	236 and info.formNamespace.startwith(namespace)):
f08f8536cab8 mod delegation: extensions management (XEP-0128) Goffi <goffi@goffi.org> parents: 288 diff changeset	237 # extensions management (XEP-0128)
f08f8536cab8 mod delegation: extensions management (XEP-0128) Goffi <goffi@goffi.org> parents: 288 diff changeset	238 ns_features.append(info)
288 073161f6f143 namespace delegation: disco nesting management Goffi <goffi@goffi.org> parents: 287 diff changeset	239
073161f6f143 namespace delegation: disco nesting management Goffi <goffi@goffi.org> parents: 287 diff changeset	240 if namespace == pubsub.NS_PUBSUB:
073161f6f143 namespace delegation: disco nesting management Goffi <goffi@goffi.org> parents: 287 diff changeset	241 ns_features.append(disco.DiscoIdentity('pubsub', 'pep'))
073161f6f143 namespace delegation: disco nesting management Goffi <goffi@goffi.org> parents: 287 diff changeset	242
073161f6f143 namespace delegation: disco nesting management Goffi <goffi@goffi.org> parents: 287 diff changeset	243 return ns_features
073161f6f143 namespace delegation: disco nesting management Goffi <goffi@goffi.org> parents: 287 diff changeset	244
073161f6f143 namespace delegation: disco nesting management Goffi <goffi@goffi.org> parents: 287 diff changeset	245 for handler in self.parent.handlers:
073161f6f143 namespace delegation: disco nesting management Goffi <goffi@goffi.org> parents: 287 diff changeset	246 if isinstance(handler, disco.DiscoHandler):
073161f6f143 namespace delegation: disco nesting management Goffi <goffi@goffi.org> parents: 287 diff changeset	247 break
073161f6f143 namespace delegation: disco nesting management Goffi <goffi@goffi.org> parents: 287 diff changeset	248
073161f6f143 namespace delegation: disco nesting management Goffi <goffi@goffi.org> parents: 287 diff changeset	249 if not isinstance(handler, disco.DiscoHandler):
073161f6f143 namespace delegation: disco nesting management Goffi <goffi@goffi.org> parents: 287 diff changeset	250 log.err("Can't find DiscoHandler")
073161f6f143 namespace delegation: disco nesting management Goffi <goffi@goffi.org> parents: 287 diff changeset	251 return []
073161f6f143 namespace delegation: disco nesting management Goffi <goffi@goffi.org> parents: 287 diff changeset	252
073161f6f143 namespace delegation: disco nesting management Goffi <goffi@goffi.org> parents: 287 diff changeset	253 d = handler.info(requestor, target, '')
073161f6f143 namespace delegation: disco nesting management Goffi <goffi@goffi.org> parents: 287 diff changeset	254 d.addCallback(gotInfos)
073161f6f143 namespace delegation: disco nesting management Goffi <goffi@goffi.org> parents: 287 diff changeset	255 return d
073161f6f143 namespace delegation: disco nesting management Goffi <goffi@goffi.org> parents: 287 diff changeset	256
073161f6f143 namespace delegation: disco nesting management Goffi <goffi@goffi.org> parents: 287 diff changeset	257 def getDiscoItems(self, requestor, target, nodeIdentifier=''):
073161f6f143 namespace delegation: disco nesting management Goffi <goffi@goffi.org> parents: 287 diff changeset	258 return []

Mercurial > libervia-pubsub

annotate sat_pubsub/delegation.py @ 318:d13526c0eb32