Add heatmap graph for 'stat' command.

Signed-off-by: Eric Wang <[email protected]>

Author: skygragon

lib/commands/stat.js

@@ -1,5 +1,6 @@
 var log = require('loglevel');
 var sprintf = require('sprintf-js').sprintf;
+var _ = require('underscore');
 
 var chalk = require('../chalk');
 var core = require('../core');
@@ -8,6 +9,12 @@ var cmd = {
   command: 'stat',
   desc:    'show statistics',
   builder: {
+    graph: {
+      alias:    'g',
+      type:     'boolean',
+      default:  false,
+      describe: 'Show graphic statistics'
+    }
   }
 };
 
@@ -53,11 +60,47 @@ function showSummary(problems) {
   log.info(prettyLine('Hard', statsNoLock.acHard, statsNoLock.allHard));
 }
 
+function showGraph(problems) {
+  var graph = [];
+  _.each(problems, function(problem) {
+    if (problem.state === 'ac') {
+      graph[problem.id] = chalk.green('█');
+    } else if (problem.state === 'notac') {
+      graph[problem.id] = chalk.red('▓');
+    } else {
+      graph[problem.id] = '░';
+    }
+  });
+
+  log.info(sprintf('%13d%13d%13d%13d%13d', 10, 20, 30, 40, 50));
+
+  var line = [sprintf(' %03d   ', 1)];
+  for (var i = 1, n = graph.length; i < n; ++i) {
+    line.push(graph[i] || ' ');
+    if (i % 10 === 0) line.push('   ');
+    if (i % 50 === 0 || i === n) {
+      log.info(line.join(''));
+      line = [sprintf(' %03d   ', i)];
+    }
+  }
+
+  log.info();
+  log.info(sprintf('%7s%s%3s%s%3s%s',
+        ' ', chalk.green('█ Accepted'),
+        ' ', chalk.red('▓ Not Accepted'),
+        ' ', '░ Remaining'));
+  log.info();
+}
+
 cmd.handler = function(argv) {
   core.getProblems(function(e, problems) {
     if (e) return log.fail(e);
 
-    showSummary(problems);
+    if (argv.graph) {
+      showGraph(problems);
+    } else {
+      showSummary(problems);
+    }
   });
 };