refactor(ES6): upgrade SetVarMainTemplatePlugin to ES6 (webpack#3726)

willmendesneto · TheLarkInn · commit 6220b4e0a4ef · 2017-01-04T12:00:09.000-06:00
diff --git a/lib/SetVarMainTemplatePlugin.js b/lib/SetVarMainTemplatePlugin.js
@@ -2,35 +2,40 @@
 	MIT License http://www.opensource.org/licenses/mit-license.php
 	Author Tobias Koppers @sokra
 */
-var ConcatSource = require("webpack-sources").ConcatSource;
+"use strict";
 
-function SetVarMainTemplatePlugin(varExpression, copyObject) {
-	this.varExpression = varExpression;
-	this.copyObject = copyObject;
+const ConcatSource = require("webpack-sources").ConcatSource;
+
+class SetVarMainTemplatePlugin {
+	constructor(varExpression, copyObject) {
+		this.varExpression = varExpression;
+		this.copyObject = copyObject;
+	}
+
+	apply(compilation) {
+		const mainTemplate = compilation.mainTemplate;
+		compilation.templatesPlugin("render-with-entry", (source, chunk, hash) => {
+			const varExpression = mainTemplate.applyPluginsWaterfall("asset-path", this.varExpression, {
+				hash,
+				chunk
+			});
+			if(this.copyObject) {
+				return new ConcatSource(`(function(e, a) { for(var i in a) e[i] = a[i]; }(${varExpression}, `, source, "))");
+			} else {
+				const prefix = `${varExpression} =\n`;
+				return new ConcatSource(prefix, source);
+			}
+		});
+		mainTemplate.plugin("global-hash-paths", function(paths) {
+			if(this.varExpression) paths.push(this.varExpression);
+			return paths;
+		});
+		mainTemplate.plugin("hash", hash => {
+			hash.update("set var");
+			hash.update(`${this.varExpression}`);
+			hash.update(`${this.copyObject}`);
+		});
+	}
 }
+
 module.exports = SetVarMainTemplatePlugin;
-SetVarMainTemplatePlugin.prototype.apply = function(compilation) {
-	var mainTemplate = compilation.mainTemplate;
-	compilation.templatesPlugin("render-with-entry", function(source, chunk, hash) {
-		var varExpression = mainTemplate.applyPluginsWaterfall("asset-path", this.varExpression, {
-			hash: hash,
-			chunk: chunk
-		});
-		if(this.copyObject) {
-			return new ConcatSource("(function(e, a) { for(var i in a) e[i] = a[i]; }(" +
-				varExpression + ", ", source, "))");
-		} else {
-			var prefix = varExpression + " =\n";
-			return new ConcatSource(prefix, source);
-		}
-	}.bind(this));
-	mainTemplate.plugin("global-hash-paths", function(paths) {
-		if(this.varExpression) paths.push(this.varExpression);
-		return paths;
-	});
-	mainTemplate.plugin("hash", function(hash) {
-		hash.update("set var");
-		hash.update(this.varExpression + "");
-		hash.update(this.copyObject + "");
-	}.bind(this));
-};
