Add --no-lock option in show stats without locked questions.

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

Author: skygragon

lib/commands/stat.js

@@ -20,16 +20,23 @@ var cmd = {
         default:  false,
         describe: 'Show graphic statistics'
       })
+      .option('l', {
+        alias:    'lock',
+        type:     'boolean',
+        default:  true,
+        describe: 'Include locked questions'
+      })
       .option('t', {
         alias:    'tag',
         type:     'string',
         default:  'all',
         describe: 'Show statistics for given tag',
         choices:  ['all'].concat(config.sys.categories)
       })
-      .example(chalk.yellow('leetcode stat'), 'Show summarised progress')
-      .example(chalk.yellow('leetcode stat -t algorithms'), 'Show summarised progress of algorithms questions')
-      .example(chalk.yellow('leetcode stat -g'), 'Show detailed status of all questions');
+      .example(chalk.yellow('leetcode stat'), 'Show progress status')
+      .example(chalk.yellow('leetcode stat --no-lock'), 'Show progress status without locked questions')
+      .example(chalk.yellow('leetcode stat -t algorithms'), 'Show progress status of algorithms questions')
+      .example(chalk.yellow('leetcode stat -g'), 'Show detailed status in graph');
   }
 };
 
@@ -51,40 +58,27 @@ function printLine(key, done, all) {
       chalk.red(bar('░', n - x)));
 }
 
-function showSummary(problems) {
+function showProgress(problems) {
   var stats = {
     easy:   {all: 0, ac: 0},
     medium: {all: 0, ac: 0},
     hard:   {all: 0, ac: 0}
   };
-  var statsNoLock = {
-    easy:   {all: 0, ac: 0},
-    medium: {all: 0, ac: 0},
-    hard:   {all: 0, ac: 0}
-  };
 
   problems.forEach(function(problem) {
     var level = problem.level.toLowerCase();
     var state = problem.state.toLowerCase();
 
     if (!(level in stats)) return;
     ++stats[level].all;
-    if (!problem.locked) ++statsNoLock[level].all;
 
     if (!(state in stats[level])) return;
     ++stats[level][state];
-    if (!problem.locked) ++statsNoLock[level][state];
   });
 
   printLine('Easy', stats.easy.ac, stats.easy.all);
   printLine('Medium', stats.medium.ac, stats.medium.all);
   printLine('Hard', stats.hard.ac, stats.hard.all);
-
-  log.info();
-  log.info('Without Locked:');
-  printLine('Easy', statsNoLock.easy.ac, statsNoLock.easy.all);
-  printLine('Medium', statsNoLock.medium.ac, statsNoLock.medium.all);
-  printLine('Hard', statsNoLock.hard.ac, statsNoLock.hard.all);
 }
 
 function showGraph(problems) {
@@ -143,11 +137,12 @@ cmd.handler = function(argv) {
       problems = _.filter(problems, function(x) { return x.category === argv.tag; });
     }
 
-    if (argv.graph) {
-      showGraph(problems);
-    } else {
-      showSummary(problems);
+    if (!argv.lock) {
+      problems = _.filter(problems, function(x) { return !x.locked; });
     }
+
+    if (argv.graph) return showGraph(problems);
+    showProgress(problems);
   });
 };