colorize/lib/colorize.js

An expressive interface for ANSI colored strings and terminal output.

/*
	Terminal coloring for Node.js
	The syntax looks like this:
	
		This word is #red[red].
	
	You can do nesting and all kinds of fun things.
	
		For when #bold[you #blink[really]] want to make a point.
	
	You can transform a string using .ansify(str), or use
	the modified version of console to output colored strings
	automatically.
*/

exports.console = {
	_forward: function(method, args){
		args = Array.prototype.slice.call(args);
		
		// process the first argument for tags
		if (args.length > 0)
			args[0] = exports.ansify(args[0]);

		// route it to the console
		method.apply(console, args);
	},
	log: function(){
		this._forward(console.log, arguments);
	},
	info: function(){
		this._forward(console.info, arguments);
	},
	warn: function(){
		this._forward(console.warn, arguments);
	},
	error: function(){
		this._forward(console.error, arguments);
	}
};

exports.enabled = true;

exports.ansicodes = {
	'reset': '\033[0m',
	'bold': '\033[1m',
	'italic': '\033[3m',
	'underline': '\033[4m',
	'blink': '\033[5m',
	'black': '\033[30m',
	'red': '\033[31m',
	'green': '\033[32m',
	'yellow': '\033[33m',
	'blue': '\033[34m',
	'magenta': '\033[35m',
	'cyan': '\033[36m',
	'white': '\033[37m',
	
};

exports.ansify = function(str){
	var tag = /#([a-z]+)\[|\]/;
	var cstack = [];
	
	while (tag.test(str))
	{
		var matches = tag.exec(str);
		var orig = matches[0];
		
		if (this.enabled)
		{
			if (orig == ']')
				cstack.pop();
			else
			{
				var name = matches[1];
				if (name in this.ansicodes)
				{
					var code = this.ansicodes[name];
					cstack.push(code);
				}
			}
			
			str = str.replace(orig, this.ansicodes.reset + cstack.join(''));
		}
		else
			str = str.replace(orig, '');
	}
	return str;
};