mirror of https://github.com/webpack/webpack.git
				
				
				
			
		
			
				
	
	
		
			237 lines
		
	
	
		
			9.0 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
			
		
		
	
	
			237 lines
		
	
	
		
			9.0 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
/*
 | 
						|
	MIT License http://www.opensource.org/licenses/mit-license.php
 | 
						|
	Author Tobias Koppers @sokra
 | 
						|
*/
 | 
						|
"use strict";
 | 
						|
 | 
						|
const SourceMapConsumer = require("source-map").SourceMapConsumer;
 | 
						|
const SourceMapSource = require("webpack-sources").SourceMapSource;
 | 
						|
const RawSource = require("webpack-sources").RawSource;
 | 
						|
const ConcatSource = require("webpack-sources").ConcatSource;
 | 
						|
const RequestShortener = require("../RequestShortener");
 | 
						|
const ModuleFilenameHelpers = require("../ModuleFilenameHelpers");
 | 
						|
const uglify = require("uglify-js");
 | 
						|
 | 
						|
class UglifyJsPlugin {
 | 
						|
	constructor(options) {
 | 
						|
		if(typeof options !== "object" || Array.isArray(options)) options = {};
 | 
						|
		if(typeof options.compressor !== "undefined") options.compress = options.compressor;
 | 
						|
		this.options = options;
 | 
						|
	}
 | 
						|
 | 
						|
	apply(compiler) {
 | 
						|
		const options = this.options;
 | 
						|
		options.test = options.test || /\.js($|\?)/i;
 | 
						|
		const warningsFilter = options.warningsFilter || (() => true);
 | 
						|
 | 
						|
		const requestShortener = new RequestShortener(compiler.context);
 | 
						|
		compiler.plugin("compilation", (compilation) => {
 | 
						|
			if(options.sourceMap) {
 | 
						|
				compilation.plugin("build-module", (module) => {
 | 
						|
					// to get detailed location info about errors
 | 
						|
					module.useSourceMap = true;
 | 
						|
				});
 | 
						|
			}
 | 
						|
			compilation.plugin("optimize-chunk-assets", (chunks, callback) => {
 | 
						|
				const files = [];
 | 
						|
				chunks.forEach((chunk) => files.push.apply(files, chunk.files));
 | 
						|
				files.push.apply(files, compilation.additionalChunkAssets);
 | 
						|
				const filterdFiles = files.filter(ModuleFilenameHelpers.matchObject.bind(undefined, options));
 | 
						|
				filterdFiles.forEach((file) => {
 | 
						|
					const oldWarnFunction = uglify.AST_Node.warn_function;
 | 
						|
					const warnings = [];
 | 
						|
					let sourceMap;
 | 
						|
					try {
 | 
						|
						const asset = compilation.assets[file];
 | 
						|
						if(asset.__UglifyJsPlugin) {
 | 
						|
							compilation.assets[file] = asset.__UglifyJsPlugin;
 | 
						|
							return;
 | 
						|
						}
 | 
						|
						let input;
 | 
						|
						let inputSourceMap;
 | 
						|
						if(options.sourceMap) {
 | 
						|
							if(asset.sourceAndMap) {
 | 
						|
								const sourceAndMap = asset.sourceAndMap();
 | 
						|
								inputSourceMap = sourceAndMap.map;
 | 
						|
								input = sourceAndMap.source;
 | 
						|
							} else {
 | 
						|
								inputSourceMap = asset.map();
 | 
						|
								input = asset.source();
 | 
						|
							}
 | 
						|
							sourceMap = new SourceMapConsumer(inputSourceMap);
 | 
						|
							uglify.AST_Node.warn_function = (warning) => { // eslint-disable-line camelcase
 | 
						|
								const match = /\[.+:([0-9]+),([0-9]+)\]/.exec(warning);
 | 
						|
								const line = +match[1];
 | 
						|
								const column = +match[2];
 | 
						|
								const original = sourceMap.originalPositionFor({
 | 
						|
									line: line,
 | 
						|
									column: column
 | 
						|
								});
 | 
						|
								if(!original || !original.source || original.source === file) return;
 | 
						|
								if(!warningsFilter(original.source)) return;
 | 
						|
								warnings.push(warning.replace(/\[.+:([0-9]+),([0-9]+)\]/, "") +
 | 
						|
									"[" + requestShortener.shorten(original.source) + ":" + original.line + "," + original.column + "]");
 | 
						|
							};
 | 
						|
						} else {
 | 
						|
							input = asset.source();
 | 
						|
							uglify.AST_Node.warn_function = (warning) => { // eslint-disable-line camelcase
 | 
						|
								warnings.push(warning);
 | 
						|
							};
 | 
						|
						}
 | 
						|
						uglify.base54.reset();
 | 
						|
						let ast = uglify.parse(input, {
 | 
						|
							filename: file
 | 
						|
						});
 | 
						|
						if(options.compress !== false) {
 | 
						|
							ast.figure_out_scope();
 | 
						|
							const compress = uglify.Compressor(options.compress || {
 | 
						|
								warnings: false
 | 
						|
							}); // eslint-disable-line new-cap
 | 
						|
							ast = ast.transform(compress);
 | 
						|
						}
 | 
						|
						if(options.mangle !== false) {
 | 
						|
							ast.figure_out_scope(options.mangle || {});
 | 
						|
							ast.compute_char_frequency(options.mangle || {});
 | 
						|
							ast.mangle_names(options.mangle || {});
 | 
						|
							if(options.mangle && options.mangle.props) {
 | 
						|
								uglify.mangle_properties(ast, options.mangle.props);
 | 
						|
							}
 | 
						|
						}
 | 
						|
						const output = {};
 | 
						|
						output.comments = Object.prototype.hasOwnProperty.call(options, "comments") ? options.comments : /^\**!|@preserve|@license/;
 | 
						|
						output.beautify = options.beautify;
 | 
						|
						for(let k in options.output) {
 | 
						|
							output[k] = options.output[k];
 | 
						|
						}
 | 
						|
						const extractedComments = [];
 | 
						|
						if(options.extractComments) {
 | 
						|
							const condition = {};
 | 
						|
							if(typeof options.extractComments === "string" || options.extractComments instanceof RegExp) {
 | 
						|
								// extractComments specifies the extract condition and output.comments specifies the preserve condition
 | 
						|
								condition.preserve = output.comments;
 | 
						|
								condition.extract = options.extractComments;
 | 
						|
							} else if(Object.prototype.hasOwnProperty.call(options.extractComments, "condition")) {
 | 
						|
								// Extract condition is given in extractComments.condition
 | 
						|
								condition.preserve = output.comments;
 | 
						|
								condition.extract = options.extractComments.condition;
 | 
						|
							} else {
 | 
						|
								// No extract condition is given. Extract comments that match output.comments instead of preserving them
 | 
						|
								condition.preserve = false;
 | 
						|
								condition.extract = output.comments;
 | 
						|
							}
 | 
						|
 | 
						|
							// Ensure that both conditions are functions
 | 
						|
							["preserve", "extract"].forEach(key => {
 | 
						|
								switch(typeof condition[key]) {
 | 
						|
									case "boolean":
 | 
						|
										var b = condition[key];
 | 
						|
										condition[key] = () => b;
 | 
						|
										break;
 | 
						|
									case "function":
 | 
						|
										break;
 | 
						|
									case "string":
 | 
						|
										if(condition[key] === "all") {
 | 
						|
											condition[key] = () => true;
 | 
						|
											break;
 | 
						|
										}
 | 
						|
										var regex = new RegExp(condition[key]);
 | 
						|
										condition[key] = (astNode, comment) => regex.test(comment.value);
 | 
						|
										break;
 | 
						|
									default:
 | 
						|
										regex = condition[key];
 | 
						|
										condition[key] = (astNode, comment) => regex.test(comment.value);
 | 
						|
								}
 | 
						|
							});
 | 
						|
 | 
						|
							// Redefine the comments function to extract and preserve
 | 
						|
							// comments according to the two conditions
 | 
						|
							output.comments = (astNode, comment) => {
 | 
						|
								if(condition.extract(astNode, comment)) {
 | 
						|
									extractedComments.push(
 | 
						|
										comment.type === "comment2" ? "/*" + comment.value + "*/" : "//" + comment.value
 | 
						|
									);
 | 
						|
								}
 | 
						|
								return condition.preserve(astNode, comment);
 | 
						|
							};
 | 
						|
						}
 | 
						|
						let map;
 | 
						|
						if(options.sourceMap) {
 | 
						|
							map = uglify.SourceMap({ // eslint-disable-line new-cap
 | 
						|
								file: file,
 | 
						|
								root: ""
 | 
						|
							});
 | 
						|
							output.source_map = map; // eslint-disable-line camelcase
 | 
						|
						}
 | 
						|
						const stream = uglify.OutputStream(output); // eslint-disable-line new-cap
 | 
						|
						ast.print(stream);
 | 
						|
						if(map) map = map + "";
 | 
						|
						const stringifiedStream = stream + "";
 | 
						|
						let outputSource = (map ?
 | 
						|
							new SourceMapSource(stringifiedStream, file, JSON.parse(map), input, inputSourceMap) :
 | 
						|
							new RawSource(stringifiedStream));
 | 
						|
						if(extractedComments.length > 0) {
 | 
						|
							let commentsFile = options.extractComments.filename || file + ".LICENSE";
 | 
						|
							if(typeof commentsFile === "function") {
 | 
						|
								commentsFile = commentsFile(file);
 | 
						|
							}
 | 
						|
 | 
						|
							// Write extracted comments to commentsFile
 | 
						|
							const commentsSource = new RawSource(extractedComments.join("\n\n") + "\n");
 | 
						|
							if(commentsFile in compilation.assets) {
 | 
						|
								// commentsFile already exists, append new comments...
 | 
						|
								if(compilation.assets[commentsFile] instanceof ConcatSource) {
 | 
						|
									compilation.assets[commentsFile].add("\n");
 | 
						|
									compilation.assets[commentsFile].add(commentsSource);
 | 
						|
								} else {
 | 
						|
									compilation.assets[commentsFile] = new ConcatSource(
 | 
						|
										compilation.assets[commentsFile], "\n", commentsSource
 | 
						|
									);
 | 
						|
								}
 | 
						|
							} else {
 | 
						|
								compilation.assets[commentsFile] = commentsSource;
 | 
						|
							}
 | 
						|
 | 
						|
							// Add a banner to the original file
 | 
						|
							if(options.extractComments.banner !== false) {
 | 
						|
								let banner = options.extractComments.banner || "For license information please see " + commentsFile;
 | 
						|
								if(typeof banner === "function") {
 | 
						|
									banner = banner(commentsFile);
 | 
						|
								}
 | 
						|
								if(banner) {
 | 
						|
									outputSource = new ConcatSource(
 | 
						|
										"/*! " + banner + " */\n", outputSource
 | 
						|
									);
 | 
						|
								}
 | 
						|
							}
 | 
						|
						}
 | 
						|
						asset.__UglifyJsPlugin = compilation.assets[file] = outputSource;
 | 
						|
						if(warnings.length > 0) {
 | 
						|
							compilation.warnings.push(new Error(file + " from UglifyJs\n" + warnings.join("\n")));
 | 
						|
						}
 | 
						|
					} catch(err) {
 | 
						|
						if(err.line) {
 | 
						|
							const original = sourceMap && sourceMap.originalPositionFor({
 | 
						|
								line: err.line,
 | 
						|
								column: err.col
 | 
						|
							});
 | 
						|
							if(original && original.source) {
 | 
						|
								compilation.errors.push(new Error(file + " from UglifyJs\n" + err.message + " [" + requestShortener.shorten(original.source) + ":" + original.line + "," + original.column + "][" + file + ":" + err.line + "," + err.col + "]"));
 | 
						|
							} else {
 | 
						|
								compilation.errors.push(new Error(file + " from UglifyJs\n" + err.message + " [" + file + ":" + err.line + "," + err.col + "]"));
 | 
						|
							}
 | 
						|
						} else if(err.msg) {
 | 
						|
							compilation.errors.push(new Error(file + " from UglifyJs\n" + err.msg));
 | 
						|
						} else
 | 
						|
							compilation.errors.push(new Error(file + " from UglifyJs\n" + err.stack));
 | 
						|
					} finally {
 | 
						|
						uglify.AST_Node.warn_function = oldWarnFunction; // eslint-disable-line camelcase
 | 
						|
					}
 | 
						|
				});
 | 
						|
				callback();
 | 
						|
			});
 | 
						|
		});
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
module.exports = UglifyJsPlugin;
 |