feat: add finalize hook

unplugin · May 3, 2024 · d787e5f · d787e5f
1 parent 7c7a90e
commit d787e5f
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/src/core/transform.ts b/src/core/transform.ts
@@ -92,5 +92,9 @@ export async function transform(
     }
   }
 
+  for (const { transformer } of transformers) {
+    await transformer.finalize?.(s)
+  }
+
   return generateTransform(s, id)
 }
diff --git a/src/core/types.ts b/src/core/types.ts
@@ -1,3 +1,4 @@
+import type { MagicStringAST } from 'magic-string-ast'
 import type { Awaitable } from '@antfu/utils'
 import type { Node } from '@babel/types'
 
@@ -33,4 +34,5 @@ export interface Transformer<T extends Node = Node> {
       id: string
     },
   ) => Awaitable<string | Node | false | undefined | null>
+  finalize?: (s: MagicStringAST) => Awaitable<void>
 }