everycss/lib/stringifier.js

var Node          = require('./node'),
    stringifiers  = {},
    styleStack    = Array(1),
    toString,
    set,
    get;

// return string representation of a class
toString = function (style) {
  var stringifier, string = '';

  styleStack.unshift(style || styleStack[0]);
  stringifier = get(this.type, styleStack[0]);
  if (stringifier) {
    string = stringifier.call(this, styleStack[0]);
  }
  styleStack.shift();

  return string;
};

get = function (type, style) {
  if (stringifiers[type]) {
    if (style && stringifiers[type].styles[style]) {
      return stringifiers[type].styles[style];
    } else {
      return stringifiers[type].main;
    }
  }
};

set = function (type, func, style, main) {
  var i;
  if (!Node.isSet(type)) {
    throw 'Unknown type `' + type + '`';
  }

  // first stringifier for this type
  if (!stringifiers[type]) {
    stringifiers[type] = {
      // set first stringifier as main by default
      main: func,
      styles: {}
    };

    Node.getTypeClass(type).prototype.toString = toString;
  }
  // main stringifier
  else if (main) {
    stringifiers[type].main = func;
  }
  
  // link stringifier to its name
  if (typeof style === 'string') {
    stringifiers[type].styles[style] = func;
  } else if(style instanceof Array) {
    for (i = 0; i < style.length; i++) {
      stringifiers[type].styles[style[i]] = func;
    }
  }
};

exports.set = set;
exports.get = get;

// return indent string
exports.getIndent = function (n) {
  return Array(n + 1).join(' ');
}