feat(rehype-proc-term): termについてそれを直接囲む要素を表すtermContainerを追加(一部のタグのみ)

Author: LumaKernel

examples/render/output/def.mdx.js

@@ -44,6 +44,7 @@ function _createMdxContent(props) {
                 text: "用語",
                 reference: "用語",
                 refIndex: JSON.parse("0"),
+                termContainer: JSON.parse("null"),
               }),
               "を",
               "展",
@@ -85,6 +86,7 @@ function _createMdxContent(props) {
                       text: "用語",
                       reference: "用語",
                       refIndex: JSON.parse("1"),
+                      termContainer: JSON.parse("null"),
                     }),
                   ],
                 }),
@@ -103,12 +105,14 @@ function _createMdxContent(props) {
               text: "用語",
               reference: "用語",
               refIndex: JSON.parse("2"),
+              termContainer: JSON.parse("null"),
             }),
             "の",
             _jsx(Term, {
               text: "定義",
               reference: "定義",
               refIndex: JSON.parse("0"),
+              termContainer: JSON.parse("null"),
             }),
             "を",
             "す",
@@ -133,6 +137,7 @@ function _createMdxContent(props) {
               text: "用語",
               reference: "用語",
               refIndex: JSON.parse("3"),
+              termContainer: JSON.parse('"h2"'),
             }),
             "も",
             "置",

examples/render/output/save-0.mdx.js

@@ -34,6 +34,7 @@ function _createMdxContent(props) {
                 text: "定義",
                 reference: "定義",
                 refIndex: JSON.parse("0"),
+                termContainer: JSON.parse("null"),
               }),
               "は",
             ],
@@ -65,6 +66,7 @@ function _createMdxContent(props) {
                 text: "定義",
                 reference: "定義",
                 refIndex: JSON.parse("1"),
+                termContainer: JSON.parse("null"),
               }),
               "は",
             ],
@@ -96,6 +98,7 @@ function _createMdxContent(props) {
                 text: "定義",
                 reference: "定義",
                 refIndex: JSON.parse("2"),
+                termContainer: JSON.parse("null"),
               }),
               "は",
             ],
@@ -127,6 +130,7 @@ function _createMdxContent(props) {
                 text: "定義",
                 reference: "定義",
                 refIndex: JSON.parse("3"),
+                termContainer: JSON.parse("null"),
               }),
               "は",
             ],

src/rehype-proc-term.ts

@@ -31,14 +31,33 @@ export type RehypeProcTermPluginParams = {
   readonly termProcessor: TermProcessorProtocol;
 };
 
+export type TermContainer = "h1" | "h2" | "h3" | "h4" | "h5" | "h6";
+export const parseConatiner = (parent: unknown): TermContainer | null => {
+  if (typeof parent !== "object" || parent === null) return null;
+  if (!("type" in parent) || typeof parent.type !== "string") return null;
+  if (!("tagName" in parent) || typeof parent.tagName !== "string") return null;
+  if (parent.type !== "element") return null;
+
+  switch (parent.tagName) {
+    case "h1":
+    case "h2":
+    case "h3":
+    case "h4":
+    case "h5":
+    case "h6":
+      return parent.tagName;
+  }
+  return null;
+};
+
 export type RehypeProcTermPlugin = import("unified").Plugin<
   [RehypeProcTermPluginParams],
   Root
 >;
 const rehypeProcTerm: RehypeProcTermPlugin = ({ termProcessor }) => {
   return async (tree) => {
     const refCount = new Map<string, number>();
-    await visitAsync(tree, async (node) => {
+    await visitAsync(tree, async (node, parents) => {
       if (node.type !== "text" || typeof node.value !== "string") return;
 
       const newChildren: MdxJsxTextElement["children"] = [];
@@ -59,6 +78,7 @@ const rehypeProcTerm: RehypeProcTermPlugin = ({ termProcessor }) => {
             break;
           case "term": {
             const refIndex = refCount.get(parsed.term) ?? 0;
+            const termContainer = parseConatiner(parents.at(-1)?.node);
             refCount.set(parsed.term, refIndex + 1);
             newChildren.push({
               type: "mdxJsxTextElement",
@@ -81,6 +101,17 @@ const rehypeProcTerm: RehypeProcTermPlugin = ({ termProcessor }) => {
                     },
                   },
                 },
+                {
+                  type: "mdxJsxAttribute",
+                  name: "termContainer",
+                  value: {
+                    type: "mdxJsxAttributeValueExpression",
+                    value: "",
+                    data: {
+                      estree: estreeJsonParseOf(termContainer),
+                    },
+                  },
+                },
               ],
               children: [],
             });