Add stat -c to show accepted status by date.

Signed-off-by: Eric Wang <skygragon@gmail.com>

Author: skygragon

lib/chalk.js

@@ -29,6 +29,11 @@ chalk.setTheme = function(name) {
   this.theme = _.extendOwn(DEFAULT, theme);
 };
 
+chalk.sprint = function(s, hex) {
+  var color = chalk.use256 ? style.color.ansi256.hex(hex) : style.color.ansi.hex(hex);
+  return color + s + style.color.close;
+};
+
 chalk.print = function(s) {
   s = this.enabled ? pres.join('') + s + posts.join('') : s;
   pres.length = posts.length = 0;

lib/commands/stat.js

@@ -1,3 +1,4 @@
+var moment = require('moment');
 var sprintf = require('sprintf-js').sprintf;
 var _ = require('underscore');
 
@@ -14,6 +15,12 @@ var cmd = {
   aliases: ['stats', 'progress', 'report'],
   builder: function(yargs) {
     return yargs
+      .option('c', {
+        alias:    'cal',
+        type:     'boolean',
+        default:  false,
+        describe: 'Show calendar statistics'
+      })
       .option('g', {
         alias:    'graph',
         type:     'boolean',
@@ -42,9 +49,7 @@ var cmd = {
 
 function bar(c, n) {
   return _.range(n)
-    .map(function(i) {
-      return c;
-    })
+    .map(function(i) { return c; })
     .join('');
 }
 
@@ -81,11 +86,17 @@ function showProgress(problems) {
   printLine('Hard', stats.hard.ac, stats.hard.all);
 }
 
+var CHARS = {
+  ac:    h.isWindows() ? 'O ' : '▣ ',
+  notac: h.isWindows() ? 'X ' : '▤ ',
+  none:  h.isWindows() ? 'o ' : '⬚ ',
+};
+
 function showGraph(problems) {
-  var icons = {
-    ac:    chalk.green(h.isWindows() ? 'O ' : '▣ '),
-    notac: chalk.red(h.isWindows() ? 'X ' : '▤ '),
-    none:  chalk.gray(h.isWindows() ? 'o ' : '⬚ '),
+  var ICONS = {
+    ac:    chalk.green(CHARS.ac),
+    notac: chalk.red(CHARS.notac),
+    none:  chalk.gray(CHARS.none),
     empty: '  '
   };
 
@@ -103,15 +114,15 @@ function showGraph(problems) {
 
   var graph = [];
   _.each(problems, function(problem) {
-    graph[problem.id] = icons[problem.state] || icons.none;
+    graph[problem.id] = ICONS[problem.state] || ICONS.none;
   });
 
   var line = [sprintf(' %03d', 0)];
   for (var i = 1, n = graph.length; i <= n; ++i) {
     // padding before group
     if (i % 10 === 1) line.push('   ');
 
-    line.push(graph[i] || icons.empty);
+    line.push(graph[i] || ICONS.empty);
 
     // time to start new row
     if (i % (10 * groups) === 0 || i === n) {
@@ -122,10 +133,83 @@ function showGraph(problems) {
 
   log.info();
   log.printf('%7s%s%3s%s%3s%s',
-      ' ', icons.ac + chalk.green(' Accepted'),
-      ' ', icons.notac + chalk.red(' Not Accepted'),
-      ' ', icons.none + ' Remaining');
+      ' ', ICONS.ac + chalk.green(' Accepted'),
+      ' ', ICONS.notac + chalk.red(' Not Accepted'),
+      ' ', ICONS.none + ' Remaining');
+}
+
+function showCal() {
+  var MONTHS = ['Jan', 'Feb', 'Mar', 'Apr', 'May', 'Jun', 'Jul', 'Aug', 'Sep', 'Oct', 'Nov', 'Dec'];
+  var WEEKDAYS = ['Sun', 'Mon', 'Tue', 'Wed', 'Thu', 'Fri', 'Sat'];
+  var ICONS = [
+    CHARS.none,
+    chalk.sprint(CHARS.ac, '#ffffcc'),
+    chalk.sprint(CHARS.ac, '#ccff66'),
+    chalk.sprint(CHARS.ac, '#66cc33'),
+    chalk.sprint(CHARS.ac, '#00ff00')
+  ];
+
+  var N_MONTHS = 12;
+  var N_WEEKS = 53;
+  var N_WEEKDAYS = 7;
+
+  var now = moment();
+
+  // load historical stats
+  var graph = [];
+  var stats = require('../cache').get(h.KEYS.stat) || {};
+  _.keys(stats).forEach(function(k) {
+    var v = stats[k].ac || 0;
+    if (v === 0) return;
+
+    var d = moment(k, 'YYYY-MM-DD');
+    graph[now.diff(d, 'days')] = v;
+  });
+
+  // print header
+  var buf = Buffer.alloc(120, ' ', 'ascii');
+  for (var i = 0; i <= N_MONTHS; ++i) {
+    // for day 1 in each month, calculate its column position in graph
+    var d = now.clone().subtract(i, 'months').date(1);
+    var idx = now.diff(d, 'days');
+
+    var j = (N_WEEKS - idx / N_WEEKDAYS + 1) * 2;
+    if (j >= 0) buf.write(MONTHS[d.month()], j);
+  }
+  log.printf('%7s%s', ' ', buf.toString());
+
+  // print graph
+  var idx;
+  for (var i = 0; i < N_WEEKDAYS; ++i) {
+    var line = [];
+    // print day in week
+    idx = (now.day() + i + 1) % N_WEEKDAYS;
+    line.push(sprintf('%4s   ', WEEKDAYS[idx]));
+
+    for (var j = 0; j < N_WEEKS; ++j) {
+      idx = (N_WEEKS - j - 1) * N_WEEKDAYS + N_WEEKDAYS - i - 1;
+      var d = now.clone().subtract(idx, 'days');
+
+      // map count to icons index:
+      // [0] => 0, [1,5] => 1, [6,10] => 2, [11,15] => 3, [16,) => 4
+      var count = graph[idx] || 0;
+      idx = Math.floor((count - 1) / 5) + 1;
+      if (idx > 4) idx = 4;
+
+      var icon = ICONS[idx];
+      // use different colors for adjacent months
+      if (idx === 0 && d.month() % 2) icon = chalk.gray(icon);
+      line.push(icon);
+    }
+    log.info(line.join(''));
+  }
+
   log.info();
+  log.printf('%7s%s%3s%s%3s%s%3s%s',
+    ' ', ICONS[1] + ' 1~5',
+    ' ', ICONS[2] + ' 6~10',
+    ' ', ICONS[3] + ' 11~15',
+    ' ', ICONS[4] + ' 16+');
 }
 
 cmd.handler = function(argv) {
@@ -141,8 +225,11 @@ cmd.handler = function(argv) {
       problems = _.filter(problems, function(x) { return !x.locked; });
     }
 
-    if (argv.graph) return showGraph(problems);
-    showProgress(problems);
+    log.info();
+    if (argv.graph)    showGraph(problems);
+    else if (argv.cal) showCal();
+    else               showProgress(problems);
+    log.info();
   });
 };
 

lib/helper.js

@@ -51,6 +51,7 @@ var h = {};
 
 h.KEYS = {
   user:     '../user',
+  stat:     '../stat',
   problems: 'problems',
   problem:  function(p) { return p.id + '.' + p.slug + '.' + p.category; }
 };

package.json

@@ -34,6 +34,7 @@
     "cheerio": "0.20.0",
     "he": "1.1.1",
     "mkdirp": "0.5.1",
+    "moment": "^2.20.1",
     "nconf": "0.10.0",
     "ora": "^1.3.0",
     "prompt": "1.0.0",