Use 256 colors only if terminal supports it.

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

Author: skygragon

lib/chalk.js

@@ -1,8 +1,10 @@
 var _ = require('underscore');
 var style = require('ansi-styles');
+var supportsColor = require('supports-color');
 
 var chalk = {
-  enabled: true,
+  enabled: supportsColor,
+  use256:  supportsColor && supportsColor.has256,
   themes:  {},
   theme:   {}
 };
@@ -51,7 +53,7 @@ chalk.init = function() {
   _.each(fs.readdirSync(dir), function(f) {
     var theme = JSON.parse(fs.readFileSync(path.join(dir, f)));
     chalk.themes[path.basename(f, '.json')] = _.mapObject(theme, function(v, k) {
-      return style.color.ansi256.hex(v);
+      return chalk.use256 ? style.color.ansi256.hex(v) : style.color.ansi.hex(v);
     });
   });
 
@@ -74,5 +76,4 @@ chalk.init = function() {
   });
 };
 
-chalk.init();
 module.exports = chalk;

lib/cli.js

@@ -11,11 +11,11 @@ var config = require('./config');
 // global config < local config < cli params
 // Color is a tricky one so we manually handle it here.
 function setColorMode() {
-  var useColor = config.USE_COLOR || false;
-  if (process.argv.indexOf('--color') >= 0) useColor = true;
-  if (process.argv.indexOf('--no-color') >= 0) useColor = false;
+  // FIXME: delete this hack when supports-color handles it well.
+  if (process.env.TERM_PROGRAM === 'iTerm.app') chalk.use256 = true;
 
-  chalk.enabled = useColor;
+  chalk.enabled = config.USE_COLOR && chalk.enabled;
+  chalk.init();
   chalk.setTheme(config.COLOR_THEME);
 }
 

package.json

@@ -37,6 +37,7 @@
     "prompt": "^1.0.0",
     "request": "^2.74.0",
     "sprintf-js": "^1.0.3",
+    "supports-color": "^3.2.3",
     "underscore": "^1.8.3",
     "wordwrap": "^1.0.0",
     "yargs": "^5.0.0"

test/test_helper.js

@@ -3,6 +3,8 @@ var assert = require('chai').assert;
 var chalk = require('../lib/chalk');
 var h = require('../lib/helper');
 
+chalk.init();
+
 describe('helper', function() {
   describe('#prettyState', function() {
     it('should ok w/ color', function() {