bridge.py

########################################
#   Bitbucket Mattermost Bridge        #
#                                      #
#   Written by Daniel Kappelle, 2016   #
########################################

# import modules
import json
import requests
from flask import Flask, request
from pyxtension.Json import Json

# import user config
import config
import payload

# Initialize flask app
app = Flask(__name__)

# Folder for the template files
template_folder = "templates/"


@app.route("/")
def home_page():
    tpl = '''
    <h1>Bitbucket Mattermost Bridge</h1>
    <p>Please refer to the repo on
    <a href='https://github.com/danielkappelle/bitbucket-mattermost-bridge'>
        Github
    </a> for the Readme.</p>
    '''
    return tpl


@app.route("/hooks/<hook>", methods=['GET', 'POST'])
def bridge_hook(hook):
        # This function does all the magic

        # The event key is used to determine the type of event
        # e.g. repo:push, issue:created, etc.
        event = request.headers.get('X-Event-Key')

        # The template folder is searched for a template file
        # that matches thee event-key, (: replaced by -), e.g.
        # repo-push
        payload_name = event.replace(":", "_")
        payload_func = getattr(payload, payload_name, None)

        if payload_func:
            # Parse the json data from the webhook
            data = Json(request.get_json())
            output = payload_func(data)
            # Submit the new, bridged, webhook to the mattermost
            # incoming webhook
            submit_hook(config.webhook_url + hook, output)
            return "Done"
        else:
            # In case there's no templat for the event
            # throw an error
            print(event)
            return "Couldn't handle this event", 501


def submit_hook(url, hook_data):
    # This function submits the new hook to mattermost
    data = {
        'attachments': [hook_data],
        'username': config.username,
        'icon_url': config.icon,
    }
    # Post the webhook
    response = requests.post(
        url,
        data=json.dumps(data),
        headers={'Content-Type': 'application/json'}
    )
    if response.status_code != 200:
        err = 'Request to mattermost returned an error %s, the response is:\n%s'
        raise ValueError(err % (response.status_code, response.text))

if __name__ == "__main__":
        # Run flask app on host, this is set in config.py
        app.run(host=config.host, port=config.port)