54 lines
1.7 KiB
JavaScript
54 lines
1.7 KiB
JavaScript
import { fromHtml } from "hast-util-from-html";
|
|
import { toText } from "hast-util-to-text";
|
|
import { removePosition } from "unist-util-remove-position";
|
|
import { visitParents } from "unist-util-visit-parents";
|
|
const languagePattern = /\blanguage-(\S+)\b/;
|
|
async function highlightCodeBlocks(tree, highlighter) {
|
|
const nodes = [];
|
|
visitParents(tree, { type: "element", tagName: "code" }, (node, ancestors) => {
|
|
const parent = ancestors.at(-1);
|
|
if (parent?.type !== "element" || parent.tagName !== "pre") {
|
|
return;
|
|
}
|
|
if (parent.children.length !== 1) {
|
|
return;
|
|
}
|
|
let languageMatch;
|
|
let { className } = node.properties;
|
|
if (typeof className === "string") {
|
|
languageMatch = languagePattern.exec(className);
|
|
} else if (Array.isArray(className)) {
|
|
for (const cls of className) {
|
|
if (typeof cls !== "string") {
|
|
continue;
|
|
}
|
|
languageMatch = languagePattern.exec(cls);
|
|
if (languageMatch) {
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
if (languageMatch?.[1] === "math") {
|
|
return;
|
|
}
|
|
nodes.push({
|
|
node,
|
|
language: languageMatch?.[1] || "plaintext",
|
|
parent,
|
|
grandParent: ancestors.at(-2)
|
|
});
|
|
});
|
|
for (const { node, language, grandParent, parent } of nodes) {
|
|
const meta = node.data?.meta ?? node.properties.metastring ?? void 0;
|
|
const code = toText(node, { whitespace: "pre" });
|
|
const html = await highlighter(code, language, { meta });
|
|
const replacement = fromHtml(html, { fragment: true }).children[0];
|
|
removePosition(replacement);
|
|
const index = grandParent.children.indexOf(parent);
|
|
grandParent.children[index] = replacement;
|
|
}
|
|
}
|
|
export {
|
|
highlightCodeBlocks
|
|
};
|