backend/src/appenlight/models/integrations/flowdock.py

# -*- coding: utf-8 -*-

# Copyright 2010 - 2017 RhodeCode GmbH and the AppEnlight project authors
#
# 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.

import logging

import requests
from requests.auth import HTTPBasicAuth
import simplejson as json

from appenlight.models.integrations import IntegrationBase, IntegrationException

_ = str

log = logging.getLogger(__name__)


class NotFoundException(Exception):
    pass


class FlowdockIntegration(IntegrationBase):
    __mapper_args__ = {"polymorphic_identity": "flowdock"}
    front_visible = False
    as_alert_channel = True
    supports_report_alerting = True
    action_notification = True
    integration_action = "Message via Flowdock"

    @classmethod
    def create_client(cls, api_token):
        client = FlowdockClient(api_token)
        return client


class FlowdockClient(object):
    def __init__(self, api_token):
        self.auth = HTTPBasicAuth(api_token, "")
        self.api_token = api_token
        self.api_url = "https://api.flowdock.com/v1/messages"

    def make_request(self, url, method="get", data=None):
        headers = {
            "Content-Type": "application/json",
            "User-Agent": "appenlight-flowdock",
        }
        try:
            if data:
                data = json.dumps(data)
            resp = getattr(requests, method)(
                url, data=data, headers=headers, timeout=10
            )
        except Exception as e:
            raise IntegrationException(
                _("Error communicating with Flowdock: %s") % (e,)
            )
        if resp.status_code > 299:
            raise IntegrationException(resp.text)
        return resp

    def send_to_chat(self, payload):
        url = "%(api_url)s/chat/%(api_token)s" % {
            "api_url": self.api_url,
            "api_token": self.api_token,
        }
        return self.make_request(url, method="post", data=payload).json()

    def send_to_inbox(self, payload):
        f_args = {"api_url": self.api_url, "api_token": self.api_token}
        url = "%(api_url)s/team_inbox/%(api_token)s" % f_args
        return self.make_request(url, method="post", data=payload).json()