hexo-util
Version:
Utilities for Hexo.
110 lines • 4.54 kB
JavaScript
;
var __importDefault = (this && this.__importDefault) || function (mod) {
return (mod && mod.__esModule) ? mod : { "default": mod };
};
const strip_indent_1 = __importDefault(require("strip-indent"));
const components_1 = __importDefault(require("prismjs/components/"));
let Prism;
// https://github.com/PrismJS/prism/issues/2145
const components_2 = __importDefault(require("prismjs/components"));
const prismAlias = Object.entries(components_2.default.languages).reduce((acc, [key, value]) => {
if (value.alias) {
if (Array.isArray(value.alias)) {
value.alias.forEach(alias => (acc[alias] = key));
}
else if (typeof value.alias === 'string') {
acc[value.alias] = key;
}
}
return acc;
}, {});
const prismSupportedLanguages = Object.keys(components_2.default.languages).concat(Object.keys(prismAlias));
const escape_html_1 = __importDefault(require("./escape_html"));
/**
* Wrapper of Prism.highlight()
* @param {String} code
* @param {String} language
*/
function prismHighlight(code, language) {
if (!Prism)
Prism = require('prismjs');
// Prism has not load the language pattern
if (!Prism.languages[language] && prismSupportedLanguages.includes(language))
(0, components_1.default)(language);
if (Prism.languages[language]) {
// Prism escapes output by default
return Prism.highlight(code, Prism.languages[language], language);
}
// Current language is not supported by Prism, return origin code;
return (0, escape_html_1.default)(code);
}
/**
* Generate line number required HTML snippet
* @param {String} code - Highlighted code
*/
function lineNumberUtil(code) {
const matched = code.match(/\n(?!$)/g);
const num = matched ? matched.length + 1 : 1;
const lines = new Array(num + 1).join('<span></span>');
return `<span aria-hidden="true" class="line-numbers-rows">${lines}</span>`;
}
function replaceTabs(str, tab) {
return str.replace(/^\t+/gm, match => tab.repeat(match.length));
}
function PrismUtil(str, options = {}) {
if (typeof str !== 'string')
throw new TypeError('str must be a string!');
const { lineNumber = true, lang = 'none', tab, mark, firstLine, isPreprocess = true, caption, stripIndent: enableStripIndent = true } = options;
if (enableStripIndent) {
str = (0, strip_indent_1.default)(str);
}
// To be consistent with highlight.js
let language = lang === 'plaintext' || lang === 'none' ? 'none' : lang;
if (prismAlias[language])
language = prismAlias[language];
const preTagClassArr = [];
const preTagAttrArr = [];
let preTagAttr = '';
if (lineNumber)
preTagClassArr.push('line-numbers');
preTagClassArr.push(`language-${language}`);
// Show Languages plugin
// https://prismjs.com/plugins/show-language/
if (language !== 'none')
preTagAttrArr.push(`data-language="${language}"`);
if (!isPreprocess) {
// Shift Line Numbers ('firstLine' option) should only be added under non-preprocess mode
// https://prismjs.com/plugins/line-numbers/
if (lineNumber && firstLine)
preTagAttrArr.push(`data-start="${firstLine}"`);
// Line Highlight ('mark' option) should only be added under non-preprocess mode
// https://prismjs.com/plugins/line-highlight/
if (mark)
preTagAttrArr.push(`data-line="${mark}"`);
// Apply offset for 'mark' option
// https://github.com/hexojs/hexo-util/pull/172#issuecomment-571882480
if (firstLine && mark)
preTagAttrArr.push(`data-line-offset="${firstLine - 1}"`);
}
if (preTagAttrArr.length)
preTagAttr = ' ' + preTagAttrArr.join(' ');
if (tab)
str = replaceTabs(str, tab);
const codeCaption = caption ? `<div class="caption">${caption}</div>` : '';
const startTag = `<pre class="${preTagClassArr.join(' ')}"${preTagAttr}>${codeCaption}<code class="language-${language}">`;
const endTag = '</code></pre>';
let parsedCode = '';
if (language === 'none' || !isPreprocess) {
parsedCode = (0, escape_html_1.default)(str);
}
else {
parsedCode = prismHighlight(str, language);
}
// lineNumberUtil() should be used only under preprocess mode
if (lineNumber && isPreprocess) {
parsedCode += lineNumberUtil(parsedCode);
}
return startTag + parsedCode + endTag;
}
module.exports = PrismUtil;
//# sourceMappingURL=prism.js.map