Refactor prettyLevel.

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

Author: skygragon

Diff for: lib/commands/list.js

@@ -1,4 +1,5 @@
 var _ = require('underscore');
+var sprintf = require('sprintf-js').sprintf;
 
 var h = require('../helper');
 var chalk = require('../chalk');
@@ -132,7 +133,7 @@ cmd.handler = function(argv) {
           h.prettyState(problem.state),
           problem.id,
           problem.name,
-          h.prettyLevel(problem.level),
+          h.prettyLevel(sprintf('%-6s', problem.level)),
           problem.percent);
     });
 

Diff for: lib/helper.js

@@ -89,10 +89,10 @@ h.prettyTime = function(n) {
 
 h.prettyLevel = function(level) {
   var chalk = require('./chalk');
-  switch (level.toLowerCase()) {
-    case 'easy':   return chalk.green(level + '  ');
+  switch (level.toLowerCase().trim()) {
+    case 'easy':   return chalk.green(level);
     case 'medium': return chalk.yellow(level);
-    case 'hard':   return chalk.red(level + '  ');
+    case 'hard':   return chalk.red(level);
     default:       return level;
   }
 };

Diff for: test/test_helper.js

@@ -50,12 +50,12 @@ describe('helper', function() {
     it('should ok w/ color', function() {
       chalk.enabled = true;
 
-      assert.equal(h.prettyLevel('Easy'), chalk.green('Easy  '));
+      assert.equal(h.prettyLevel('Easy'), chalk.green('Easy'));
       assert.equal(h.prettyLevel('Medium'), chalk.yellow('Medium'));
-      assert.equal(h.prettyLevel('Hard'), chalk.red('Hard  '));
-      assert.equal(h.prettyLevel('easy'), chalk.green('easy  '));
+      assert.equal(h.prettyLevel('Hard'), chalk.red('Hard'));
+      assert.equal(h.prettyLevel('easy  '), chalk.green('easy  '));
       assert.equal(h.prettyLevel('medium'), chalk.yellow('medium'));
-      assert.equal(h.prettyLevel('hard'), chalk.red('hard  '));
+      assert.equal(h.prettyLevel('hard  '), chalk.red('hard  '));
       assert.equal(h.prettyLevel('unknown'), 'unknown');
     });
   }); // #prettyLevel