wat/dist/cosmetician/highlighter.js

Community-controlled cheat sheets for every coder.

'use strict';

var _ = require('lodash');
var chalk = require('chalk');
var fs = require('fs');
var hljs = require('highlight.js');
var path = require('path');

var highlighter = {

  configPath: __dirname + '/../../config/themes',

  allClasses: ['class', 'comment', 'constant', 'function', 'keyword', 'number', 'regexp', 'string', 'subst', 'symbol', 'title', 'variable', 'addition', 'annotaion', 'annotation', 'argument', 'array', 'aspect', 'at_rule', 'atom', 'attr_selector', 'attribute', 'begin', 'blockquote', 'body', 'built_in', 'bullet', 'cbracket', 'cdata', 'cell', 'change', 'char', 'characteristic', 'chunk', 'code', 'collection', 'command', 'commands', 'component', 'container', 'data', 'date', 'decorator', 'default', 'deletion', 'doctag', 'doctype', 'emphasis', 'end', 'envvar', 'expression', 'facet', 'filename', 'filter', 'flow', 'foreign', 'formula', 'func', 'function_name', 'generics', 'header', 'hexcolor', 'horizontal_rule', 'id', 'import', 'important', 'infix', 'inheritance', 'input', 'instance', 'instruction', 'io', 'keywords', 'kind', 'label', 'link_label', 'link_reference', 'link_url', 'list', 'literal', 'localvars', 'long_brackets', 'matrix', 'misc_keyword', 'module', 'operator', 'output', 'package', 'param', 'parameter', 'params', 'parent', 'pi', 'pod', 'pp', 'pragma', 'preprocessor', 'prompt', 'property', 'pseudo', 'quoted', 'record_name', 'regex', 'request', 'reserved', 'rest_arg', 'rule', 'rules', 'section', 'shader', 'shading', 'shebang', 'special', 'sqbracket', 'status', 'stream', 'strong', 'sub', 'summary', 'tag', 'template_tag', 'type', 'typedef', 'typename', 'units', 'value', 'var_expand', 'verb', 'winutils', 'xmlDocTag'],

  theme: function theme(str) {
    var theme = str || 'default';
    var file = highlighter.configPath + '/' + theme + '.json';
    var config = undefined;
    try {
      config = require(file);
    } catch (e) {
      return false;
    }

    this.mapping = {};
    this.mapping.fallback = {};
    for (var i = 0; i < this.allClasses.length; ++i) {
      this.mapping.fallback['reset' + i] = new RegExp('<span class="hljs-' + this.allClasses[i] + '">((.|\n)*?)</span>', 'g');
    }

    for (var lang in config) {
      if (config.hasOwnProperty(lang)) {
        this.mapping[lang] = this.mapping[lang] || {};
        var ctr = 0;
        for (var item in config[lang]) {
          if (config[lang].hasOwnProperty(item)) {
            ctr++;
            var styles = config[lang][item];
            styles = _.isArray(styles) ? styles : [styles];
            for (var j = 0; j < styles.length; ++j) {
              if (lang === 'markdown') {
                this.mapping[lang][styles[j] + ctr] = new RegExp('<md-' + item + '>((.|\n)*?)</md>', 'g');
              } else {
                this.mapping[lang][styles[j] + ctr] = new RegExp('<span class="hljs-' + item + '">((.|\n)*?)</span>', 'g');
              }
            }
          }
        }
      }
    }

    return this.mapping;
  },

  getThemes: function getThemes() {
    var themes = [];
    fs.readdirSync(path.normalize(highlighter.configPath)).forEach(function (name) {
      var parts = String(name).split('.');
      if (parts[parts.length - 1] === 'json') {
        parts.pop();
        themes.push(parts.join('.'));
      }
    });
    return themes;
  },

  highlight: function highlight(data, lang) {
    var fallback = !this.mapping[lang] ? 'default' : undefined;
    var hl = this.unescape(data);

    if (lang !== 'markdown') {
      if (!lang) {
        hl = hljs.highlightAuto(hl);
      } else {
        try {
          hl = hljs.highlight(lang, hl);
        } catch (e) {
          hl = hljs.highlightAuto(hl);
        }
      }
      if (hl.language) {
        fallback = hl.language;
      }
      hl = hl.value;
    }

    hl = this.format(hl);

    var mappingLang = fallback || lang;
    mappingLang = !this.mapping[mappingLang] ? 'default' : mappingLang;

    // If a custom language is detected, apply its styles.
    for (var color in this.mapping[mappingLang]) {
      if (this.mapping[mappingLang].hasOwnProperty(color)) {
        var clr = String(color).replace(/[0-9]/g, '');
        hl = String(hl).replace(this.mapping[mappingLang][color], chalk[clr]('$1'));
      }
    }

    // If the "default" styles weren't applied, apply them now.
    if (mappingLang !== 'default') {
      for (var color in this.mapping['default']) {
        if (this.mapping['default'].hasOwnProperty(color)) {
          var clr = String(color).replace(/[0-9]/g, '');
          hl = String(hl).replace(this.mapping['default'][color], chalk[clr]('$1'));
        }
      }
    }

    // Catch any highlighting tags not given in
    // that theme file, and reset any color on them.
    for (var style in this.mapping.fallback) {
      if (this.mapping.fallback.hasOwnProperty(style)) {
        hl = String(hl).replace(this.mapping.fallback[style], chalk.reset('$1'));
      }
    }

    return hl;
  },

  format: function format(str) {
    return str;
  },

  unescape: function unescape(data) {
    data = data.replace(/&lt;/g, '<');
    data = data.replace(/&gt;/g, '>');
    data = data.replace(/&apos;/g, '\'');
    data = data.replace(/&quot;/g, '"');
    data = data.replace(/&amp;/g, '&');
    data = data.replace(/&#39;/g, '\'');
    return data;
  }
};

module.exports = highlighter;