26 lines
947 B
TypeScript
26 lines
947 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 MarkdownIt from 'markdown-it';
|
||
|
|
import { 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>`;
|
||
|
|
};
|
||
|
|
};
|