test: test highlighted lines in code blocks

test/markdown/code-block-highlighted-lines.test.ts

@@ -0,0 +1,24 @@
+import { expect, it } from 'vitest'
+import { parseMarkdown } from '../utils/parser'
+
+const md = `
+\`\`\`ts {2,3}
+class C {
+  private name: string = "foo"
+  private age: number = 18
+}
+\`\`\`
+`.trim()
+
+it('Highlighted code block', async () => {
+  const { body } = await parseMarkdown(md, {
+    highlight: {
+      theme: 'github-dark'
+    }
+  })
+  expect(body).toHaveProperty('type', 'root')
+  expect(body.children).toHaveLength(2)
+  expect((body.children[0] as any).props?.highlights).toEqual([2,3])
+  expect((body.children[0] as any).children[0].children[1].props).haveOwnProperty('class', 'line highlight')
+  expect((body.children[0] as any).children[0].children[2].props).haveOwnProperty('class', 'line highlight')
+})

test/utils/parser.ts

@@ -17,12 +17,12 @@ export const parseMarkdown = (md: string, options: MDCParseOptions = {}) => {
     options.highlight = options.highlight || {}
     options.highlight.theme = options.highlight.theme || 'github-light'
 
-    options.highlight.highlighter = async (code: string, lang: string, theme: Theme) => {
+    options.highlight.highlighter = async (code: string, lang: string, theme: Theme, highlights) => {
       const shikiHighlighter = useShikiHighlighter({})
 
       const styleMap: TokenStyleMap = {}
 
-      const { tree, className } = await shikiHighlighter.getHighlightedAST(code as string, lang as any, theme as Theme, { styleMap })
+      const { tree, className } = await shikiHighlighter.getHighlightedAST(code as string, lang as any, theme as Theme, { styleMap, highlights })
 
       return {
         tree,