Disable/enable color in the very beginning.

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

Author: skygragon

bin/lc

@@ -1,3 +1,3 @@
 #!/usr/bin/env node
 
-require('../lib/cli');
+require('../lib/cli').run();

lib/cli.js

@@ -1,7 +1,25 @@
-var yargs = require('yargs');
+// We are expecting a tier configuration like:
+// global config < local config < cli params
+// Color is a tricky one so we manually handle it here.
+function setColorMode() {
+  var useColor = require('./config').USE_COLOR || false;
+  if (process.argv.indexOf('--color') >= 0) useColor = true;
+  if (process.argv.indexOf('--no-color') >= 0) useColor = false;
 
-yargs.commandDir('commands')
-  .completion()
-  .help()
-  .strict()
-  .argv;
+  require('chalk').enabled = useColor;
+}
+
+var cli = {};
+
+cli.run = function() {
+  setColorMode();
+
+  require('yargs')
+    .commandDir('commands')
+    .completion()
+    .help()
+    .strict()
+    .argv;
+};
+
+module.exports = cli;

lib/config.js

@@ -8,5 +8,6 @@ module.exports = {
   VERIFY_URL:   'https://leetcode.com/submissions/detail/$id/check/',
 
   // but you will want change these
-  LANG: 'cpp' // others: c,csharp,golang,java,javascript,python,ruby,swift
+  LANG:      'cpp', // avail: [c,cpp,csharp,golang,java,javascript,python,ruby,swift]
+  USE_COLOR: false
 };