Johnsonys
diff --git a/‎lib/commands/user.js
Lines changed: 41 additions & 1 deletion b/‎lib/commands/user.js
Lines changed: 41 additions & 1 deletion
@@ -1,5 +1,6 @@
 var chalk = require('chalk');
 var prompt = require('prompt');
+var sprintf = require('sprintf-js').sprintf;
 
 var core = require('../core');
 
@@ -9,15 +10,35 @@ var cmd = {
   builder: {
     login: {
       alias:    'l',
+      type:     'boolean',
       describe: 'Login leetcode.'
     },
     logout: {
       alias:    'L',
+      type:     'boolean',
       describe: 'Logout leetcode.'
+    },
+    stat: {
+      alias:    's',
+      type:     'boolean',
+      describe: 'Show user stats.'
     }
   }
 };
 
+function bar(c, n) {
+  return new Buffer(n).fill(c).toString();
+}
+
+function prettyLine(key, done, all) {
+  var n = 30;
+  var x = Math.ceil(n * done / all);
+  return sprintf('%-8s %3d/%-3d (%.2f%%)\t%s%s',
+      key, done, all, done * 100 / all,
+      chalk.green('[' + bar(chalk.enabled ? '.' : '+', x)),
+      chalk.red(bar('.', n - x) + ']'));
+}
+
 cmd.handler = function(argv) {
   var user = null;
   if (argv.login) {
@@ -50,7 +71,26 @@ cmd.handler = function(argv) {
     if (user)
       console.log('You are now login as', chalk.yellow(user.name));
     else
-      console.log('You are not login yet?');
+      return console.log('You are not login yet?');
+
+    if (argv.stat) {
+      core.getProblems(function(e, problems) {
+        if (e) return console.log('Get stats failed:', e);
+
+        var stats = {};
+        problems.forEach(function(problem) {
+          var keyAll = 'all' + problem.level;
+          var keyAC = problem.state + problem.level;
+          stats[keyAll] = (stats[keyAll] || 0) + 1;
+          stats[keyAC] = (stats[keyAC] || 0) + 1;
+        });
+
+        console.log();
+        console.log(prettyLine('Easy', stats.acEasy, stats.allEasy));
+        console.log(prettyLine('Medium', stats.acMedium, stats.allMedium));
+        console.log(prettyLine('Hard', stats.acHard, stats.allHard));
+      });
+    }
   }
 };