Implement colorful output.

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

Author: skygragon

README.md

@@ -58,6 +58,7 @@ Navigate all the problems. The heading `✔` means you have AC-ed the problem.
 * `-l` to filter by level.
 * `-s` to show statistic counters.
 * `lc list <keyword>` to search by keyword.
+* `--no-color` to disable colorful output.
 
 ### 3. Prepare
 

lib/commands/show.js

@@ -1,4 +1,5 @@
 var _ = require('underscore');
+var chalk = require('chalk');
 var fs = require('fs');
 
 var sprintf = require('sprintf-js').sprintf;
@@ -43,11 +44,11 @@ cmd.handler = function(argv) {
 
       var f = problem.key + h.langToExt(argv.lang);
       fs.writeFileSync(f, template.defaultCode);
-      msg = sprintf('(File: %s)', f);
+      msg = sprintf('(File: %s)', chalk.yellow.underline(f));
     }
 
     console.log(sprintf('[%d] %s\t%s\n', problem.id, problem.name, msg));
-    console.log(sprintf('%s\n', problem.link));
+    console.log(sprintf('%s\n', chalk.underline(problem.link)));
     console.log(sprintf('* %s (%s)', problem.level, problem.percent));
     console.log(sprintf('* Total Accepted: %d', problem.totalAC));
     console.log(sprintf('* Total Submissions: %d\n', problem.totalSubmit));

lib/commands/test.js

@@ -1,4 +1,5 @@
 var _ = require('underscore');
+var chalk = require('chalk');
 var util = require('util');
 
 var core = require('../core');
@@ -31,7 +32,8 @@ function prettyLine(actual, expected, key) {
            !_.isEqual(actual[key], expected[key])) {
     sym = '✘';
   }
-  return util.format('    %s %s: %s', sym, key.split('_').pop(), actual[key]);
+  var line = util.format('    %s %s: %s', sym, key.split('_').pop(), actual[key]);
+  return (sym === '✔') ? chalk.green(line) : chalk.red(line);
 }
 
 cmd.handler = function(argv) {
@@ -40,6 +42,9 @@ cmd.handler = function(argv) {
     testcase = h.readStdin();
   }
 
+  if (!testcase || testcase === '')
+    return console.log('ERROR: missing testcase?');
+
   var keyword = h.getFilename(argv.filename);
   core.getProblem(keyword, function(e, problem) {
     if (e) return console.log('ERROR:', e);
@@ -55,7 +60,7 @@ cmd.handler = function(argv) {
 
       for (var i = 0; i < results.length; ++i) {
         console.log();
-        console.log(results[i].name);
+        console.log(chalk.yellow(results[i].name));
         console.log(prettyLine(results[i], results[i + 1], 'status_runtime'));
         console.log(prettyLine(results[i], results[i + 1], 'code_answer'));
         console.log(prettyLine(results[i], results[i + 1], 'code_output'));

lib/commands/user.js

@@ -1,3 +1,4 @@
+var chalk = require('chalk');
 var prompt = require('prompt');
 
 var core = require('../core');
@@ -33,21 +34,21 @@ cmd.handler = function(argv) {
       core.login(user, function(e, user) {
         if (e) return console.log('Login failed:', e);
 
-        console.log('Successfully login as', user.name);
+        console.log('Successfully login as', chalk.yellow(user.name));
       });
     });
   } else if (argv.logout) {
     // logout
     user = core.logout(null);
     if (user)
-      console.log('Successfully logout as', user.name);
+      console.log('Successfully logout as', chalk.yellow(user.name));
     else
       console.log('You are not login yet?');
   } else {
     // show current user
     user = core.getUser();
     if (user)
-      console.log('You are now login as', user.name);
+      console.log('You are now login as', chalk.yellow(user.name));
     else
       console.log('You are not login yet?');
   }

lib/helper.js

@@ -1,12 +1,13 @@
+var chalk = require('chalk');
 var fs = require('fs');
 var path = require('path');
 
 var h = {};
 
 h.prettyState = function(state) {
   switch (state) {
-    case 'ac':    return '✔';
-    case 'notac': return '✘';
+    case 'ac':    return chalk.green('✔');
+    case 'notac': return chalk.red('✘');
     default:      return ' ';
   }
 };

package.json

@@ -30,6 +30,7 @@
   },
   "homepage": "https://github.com/skygragon/leetcode-cli#readme",
   "dependencies": {
+    "chalk": "^1.1.3",
     "cheerio": "^0.20.0",
     "prompt": "^1.0.0",
     "request": "^2.74.0",