Spark-api/travis.js at master · Spark-Core/Spark-api · 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
const discord = require("discord.js")
const options = require("./webhook-options.json")
const hook = new discord.WebhookClient(options.hookId, options.hookToken);
const request = require("request-promise")
require("request")
const crypto = require("crypto")
const moment = require("moment")

module.exports = async (req, res) => {
    if (!req.body.payload) {
        return res.sendStatus(404)
    }
    try {
        payload = JSON.parse(req.body.payload)
        payload.signature = Buffer.from(req.headers.signature, 'base64');
        var key;
        if (payload.build_url.startsWith("https://travis-ci.com/")) {
            key = await request({
                uri: "https://api.travis-ci.com/config",
                json: true
            })
        } else {
            key = await request({
                uri: "https://api.travis-ci.org/config",
                json: true
            })
        }
        key = key.config.notifications.webhook.public_key;
        let verifier = crypto.createVerify('sha1');
        verifier.update(req.body.payload);
        status = verifier.verify(key, payload.signature);
        if (!status) {
            return res.sendStatus(403)
        } else if (options.whitelist.length != 0 && !options.whitelist.includes(payload.repository.id)) {
            return res.sendStatus(401)
        }
        req.buildInfo = payload;
        next(req, res)
    } catch (e) {
        console.log(e)
        return res.sendStatus(403)
    }
}

function next(req, res) {
    const embed = new discord.RichEmbed;
    hook.edit(options.name || "Travis status update", options.icon || "https://cdn.discordapp.com/attachments/347923404357107712/433693539805691905/travis4x.png")

    if (req.buildInfo.pull_request) {
        embed.setFooter("Pull request #" + req.buildInfo.pull_request_number + " (" + req.buildInfo.commit + ")", "https://cdn.discordapp.com/attachments/347923404357107712/433673027385819148/Scheme-256.png")
    } else {
        embed.setFooter("Push (" + req.buildInfo.commit + ")", "https://cdn.discordapp.com/attachments/347923404357107712/433673144218419210/commit-128.png")
    }
    if (req.buildInfo.state == "started") {
        embed.setAuthor("Build #" + req.buildInfo.number + " has started! | Repository: " + req.buildInfo.repository.name, "https://cdn.discordapp.com/attachments/347923404357107712/462750913832943616/68747470733a2f2f7472617669732d63692e636f6d2f696d616765732f7374726f6b652d69636f6e732f69636f6e2d72756e.png", req.buildInfo.build_url)
        message = req.buildInfo.message.substring(0, 2000)
        embed.setDescription("Commit body:\n" + message)
        embed.setColor(0xedde3f)
    } else if (req.buildInfo.state == "passed") {
        embed.setAuthor("Build #" + req.buildInfo.number + " has completed succesfully. | Repository: " + req.buildInfo.repository.name, "https://cdn.discordapp.com/attachments/347923404357107712/462752187135557645/check.png", req.buildInfo.build_url)
        try {
            var duration = moment.duration(req.buildInfo.duration, "seconds")
            var time = Math.floor(duration.asHours()) + moment.utc(duration.asMilliseconds()).format(":mm:ss")
            embed.setDescription("Duration: \n**" + time + "**")
        } catch (e) {
            console.log("Couldn't parse duration value.", e)
            embed.setDescription("Couldn't parse duration value.")
        }
        embed.setColor(0x39aa56)
    } else if (req.buildInfo.state == "failed") {
        embed.setAuthor("Build #" + req.buildInfo.number + " has failed. | Repository: " + req.buildInfo.repository.name, "https://cdn.discordapp.com/attachments/347923404357107712/462752914369150986/cross.png", req.buildInfo.build_url)
        embed.setDescription("Check [your console](" + req.buildInfo.build_url + ") for errors")
        embed.setColor(0xdb4545)
    } else if (req.buildInfo.state == "canceled") {
        embed.setTitle("Build #" + req.buildInfo.number + " has been canceled. | Repository: " + req.buildInfo.repository.name)
        embed.setDescription("This build has been canceled.")
        embed.setColor(0x9d9d9d)
    } else {
        embed.setTitle("Invalid state (" + req.buildInfo.state + ") | Repository: " + req.buildInfo.repository.name)
        embed.setDescription("Please see the console for more information.")
    }
    embed.addField("Links", "[Commit **" + req.buildInfo.commit.substring(0, 6) + "**](https://github.com/" + req.buildInfo.repository.owner_name + "/" + req.buildInfo.repository.name + "/commit/" + req.buildInfo.commit + ")\n[Compare](" + req.buildInfo.compare_url + ")\n[Build info](" + req.buildInfo.build_url + ")\n[Github repo](https://github.com/" +
        req.buildInfo.repository.owner_name + "/" + req.buildInfo.repository.name + ")")
    hook.send("", {
            embeds: [embed]
        })
        .then(() => {
            return res.sendStatus(200)
        })
        .catch(() => {
            return res.sendStatus(500)
        })
}