Colorful stdout, with ability to turn colors off/on

Couto · Feb 12, 2012 · 406fea4 · 406fea4
1 parent f663cfe
commit 406fea4
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 3 deletions.
diff --git a/examples/file.js b/examples/file.js
@@ -6,13 +6,19 @@
 var Log = require('../lib/log')
   , fs = require('fs')
   // , stream = fs.createWriteStream(__dirname + '/file.log', { flags: 'a' })
-  , log = new Log('debug');
+  , log = new Log('debug').colorful({
+    DEBUG : '\033[1;30m',
+    NOTICE: '\033[0;32m'
+  });
 
 log.debug('a debug message');
 log.info('a info message');
 log.notice('a notice message');
 log.warning('a warning message');
+
+log.colorful(false);
+
 log.error('a error message');
 log.critical('a critical message');
 log.alert('a alert message');
-log.emergency('a emergency message');
+log.emergency('a emergency message');
diff --git a/lib/log.js b/lib/log.js
@@ -178,7 +178,7 @@ Log.prototype = {
       var msg = args[0].replace(/%s/g, function(){
         return args[i++];
       });
-      if (this.stream === process.stdout) {
+      if (this.stream === process.stdout && this.useColors) {
           this.stream.write(
             exports.colors[levelStr]
             + '[' + new Date + ']'
@@ -198,6 +198,20 @@ Log.prototype = {
     }
   },
 
+  colorful : function (colors) {
+      var k;
+      if (Object.prototype.toString.call(colors) === '[object Object]') {
+          for (k in colors) {
+              if (colors.hasOwnProperty(k) && exports.colors.hasOwnProperty(k)) {
+                exports.colors[k] = colors[k];
+              }
+          }
+          this.useColors = true;
+      } else if (colors === false) { this.useColors = false; }
+      else { this.useColors = true; }
+      return this;
+  },
+
   /**
    * Log emergency `msg`.
    *