Initial commit

Author: phillipj

.gitignore

@@ -0,0 +1,4 @@
+node_modules
+npm-debug.log
+
+.DS_Store

README.md

@@ -0,0 +1,11 @@
+# Node.js GitHub Bot
+
+What's this?? ATM an experiment to see if a bot could be helpful for the Node.js project, to automate all the things!
+
+## Contributing
+
+Please do, contributions are more than welcome!
+
+## License
+
+MIT

lib/pollTravis.js

@@ -0,0 +1,85 @@
+'use strict'
+
+const GitHub = require('github')
+const Travis = require('travis-ci')
+
+const travisClient = new Travis({
+  version: '2.0.0',
+  access_token: process.env.TRAVIS_CI_TOKEN
+})
+const githubClient = new GitHub({
+  version: '3.0.0',
+  protocol: 'https',
+  host: 'api.github.com',
+  timeout: 5 * 1000,
+  headers: {
+    'user-agent': 'Node.js GitHub Bot v1.0-beta'
+  }
+})
+
+githubClient.authenticate({
+  type: 'oauth',
+  token: process.env.GITHUB_TOKEN
+})
+
+function pollAndComment (owner, repoName, prId, checkNumber) {
+  checkNumber = checkNumber || 1
+
+  const prInfo = prInfoStr({ owner, repoName, prId })
+  const createGhComment = createGhCommentFn({ owner, repoName, prId })
+
+  if (checkNumber > 100) {
+    console.warn(`* ${prInfo} Was not able to find matching build for PR, stopping poll now :(`)
+    return
+  }
+
+  travisClient.repos(owner, repoName).builds.get((err, res) => {
+    if (err) {
+      return console.error(`! ${prInfo} Error while retrieving builds`, err.stack)
+    }
+
+    const lastBuildForPr = res.builds.find((build) => build.pull_request_number === prId)
+
+    if (lastBuildForPr) {
+      const lastState = lastBuildForPr.state
+
+      if (lastState === 'passed') {
+        return createGhComment(`[Travis build passed](https://travis-ci.org/${owner}/${repoName}/builds/${lastBuildForPr.id}) :+1:`)
+      } else if (lastState === 'failed') {
+        return createGhComment(`[Travis build failed](https://travis-ci.org/${owner}/${repoName}/builds/${lastBuildForPr.id}) :-1:`)
+      } else if (~['created', 'started'].indexOf(lastState)) {
+        console.log(`* ${prInfo} "${lastState}" build found, will do check #${checkNumber + 1} in 30 seconds`)
+      } else {
+        return console.log(`* ${prInfo} Unknown build state: "${lastState}", stopping polling`)
+      }
+    } else {
+      console.warn(`! ${prInfo} Was not able to find matching build, will do check #${checkNumber + 1} in 30 seconds`)
+    }
+
+    setTimeout(pollAndComment, 30 * 1000, owner, repoName, prId, checkNumber + 1)
+  })
+}
+
+function createGhCommentFn (options) {
+  const prInfo = prInfoStr(options)
+
+  return (message, cb) => {
+    githubClient.issues.createComment({
+      user: options.owner,
+      repo: options.repoName,
+      number: options.prId,
+      body: message
+    }, (err, res) => {
+      if (err) {
+        return console.error(`! ${prInfo} Error from GitHub`, err.stack)
+      }
+      console.log(`* ${prInfo} Github comment created`)
+    })
+  }
+}
+
+function prInfoStr (options) {
+  return `${options.owner}/${options.repoName}/#${options.prId}`
+}
+
+exports.pollAndComment = pollAndComment

package.json

@@ -0,0 +1,23 @@
+{
+  "name": "nodejs-github-bot",
+  "version": "1.0.0-beta1",
+  "description": "Node.js GitHub Bot",
+  "scripts": {
+    "start": "node server.js",
+    "test": "standard"
+  },
+  "engines": {
+    "node": ">= 4.2.0"
+  },
+  "private": true,
+  "license": "MIT",
+  "dependencies": {
+    "body-parser": "^1.15.0",
+    "express": "^4.13.4",
+    "github": "^0.2.4",
+    "travis-ci": "^2.1.0"
+  },
+  "devDependencies": {
+    "standard": "^6.0.7"
+  }
+}

server.js

@@ -0,0 +1,39 @@
+'use strict'
+
+const express = require('express')
+const bodyParser = require('body-parser')
+
+const pollTravis = require('./lib/pollTravis')
+
+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.pollAndComment(repo.owner.login, repo.name, parseInt(req.body.number))
+  }
+
+  res.end()
+})
+
+// to trigger polling manually
+app.get('/pr/:owner/:repo/:prId', (req, res) => {
+  pollTravis.pollAndComment(req.params.owner, req.params.repo, parseInt(req.params.prId))
+  res.end()
+})
+
+app.listen(process.env.PORT || 3000, () => {
+  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'
+}