Make removeAssets more flexible (#40)

Author: jescalan

lib/index.js

@@ -121,9 +121,14 @@ module.exports = class SpikeUtils {
     })
 
     // first we remove the files from webpack's `assets`
-    const filesDotJs = files.map((f) => `${f}.js`)
-    for (const a in compilation.assets) {
-      if (filesDotJs.indexOf(a) > -1) { delete compilation.assets[a] }
+    const chunksByName = compilation.records.chunks.byName
+    const chunkNames = Object.keys(compilation.assets)
+    for (let k in chunksByName) {
+      // if the chunk name is in our spike-processed files, we remove
+      // that chunk from the assets to be written
+      if (files.indexOf(k) > -1) {
+        delete compilation.assets[chunkNames[chunksByName[k]]]
+      }
     }
 
     // Now we remove any chunks that are not further processed.