app.py

#!/usr/bin/env python3

import json
from flask import Flask
from flask import request
from flask import jsonify
from bot import Bot
app = Flask(__name__)

class GitlabBot(Bot):
    def __init__(self):
        try:
            self.authmsg = open('authmsg').read().strip()
        except:
            raise Exception("The authorization messsage file is invalid")

        super(GitlabBot, self).__init__()
        self.chats = {}
        try:
            chats = open('chats', 'r').read()
            self.chats = json.loads(chats)
        except:
            open('chats', 'w').write(json.dumps(self.chats))

        self.send_to_all('Hi !')

    def text_recv(self, txt, chatid):
        ''' registering chats '''
        txt = txt.strip()
        if txt.startswith('/'):
            txt = txt[1:]
        if txt == self.authmsg:
            if str(chatid) in self.chats:
                self.reply(chatid, "\U0001F60E  boy, you already got the power.")
            else:
                self.reply(chatid, "\U0001F60E  Ok boy, you got the power !")
                self.chats[chatid] = True
                open('chats', 'w').write(json.dumps(self.chats))
        elif txt == 'shutupbot':
            del self.chats[chatid]
            self.reply(chatid, "\U0001F63F Ok, take it easy\nbye.")
            open('chats', 'w').write(json.dumps(self.chats))

    def send_to_all(self, msg):
        for c in self.chats:
            self.reply(c, msg)


b = GitlabBot()


@app.route("/", methods=['GET', 'POST'])
def webhook():
    data = request.json
    # json contains an attribute that differenciates between the types, see
    # https://docs.gitlab.com/ce/user/project/integrations/webhooks.html
    # for more infos
    kind = data['object_kind']
    if kind == 'push':
        msg = generatePushMsg(data)
    elif kind == 'tag_push':
        msg = generatePushMsg(data)  # TODO:Make own function for this
    elif kind == 'issue':
        msg = generateIssueMsg(data)
    elif kind == 'note':
        msg = generateCommentMsg(data)
    elif kind == 'merge_request':
        msg = generateMergeRequestMsg(data)
    elif kind == 'wiki_page':
        msg = generateWikiMsg(data)
    elif kind == 'pipeline':
        msg = generatePipelineMsg(data)
    elif kind == 'build':
        msg = generateBuildMsg(data)
    b.send_to_all(msg)
    return jsonify({'status': 'ok'})


def getAssignees(data):
    assignees = ''
    for assignee in data.get('assignees', []):
        assignees += assignee['name'] + ' '
    return assignees


def generatePushMsg(data):
    msg = '*{0} ({1}) - {2} new commits*\n'\
        .format(data['project']['name'], data['project']['default_branch'], data['total_commits_count'])
    for commit in data['commits']:
        msg = msg + '----------------------------------------------------------------\n'
        msg = msg + commit['message'].rstrip()
        msg = msg + '\n' + commit['url'].replace("_", "\_") + '\n'
    msg = msg + '----------------------------------------------------------------\n'
    return msg


def generateIssueMsg(data):
    object_attributes = data['object_attributes']
    action = object_attributes['action']
    if action == 'open':
        assignees = getAssignees(data)
        msg = '*{0}* new issue for *{2}*:\n'\
            .format(
                data['project']['name'],
                object_attributes['iid'],
                assignees
            )
    elif action == 'reopen':
        assignees = getAssignees(data)
        msg = '*{0}* issue re-opened by *{1}* for *{2}*:\n'\
            .format(
                data['project']['name'],
                data['user']['name'],
                assignees
            )
    elif action == 'close':
        msg = '*{0}* issue closed by *{1}*:\n'\
            .format(
                data['project']['name'],
                data['user']['name']
            )
    elif action == 'update':
        msg = '*{0}* issue updated by *{1}*:\n'\
            .format(
                data['project']['name'],
                data['user']['name']
            )

    msg = msg + '[{0}]({1})'\
        .format(object_attributes['title'], object_attributes['url'])
    return msg


def generateCommentMsg(data):
    object_attributes = data['object_attributes']
    ntype = object_attributes['noteable_type']
    if ntype == 'Commit':
        msg = 'note to commit'
    elif ntype == 'MergeRequest':
        msg = 'note to MergeRequest'
    elif ntype == 'Issue':
        issue = data['issue']
        msg = 'New comment from *{0}*:\n{1}\n'\
            .format(
                data['user']['name'],
                object_attributes["description"]
            )
        msg = msg + '\nIssue #{2} - [{0}]({1})'\
            .format(
                issue['title'],
                object_attributes['url'],
                issue['iid']
            )
    elif ntype == 'Snippet':
        msg = 'note on code snippet'
    return msg


def generateMergeRequestMsg(data):
    return 'new MergeRequest'


def generateWikiMsg(data):
    return 'new wiki stuff'


def generatePipelineMsg(data):
    return 'new pipeline stuff'


def generateBuildMsg(data):
    return 'new build stuff'


if __name__ == "__main__":
    b.run_threaded()
    app.run(host='0.0.0.0', port=10111)