Files
documentation/node_modules/@docusaurus/theme-common/lib/utils/codeBlockUtils.js
2024-03-22 03:47:51 +05:30

210 lines
8.6 KiB
JavaScript
Raw Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

/**
* Copyright (c) Facebook, Inc. and its affiliates.
*
* This source code is licensed under the MIT license found in the
* LICENSE file in the root directory of this source tree.
*/
import rangeParser from 'parse-numeric-range';
const codeBlockTitleRegex = /title=(?<quote>["'])(?<title>.*?)\1/;
const metastringLinesRangeRegex = /\{(?<range>[\d,-]+)\}/;
// Supported types of highlight comments
const popularCommentPatterns = {
js: { start: '\\/\\/', end: '' },
jsBlock: { start: '\\/\\*', end: '\\*\\/' },
jsx: { start: '\\{\\s*\\/\\*', end: '\\*\\/\\s*\\}' },
bash: { start: '#', end: '' },
html: { start: '<!--', end: '-->' },
};
const commentPatterns = {
...popularCommentPatterns,
// minor comment styles
lua: { start: '--', end: '' },
wasm: { start: '\\;\\;', end: '' },
tex: { start: '%', end: '' },
vb: { start: "[']", end: '' },
vbnet: { start: "(?:_\\s*)?[']", end: '' },
rem: { start: '[Rr][Ee][Mm]\\b', end: '' },
f90: { start: '!', end: '' },
ml: { start: '\\(\\*', end: '\\*\\)' },
cobol: { start: '\\*>', end: '' }, // Free format only
};
const popularCommentTypes = Object.keys(popularCommentPatterns);
function getCommentPattern(languages, magicCommentDirectives) {
// To be more reliable, the opening and closing comment must match
const commentPattern = languages
.map((lang) => {
const { start, end } = commentPatterns[lang];
return `(?:${start}\\s*(${magicCommentDirectives
.flatMap((d) => [d.line, d.block?.start, d.block?.end].filter(Boolean))
.join('|')})\\s*${end})`;
})
.join('|');
// White space is allowed, but otherwise it should be on it's own line
return new RegExp(`^\\s*(?:${commentPattern})\\s*$`);
}
/**
* Select comment styles based on language
*/
function getAllMagicCommentDirectiveStyles(lang, magicCommentDirectives) {
switch (lang) {
case 'js':
case 'javascript':
case 'ts':
case 'typescript':
return getCommentPattern(['js', 'jsBlock'], magicCommentDirectives);
case 'jsx':
case 'tsx':
return getCommentPattern(['js', 'jsBlock', 'jsx'], magicCommentDirectives);
case 'html':
return getCommentPattern(['js', 'jsBlock', 'html'], magicCommentDirectives);
case 'python':
case 'py':
case 'bash':
return getCommentPattern(['bash'], magicCommentDirectives);
case 'markdown':
case 'md':
// Text uses HTML, front matter uses bash
return getCommentPattern(['html', 'jsx', 'bash'], magicCommentDirectives);
case 'tex':
case 'latex':
case 'matlab':
return getCommentPattern(['tex'], magicCommentDirectives);
case 'lua':
case 'haskell':
case 'sql':
return getCommentPattern(['lua'], magicCommentDirectives);
case 'wasm':
return getCommentPattern(['wasm'], magicCommentDirectives);
case 'vb':
case 'vba':
case 'visual-basic':
return getCommentPattern(['vb', 'rem'], magicCommentDirectives);
case 'vbnet':
return getCommentPattern(['vbnet', 'rem'], magicCommentDirectives);
case 'batch':
return getCommentPattern(['rem'], magicCommentDirectives);
case 'basic': // https://github.com/PrismJS/prism/blob/master/components/prism-basic.js#L3
return getCommentPattern(['rem', 'f90'], magicCommentDirectives);
case 'fsharp':
return getCommentPattern(['js', 'ml'], magicCommentDirectives);
case 'ocaml':
case 'sml':
return getCommentPattern(['ml'], magicCommentDirectives);
case 'fortran':
return getCommentPattern(['f90'], magicCommentDirectives);
case 'cobol':
return getCommentPattern(['cobol'], magicCommentDirectives);
default:
// All popular comment types
return getCommentPattern(popularCommentTypes, magicCommentDirectives);
}
}
export function parseCodeBlockTitle(metastring) {
return metastring?.match(codeBlockTitleRegex)?.groups.title ?? '';
}
export function containsLineNumbers(metastring) {
return Boolean(metastring?.includes('showLineNumbers'));
}
/**
* Gets the language name from the class name (set by MDX).
* e.g. `"language-javascript"` => `"javascript"`.
* Returns undefined if there is no language class name.
*/
export function parseLanguage(className) {
const languageClassName = className
.split(' ')
.find((str) => str.startsWith('language-'));
return languageClassName?.replace(/language-/, '');
}
/**
* Parses the code content, strips away any magic comments, and returns the
* clean content and the highlighted lines marked by the comments or metastring.
*
* If the metastring contains a range, the `content` will be returned as-is
* without any parsing. The returned `lineClassNames` will be a map from that
* number range to the first magic comment config entry (which _should_ be for
* line highlight directives.)
*
* @param content The raw code with magic comments. Trailing newline will be
* trimmed upfront.
* @param options Options for parsing behavior.
*/
export function parseLines(content, options) {
let code = content.replace(/\n$/, '');
const { language, magicComments, metastring } = options;
// Highlighted lines specified in props: don't parse the content
if (metastring && metastringLinesRangeRegex.test(metastring)) {
const linesRange = metastring.match(metastringLinesRangeRegex).groups
.range;
if (magicComments.length === 0) {
throw new Error(`A highlight range has been given in code block's metastring (\`\`\` ${metastring}), but no magic comment config is available. Docusaurus applies the first magic comment entry's className for metastring ranges.`);
}
const metastringRangeClassName = magicComments[0].className;
const lines = rangeParser(linesRange)
.filter((n) => n > 0)
.map((n) => [n - 1, [metastringRangeClassName]]);
return { lineClassNames: Object.fromEntries(lines), code };
}
if (language === undefined) {
return { lineClassNames: {}, code };
}
const directiveRegex = getAllMagicCommentDirectiveStyles(language, magicComments);
// Go through line by line
const lines = code.split('\n');
const blocks = Object.fromEntries(magicComments.map((d) => [d.className, { start: 0, range: '' }]));
const lineToClassName = Object.fromEntries(magicComments
.filter((d) => d.line)
.map(({ className, line }) => [line, className]));
const blockStartToClassName = Object.fromEntries(magicComments
.filter((d) => d.block)
.map(({ className, block }) => [block.start, className]));
const blockEndToClassName = Object.fromEntries(magicComments
.filter((d) => d.block)
.map(({ className, block }) => [block.end, className]));
for (let lineNumber = 0; lineNumber < lines.length;) {
const line = lines[lineNumber];
const match = line.match(directiveRegex);
if (!match) {
// Lines without directives are unchanged
lineNumber += 1;
continue;
}
const directive = match
.slice(1)
.find((item) => item !== undefined);
if (lineToClassName[directive]) {
blocks[lineToClassName[directive]].range += `${lineNumber},`;
}
else if (blockStartToClassName[directive]) {
blocks[blockStartToClassName[directive]].start = lineNumber;
}
else if (blockEndToClassName[directive]) {
blocks[blockEndToClassName[directive]].range += `${blocks[blockEndToClassName[directive]].start}-${lineNumber - 1},`;
}
lines.splice(lineNumber, 1);
}
code = lines.join('\n');
const lineClassNames = {};
Object.entries(blocks).forEach(([className, { range }]) => {
rangeParser(range).forEach((l) => {
lineClassNames[l] ??= [];
lineClassNames[l].push(className);
});
});
return { lineClassNames, code };
}
export function getPrismCssVariables(prismTheme) {
const mapping = {
color: '--prism-color',
backgroundColor: '--prism-background-color',
};
const properties = {};
Object.entries(prismTheme.plain).forEach(([key, value]) => {
const varName = mapping[key];
if (varName && typeof value === 'string') {
properties[varName] = value;
}
});
return properties;
}
//# sourceMappingURL=codeBlockUtils.js.map