packages/markdown/remark/src/remark-shiki.ts


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21

import { visit } from 'unist-util-visit';
import { createShikiHighlighter, type ShikiHighlighter } from './shiki.js';
import type { RemarkPlugin, ShikiConfig } from './types.js';

export function remarkShiki(config?: ShikiConfig): ReturnType<RemarkPlugin> {
	let highlighterAsync: Promise<ShikiHighlighter> | undefined;

	return async (tree: any) => {
		highlighterAsync ??= createShikiHighlighter(config);
		const highlighter = await highlighterAsync;

		visit(tree, 'code', (node) => {
			const lang = typeof node.lang === 'string' ? node.lang : 'plaintext';
			const html = highlighter.highlight(node.value, lang);

			node.type = 'html';
			node.value = html;
			node.children = [];
		});
	};
}