forked-synapse/synapse/push/httppusher.py

# -*- coding: utf-8 -*-
# Copyright 2014 OpenMarket Ltd
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

from synapse.push import Pusher, PusherConfigException
from synapse.http.client import SimpleHttpClient

from twisted.internet import defer

import logging

logger = logging.getLogger(__name__)


class HttpPusher(Pusher):
    def __init__(self, _hs, user_name, app_id,
                 app_display_name, device_display_name, pushkey, data,
                 last_token, last_success, failing_since):
        super(HttpPusher, self).__init__(
            _hs,
            user_name,
            app_id,
            app_display_name,
            device_display_name,
            pushkey,
            data,
            last_token,
            last_success,
            failing_since
        )
        if 'url' not in data:
            raise PusherConfigException(
                "'url' required in data for HTTP pusher"
            )
        self.url = data['url']
        self.httpCli = SimpleHttpClient(self.hs)
        self.data_minus_url = {}
        self.data_minus_url.update(self.data)
        del self.data_minus_url['url']

    def _build_notification_dict(self, event):
        # we probably do not want to push for every presence update
        # (we may want to be able to set up notifications when specific
        # people sign in, but we'd want to only deliver the pertinent ones)
        # Actually, presence events will not get this far now because we
        # need to filter them out in the main Pusher code.
        if 'event_id' not in event:
            return None

        return {
            'notification': {
                'transition': 'new',
                # everything is new for now: we don't have read receipts
                'id': event['event_id'],
                'type': event['type'],
                'from': event['user_id'],
                # we may have to fetch this over federation and we
                # can't trust it anyway: is it worth it?
                #'fromDisplayName': 'Steve Stevington'
                #'counts': { -- we don't mark messages as read yet so
                # we have no way of knowing
                #    'unread': 1,
                #    'missedCalls': 2
                # },
                'devices': [
                    {
                        'app_id': self.app_id,
                        'pushkey': self.pushkey,
                        'data': self.data_minus_url
                    }
                ]
            }
        }

    @defer.inlineCallbacks
    def dispatch_push(self, event):
        notification_dict = self._build_notification_dict(event)
        if not notification_dict:
            defer.returnValue(True)
        try:
            yield self.httpCli.post_json_get_json(self.url, notification_dict)
        except:
            logger.exception("Failed to push %s ", self.url)
            defer.returnValue(False)
        defer.returnValue(True)