updatebot/bot.py at master · genesisproject/updatebot · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
#!/usr/bin/env python3
"""
A bot for pushing Github updates to a Discord channel

Author: Erin <sylphofelectricity@gmail.com>
"""

import os
import asyncio
import threading
import functools
from flask import Flask, request
import discord
import requests

DISCORD_SERVER = os.environ["DISCORD_SERVER"]
DISCORD_CHANNEL = os.environ["DISCORD_CHANNEL"]
DISCORD_TOKEN = os.environ["DISCORD_TOKEN"]

def shorten_url(url):
    r = requests.post('https://v.gd/create.php', data={'format': 'simple', 'url': url})
    return r.text

def gl_push_message(data):
    repo = data['repository']['name']
    commits = data['commits']
    branch = data['ref'].split('/')[-1]

    message = "[**{repo}**][*{branch}*] {n_commits} new {commit}:".format(repo=repo,
                        n_commits=len(commits), branch=branch,
                        commit="commits" if len(commits) > 1 else "commit")
    for commit in commits:
        author = commit['author']
        shorturl = shorten_url(commit['url'])
        message += "\n<{url}>: {hash} ↪ {description} ↬ {branch} ↯ {user}".format(url=shorturl,
                        hash=commit['id'][0:7],
                        description=commit['message'].splitlines()[0],
                        branch=branch,
                        user=author['name'])

    return message

gl_message_template_functions = {
    "push hook": gl_push_message
}

app = Flask(__name__)
dc = discord.Client()
loop = asyncio.new_event_loop()
def start_discord_client():
    asyncio.set_event_loop(loop)
    print("Starting Discord client")
    dc.run(DISCORD_TOKEN)
t = threading.Thread(target=start_discord_client)
t.start()

@dc.event
@asyncio.coroutine
def on_ready():
    print('Logged in as')
    print(client.user.name)
    print(client.user.id)
    print('------')

@app.route('/gl-hook', methods=["POST"])
def gl_hook():
    event_type = request.headers.get('X-Gitlab-Event').lower()
    data = request.get_json()
    print('Gitlab hook route called.')
    print('Got a webhook request for event type: %s' % event_type)

    # get a message to send to the Discord channel
    if event_type in gl_message_template_functions:
        message = gl_message_template_functions[event_type](data)
    else:
        print('Unhandled event of type %s' % event_type)
        return ''

    print(message)
    server = discord.utils.get(dc.servers, name=DISCORD_SERVER)
    channel = discord.utils.get(server.channels, name=DISCORD_CHANNEL)
    print("Sending message to channel %s on server %s" % (channel, server))

    asyncio.run_coroutine_threadsafe(dc.send_message(channel, message), dc.loop).result()

    return ''


def ucb_build_status(status, data):
    project = data['projectName']
    build_target = data['buildTargetName']
    build_number = data['buildNumber']

    status_emoji = {'succeeded': '☑', 'failed': '☒', 'canceled': '©'}

    return "{status_emoji} [**{project}**] Build #{build_n} (target:{target}) {status}".format(status_emoji=status_emoji.get(status, ''),
                            project=project, build_n=build_number, target=build_target, status=status)

ucb_message_template_functions = {
    'ProjectBuildStarted': functools.partial(ucb_build_status, 'started'),
    'ProjectBuildSuccess': functools.partial(ucb_build_status, 'succeeded'),
    'ProjectBuildFailure': functools.partial(ucb_build_status, 'failed'),
    'ProjectBuildCanceled': functools.partial(ucb_build_status, 'canceled')
}

@app.route('/ucb-hook', methods=["POST"])
def ucb_hook():
    event_type = request.headers.get('X-UnityCloudBuild-Event')
    data = request.get_json()
    print('Unity Cloud Build hook route called')
    print('Got a webhook request for event type: %s' % event_type)

    # get a message to send to the Discord channel
    if event_type in ucb_message_template_functions:
        message = ucb_message_template_functions[event_type](data)
    else:
        print('Unhandled event of type %s' % event_type)
        return ''

    print(message)
    server = discord.utils.get(dc.servers, name=DISCORD_SERVER)
    channel = discord.utils.get(server.channels, name=DISCORD_CHANNEL)
    print("Sending message to channel %s on server %s" % (channel, server))

    asyncio.run_coroutine_threadsafe(dc.send_message(channel, message), dc.loop).result()

    return ''