dynamic-entry-plugin: improve

nkzawa · nkzawa · commit 301cd2623507 · 2016-10-24T11:38:55.000+09:00
diff --git a/server/build/plugins/dynamic-entry-plugin.js b/server/build/plugins/dynamic-entry-plugin.js
@@ -7,7 +7,10 @@ export default class DynamicEntryPlugin {
     compiler.addEntry = addEntry
     compiler.removeEntry = removeEntry
     compiler.hasEntry = hasEntry
-    compiler.createCompilation = createCompilation(compiler.createCompilation)
+
+    compiler.plugin('compilation', (compilation) => {
+      compilation.addEntry = compilationAddEntry(compilation.addEntry)
+    })
   }
 }
 
@@ -41,14 +44,6 @@ function hasEntry (name = 'main') {
   return this.entryNames.has(name)
 }
 
-function createCompilation (original) {
-  return function (...args) {
-    const compilation = original.apply(this, args)
-    compilation.addEntry = compilationAddEntry(compilation.addEntry)
-    return compilation
-  }
-}
-
 function compilationAddEntry (original) {
   return function (context, entry, name, callback) {
     if (!this.compiler.entryNames.has(name)) {
