Mercurial > libervia-backend
view sat/bridge/pb.py @ 2776:838f53730ce4
plugin pubsub admin: pubsub administrator first draft:
This experimental protocol allows to publish items where publishers can be specified.
author | Goffi <goffi@goffi.org> |
---|---|
date | Tue, 15 Jan 2019 08:51:54 +0100 |
parents | 003b8b4b56a7 |
children | 0f277708e2ae |
line wrap: on
line source
#!/usr/bin/env python2 # -*- coding: utf-8 -*- # SAT: a jabber client # Copyright (C) 2009-2019 Jérôme Poisson (goffi@goffi.org) # This program is free software: you can redistribute it and/or modify # it under the terms of the GNU Affero General Public License as published by # the Free Software Foundation, either version 3 of the License, or # (at your option) any later version. # This program 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 Affero General Public License for more details. # You should have received a copy of the GNU Affero General Public License # along with this program. If not, see <http://www.gnu.org/licenses/>. from sat.core.log import getLogger log = getLogger(__name__) from twisted.spread import jelly, pb from twisted.internet import reactor ## jelly hack # we monkey patch jelly to handle namedtuple ori_jelly = jelly._Jellier.jelly def fixed_jelly(self, obj): """this method fix handling of namedtuple""" if isinstance(obj, tuple) and not obj is tuple: obj = tuple(obj) return ori_jelly(self, obj) jelly._Jellier.jelly = fixed_jelly class PBRoot(pb.Root): def __init__(self): self.signals_handlers = [] def remote_initBridge(self, signals_handler): self.signals_handlers.append(signals_handler) log.info(u"registered signal handler") def sendSignalEb(self, failure, signal_name): log.error( u"Error while sending signal {name}: {msg}".format( name=signal_name, msg=failure ) ) def sendSignal(self, name, args, kwargs): to_remove = [] for handler in self.signals_handlers: try: d = handler.callRemote(name, *args, **kwargs) except pb.DeadReferenceError: to_remove.append(handler) else: d.addErrback(self.sendSignalEb, name) if to_remove: for handler in to_remove: log.debug(u"Removing signal handler for dead frontend") self.signals_handlers.remove(handler) ##METHODS_PART## class Bridge(object): def __init__(self): log.info("Init Perspective Broker...") self.root = PBRoot() reactor.listenTCP(8789, pb.PBServerFactory(self.root)) def sendSignal(self, name, *args, **kwargs): self.root.sendSignal(name, args, kwargs) def remote_initBridge(self, signals_handler): self.signals_handlers.append(signals_handler) log.info(u"registered signal handler") def register_method(self, name, callback): log.debug("registering PB bridge method [%s]" % name) setattr(self.root, "remote_" + name, callback) # self.root.register_method(name, callback) def addMethod(self, name, int_suffix, in_sign, out_sign, method, async=False, doc={}): """Dynamically add a method to PB Bridge""" # FIXME: doc parameter is kept only temporary, the time to remove it from calls log.debug("Adding method {name} to PB bridge".format(name=name)) self.register_method(name, method) def addSignal(self, name, int_suffix, signature, doc={}): log.debug("Adding signal {name} to PB bridge".format(name=name)) setattr( self, name, lambda *args, **kwargs: self.sendSignal(name, *args, **kwargs) ) def actionNew(self, action_data, id, security_limit, profile): self.sendSignal("actionNew", action_data, id, security_limit, profile) def connected(self, profile, jid_s): self.sendSignal("connected", profile, jid_s) def contactDeleted(self, entity_jid, profile): self.sendSignal("contactDeleted", entity_jid, profile) def disconnected(self, profile): self.sendSignal("disconnected", profile) def entityDataUpdated(self, jid, name, value, profile): self.sendSignal("entityDataUpdated", jid, name, value, profile) def messageEncryptionStarted(self, to_jid, encryption_data, profile_key): self.sendSignal("messageEncryptionStarted", to_jid, encryption_data, profile_key) def messageEncryptionStopped(self, to_jid, encryption_data, profile_key): self.sendSignal("messageEncryptionStopped", to_jid, encryption_data, profile_key) def messageNew(self, uid, timestamp, from_jid, to_jid, message, subject, mess_type, extra, profile): self.sendSignal("messageNew", uid, timestamp, from_jid, to_jid, message, subject, mess_type, extra, profile) def newContact(self, contact_jid, attributes, groups, profile): self.sendSignal("newContact", contact_jid, attributes, groups, profile) def paramUpdate(self, name, value, category, profile): self.sendSignal("paramUpdate", name, value, category, profile) def presenceUpdate(self, entity_jid, show, priority, statuses, profile): self.sendSignal("presenceUpdate", entity_jid, show, priority, statuses, profile) def progressError(self, id, error, profile): self.sendSignal("progressError", id, error, profile) def progressFinished(self, id, metadata, profile): self.sendSignal("progressFinished", id, metadata, profile) def progressStarted(self, id, metadata, profile): self.sendSignal("progressStarted", id, metadata, profile) def subscribe(self, sub_type, entity_jid, profile): self.sendSignal("subscribe", sub_type, entity_jid, profile)