fix: meta should not be removed by rehypeFernCode inside CodeBlocks

fern-api · Mar 26, 2024 · 1299723 · 1299723
1 parent 4bad116
commit 1299723
Showing 1 changed file with 13 additions and 3 deletions.
diff --git a/packages/ui/app/src/mdx/plugins/rehypeFernCode.ts b/packages/ui/app/src/mdx/plugins/rehypeFernCode.ts
@@ -78,7 +78,10 @@ export function rehypeFernCode(): (tree: Root) => void {
             }
 
             if (isBlockCode(node) && node.data?.visited !== true) {
-                node.data = { visited: true };
+                if (node.data == null) {
+                    node.data = {};
+                }
+                node.data.visited = true;
                 const head = node.children.filter(isElement).find((child) => child.tagName === "code");
                 if (head != null) {
                     const code = getCode(head);
@@ -131,7 +134,10 @@ function visitCodeBlockNodes(nodeToVisit: MdxJsxFlowElementHast) {
             const title = jsxAttributes.find((attr) => attr.name === "title");
             visit(node, "element", (child) => {
                 if (child.tagName === "code" && child.data?.visited !== true) {
-                    child.data = { visited: true };
+                    if (child.data == null) {
+                        child.data = {};
+                    }
+                    child.data.visited = true;
                     const code = getCode(child);
                     const meta = parseBlockMetaString(child, "plaintext");
                     if (code != null) {
@@ -157,7 +163,10 @@ function visitCodeBlockNodes(nodeToVisit: MdxJsxFlowElementHast) {
 
     visit(nodeToVisit, "element", (child) => {
         if (child.tagName === "code" && child.data?.visited !== true) {
-            child.data = { visited: true };
+            if (child.data == null) {
+                child.data = {};
+            }
+            child.data.visited = true;
             const code = getCode(child);
             const meta = parseBlockMetaString(child, "plaintext");
             if (code != null) {
@@ -212,6 +221,7 @@ function maybeParseInt(str: string | null | undefined): number | undefined {
 
 export function parseBlockMetaString(element: Element, defaultFallback: string): FernCodeMeta {
     let meta: string = unknownToString(element.data?.meta ?? element.properties?.metastring ?? "");
+    console.log(meta);
 
     const titleMatch = meta.match(/title="([^"]*)"/);
     const title = titleMatch?.[1];