From 78d206dd014ce8d0389b8ecbf36fb0bdbe71d7f1 Mon Sep 17 00:00:00 2001
From: Vijay Jain <mnnit.vijay@gmail.com>
Date: Fri, 28 Aug 2020 19:30:22 +0530
Subject: [PATCH] Update index.js

---
 src/index.js | 15 +++++++++++----
 1 file changed, 11 insertions(+), 4 deletions(-)

diff --git a/src/index.js b/src/index.js
index dc5b66e..2f0a297 100644
--- a/src/index.js
+++ b/src/index.js
@@ -107,12 +107,19 @@ class CompressionPlugin {
     }
   }
 
-  static emitAsset(compilation, name, source, assetInfo) {
+  static emitAsset(compilation, name, source, assetInfo, oldName) {
     // New API
     if (compilation.emitAsset) {
       compilation.emitAsset(name, source, assetInfo);
     }
-
+    
+    // Update chunks to include new asset name as well
+    compilation.chunks.forEach((chunk) => {
+      if(chunk.files.includes(oldName)) {
+        chunk.files.push(name);
+      }
+    });
+ 
     // eslint-disable-next-line no-param-reassign
     compilation.assets[name] = source;
   }
@@ -122,7 +129,7 @@ class CompressionPlugin {
     if (compilation.updateAsset) {
       compilation.updateAsset(name, newSource, assetInfo);
     }
-
+    
     // eslint-disable-next-line no-param-reassign
     compilation.assets[name] = newSource;
   }
@@ -224,7 +231,7 @@ class CompressionPlugin {
 
     CompressionPlugin.emitAsset(compilation, newAssetName, output, {
       compressed: true,
-    });
+    }, assetName);
 
     if (this.options.deleteOriginalAssets) {
       // eslint-disable-next-line no-param-reassign