documentation/lib/output/highlighter.js

a documentation generator

'use strict';

const visit = require('unist-util-visit');
const hljs = require('highlight.js');

/**
 * Adapted from remark-highlight.js
 * https://github.com/ben-eb/remark-highlight.js
 * @param {Object} node AST node
 * @returns {undefined} modifies the node by reference
 * @private
 */
function visitor(node) {
  if (node.lang) {
    node.type = 'html';
    node.value = "<pre class='hljs'>" + hljs.highlightAuto(node.value, [node.lang]).value + '</pre>';
  }
}

module.exports = function (ast) {
  visit(ast, 'code', visitor);
  return ast;
};