Add color theme.

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

Author: skygragon

lib/chalk.js

@@ -5,24 +5,51 @@ function rgb(r, g, b) {
   return style.color.ansi256.rgb(r, g, b);
 }
 
-var colors = {
-  black:   rgb(0, 0, 0),
-  blue:    rgb(0, 0, 255),
-  cyan:    rgb(0, 255, 255),
-  green:   rgb(0, 255, 0),
-  magenta: rgb(255, 0, 255),
-  red:     rgb(255, 0, 0),
-  white:   rgb(255, 255, 255),
-  yellow:  rgb(255, 255, 0)
+var THEMES = {
+  'default': {
+    black:   rgb(0, 0, 0),
+    blue:    rgb(0, 0, 255),
+    cyan:    rgb(0, 255, 255),
+    green:   rgb(0, 255, 0),
+    magenta: rgb(255, 0, 255),
+    red:     rgb(255, 0, 0),
+    white:   rgb(255, 255, 255),
+    yellow:  rgb(255, 255, 0)
+  },
+  'dark': {
+    black:   rgb(0, 0, 0),
+    blue:    rgb(0, 0, 153),
+    cyan:    rgb(0, 153, 153),
+    green:   rgb(0, 153, 0),
+    magenta: rgb(153, 0, 153),
+    red:     rgb(153, 0, 0),
+    white:   rgb(255, 255, 255),
+    yellow:  rgb(153, 153, 0)
+  },
+  'pink': {
+    black:   rgb(0, 0, 0),
+    blue:    rgb(0, 0, 153),
+    cyan:    rgb(0, 153, 153),
+    green:   rgb(255, 20, 147),
+    magenta: rgb(153, 0, 153),
+    red:     rgb(220, 20, 60),
+    white:   rgb(255, 255, 255),
+    yellow:  rgb(255, 69, 0)
+  }
 };
 
 var chalk = {
-  enabled: true
+  enabled: true,
+  theme:   THEMES.default
 };
 
 var pres = [];
 var posts = [];
 
+chalk.setTheme = function(name) {
+  this.theme = THEMES[name] || THEMES.default;
+};
+
 chalk.print = function(s) {
   s = this.enabled ? pres.join('') + s + posts.join('') : s;
   pres.length = posts.length = 0;
@@ -43,7 +70,7 @@ _.chain(['black', 'blue', 'cyan', 'green', 'magenta', 'red', 'white', 'yellow'])
 .each(function(color) {
   Object.defineProperty(chalk, color, {
     get: function() {
-      return chalk.wrap(colors[color], style.color.close);
+      return chalk.wrap(chalk.theme[color], style.color.close);
     }
   });
 });

lib/cli.js

@@ -4,7 +4,6 @@ var log = require('loglevel');
 
 var chalk = require('./chalk');
 var config = require('./config');
-var h = require('./helper');
 
 // We are expecting a tier configuration like:
 // global config < local config < cli params
@@ -15,6 +14,7 @@ function setColorMode() {
   if (process.argv.indexOf('--no-color') >= 0) useColor = false;
 
   chalk.enabled = useColor;
+  chalk.setTheme(config.COLOR_THEME);
 }
 
 function setLogLevel() {
@@ -30,6 +30,7 @@ function setLogLevel() {
 }
 
 function checkCache() {
+  var h = require('./helper');
   var cacheDir = h.getCacheDir();
 
   if (!fs.existsSync(cacheDir))

lib/config.js

@@ -18,6 +18,7 @@ var DEFAULT_CONFIG = {
   // but you will want change these
   LANG:        'cpp', // avail: [c,cpp,csharp,golang,java,javascript,python,ruby,swift]
   USE_COLOR:   true,
+  COLOR_THEME: 'default',
   AUTO_LOGIN:  false,
   MAX_WORKERS: 10
 };

lib/helper.js

@@ -1,8 +1,6 @@
 var fs = require('fs');
 var path = require('path');
 
-var chalk = require('./chalk');
-
 var h = {};
 
 h.prettyState = function(state) {
@@ -14,6 +12,7 @@ h.prettyState = function(state) {
 };
 
 h.prettyText = function(text, yesNo) {
+  var chalk = require('./chalk');
   switch (yesNo) {
     case true:  return chalk.green('✔' + text);
     case false: return chalk.red('✘' + text);