refactor(BannerPlugin): update BannerPlugin to es2015

hiroppy · hiroppy · commit f73e87dfc5ea · 2016-12-29T15:30:22.000+09:00
diff --git a/lib/BannerPlugin.js b/lib/BannerPlugin.js
@@ -1,40 +1,51 @@
 /*
-	MIT License http://www.opensource.org/licenses/mit-license.php
-	Author Tobias Koppers @sokra
-*/
-var ConcatSource = require("webpack-sources").ConcatSource;
-var ModuleFilenameHelpers = require("./ModuleFilenameHelpers");
+ MIT License http://www.opensource.org/licenses/mit-license.php
+ Author Tobias Koppers @sokra
+ */
+
+"use strict";
+
+const ConcatSource = require("webpack-sources").ConcatSource;
+const ModuleFilenameHelpers = require("./ModuleFilenameHelpers");
 
 function wrapComment(str) {
-	if(str.indexOf("\n") < 0) return "/*! " + str + " */";
-	return "/*!\n * " + str.split("\n").join("\n * ") + "\n */";
+	if(!str.includes("\n")) return `/*! ${str} */`;
+	return `/*!\n * ${str.split("\n").join("\n * ")}\n */`;
 }
 
-function BannerPlugin(options) {
-	if(arguments.length > 1)
-		throw new Error("BannerPlugin only takes one argument (pass an options object)");
-	if(typeof options === "string")
-		options = {
-			banner: options
-		};
-	this.options = options || {};
-	this.banner = this.options.raw ? options.banner : wrapComment(options.banner);
-}
-module.exports = BannerPlugin;
+class BannerPlugin {
+	constructor(options) {
+		if(arguments.length > 1)
+			throw new Error("BannerPlugin only takes one argument (pass an options object)");
+		if(typeof options === "string")
+			options = {
+				banner: options
+			};
+		this.options = options || {};
+		this.banner = this.options.raw ? options.banner : wrapComment(options.banner);
+	}
 
-BannerPlugin.prototype.apply = function(compiler) {
-	var options = this.options;
-	var banner = this.banner;
+	apply(compiler) {
+		let options = this.options;
+		let banner = this.banner;
 
-	compiler.plugin("compilation", function(compilation) {
-		compilation.plugin("optimize-chunk-assets", function(chunks, callback) {
-			chunks.forEach(function(chunk) {
-				if(options.entryOnly && !chunk.isInitial()) return;
-				chunk.files.filter(ModuleFilenameHelpers.matchObject.bind(undefined, options)).forEach(function(file) {
-					compilation.assets[file] = new ConcatSource(banner, "\n", compilation.assets[file]);
+		compiler.plugin("compilation", (compilation) => {
+			compilation.plugin("optimize-chunk-assets", (chunks, callback) => {
+				chunks.forEach((chunk) => {
+					if(options.entryOnly && !chunk.isInitial()) return;
+
+					chunk.files
+						.filter(ModuleFilenameHelpers.matchObject.bind(undefined, options))
+						.forEach((file) =>
+							compilation.assets[file] = new ConcatSource(
+								banner, "\n", compilation.assets[file]
+							)
+						);
 				});
+				callback();
 			});
-			callback();
 		});
-	});
-};
+	}
+}
+
+module.exports = BannerPlugin;
