refactor(es6): Upgrade AutomaticPrefetchPlugin to es6 (webpack#3682)

chicoxyzzy · TheLarkInn · commit e52702050a4b · 2017-01-03T14:19:06.000-06:00
diff --git a/lib/AutomaticPrefetchPlugin.js b/lib/AutomaticPrefetchPlugin.js
@@ -2,34 +2,35 @@
 	MIT License http://www.opensource.org/licenses/mit-license.php
 	Author Tobias Koppers @sokra
 */
-var async = require("async");
-var PrefetchDependency = require("./dependencies/PrefetchDependency");
-var NormalModule = require("./NormalModule");
+"use strict";
 
-function AutomaticPrefetchPlugin() {}
-module.exports = AutomaticPrefetchPlugin;
-AutomaticPrefetchPlugin.prototype.apply = function(compiler) {
-	compiler.plugin("compilation", function(compilation, params) {
-		var normalModuleFactory = params.normalModuleFactory;
+const async = require("async");
+const PrefetchDependency = require("./dependencies/PrefetchDependency");
+const NormalModule = require("./NormalModule");
+
+class AutomaticPrefetchPlugin {
+	apply(compiler) {
+		compiler.plugin("compilation", (compilation, params) => {
+			const normalModuleFactory = params.normalModuleFactory;
 
-		compilation.dependencyFactories.set(PrefetchDependency, normalModuleFactory);
-	});
-	var lastModules = null;
-	compiler.plugin("after-compile", function(compilation, callback) {
-		lastModules = compilation.modules.filter(function(m) {
-			return m instanceof NormalModule;
-		}).map(function(m) {
-			return {
-				context: m.context,
-				request: m.request
-			};
+			compilation.dependencyFactories.set(PrefetchDependency, normalModuleFactory);
+		});
+		let lastModules = null;
+		compiler.plugin("after-compile", (compilation, callback) => {
+			lastModules = compilation.modules
+				.filter(m => m instanceof NormalModule)
+				.map(m => ({
+					context: m.context,
+					request: m.request
+				}));
+			callback();
 		});
-		callback();
-	});
-	compiler.plugin("make", function(compilation, callback) {
-		if(!lastModules) return callback();
-		async.forEach(lastModules, function(m, callback) {
-			compilation.prefetch(m.context || compiler.context, new PrefetchDependency(m.request), callback);
-		}, callback);
-	});
-};
+		compiler.plugin("make", (compilation, callback) => {
+			if(!lastModules) return callback();
+			async.forEach(lastModules, (m, callback) => {
+				compilation.prefetch(m.context || compiler.context, new PrefetchDependency(m.request), callback);
+			}, callback);
+		});
+	}
+}
+module.exports = AutomaticPrefetchPlugin;
