Merge pull request #5 from speedtracker/fix/connect_headers

Add Accept header to connect endpoint

Author: eduardoboucas

index.js

@@ -82,7 +82,7 @@ server.post('/v1/test/:user/:repo/:branch/:profile', testHandler)
 // ------------------------------------
 
 server.get('/v1/connect/:user/:repo', (req, res) => {
-  const github = new GitHub()
+  const github = new GitHub(GitHub.GITHUB_CONNECT)
 
   github.authenticate(config.get('githubToken'))
 
@@ -103,12 +103,14 @@ server.get('/v1/connect/:user/:repo', (req, res) => {
     } else {
       return Promise.reject()
     }
-  }).then((response) => {
+  }).then(response => {
     // Track event
     new Analytics().track(Analytics.Events.CONNECT)
 
     res.send('OK!')
-  }).catch((err) => {
+  }).catch(err => {
+    console.log(err.stack || err)
+
     res.status(500).send('Invitation not found.')
   })  
 })

lib/GitHub.js

@@ -1,15 +1,26 @@
 const GitHubApi = require('github')
 
-const GitHub = function () {
+const GITHUB_CONNECT = 1
+
+const GitHub = function (type) {
+  let headers = {
+    'user-agent': 'SpeedTracker agent'
+  }
+
+  switch (type) {
+    case GITHUB_CONNECT:
+      headers['Accept'] = 'application/vnd.github.swamp-thing-preview+json'
+
+      break
+  }
+
   this.api = new GitHubApi({
     debug: (process.env.NODE_ENV !== 'production'),
     debug: false,
     protocol: 'https',
     host: 'api.github.com',
     pathPrefix: '',
-    headers: {
-      'user-agent': 'SpeedTracker agent'
-    },
+    headers,
     timeout: 5000,
     Promise: Promise
   })
@@ -25,3 +36,5 @@ GitHub.prototype.authenticate = function (token) {
 }
 
 module.exports = GitHub
+
+module.exports.GITHUB_CONNECT = GITHUB_CONNECT

package.json

@@ -1,6 +1,6 @@
 {
   "name": "speedtracker-api",
-  "version": "1.0.1",
+  "version": "1.0.2",
   "description": "",
   "main": "index.js",
   "scripts": {