chore(index): clean up code

webpack-contrib · Mar 8, 2018 · 527db3e · 527db3e
1 parent 57271de
commit 527db3e
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/src/index.js b/src/index.js
@@ -58,8 +58,8 @@ function optimizeChunkAssets(compilation, options, chunks) {
     });
 }
 
-function compilationFn(compilation, plugin) {
-  const { options } = plugin;
+function compilationFn(compilation) {
+  const { options, plugin } = this;
 
   if (compilation.hooks) {
     compilation
@@ -112,16 +112,17 @@ export default class BabelMinifyPlugin {
 
   apply(compiler) {
     const { options } = this;
-
     // if sourcemap is not set
     options.sourceMap = getDefault(options.sourceMap, !!compiler.options.devtool);
 
+    const boundCompilationFn = compilationFn.bind(this);
+
     if (compiler.hooks) {
       const { compilation } = compiler.hooks;
 
-      compilation.tap(this.plugin, compilation => compilationFn(compilation, this));
+      compilation.tap(this.plugin, boundCompilationFn);
     } else {
-      compiler.plugin('compilation', compilation => compilationFn(compilation, this));
+      compiler.plugin('compilation', boundCompilationFn);
     }
   }
 }