fix: keep element props (#139)

* fix: keep element props

* fix: add code.data

* refactor

src/index.ts

@@ -1,4 +1,4 @@
-import type { Element, ElementContent, Root } from 'hast';
+import type { Element, ElementContent, ElementData, Root } from 'hast';
 import type { Options, Theme } from '../';
 import type { CharsHighlighterOptions } from './types';
 import type { Highlighter, CodeToHastOptions } from 'shikiji';
@@ -54,8 +54,10 @@ function apply(
   }: ApplyProps,
 ) {
   element.tagName = inline ? 'span' : 'figure';
-  // User can replace this with a real Fragment at runtime
-  element.properties = { 'data-rehype-pretty-code-figure': '' };
+  element.properties['data-rehype-pretty-code-figure'] = '';
+
+  const codeData = element.children[0]?.data as ElementData | undefined;
+
   element.children = [tree]
     .map((tree) => {
       const pre = tree.children[0];
@@ -95,6 +97,7 @@ function apply(
 
       code.properties['data-language'] = lang;
       code.properties['data-theme'] = themeNamesString;
+      code.data = codeData;
 
       if (inline) {
         if (keepBackground) {