feat: support buildEnd hook, close #27 (#34)

Co-authored-by: Anthony Fu <[email protected]>
unjs · Oct 25, 2021 · c95f42d · c95f42d
1 parent 9d0344c
commit c95f42d
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -20,6 +20,7 @@ Currently supports:
 | Hook | Rollup | Vite | Webpack 4 | Webpack 5 |
 | ---- | :----: | :--: | :-------: | :-------: |
 | [`buildStart`](https://rollupjs.org/guide/en/#buildstart) | ✅ | ✅ | ✅ | ✅ |
+| [`buildEnd`](https://rollupjs.org/guide/en/#buildend) | ✅ | ✅ | ✅ | ✅ |
 | `transformInclude`* | ✅ | ✅ | ✅ | ✅ |
 | [`transform`](https://rollupjs.org/guide/en/#transformers) | ✅ | ✅ | ✅ | ✅ |
 | [`enforce`](https://rollupjs.org/guide/en/#enforce) | ❌\*\* | ✅ | ✅ | ✅ |

diff --git a/src/types.ts b/src/types.ts
@@ -19,6 +19,7 @@ export interface UnpluginOptions {
   name: string;
   enforce?: 'post' | 'pre' | undefined;
   buildStart?: () => Promise<void> | void;
+  buildEnd?: () => Promise<void> | void;
   transformInclude?: (id: string) => boolean;
   transform?: (this: UnpluginContext, code: string, id: string) => Thenable<TransformResult>;
   load?: (this: UnpluginContext, id: string) => Thenable<TransformResult>

diff --git a/src/webpack/index.ts b/src/webpack/index.ts
@@ -158,6 +158,12 @@ export function getWebpackPlugin<UserOptions = {}> (
         }
 
         plugin.buildStart?.()
+
+        if (plugin.buildEnd) {
+          compiler.hooks.done.tapAsync(plugin.name, () => {
+            plugin.buildEnd!()
+          })
+        }
       }
     }
   }