Merge pull request #8 from williamkapke/evented-approach

emit github events

Author: williamkapke

Procfile

@@ -0,0 +1 @@
+web: node server

lib/github-events.js

@@ -0,0 +1,39 @@
+const crypto = require('crypto')
+const debug = require('debug')('github')
+
+const secret = process.env.GITHUB_WEBHOOK_SECRET || 'hush-hush'
+
+const sign = (secret, data) => {
+  const buffer = new Buffer(data, 'utf8')
+  return 'sha1=' + crypto.createHmac('sha1', secret).update(buffer).digest('hex')
+}
+
+module.exports = (app) => {
+  app.post('/hooks/github', (req, res) => {
+    const event = req.headers['x-github-event']
+    if (!event) {
+      res.writeHead(400, 'Event Header Missing')
+      return res.end()
+    }
+
+    const signature = req.headers['x-hub-signature']
+    if (!signature || signature !== sign(secret, req.raw)) {
+      res.writeHead(401, 'Invalid Signature')
+      return res.end()
+    }
+
+    res.end()
+
+    const data = req.body
+    data.action = data.action ? event + '.' + data.action : event
+
+    var source = data.repository ? data.repository.full_name : data.organization.login
+    console.log('event@%s: %s', source, data.action)
+
+    if (debug.enabled) {
+      debug(JSON.stringify(data, null, 2))
+    }
+
+    app.emit(data.action, data)
+  })
+}

package.json

@@ -13,6 +13,8 @@
   "license": "MIT",
   "dependencies": {
     "body-parser": "^1.15.0",
+    "debug": "^2.2.0",
+    "dotenv": "^2.0.0",
     "express": "^4.13.4",
     "github": "^0.2.4",
     "travis-ci": "^2.1.0"

scripts/display-travis-status.js

@@ -0,0 +1,27 @@
+'use strict'
+
+const debug = require('debug')('display_travis_status')
+const pollTravis = require('../lib/pollTravis')
+const enabledRepos = ['citgm', 'readable-stream', 'nodejs.org']
+
+module.exports = function (app) {
+  app.on('pull_request.opened', (event) => {
+    const owner = event.repository.owner.login
+    const repo = event.repository.name
+    if (!~enabledRepos.indexOf(repo)) return
+
+    debug(`/${owner}/${repo}/pull/${event.number} opened`)
+    pollTravis.pollThenComment(owner, repo, event.number)
+  })
+
+  // to trigger polling manually
+  app.get('/pr/:owner/:repo/:id', (req, res) => {
+    const owner = req.params.owner
+    const repo = req.params.repo
+    const id = req.params.id
+    if (~enabledRepos.indexOf(repo)) {
+      pollTravis.pollThenComment(owner, repo, parseInt(id, 10))
+    }
+    res.end()
+  })
+}

server.js

@@ -1,39 +1,23 @@
 'use strict'
 
+require('dotenv').load({ silent: true })
+
 const express = require('express')
 const bodyParser = require('body-parser')
-
-const pollTravis = require('./lib/pollTravis')
+const captureRaw = (req, res, buffer) => { req.raw = buffer }
 
 const app = express()
-
-const port = process.env.PORT || 3000
-
-app.use(bodyParser.json())
-
-app.all('/hooks/github', (req, res) => {
-  if (wasPullRequestOpened(req)) {
-    const repo = req.body.repository
-
-    console.log(`* ${repo.owner.login}/${repo.name}/#${req.body.number} Opened, starting build checks!`)
-    pollTravis.pollThenComment(repo.owner.login, repo.name, parseInt(req.body.number))
-  }
-
-  res.end()
-})
-
-// to trigger polling manually
-app.get('/pr/:owner/:repo/:prId', (req, res) => {
-  pollTravis.pollThenComment(req.params.owner, req.params.repo, parseInt(req.params.prId))
-  res.end()
+app.use(bodyParser.json({ verify: captureRaw }))
+require('./lib/github-events.js')(app)
+
+// load all the files in the scripts folder
+require('fs').readdirSync('./scripts').forEach((file) => {
+  file = './scripts/' + file
+  console.log('loading:', file)
+  require(file)(app)
 })
 
-app.listen(process.env.PORT || 3000, () => {
+const port = process.env.PORT || 3000
+app.listen(port, () => {
   console.log('Example app listening on port', port)
 })
-
-function wasPullRequestOpened (req) {
-  const githubEvent = req.headers['x-github-event'] || ''
-  const githubAction = req.body.action || ''
-  return githubEvent === 'pull_request' && githubAction === 'opened'
-}