index.js

'use strict';

var align = require('lpad-align');
var chalk = require('chalk');
var EventEmitter = require('events').EventEmitter;
var fmt = require('util').format;
var inherits = require('util').inherits;

/**
 * Initialize a new `Squeak`
 *
 * @param {Object} opts
 * @api public
 */

function Squeak(opts) {
	if (!(this instanceof Squeak)) {
		return new Squeak(opts);
	}

	EventEmitter.call(this);

	this.opts = opts || {};
	this.indent = this.opts.indent || 2;
	this.separator = this.opts.separator || ' : ';
	this.stream = this.opts.stream || process.stderr;
	this.types = [];
}

/**
 * Inherit from `EventEmitter`
 */

inherits(Squeak, EventEmitter);

/**
 * Write args to stream
 *
 * @api public
 */

Squeak.prototype.write = function () {
	this.log([].slice.call(arguments));
	return this;
};

/**
 * Write args and new line to stream
 *
 * @api public
 */

Squeak.prototype.writeln = function () {
	this.log([].slice.call(arguments));
	this.stream.write('\n');
	return this;
};

/**
 * Pad and write args to stream
 *
 * @api public
 */

Squeak.prototype.writelpad = function () {
	var args = [].slice.call(arguments);
	var pad = new Array(this.indent + 1).join(' ');

	if (args.length) {
		args[0] = pad + args[0];
	}

	this.log(args);
	return this;
};

/**
 * Add type
 *
 * @param {String} type
 * @param {Object} opts
 * @param {Function} cb
 * @api public
 */

Squeak.prototype.type = function (type, opts, cb) {
	if (!cb && typeof opts === 'function') {
		cb = opts;
		opts = {};
	}

	opts = opts || {};
	opts.color = opts.color || 'reset';
	opts.prefix = opts.prefix || type;

	this.types.push(opts.prefix);
	this[type] = function () {
		this.log([].slice.call(arguments), opts);

		if (cb) {
			cb();
		}
	};

	return this;
};

/**
 * End
 *
 * @param {Function} cb
 * @api public
 */

Squeak.prototype.end = function (cb) {
	this.stream.write('\n');

	if (cb) {
		cb();
	}

	return this;
};

/**
 * Log
 *
 * @param {Array} args
 * @param {Object} opts
 * @api private
 */

Squeak.prototype.log = function (args, opts) {
	opts = opts || {};

	var msg = [fmt.apply(null, args)];
	var color = opts.color;
	var prefix = opts.prefix;

	if (prefix) {
		msg.unshift(chalk[color](align(prefix, this.types, this.indent)));
	}

	this.stream.write(msg.join(this.separator));
	this.stream.write('\n');

	return this;
};

/**
 * Module exports
 */

module.exports = Squeak;