Merge pull request #205 from qkdreyer/patch-1

Fixed #204
chrisvfritz · Jun 1, 2018 · cbdc72a · cbdc72a
2 parents 3bc77eb + 19698ab
commit cbdc72a
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/es6/index.js b/es6/index.js
@@ -54,7 +54,7 @@ function PrerenderSPAPlugin (...args) {
 }
 
 PrerenderSPAPlugin.prototype.apply = function (compiler) {
-  compiler.plugin('after-emit', (compilation, done) => {
+  const afterEmit = (compilation, done) => {
     const PrerendererInstance = new Prerenderer(this._options)
 
     PrerendererInstance.initialize()
@@ -139,7 +139,14 @@ PrerenderSPAPlugin.prototype.apply = function (compiler) {
         console.error('[prerender-spa-plugin] Unable to prerender all routes!')
         throw err
       })
-  })
+  }
+
+  if (compiler.hooks) {
+    const plugin = { name: 'PrerenderSPAPlugin' }
+    compiler.hooks.afterEmit.tapAsync(plugin, afterEmit)
+  } else {
+    compiler.plugin('after-emit', afterEmit)
+  }
 }
 
 PrerenderSPAPlugin.PuppeteerRenderer = PuppeteerRenderer