fix Transport#prompt() to handle --no-color (fixes #70)

pstadler · Feb 4, 2015 · d4bfae7 · d4bfae7
1 parent bc5e811
commit d4bfae7
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 6 deletions.
diff --git a/lib/index.js b/lib/index.js
@@ -5,7 +5,8 @@ var format = require('util').format
   , logger = require('./logger')()
   , errors = require('./errors')
   , flight = require('./flight')
-  , prettyTime = require('pretty-hrtime');
+  , prettyTime = require('pretty-hrtime')
+  , chalk = require('chalk');
 
 var DEFAULT_TASK = 'default';
 
@@ -312,7 +313,7 @@ Flightplan.prototype.run = function(task, target, options) {
   options = options || {};
 
   if(options.color !== undefined) {
-    logger.useColors(options.color);
+    chalk.enabled = options.color;
   }
 
   if(Object.keys(this._targets).indexOf(target) === -1) {

diff --git a/lib/logger/index.js b/lib/logger/index.js
@@ -24,10 +24,6 @@ function Logger(options) {
   }
 }
 
-Logger.prototype.useColors = function(flag) {
-  chalk.enabled = flag;
-};
-
 Logger.prototype.user = function(message) {
   this._log(chalk.cyan(message));
 };

diff --git a/lib/transport/index.js b/lib/transport/index.js
@@ -306,6 +306,7 @@ Transport.prototype.prompt = (function() {
 
     return this.waitFor(function(done) {
       Queue.push(function() {
+        prompt.colors = false;
         prompt.delimiter = '';
         prompt.message = chalk.gray(self._context.remote.host);
         prompt.start();