// we need a custom renderer for code blocks cause we have a slightly non compliant // format with special handling for text and so on const TEXT_CODE_CLASSES = ["text", "pre", "plain"]; function extractTokenInfo(info, md) { if (!info) { return; } info = info.trim(); const matches = info.match(/(^\s*\S*)\s*(.*)/i); if (!matches) { return; } // ensure the token has only valid chars // c++, strucuted-text and p91, are all valid if (!/^[\w+-]*$/i.test(matches[1])) { return; } const ASCII_REGEX = /[^\x00-\x7F]/; const tag = md.utils.unescapeAll(matches[1].replace(ASCII_REGEX, "")); const extractedData = { tag, attributes: {} }; if (matches[2]?.length) { md.utils .unescapeAll(matches[2].replace(ASCII_REGEX, "")) .split(",") .forEach((potentialPair) => { const [key, value] = potentialPair.trim().split(/\s+/g)[0].split("="); // invalid pairs would get caught here and not used, eg `foo=` if (key && value) { extractedData.attributes[key] = value; } }); } return extractedData; } function render(tokens, idx, options, env, slf, md) { const token = tokens[idx]; const escapedContent = md.utils.escapeHtml(token.content); const tokenInfo = extractTokenInfo(token.info, md); const tag = tokenInfo?.tag || md.options.discourse.defaultCodeLang; const attributes = tokenInfo?.attributes || {}; let className; const acceptableCodeClasses = md.options.discourse.acceptableCodeClasses || []; if (TEXT_CODE_CLASSES.indexOf(tag) > -1) { className = "lang-nohighlight"; } else if (acceptableCodeClasses.indexOf(tag) > -1) { className = `lang-${tag}`; } else { className = "lang-nohighlight"; attributes["wrap"] = tag; } const dataAttributes = Object.keys(attributes) .map((key) => { const value = md.utils.escapeHtml(attributes[key]); key = md.utils.escapeHtml(key); return `data-code-${key}="${value}"`; }) .join(" "); return `
${escapedContent}
\n`;
}
export function setup(helper) {
helper.registerOptions((opts, siteSettings) => {
opts.defaultCodeLang = siteSettings.default_code_lang;
opts.acceptableCodeClasses = (siteSettings.highlighted_languages || "")
.split("|")
.filter(Boolean)
.concat(["auto", "nohighlight"]);
});
helper.allowList(["pre[data-code-*]"]);
helper.allowList({
custom(tag, name, value) {
if (tag === "code" && name === "class") {
const m = /^lang\-(.+)$/.exec(value);
if (m) {
return helper.getOptions().acceptableCodeClasses.indexOf(m[1]) !== -1;
}
}
},
});
helper.registerPlugin((md) => {
md.renderer.rules.fence = (tokens, idx, options, env, slf) =>
render(tokens, idx, options, env, slf, md);
});
}