support replacement as a function (#76)

* support replacement as a function * fix typo * Create rotten-seahorses-occur.md --------- Co-authored-by: Oleksandr Fediashov <[email protected]>
Anber · Apr 25, 2024 · 9096ba1 · 9096ba1
1 parent 99a3634
commit 9096ba1
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 4 deletions.
diff --git a/.changeset/rotten-seahorses-occur.md b/.changeset/rotten-seahorses-occur.md
@@ -0,0 +1,5 @@
+---
+"@wyw-in-js/processor-utils": patch
+---
+
+feat: support `replacement` as a function
diff --git a/packages/processor-utils/src/BaseProcessor.ts b/packages/processor-utils/src/BaseProcessor.ts
@@ -1,5 +1,5 @@
 /* eslint-disable class-methods-use-this */
-import type { types as t } from '@babel/core';
+import type { NodePath, types as t } from '@babel/core';
 import generator from '@babel/generator';
 import type {
   Expression,
@@ -61,7 +61,7 @@ export abstract class BaseProcessor {
     },
     public readonly location: SourceLocation | null,
     protected readonly replacer: (
-      replacement: Expression,
+      replacement: Expression | ((tagPath: NodePath) => Expression),
       isPure: boolean
     ) => void,
     public readonly displayName: string,

diff --git a/packages/transform/src/utils/getTagProcessor.ts b/packages/transform/src/utils/getTagProcessor.ts
@@ -245,9 +245,14 @@ function getBuilderForIdentifier(
     break;
   }
 
-  const replacer = (replacement: Expression, isPure: boolean) => {
+  const replacer = (
+    replacement: Expression | ((tagPath: NodePath) => Expression),
+    isPure: boolean
+  ) => {
     mutate(prev, (p) => {
-      p.replaceWith(replacement);
+      p.replaceWith(
+        typeof replacement === 'function' ? replacement(p) : replacement
+      );
       if (isPure) {
         p.addComment('leading', '#__PURE__');
       }