Anotate code nodes in html with data-lang

zephraph · Oct 4, 2024 · 8be58c4 · 8be58c4
1 parent 76d0c63
commit 8be58c4
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 2 deletions.
diff --git a/packages/remark-obsidian/index.ts b/packages/remark-obsidian/index.ts
@@ -1,9 +1,9 @@
-import type { RemarkPlugin } from "@astrojs/markdown-remark";
+import type { Node, RemarkPlugin } from "@astrojs/markdown-remark";
 import { visit } from "unist-util-visit";
 import internalLinkPlugin from "./internal-link";
 import embedPlugin from "./embed";
 import { href } from "./internal-link/utils";
-import type { EmbedNode, InternalLinkNode } from "./types";
+import type { CodeNode, EmbedNode, InternalLinkNode } from "./types";
 
 interface RemarkObsidianOptions {
   fileResolver?: (path: string) => Promise<string>;
@@ -48,6 +48,16 @@ export const remarkObsidian = (
             break;
         }
       });
+
+      visit(root, "code", (node: CodeNode) => {
+        if (node.lang) {
+          node.data = {
+            hProperties: {
+              ["data-lang"]: node.lang,
+            },
+          };
+        }
+      });
     };
   };
 

diff --git a/packages/remark-obsidian/types.d.ts b/packages/remark-obsidian/types.d.ts
@@ -26,6 +26,13 @@ export interface EmbedNode {
   data?: Record<string, any>;
 }
 
+export interface CodeNode {
+  type: "code";
+  lang?: string | null;
+  value?: string;
+  data?: Record<string, any>;
+}
+
 declare module "micromark-util-types" {
   export interface TokenTypeMap extends TokenTypeMap {
     internalLink: "internalLink";