prettify/prettify.js

{{prettify}} handlebars helper for formatting (beautifying) HTML, JavaScript and CSS.

/**
 * {{prettify}} by Jon Schlinkert
 * http://github.com/helpers/prettify
 *
 * Copyright (c) 2013 Jon Schlinkert
 * MIT License
 */


'use strict';

var _ = require('lodash');
var engineOpts = require('./lib/defaults');

function condense(str) {
  return str.replace(/(\n|\r){2,}/g, '\n');
}

function padcomments(str) {
  return str.replace(/(\s*<!--)/g, '\n$1');
}

function fixspaces(str) {
  return str.replace(/(<\/(a|span|strong|h1|h2|h3|h4|h5|h6)>(?!(,|\.|!|\?|;|:)))/g, '$1 ');
}

module.exports.register = function (Handlebars, options) {

  // If the 'assemble.options' object exists, use it. Otherwise use an empty object.
  var o = options || {};


  /**
   * Prettify
   * @param  {[type]} options [description]
   * @return {[type]}         [description]
   */
  Handlebars.registerHelper('prettify', function (options) {
    var opts = {};
    var content = options.fn(this);

    // Prettify using js, CSS or HTML mode. 'html' is the default.
    function prettifyBasedOnMode() {
      opts.mode = 'html';
      o.prettify = (opts.mode === 'html') ? o.prettify : o.prettify[opts.mode];
      opts = _.defaults(options.hash, _.extend({}, opts, engineOpts[opts.mode], o.prettify));
      opts.indent_size = opts.indent;
      content = require('js-prettify')[opts.mode](content, opts);
    }

    // Make it so.
    prettifyBasedOnMode();

    // Clean up new lines after prettification.
    if(opts.mode === 'html') {
      // Reduce multiple newlines to a single newline
      content = (opts.condense === true) ? condense(content) : content;
      // Add a single newline above code comments.
      content = (opts.padcomments === true) ? padcomments(content) : content;
    }

    return new Handlebars.SafeString(fixspaces(content));
  });
};