From a279174abcbe970c9e7d32338882b2c58af6f1a4 Mon Sep 17 00:00:00 2001
From: Thaddee Tyl <thaddee.tyl@gmail.com>
Date: Wed, 22 Jan 2014 09:21:05 +0100
Subject: [PATCH] PyPI Integration.

Part of issue #16.
---
 server.js | 49 +++++++++++++++++++++++++++++++++++++++++++++++--
 1 file changed, 47 insertions(+), 2 deletions(-)

diff --git a/server.js b/server.js
index 4b64561..22d6258 100644
--- a/server.js
+++ b/server.js
@@ -120,8 +120,10 @@ function(data, match, end, ask) {
         badgeData.colorscheme = 'yellow';
       } else if (monthly < 100) {
         badgeData.colorscheme = 'yellowgreen';
-      } else {
+      } else if (monthly < 1000) {
         badgeData.colorscheme = 'green';
+      } else {
+        badgeData.colorscheme = 'brightgreen';
       }
       badge(badgeData, makeSend(format, ask.res, end));
     });
@@ -171,8 +173,10 @@ function(data, match, end, ask) {
         badgeData.colorscheme = 'yellow';
       } else if (monthly < 100) {
         badgeData.colorscheme = 'yellowgreen';
-      } else {
+      } else if (monthly < 1000) {
         badgeData.colorscheme = 'green';
+      } else {
+        badgeData.colorscheme = 'brightgreen';
       }
       badge(badgeData, makeSend(format, ask.res, end));
     });
@@ -252,6 +256,47 @@ function(data, match, end, ask) {
   });
 });
 
+// PyPI integration.
+camp.route(/^\/pypi\/dm\/(.*)\.(svg|png|gif|jpg)$/,
+function(data, match, end, ask) {
+  var egg = match[1];  // eg, `gevent`.
+  var format = match[2];
+  var apiUrl = 'https://pypi.python.org/pypi/' + egg + '/json';
+  var label = getLabel('downloads', data);
+  var badgeData = {text:[label, 'n/a'], colorscheme:'lightgrey'};
+  https.get(apiUrl, function(res) {
+    var buffer = '';
+    res.on('data', function(chunk) { buffer += ''+chunk; });
+    res.on('end', function(chunk) {
+      if (chunk) { buffer += ''+chunk; }
+      try {
+        var data = JSON.parse(buffer);
+        var monthly = data.info.downloads.last_month;
+      } catch(e) {
+        badgeData.text[1] = 'invalid';
+        badge(badgeData, makeSend(format, ask.res, end));
+        return;
+      }
+      badgeData.text[1] = metric(monthly) + '/month';
+      if (monthly === 0) {
+        badgeData.colorscheme = 'red';
+      } else if (monthly < 10) {
+        badgeData.colorscheme = 'yellow';
+      } else if (monthly < 100) {
+        badgeData.colorscheme = 'yellowgreen';
+      } else if (monthly < 1000) {
+        badgeData.colorscheme = 'green';
+      } else {
+        badgeData.colorscheme = 'brightgreen';
+      }
+      badge(badgeData, makeSend(format, ask.res, end));
+    });
+  }).on('error', function(e) {
+    badgeData.text[1] = 'inaccessible';
+    badge(badgeData, makeSend(format, ask.res, end));
+  });
+});
+
 // Coveralls integration.
 camp.route(/^\/coveralls\/([^\/]+\/[^\/]+)(?:\/(.+))?\.(svg|png|gif|jpg)$/,
 function(data, match, end, ask) {