Rename transform files

Author: pomber

packages/mdx/src/remark/code.ts

@@ -18,22 +18,6 @@ import { mergeFocus } from "../utils"
 import { CodeNode, SuperNode } from "./nodes"
 import { CodeHikeConfig } from "./config"
 
-export async function transformCodeNodes(
-  tree: SuperNode,
-  config: CodeHikeConfig
-) {
-  await visitAsync(
-    tree,
-    "code",
-    async (node: CodeNode, index, parent) => {
-      await transformCode(
-        { node, index, parent: parent! },
-        config
-      )
-    }
-  )
-}
-
 export function isEditorNode(node: SuperNode) {
   return (
     node.type === "code" ||
@@ -42,25 +26,6 @@ export function isEditorNode(node: SuperNode) {
   )
 }
 
-async function transformCode(
-  nodeInfo: NodeInfo<CodeNode>,
-  config: CodeHikeConfig
-) {
-  toJSX(nodeInfo.node, {
-    name: "CH.Code",
-    props: await mapCode(nodeInfo, config),
-  })
-}
-export async function transformEditor(
-  nodeInfo: NodeInfo,
-  config: CodeHikeConfig
-) {
-  toJSX(nodeInfo.node, {
-    name: "CH.Code",
-    props: await mapEditor(nodeInfo, config),
-  })
-}
-
 export async function mapAnyCodeNode(
   nodeInfo: NodeInfo,
   config: CodeHikeConfig

packages/mdx/src/remark/editor.tsx

@@ -0,0 +1,36 @@
+import { NodeInfo, toJSX, visitAsync } from "./unist-utils"
+import { mapEditor } from "./code"
+import { CodeNode, JsxNode, SuperNode } from "./nodes"
+import { CodeHikeConfig } from "./config"
+
+export async function transformCodes(
+  tree: SuperNode,
+  config: CodeHikeConfig
+) {
+  await visitAsync(
+    tree,
+    "mdxJsxFlowElement",
+    async (node: JsxNode, index, parent) => {
+      if (node.name === "CH.Code") {
+        await transformCode({ node, index, parent }, config)
+      }
+    }
+  )
+  await visitAsync(
+    tree,
+    "code",
+    async (node: CodeNode, index, parent) => {
+      await transformCode({ node, index, parent }, config)
+    }
+  )
+}
+
+async function transformCode(
+  nodeInfo: NodeInfo,
+  config: CodeHikeConfig
+) {
+  toJSX(nodeInfo.node, {
+    name: "CH.Code",
+    props: await mapEditor(nodeInfo, config),
+  })
+}

packages/mdx/src/remark/transform.code.ts

@@ -5,7 +5,7 @@ import {
 } from "./unist-utils"
 import { extractStepsInfo } from "./steps"
 import { getPresetConfig } from "./transform.preview"
-import { transformLinks } from "./section"
+import { transformLinks } from "./transform.section"
 import { SuperNode } from "./nodes"
 import { CodeHikeConfig } from "./config"
 

packages/mdx/src/remark/inline-code.ts renamed to packages/mdx/src/remark/transform.inline-code.ts

@@ -1,10 +1,9 @@
-import { transformCodeNodes } from "./code"
-import { transformEditorNodes } from "./editor"
-import { transformSections } from "./section"
-import { transformSpotlights } from "./spotlight"
-import { transformScrollycodings } from "./scrollycoding"
-import { transformSlideshows } from "./slideshow"
-import { transformInlineCodes } from "./inline-code"
+import { transformCodes } from "./transform.code"
+import { transformSections } from "./transform.section"
+import { transformSpotlights } from "./transform.spotlight"
+import { transformScrollycodings } from "./transform.scrollycoding"
+import { transformSlideshows } from "./transform.slideshow"
+import { transformInlineCodes } from "./transform.inline-code"
 import { transformPreviews } from "./transform.preview"
 
 import { valueToEstree } from "./to-estree"
@@ -19,8 +18,7 @@ const transforms = [
   transformSlideshows,
   transformSections,
   transformInlineCodes,
-  transformEditorNodes,
-  transformCodeNodes,
+  transformCodes,
 ]
 
 export function transform(unsafeConfig: CodeHikeConfig) {
@@ -31,20 +29,20 @@ export function transform(unsafeConfig: CodeHikeConfig) {
       for (const transform of transforms) {
         await transform(tree, config)
       }
-    } catch (e) {
-      console.error("error running remarkCodeHike", e)
-      throw e
-    }
 
-    const usedCodeHikeComponents =
-      getUsedCodeHikeComponentNames(tree)
+      const usedCodeHikeComponents =
+        getUsedCodeHikeComponentNames(tree)
 
-    if (usedCodeHikeComponents.length > 0) {
-      addConfig(tree, config)
+      if (usedCodeHikeComponents.length > 0) {
+        addConfig(tree, config)
 
-      if (config.autoImport) {
-        addSmartImport(tree, usedCodeHikeComponents)
+        if (config.autoImport) {
+          addSmartImport(tree, usedCodeHikeComponents)
+        }
       }
+    } catch (e) {
+      console.error("error running remarkCodeHike", e)
+      throw e
     }
   }
 }