hifile/index.js

generate markup for highlight.js file viewer

'use strict';

var hl = require('highlight.js');
// map file extensions to highlight.js languages
var exts = {
  'js': 'javascript',
  'sh': 'bash',
  'md': 'markdown'
};

/* attempt to fix multiline comments by adding extra spans */
function multiLineComments (match) {
  var ret = match;
  var lines = match.split('\n');
  var ln = lines.length;

  if (ln > 1) {
    for (var i = 0; i < ln; i++) {
      if (i === 0) {
        ret = lines[i] + '</span>';
      } else if (i === ln) {
        ret = ret + '<span class="comment">' + lines[i];
      } else {
        ret = ret + '\n';
        if (lines[i] !== '') {
          ret = ret + '<span class="comment">' + lines[i] + '</span>';
        }
      }
    }
  }
  return ret;
}

module.exports = function hifile (str, ext) {
  var code;
  var col = '';
  var ln;

  ext = exts[ext] || ext;
  try {
    if (ext) {
      code = hl.highlight(exts[ext] || ext, str);
    } else {
      code = hl.highlightAuto(str);
    }
  } catch (e) {
    code = hl.highlightAuto(str);
  }
  code.value = code.value.replace(/<span class="comment">(.|\n)+?<\/span>/gm, multiLineComments);
  var lines = code.value.split('\n');
  lines.forEach(function (line, i, arr) {
    ln = i + 1;
    arr[i] = '<div id="H' + ln + '" class="cl">' + arr[i] + '</div>';
    col = col + '<div id="L' + ln + '" class="ln"><a href="#L' + ln + '">' + ln + '</a></div>';
  });
  code = lines.join('\n');
  return '<div class="hifile hljs"><div id="col">' + col + '</div><div id="code">' + code + '</div></div>';
};