Function rehypeShiki

apps/baseai.dev/src/mdx/rehype.ts:23–59

()

let highlighter: any;

export function rehypeShiki() {
	return async (tree: any) => {
		highlighter =
			highlighter ??
			(await shiki.getHighlighter({
				theme: theme as any,
				langs: lang as any
			}));

		visit(tree, 'element', node => {
			if (
				node.tagName === 'pre' &&
				node.children[0]?.tagName === 'code'
			) {
				let codeNode = node.children[0];
				let textNode = codeNode.children[0];

				node.properties.code = textNode.value;

				if (node.properties.language) {
					let tokens = highlighter.codeToThemedTokens(
						textNode.value,
						node.properties.language
					);

					textNode.value = shiki.renderToHtml(tokens, {
						elements: {
							pre: ({ children }) => children,
							code: ({ children }) => children,
							line: ({ children }) => `<span>${children}</span>`
						}
					});
				}
			}
		});
	};
}

export function rehypeSlugify() {
	return (tree: any) => {