29 lines
978 B
TypeScript
29 lines
978 B
TypeScript
// markdown-it plugin for wrapping <pre> ... </pre>.
|
|
//
|
|
// If your plugin was chained before preWrapper, you can add additional element directly.
|
|
// If your plugin was chained after preWrapper, you can use these slots:
|
|
// 1. <!--beforebegin-->
|
|
// 2. <!--afterbegin-->
|
|
// 3. <!--beforeend-->
|
|
// 4. <!--afterend-->
|
|
|
|
import type MarkdownIt from 'markdown-it';
|
|
import type { MarkdownParsedData } from '../markdown';
|
|
|
|
export const preWrapperPlugin = (md: MarkdownIt) => {
|
|
const fence = md.renderer.rules.fence!;
|
|
md.renderer.rules.fence = (...args) => {
|
|
const [tokens, idx] = args;
|
|
const token = tokens[idx];
|
|
const data = (md as any).__data as MarkdownParsedData;
|
|
if (token.info?.trim() === 'vue') {
|
|
data.vueCode = token.content;
|
|
}
|
|
const rawCode = fence(...args).replace(
|
|
/<pre /g,
|
|
`<pre class="language-${token.info?.trim()}" `,
|
|
);
|
|
return rawCode; //`<div class="language-${token.info.trim()}">${rawCode}</div>`;
|
|
};
|
|
};
|