From d8743624c47bc96240a7dd33b62cbd4dc1a70c79 Mon Sep 17 00:00:00 2001
From: Chris Reeves <hello@chris.reeves.io>
Date: Tue, 26 May 2015 18:03:47 +0100
Subject: [PATCH] Added token support to CircleCI badges

---
 server.js | 29 +++++++++++++++++++++--------
 try.html  |  4 ++++
 2 files changed, 25 insertions(+), 8 deletions(-)

diff --git a/server.js b/server.js
index 1d02b9c..39ebcdc 100644
--- a/server.js
+++ b/server.js
@@ -3302,22 +3302,35 @@ cache(function(data, match, sendBadge, request) {
 
 // CircleCI build integration.
 // https://circleci.com/api/v1/project/BrightFlair/PHP.Gt?circle-token=0a5143728784b263d9f0238b8d595522689b3af2&limit=1&filter=completed
-camp.route(/^\/circleci\/project\/([^\/]+\/[^\/]+)(?:\/(.*))?\.(svg|png|gif|jpg|json)$/,
+camp.route(/^\/circleci\/(?:token\/(\w+))?[+\/]?project\/([^\/]+\/[^\/]+)(?:\/(.*))?\.(svg|png|gif|jpg|json)$/,
 cache(function(data, match, sendBadge, request) {
-  var userRepo = match[1];  // eg, `BrightFlair/PHP.Gt`.
-  var branch = match[2];
-  var format = match[3];
+  var token = match[1]
+  var userRepo = match[2];  // eg, `BrightFlair/PHP.Gt`.
+  var branch = match[3];
+  var format = match[4];
 
+  // Base API URL
   var apiUrl = 'https://circleci.com/api/v1/project/' + userRepo;
+
+  // Query Params
+  queryParams = {};
+  queryParams['limit'] = 1;
+  queryParams['filter'] = 'completed';
+
+  // Custom Banch if present
   if(branch != null) {
     apiUrl += "/tree/" + branch;
   }
-  apiUrl +=
-    '?circle-token=0a5143728784b263d9f0238b8d595522689b3af2'
-    + '&limit=1&filter=completed';
+
+  // Append Token to Query Params if present
+  if (token) {
+    queryParams['circle-token'] = token;
+  }
+
+  // Apprend query params to API URL
+  apiUrl += '?' + querystring.stringify(queryParams);
 
   var badgeData = getBadgeData('build', data);
-
   request(apiUrl, {json:true}, function(err, res, data) {
     if (err != null) {
       badgeData.text[1] = 'inaccessible';
diff --git a/try.html b/try.html
index c1fa29c..fea686f 100644
--- a/try.html
+++ b/try.html
@@ -112,6 +112,10 @@ Pixel-perfect &nbsp; Retina-ready &nbsp; Fast &nbsp; Consistent &nbsp; Hackable
       <td><img src='/circleci/project/BrightFlair/PHP.Gt/master.svg' alt=''/></td>
       <td><code>https://img.shields.io/circleci/project/BrightFlair/PHP.Gt/master.svg</code></td>
   </tr>
+  <tr><th> CircleCI token: </th>
+      <td><img src='/circleci/project/BrightFlair/PHP.Gt/master.svg' alt=''/></td>
+      <td><code>https://img.shields.io/circleci/token/YOURTOKEN/project/BrightFlair/PHP.Gt/master.svg</code></td>
+  </tr>
   <tr><th> Jenkins: </th>
     <td><img src='/jenkins/s/https/jenkins.qa.ubuntu.com/precise-desktop-amd64_default.svg' alt=''/></td>
     <td><code>https://img.shields.io/jenkins/s/https/jenkins.qa.ubuntu.com/precise-desktop-amd64_default.svg</code></td>