Adapt stat graph to the real terminal width.

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

Author: skygragon

lib/cli.js

@@ -69,13 +69,14 @@ cli.run = function() {
   });
 
   var yargs = require('yargs');
+  h.width = yargs.terminalWidth();
   yargs.commandDir('commands')
     .completion()
     .help('h')
     .alias('h', 'help')
     .version(false)
     .epilog('Seek more help at https://skygragon.github.io/leetcode-cli/commands')
-    .wrap(Math.min(yargs.terminalWidth(), 120))
+    .wrap(Math.min(h.width, 120))
     .argv;
 };
 

lib/commands/stat.js

@@ -94,20 +94,34 @@ function showGraph(problems) {
     none:  chalk.gray('⬚ ')
   };
 
+  // row header is 4 bytes
+  // each question takes 2 bytes
+  // each group has 10 questions, which takes (2*10=20) + 3 paddings
+  var groups = Math.floor((h.width - 4) / (3 + 2 * 10));
+  if (groups < 1) groups = 1;
+  if (groups > 5) groups = 5;
+
+  var header = _.range(groups)
+    .map(function(x) { return sprintf('%5d%18d', x * 10 + 1, x * 10 + 10); })
+    .join('');
+  log.info('    ' + header);
+
   var graph = [];
   _.each(problems, function(problem) {
     graph[problem.id] = icons[problem.state] || icons.none;
   });
 
-  log.printf('%8d%19d%5d%18d%5d%18d%5d%18d%5d%18d', 1, 10, 11, 20, 21, 30, 31, 40, 41, 50);
-
-  var line = [sprintf(' %03d   ', 0)];
+  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] || '  ');
-    if (i % 10 === 0) line.push('   ');
-    if (i % 50 === 0 || i === n) {
+
+    // time to start new row
+    if (i % (10 * groups) === 0 || i === n) {
       log.info(line.join(''));
-      line = [sprintf(' %03d   ', i)];
+      line = [sprintf(' %03d', i)];
     }
   }