🐛 Fix chunk optimization listeners (#35)

* 🎨 Cleanup code + fix deprecation warning

* 🎨 Remove more legacy code + fix stage

* 🎨 Remove previous hook

* 🎨 Fix requested changes

* 🐛 Fix chunk optimization

Author: ryanatlascreative

src/webpack/CompilerEventsFacade.ts

@@ -1,4 +1,4 @@
-import { Compilation, Compiler } from "webpack";
+import { Chunk, Compilation, Compiler } from "webpack";
 
 export default class CompilerEventsFacade {
   public static extensionName = "webpack-extension-reloader";
@@ -15,16 +15,21 @@ export default class CompilerEventsFacade {
     return this._compiler.hooks.compilation.tap(
       CompilerEventsFacade.extensionName,
       (comp) => {
-        const afterOptimizeChunkAssets = (chunks) => {
-          call(comp, chunks);
+        const chunks = new Set();
+        const afterOptimizeChunkAssets = (chunksAfterOptimization) => {
+          call(comp, chunksAfterOptimization);
         };
-        /* https://github.com/webpack/webpack/blob/main/lib/Compilation.js#L772-L779
-        afterOptimizeChunkAssets = PROCESS_ASSETS_STAGE_OPTIMIZE + 1
-        */
-        const stage = Compilation.PROCESS_ASSETS_STAGE_OPTIMIZE + 1;
-        (comp.hooks as any).processAssets.tap(
-          { name: CompilerEventsFacade.extensionName, stage },
-          afterOptimizeChunkAssets,
+        comp.hooks.processAssets.tap(
+          {
+            name: CompilerEventsFacade.extensionName,
+            stage: Compilation.PROCESS_ASSETS_STAGE_ANALYSE,
+          },
+          () => {
+            comp.chunks.forEach((chunk) => {
+              chunks.add(chunk);
+            });
+            afterOptimizeChunkAssets(chunks);
+          },
         );
       },
     );