dgeni-packages/base/services/encodeCodeBlock.js

A collection of dgeni packages for generating documentation from source code

const htmlEncode = require('htmlencode').htmlEncode;

module.exports = function encodeCodeBlock() {
  return (str, inline, lang) => {

    // Encode any HTML entities in the code string
    str = htmlEncode(str, true);

    // If a language is provided then attach a CSS class to the code element
    lang = lang ? ' class="lang-' + lang + '"' : '';

    str = '<code' + lang + '>' + str + '</code>';

    // If not inline then wrap the code element in a pre element
    if ( !inline ) {
      str = '<pre>' + str + '</pre>';
    }

    return str;
  };
};