Preserve HAST element data and properties

packages/core/src/highlight/code-to-hast.ts

@@ -128,6 +128,7 @@ export function tokensToHast(
     tagName: 'pre',
     properties,
     children: [],
+    data: options.data as any,
   }
 
   let codeNode: Element = {

packages/core/test/hast.test.ts

@@ -166,4 +166,23 @@ describe('merge same style', () => {
 
     expect(html).toMatchInlineSnapshot(`"<pre class="shiki min-light" style="background-color:#ffffff;color:#24292eff" tabindex="0"><code><span class="line"><span style="color:#D32F2F" class="highlighted-word">name</span><span style="color:#D32F2F">:</span><span style="color:#22863A"> CI</span></span></code></pre>"`)
   })
+
+  it('supports data', async () => {
+    using shiki = await createHighlighter({
+      themes: ['vitesse-light'],
+      langs: ['javascript'],
+    })
+
+    const hast = shiki.codeToHast('console.log', {
+      lang: 'js',
+      theme: 'vitesse-light',
+      data: {
+        meta: 'foo="bar"',
+      },
+    })
+
+    expect((hast.children[0] as any).data).toEqual({
+      meta: 'foo="bar"',
+    })
+  })
 })

packages/types/src/options.ts

@@ -144,6 +144,11 @@ export interface CodeToHastOptionsCommon<Languages extends string = string>
   DecorationOptions,
   Pick<TokenizeWithThemeOptions, 'colorReplacements' | 'tokenizeMaxLineLength' | 'tokenizeTimeLimit' | 'grammarState' | 'grammarContextCode' | 'includeExplanation'> {
 
+  /**
+   * Data to be added to the root `<pre>` element.
+   */
+  data?: Record<string, unknown>
+
   /**
    * The grammar name for the code.
    */