mirror of https://github.com/webpack/webpack.git
				
				
				
			
		
			
				
	
	
		
			784 lines
		
	
	
		
			28 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
			
		
		
	
	
			784 lines
		
	
	
		
			28 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
| /*
 | |
| 	MIT License http://www.opensource.org/licenses/mit-license.php
 | |
| 	Author Tobias Koppers @sokra
 | |
| */
 | |
| var esprima = require("esprima");
 | |
| var Tapable = require("tapable");
 | |
| var BasicEvaluatedExpression = require("./BasicEvaluatedExpression");
 | |
| 
 | |
| function Parser(options) {
 | |
| 	Tapable.call(this);
 | |
| 	this.options = options;
 | |
| 	this.initializeEvaluating();
 | |
| }
 | |
| module.exports = Parser;
 | |
| 
 | |
| // Syntax: https://developer.mozilla.org/en/SpiderMonkey/Parser_API
 | |
| 
 | |
| 
 | |
| 
 | |
| Parser.prototype = Object.create(Tapable.prototype);
 | |
| Parser.prototype.initializeEvaluating = function() {
 | |
| 	function joinRanges(startRange, endRange) {
 | |
| 		if(!endRange) return startRange;
 | |
| 		if(!startRange) return endRange;
 | |
| 		return [startRange[0], endRange[1]];
 | |
| 	}
 | |
| 	this.plugin("evaluate Literal", function(expr) {
 | |
| 		switch(typeof expr.value) {
 | |
| 		case "number":
 | |
| 			return new BasicEvaluatedExpression().setNumber(expr.value).setRange(expr.range);
 | |
| 		case "string":
 | |
| 			return new BasicEvaluatedExpression().setString(expr.value).setRange(expr.range);
 | |
| 		case "boolean":
 | |
| 			return new BasicEvaluatedExpression().setBoolean(expr.value).setRange(expr.range);
 | |
| 		}
 | |
| 		if(expr.value instanceof RegExp)
 | |
| 			return new BasicEvaluatedExpression().setRegExp(expr.value).setRange(expr.range);
 | |
| 	});
 | |
| 	this.plugin("evaluate LogicalExpression", function(expr) {
 | |
| 		if(expr.operator == "&&") {
 | |
| 			var left = this.evaluateExpression(expr.left);
 | |
| 			var leftAsBool = left && left.asBool();
 | |
| 			if(leftAsBool === false) return new BasicEvaluatedExpression().setBoolean(false).setRange(expr.range);
 | |
| 			if(leftAsBool !== true) return;
 | |
| 			var right = this.evaluateExpression(expr.right);
 | |
| 			var rightAsBool = right && right.asBool();
 | |
| 			if(typeof rightAsBool === "boolean")
 | |
| 				return new BasicEvaluatedExpression().setBoolean(rightAsBool).setRange(expr.range);
 | |
| 		} else if(expr.operator == "||") {
 | |
| 			var left = this.evaluateExpression(expr.left);
 | |
| 			var leftAsBool = left && left.asBool();
 | |
| 			if(leftAsBool === true) return new BasicEvaluatedExpression().setBoolean(true).setRange(expr.range);
 | |
| 			if(leftAsBool !== false) return;
 | |
| 			var right = this.evaluateExpression(expr.right);
 | |
| 			var rightAsBool = right && right.asBool();
 | |
| 			if(typeof rightAsBool === "boolean")
 | |
| 				return new BasicEvaluatedExpression().setBoolean(rightAsBool).setRange(expr.range);
 | |
| 		}
 | |
| 	});
 | |
| 	this.plugin("evaluate BinaryExpression", function(expr) {
 | |
| 		if(expr.operator == "+") {
 | |
| 			var left = this.evaluateExpression(expr.left);
 | |
| 			var right = this.evaluateExpression(expr.right);
 | |
| 			if(!left || !right) return;
 | |
| 			var res = new BasicEvaluatedExpression()
 | |
| 			if(left.isString()) {
 | |
| 				if(right.isString()) {
 | |
| 					res.setString(left.string + right.string);
 | |
| 				} else if(right.isNumber()) {
 | |
| 					res.setString(left.string + right.number);
 | |
| 				} else if(right.isWrapped() && right.prefix.isString()) {
 | |
| 					res.setWrapped(
 | |
| 						new BasicEvaluatedExpression()
 | |
| 							.setString(left.string + right.prefix.string)
 | |
| 							.setRange(joinRanges(left.range, right.prefix.range)),
 | |
| 						right.postfix);
 | |
| 				} else {
 | |
| 					res.setWrapped(left, new BasicEvaluatedExpression().setString(""))
 | |
| 				}
 | |
| 			} else if(left.isNumber()) {
 | |
| 				if(right.isString()) {
 | |
| 					res.setString(left.number + right.string);
 | |
| 				} else if(right.isNumber()) {
 | |
| 					res.setNumber(left.number + right.number);
 | |
| 				}
 | |
| 			} else if(left.isWrapped() && left.postfix.isString()) {
 | |
| 				if(right.isString()) {
 | |
| 					res.setWrapped(left.prefix,
 | |
| 						new BasicEvaluatedExpression()
 | |
| 							.setString(left.postfix.string + right.string)
 | |
| 							.setRange(joinRanges(left.postfix.range, right.range))
 | |
| 					);
 | |
| 				} else if(right.isNumber()) {
 | |
| 					res.setWrapped(left.prefix,
 | |
| 						new BasicEvaluatedExpression()
 | |
| 							.setString(left.postfix.string + right.number)
 | |
| 							.setRange(joinRanges(left.postfix.range, right.range))
 | |
| 					);
 | |
| 				}
 | |
| 			} else {
 | |
| 				if(right.isString()) {
 | |
| 					res.setWrapped(new BasicEvaluatedExpression().setString(""), right);
 | |
| 				}
 | |
| 			}
 | |
| 			res.setRange(expr.range);
 | |
| 			return res;
 | |
| 		} else if(expr.operator == "-") {
 | |
| 			var left = this.evaluateExpression(expr.left);
 | |
| 			var right = this.evaluateExpression(expr.right);
 | |
| 			if(!left || !right) return;
 | |
| 			if(!left.isNumber() || !right.isNumber()) return;
 | |
| 			var res = new BasicEvaluatedExpression();
 | |
| 			res.setNumber(left.number - right.number);
 | |
| 			res.setRange(expr.range);
 | |
| 			return res;
 | |
| 		} else if(expr.operator == "*") {
 | |
| 			var left = this.evaluateExpression(expr.left);
 | |
| 			var right = this.evaluateExpression(expr.right);
 | |
| 			if(!left || !right) return;
 | |
| 			if(!left.isNumber() || !right.isNumber()) return;
 | |
| 			var res = new BasicEvaluatedExpression();
 | |
| 			res.setNumber(left.number * right.number);
 | |
| 			res.setRange(expr.range);
 | |
| 			return res;
 | |
| 		} else if(expr.operator == "/") {
 | |
| 			var left = this.evaluateExpression(expr.left);
 | |
| 			var right = this.evaluateExpression(expr.right);
 | |
| 			if(!left || !right) return;
 | |
| 			if(!left.isNumber() || !right.isNumber()) return;
 | |
| 			var res = new BasicEvaluatedExpression();
 | |
| 			res.setNumber(left.number / right.number);
 | |
| 			res.setRange(expr.range);
 | |
| 			return res;
 | |
| 		} else if(expr.operator == "==" || expr.operator == "===") {
 | |
| 			var left = this.evaluateExpression(expr.left);
 | |
| 			var right = this.evaluateExpression(expr.right);
 | |
| 			if(!left || !right) return;
 | |
| 			var res = new BasicEvaluatedExpression();
 | |
| 			res.setRange(expr.range);
 | |
| 			if(left.isString() && right.isString()) {
 | |
| 				return res.setBoolean(left.string === right.string);
 | |
| 			} else if(left.isNumber() && right.isNumber()) {
 | |
| 				return res.setBoolean(left.number === right.number);
 | |
| 			} else if(left.isBoolean() && right.isBoolean()) {
 | |
| 				return res.setBoolean(left.bool === right.bool);
 | |
| 			}
 | |
| 		} else if(expr.operator == "!=" || expr.operator == "!==") {
 | |
| 			var left = this.evaluateExpression(expr.left);
 | |
| 			var right = this.evaluateExpression(expr.right);
 | |
| 			if(!left || !right) return;
 | |
| 			var res = new BasicEvaluatedExpression();
 | |
| 			res.setRange(expr.range);
 | |
| 			if(left.isString() && right.isString()) {
 | |
| 				return res.setBoolean(left.string !== right.string);
 | |
| 			} else if(left.isNumber() && right.isNumber()) {
 | |
| 				return res.setBoolean(left.number !== right.number);
 | |
| 			} else if(left.isBoolean() && right.isBoolean()) {
 | |
| 				return res.setBoolean(left.bool !== right.bool);
 | |
| 			}
 | |
| 		}
 | |
| 	});
 | |
| 	this.plugin("evaluate UnaryExpression", function(expr) {
 | |
| 		if(expr.operator == "typeof") {
 | |
| 			if(expr.argument.type == "Identifier") {
 | |
| 				var res = this.applyPluginsBailResult("evaluate typeof " + expr.argument.name, expr);
 | |
| 				if(res !== undefined) return res;
 | |
| 			}
 | |
| 			if(expr.argument.type == "MemberExpression") {
 | |
| 				var expression = expr.argument;
 | |
| 				var exprName = [];
 | |
| 				while(expression.type == "MemberExpression" && expression.property.type == "Identifier") {
 | |
| 					exprName.unshift(expression.property.name);
 | |
| 					expression = expression.object;
 | |
| 				}
 | |
| 				if(expression.type == "Identifier" && this.scope.definitions.indexOf(expression.name) == -1) {
 | |
| 					exprName.unshift(this.scope.renames["$"+expression.name] || expression.name);
 | |
| 					exprName = exprName.join(".");
 | |
| 					var res = this.applyPluginsBailResult("evaluate typeof " + exprName, expr);
 | |
| 					if(res !== undefined) return res;
 | |
| 				}
 | |
| 			}
 | |
| 			if(expr.argument.type == "FunctionExpression") {
 | |
| 				return new BasicEvaluatedExpression().setString("function").setRange(expr.range);
 | |
| 			}
 | |
| 			var arg = this.evaluateExpression(expr.argument);
 | |
| 			if(arg.isString() || arg.isWrapped()) return new BasicEvaluatedExpression().setString("string").setRange(expr.range);
 | |
| 			else if(arg.isNumber()) return new BasicEvaluatedExpression().setString("number").setRange(expr.range);
 | |
| 			else if(arg.isBoolean()) return new BasicEvaluatedExpression().setString("boolean").setRange(expr.range);
 | |
| 			else if(arg.isArray() || arg.isConstArray() || arg.isRegExp()) return new BasicEvaluatedExpression().setString("object").setRange(expr.range);
 | |
| 		} else if(expr.operator == "!") {
 | |
| 			var argument = this.evaluateExpression(expr.argument);
 | |
| 			if(!argument) return;
 | |
| 			if(argument.isBoolean()) {
 | |
| 				return new BasicEvaluatedExpression().setBoolean(!argument.bool).setRange(expr.range);
 | |
| 			} else if(argument.isString()) {
 | |
| 				return new BasicEvaluatedExpression().setBoolean(!argument.string).setRange(expr.range);
 | |
| 			} else if(argument.isNumber()) {
 | |
| 				return new BasicEvaluatedExpression().setBoolean(!argument.number).setRange(expr.range);
 | |
| 			}
 | |
| 		}
 | |
| 	});
 | |
| 	this.plugin("evaluate Identifier", function(expr) {
 | |
| 		var name = this.scope.renames["$"+expr.name] || expr.name;
 | |
| 		if(this.scope.definitions.indexOf(expr.name) == -1) {
 | |
| 			var result = this.applyPluginsBailResult("evaluate Identifier " + name, expr);
 | |
| 			if(result) return result;
 | |
| 			return new BasicEvaluatedExpression().setIdentifier(name).setRange(expr.range);
 | |
| 		} else {
 | |
| 			return this.applyPluginsBailResult("evaluate defined Identifier " + name, expr);
 | |
| 		}
 | |
| 	});
 | |
| 	this.plugin("evaluate MemberExpression", function(expression) {
 | |
| 		var expr = expression;
 | |
| 		var exprName = [];
 | |
| 		while(expr.type == "MemberExpression" && expr.property.type == "Identifier") {
 | |
| 			exprName.unshift(expr.property.name);
 | |
| 			expr = expr.object;
 | |
| 		}
 | |
| 		if(expr.type == "Identifier") {
 | |
| 			exprName.unshift(this.scope.renames["$"+expr.name] || expr.name);
 | |
| 			exprName = exprName.join(".");
 | |
| 			if(this.scope.definitions.indexOf(expr.name) == -1) {
 | |
| 				var result = this.applyPluginsBailResult("evaluate Identifier " + exprName, expression);
 | |
| 				if(result) return result;
 | |
| 				return new BasicEvaluatedExpression().setIdentifier(exprName).setRange(expression.range);
 | |
| 			} else {
 | |
| 				return this.applyPluginsBailResult("evaluate defined Identifier " + exprName, expression);
 | |
| 			}
 | |
| 		}
 | |
| 	});
 | |
| 	this.plugin("evaluate CallExpression", function(expr) {
 | |
| 		if(expr.callee.type != "MemberExpression") return;
 | |
| 		if(expr.callee.property.type != "Identifier") return;
 | |
| 		var param = this.evaluateExpression(expr.callee.object);
 | |
| 		if(!param) return;
 | |
| 		return this.applyPluginsBailResult("evaluate CallExpression ." + expr.callee.property.name, expr, param);
 | |
| 	});
 | |
| 	this.plugin("evaluate CallExpression .replace", function(expr, param) {
 | |
| 		if(!param.isString()) return;
 | |
| 		if(expr.arguments.length !== 2) return;
 | |
| 		var arg1 = this.evaluateExpression(expr.arguments[0]);
 | |
| 		var arg2 = this.evaluateExpression(expr.arguments[1]);
 | |
| 		if(!arg1.isString() && !arg1.isRegExp()) return;
 | |
| 		arg1 = arg1.regExp || arg1.string;
 | |
| 		if(!arg2.isString()) return;
 | |
| 		arg2 = arg2.string;
 | |
| 		return new BasicEvaluatedExpression().setString(param.string.replace(arg1, arg2)).setRange(expr.range);
 | |
| 	});
 | |
| 	this.plugin("evaluate CallExpression .substr", function(expr, param) {
 | |
| 		if(!param.isString()) return;
 | |
| 		var result, str = param.string;
 | |
| 		switch(expr.arguments.length) {
 | |
| 		case 1:
 | |
| 			var arg1 = this.evaluateExpression(expr.arguments[0]);
 | |
| 			if(!arg1.isNumber()) return;
 | |
| 			result = str.substr(arg1.number);
 | |
| 			break;
 | |
| 		case 2:
 | |
| 			var arg1 = this.evaluateExpression(expr.arguments[0]);
 | |
| 			var arg2 = this.evaluateExpression(expr.arguments[0]);
 | |
| 			if(!arg1.isNumber()) return;
 | |
| 			if(!arg2.isNumber()) return;
 | |
| 			result = str.substr(arg1.number, arg2.number);
 | |
| 			break;
 | |
| 		default:
 | |
| 			return;
 | |
| 		}
 | |
| 		return new BasicEvaluatedExpression().setString(result).setRange(expr.range);
 | |
| 	});
 | |
| 	this.plugin("evaluate CallExpression .substring", function(expr, param) {
 | |
| 		if(!param.isString()) return;
 | |
| 		var result, str = param.string;
 | |
| 		switch(expr.arguments.length) {
 | |
| 		case 1:
 | |
| 			var arg1 = this.evaluateExpression(expr.arguments[0]);
 | |
| 			if(!arg1.isNumber()) return;
 | |
| 			result = str.substring(arg1.number);
 | |
| 			break;
 | |
| 		case 2:
 | |
| 			var arg1 = this.evaluateExpression(expr.arguments[0]);
 | |
| 			var arg2 = this.evaluateExpression(expr.arguments[1]);
 | |
| 			if(!arg1.isNumber()) return;
 | |
| 			if(!arg2.isNumber()) return;
 | |
| 			result = str.substring(arg1.number, arg2.number);
 | |
| 			break;
 | |
| 		default:
 | |
| 			return;
 | |
| 		}
 | |
| 		return new BasicEvaluatedExpression().setString(result).setRange(expr.range);
 | |
| 	});
 | |
| 	this.plugin("evaluate CallExpression .split", function(expr, param) {
 | |
| 		if(!param.isString()) return;
 | |
| 		if(expr.arguments.length !== 1) return;
 | |
| 		var result;
 | |
| 		var arg = this.evaluateExpression(expr.arguments[0]);
 | |
| 		if(arg.isString()) {
 | |
| 			result = param.string.split(arg.string);
 | |
| 		} else if(arg.isRegExp()) {
 | |
| 			result = param.string.split(arg.regExp);
 | |
| 		} else return;
 | |
| 		return new BasicEvaluatedExpression().setArray(result).setRange(expr.range);
 | |
| 	});
 | |
| 	this.plugin("evaluate ConditionalExpression", function(expr) {
 | |
| 		var condition = this.evaluateExpression(expr.test);
 | |
| 		var conditionValue = condition.asBool();
 | |
| 		if(conditionValue === undefined) {
 | |
| 			var consequent = this.evaluateExpression(expr.consequent);
 | |
| 			var alternate = this.evaluateExpression(expr.alternate);
 | |
| 			if(!consequent || !alternate) return;
 | |
| 			var res = new BasicEvaluatedExpression();
 | |
| 			if(consequent.isConditional())
 | |
| 				res.setOptions(consequent.options);
 | |
| 			else
 | |
| 				res.setOptions([consequent]);
 | |
| 			if(alternate.isConditional())
 | |
| 				res.addOptions(alternate.options);
 | |
| 			else
 | |
| 				res.addOptions([alternate]);
 | |
| 		} else {
 | |
| 			var res = this.evaluateExpression(conditionValue ? expr.consequent : expr.alternate);
 | |
| 		}
 | |
| 		res.setRange(expr.range);
 | |
| 		return res;
 | |
| 	});
 | |
| 	this.plugin("evaluate ArrayExpression", function(expr) {
 | |
| 		var items = expr.elements.map(function(element) {
 | |
| 			return this.evaluateExpression(element);
 | |
| 		}, this);
 | |
| 		if(items.filter(function(i) { return !i; }).length > 0) return;
 | |
| 		return new BasicEvaluatedExpression().setItems(items).setRange(expr.range);
 | |
| 	});
 | |
| }
 | |
| Parser.prototype.walkStatements = function walkStatements(statements) {
 | |
| 	statements.forEach(function(statement) {
 | |
| 		this.walkStatement(statement);
 | |
| 	}, this);
 | |
| }
 | |
| 
 | |
| Parser.prototype.walkStatement = function walkStatement(statement) {
 | |
| 	if(this.applyPluginsBailResult("statement", statement) !== undefined) return;
 | |
| 	switch(statement.type) {
 | |
| 	// Real Statements
 | |
| 	case "BlockStatement":
 | |
| 		this.walkStatements(statement.body);
 | |
| 		break;
 | |
| 	case "ExpressionStatement":
 | |
| 		this.walkExpression(statement.expression);
 | |
| 		break;
 | |
| 	case "IfStatement":
 | |
| 		var result = this.applyPluginsBailResult("statement if", statement);
 | |
| 		if(result === undefined) {
 | |
| 			this.walkExpression(statement.test);
 | |
| 			this.walkStatement(statement.consequent);
 | |
| 			if(statement.alternate)
 | |
| 				this.walkStatement(statement.alternate);
 | |
| 		} else {
 | |
| 			if(result)
 | |
| 				this.walkStatement(statement.consequent);
 | |
| 			else if(statement.alternate)
 | |
| 				this.walkStatement(statement.alternate);
 | |
| 		}
 | |
| 		break;
 | |
| 	case "LabeledStatement":
 | |
| 		var result = this.applyPluginsBailResult("label " + statement.label.name, statement);
 | |
| 		if(result === true)
 | |
| 			break;
 | |
| 		this.walkStatement(statement.body);
 | |
| 		break;
 | |
| 	case "WithStatement":
 | |
| 		this.walkExpression(statement.object);
 | |
| 		this.walkStatement(statement.body);
 | |
| 		break;
 | |
| 	case "SwitchStatement":
 | |
| 		this.walkExpression(statement.discriminant);
 | |
| 		this.walkSwitchCases(statement.cases);
 | |
| 		break;
 | |
| 	case "ReturnStatement":
 | |
| 	case "ThrowStatement":
 | |
| 		if(statement.argument)
 | |
| 			this.walkExpression(statement.argument);
 | |
| 		break;
 | |
| 	case "TryStatement":
 | |
| 		if(this.scope.inTry) {
 | |
| 			this.walkStatement(statement.block);
 | |
| 		} else {
 | |
| 			this.scope.inTry = true;
 | |
| 			this.walkStatement(statement.block);
 | |
| 			this.scope.inTry = false;
 | |
| 		}
 | |
| 		this.walkCatchClauses(statement.handlers);
 | |
| 		if(statement.finalizer)
 | |
| 			this.walkStatement(statement.finalizer);
 | |
| 		break;
 | |
| 	case "WhileStatement":
 | |
| 	case "DoWhileStatement":
 | |
| 		this.walkExpression(statement.test);
 | |
| 		this.walkStatement(statement.body);
 | |
| 		break;
 | |
| 	case "ForStatement":
 | |
| 		if(statement.init) {
 | |
| 			if(statement.init.type === "VariableDeclaration")
 | |
| 				this.walkStatement(statement.init);
 | |
| 			else
 | |
| 				this.walkExpression(statement.init);
 | |
| 		}
 | |
| 		if(statement.test)
 | |
| 			this.walkExpression(statement.test);
 | |
| 		if(statement.update)
 | |
| 			this.walkExpression(statement.update);
 | |
| 		this.walkStatement(statement.body);
 | |
| 		break;
 | |
| 	case "ForInStatement":
 | |
| 		if(statement.left.type === "VariableDeclaration")
 | |
| 			this.walkStatement(statement.left);
 | |
| 		else
 | |
| 			this.walkExpression(statement.left);
 | |
| 		this.walkExpression(statement.right);
 | |
| 		this.walkStatement(statement.body);
 | |
| 		break;
 | |
| 
 | |
| 	// Declarations
 | |
| 	case "FunctionDeclaration":
 | |
| 		this.scope.renames["$"+statement.id.name] = undefined;
 | |
| 		this.scope.definitions.push(statement.id.name);
 | |
| 		this.inScope(statement.params, function() {
 | |
| 			if(statement.body.type === "BlockStatement")
 | |
| 				this.walkStatement(statement.body);
 | |
| 			else
 | |
| 				this.walkExpression(statement.body);
 | |
| 		}.bind(this));
 | |
| 		break;
 | |
| 	case "VariableDeclaration":
 | |
| 		if(statement.declarations)
 | |
| 			this.walkVariableDeclarators(statement.declarations);
 | |
| 		break;
 | |
| 	}
 | |
| }
 | |
| 
 | |
| Parser.prototype.walkSwitchCases = function walkSwitchCases(switchCases) {
 | |
| 	switchCases.forEach(function(switchCase) {
 | |
| 		if(switchCase.test)
 | |
| 			this.walkExpression(switchCase.test);
 | |
| 		this.walkStatements(switchCase.consequent);
 | |
| 	}, this);
 | |
| }
 | |
| 
 | |
| Parser.prototype.walkCatchClauses = function walkCatchClauses(catchClauses) {
 | |
| 	catchClauses.forEach(function(catchClause) {
 | |
| 		if(catchClause.guard)
 | |
| 			this.walkExpression(catchClause.guard);
 | |
| 		this.inScope([catchClause.param], function() {
 | |
| 			this.walkStatement(catchClause.body);
 | |
| 		}.bind(this));
 | |
| 	}, this);
 | |
| }
 | |
| 
 | |
| Parser.prototype.walkVariableDeclarators = function walkVariableDeclarators(declarators) {
 | |
| 	declarators.forEach(function(declarator) {
 | |
| 		switch(declarator.type) {
 | |
| 		case "VariableDeclarator":
 | |
| 			var initToIdentifier = declarator.init && declarator.init.type === "Identifier";
 | |
| 			if(initToIdentifier && declarator.id.type === "Identifier" && !this.applyPluginsBailResult("rename " + declarator.init.name, declarator.init)) {
 | |
| 				// renaming with "var a = b;"
 | |
| 				this.scope.renames["$"+declarator.id.name] = this.scope.renames["$"+declarator.init.name] || declarator.init.name;
 | |
| 				var idx = this.scope.definitions.indexOf(declarator.id.name);
 | |
| 				if(idx >= 0) this.scope.definitions.splice(idx, 1);
 | |
| 			} else if(declarator.id.type === "Identifier" && !this.applyPluginsBailResult("var " + declarator.id.name, declarator)) {
 | |
| 				this.scope.renames["$"+declarator.id.name] = undefined;
 | |
| 				this.scope.definitions.push(declarator.id.name);
 | |
| 				if(declarator.init)
 | |
| 					this.walkExpression(declarator.init);
 | |
| 			} else {
 | |
| 				this.walkExpression(declarator.id);
 | |
| 				if(declarator.init)
 | |
| 					this.walkExpression(declarator.init);
 | |
| 			}
 | |
| 			break;
 | |
| 		}
 | |
| 	}, this);
 | |
| }
 | |
| 
 | |
| Parser.prototype.walkExpressions = function walkExpressions(expressions) {
 | |
| 	expressions.forEach(function(expression) {
 | |
| 		if(expression)
 | |
| 			this.walkExpression(expression);
 | |
| 	}, this);
 | |
| }
 | |
| 
 | |
| Parser.prototype.walkExpression = function walkExpression(expression) {
 | |
| 	switch(expression.type) {
 | |
| 	case "ArrayExpression":
 | |
| 		if(expression.elements)
 | |
| 			this.walkExpressions(expression.elements);
 | |
| 		break;
 | |
| 	case "ObjectExpression":
 | |
| 		expression.properties.forEach(function(prop) {
 | |
| 			this.walkExpression(prop.value);
 | |
| 		}, this);
 | |
| 		break;
 | |
| 	case "FunctionExpression":
 | |
| 		this.inScope(expression.params, function() {
 | |
| 			if(expression.body.type === "BlockStatement")
 | |
| 				this.walkStatement(expression.body);
 | |
| 			else
 | |
| 				this.walkExpression(expression.body);
 | |
| 		}.bind(this));
 | |
| 		break;
 | |
| 	case "SequenceExpression":
 | |
| 		if(expression.expressions)
 | |
| 			this.walkExpressions(expression.expressions);
 | |
| 		break;
 | |
| 	case "UpdateExpression":
 | |
| 		this.walkExpression(expression.argument);
 | |
| 		break;
 | |
| 	case "UnaryExpression":
 | |
| 		if(expression.operator === "typeof") {
 | |
| 			var expr = expression.argument;
 | |
| 			var exprName = [];
 | |
| 			while(expr.type == "MemberExpression" && expr.property.type == "Identifier") {
 | |
| 				exprName.unshift(expr.property.name);
 | |
| 				expr = expr.object;
 | |
| 			}
 | |
| 			if(expr.type == "Identifier" && this.scope.definitions.indexOf(expr.name) == -1) {
 | |
| 				exprName.unshift(this.scope.renames["$"+expr.name] || expr.name);
 | |
| 				exprName = exprName.join(".");
 | |
| 				var result = this.applyPluginsBailResult("typeof " + exprName, expression);
 | |
| 				if(result === true)
 | |
| 					break;
 | |
| 			}
 | |
| 		}
 | |
| 		this.walkExpression(expression.argument);
 | |
| 		break;
 | |
| 	case "BinaryExpression":
 | |
| 	case "LogicalExpression":
 | |
| 		this.walkExpression(expression.left);
 | |
| 		this.walkExpression(expression.right);
 | |
| 		break;
 | |
| 	case "AssignmentExpression":
 | |
| 		if(expression.left.type === "Identifier" && expression.right.type === "Identifier" && !this.applyPluginsBailResult("rename " + expression.right.name, expression.right)) {
 | |
| 			// renaming "a = b;"
 | |
| 			this.scope.renames["$"+expression.left.name] = expression.right.name;
 | |
| 			var idx = this.scope.definitions.indexOf(expression.left.name);
 | |
| 			if(idx >= 0) this.scope.definitions.splice(idx, 1);
 | |
| 		} else if(expression.left.type === "Identifier") {
 | |
| 			if(!this.applyPluginsBailResult("assigned " + expression.left.name, expression)) {
 | |
| 				this.walkExpression(expression.right);
 | |
| 			}
 | |
| 			this.scope.renames["$"+expression.left.name] = undefined;
 | |
| 			if(!this.applyPluginsBailResult("assign " + expression.left.name, expression)) {
 | |
| 				this.walkExpression(expression.left);
 | |
| 			}
 | |
| 		} else {
 | |
| 			this.walkExpression(expression.right);
 | |
| 			this.scope.renames["$"+expression.left.name] = undefined;
 | |
| 			this.walkExpression(expression.left);
 | |
| 		}
 | |
| 		break;
 | |
| 	case "ConditionalExpression":
 | |
| 		var result = this.applyPluginsBailResult("expression ?:", expression);
 | |
| 		if(result === undefined) {
 | |
| 			this.walkExpression(expression.test);
 | |
| 			this.walkExpression(expression.consequent);
 | |
| 			if(expression.alternate)
 | |
| 				this.walkExpression(expression.alternate);
 | |
| 		} else {
 | |
| 			if(result)
 | |
| 				this.walkExpression(expression.consequent);
 | |
| 			else if(expression.alternate)
 | |
| 				this.walkExpression(expression.alternate);
 | |
| 		}
 | |
| 		break;
 | |
| 	case "NewExpression":
 | |
| 		this.walkExpression(expression.callee);
 | |
| 		if(expression.arguments)
 | |
| 			this.walkExpressions(expression.arguments);
 | |
| 		break;
 | |
| 	case "CallExpression":
 | |
| 		function walkIIFE(functionExpression, args) {
 | |
| 			var params = functionExpression.params;
 | |
| 			var args = args.map(function(arg, idx) {
 | |
| 				var result = this.evaluateExpression(arg);
 | |
| 				if(result && !result.isIdentifier()) result = undefined;
 | |
| 				if(!result) {
 | |
| 					this.walkExpression(arg);
 | |
| 					return;
 | |
| 				} else if(this.applyPluginsBailResult("rename " + result.identifier, arg)) {
 | |
| 					return;
 | |
| 				}
 | |
| 				return result.identifier;
 | |
| 			}, this);
 | |
| 			this.inScope(params.filter(function(identifier, idx) {
 | |
| 				return !args[idx];
 | |
| 			}), function() {
 | |
| 				args.forEach(function(arg, idx) {
 | |
| 					if(!arg) return;
 | |
| 					if(params[idx].type !== "Identifier") return;
 | |
| 					this.scope.renames["$"+params[idx].name] = arg;
 | |
| 				}, this);
 | |
| 				if(functionExpression.body.type === "BlockStatement")
 | |
| 					this.walkStatement(functionExpression.body);
 | |
| 				else
 | |
| 					this.walkExpression(functionExpression.body);
 | |
| 			}.bind(this));
 | |
| 		}
 | |
| 		if(expression.callee.type === "MemberExpression" && expression.callee.object.type === "FunctionExpression" && expression.callee.property.type === "Identifier" && ["call", "bind"].indexOf(expression.callee.property.name) >= 0 && expression.arguments && expression.arguments.length > 1) {
 | |
| 			// (function(...) { }.call/bind(?, ...))
 | |
| 			walkIIFE.call(this, expression.callee.object, expression.arguments.slice(1));
 | |
| 			this.walkExpression(expression.arguments[0]);
 | |
| 		} else if(expression.callee.type === "FunctionExpression" && expression.arguments) {
 | |
| 			// (function(...) { }(...))
 | |
| 			walkIIFE.call(this, expression.callee, expression.arguments);
 | |
| 		} else {
 | |
| 
 | |
| 			var callee = this.evaluateExpression(expression.callee);
 | |
| 			if(callee.isIdentifier()) {
 | |
| 				var result = this.applyPluginsBailResult("call " + callee.identifier, expression);
 | |
| 				if(result === true)
 | |
| 					break;
 | |
| 			}
 | |
| 
 | |
| 			if(expression.callee)
 | |
| 				this.walkExpression(expression.callee);
 | |
| 			if(expression.arguments)
 | |
| 				this.walkExpressions(expression.arguments);
 | |
| 		}
 | |
| 		break;
 | |
| 	case "MemberExpression":
 | |
| 		var expr = expression;
 | |
| 		var exprName = [];
 | |
| 		while(expr.type == "MemberExpression" && expr.property.type == "Identifier") {
 | |
| 			exprName.unshift(expr.property.name);
 | |
| 			expr = expr.object;
 | |
| 		}
 | |
| 		if(expr.type == "Identifier" && this.scope.definitions.indexOf(expr.name) == -1) {
 | |
| 			exprName.unshift(this.scope.renames["$"+expr.name] || expr.name);
 | |
| 			exprName = exprName.join(".");
 | |
| 			var result = this.applyPluginsBailResult("expression " + exprName, expression);
 | |
| 			if(result === true)
 | |
| 				break;
 | |
| 		}
 | |
| 		this.walkExpression(expression.object);
 | |
| 		if(expression.property.type !== "Identifier")
 | |
| 			this.walkExpression(expression.property);
 | |
| 		break;
 | |
| 	case "Identifier":
 | |
| 		if(this.scope.definitions.indexOf(expression.name) == -1) {
 | |
| 			var result = this.applyPluginsBailResult("expression " + (this.scope.renames["$"+expression.name] || expression.name), expression);
 | |
| 			if(result === true)
 | |
| 				break;
 | |
| 		}
 | |
| 		break;
 | |
| 	}
 | |
| }
 | |
| 
 | |
| Parser.prototype.inScope = function inScope(params, fn) {
 | |
| 	var oldScope = this.scope;
 | |
| 	this.scope = {
 | |
| 		inTry: false,
 | |
| 		definitions: oldScope.definitions.slice(),
 | |
| 		renames: Object.create(oldScope.renames)
 | |
| 	};
 | |
| 	params.forEach(function(param) {
 | |
| 		if(typeof param !== "string") {
 | |
| 			if(param.type !== "Identifier")
 | |
| 				return;
 | |
| 			param = param.name;
 | |
| 		}
 | |
| 		this.scope.renames["$"+param] = undefined;
 | |
| 		this.scope.definitions.push(param);
 | |
| 	}, this);
 | |
| 	fn();
 | |
| 	this.scope = oldScope;
 | |
| }
 | |
| 
 | |
| Parser.prototype.evaluateExpression = function evaluateExpression(expression) {
 | |
| 	var result = this.applyPluginsBailResult("evaluate " + expression.type, expression);
 | |
| 	if(result !== undefined)
 | |
| 		return result;
 | |
| 	return new BasicEvaluatedExpression().setRange(expression.range);
 | |
| }
 | |
| 
 | |
| Parser.prototype.parseString = function parseString(expression) {
 | |
| 	switch(expression.type) {
 | |
| 	case "BinaryExpression":
 | |
| 		if(expression.operator == "+")
 | |
| 			return this.parseString(expression.left) + this.parseString(expression.right);
 | |
| 		break;
 | |
| 	case "Literal":
 | |
| 		return expression.value+"";
 | |
| 	}
 | |
| 	throw new Error(expression.type + " is not supported as parameter for require");
 | |
| }
 | |
| 
 | |
| Parser.prototype.parseCalculatedString = function parseCalculatedString(expression) {
 | |
| 	switch(expression.type) {
 | |
| 	case "BinaryExpression":
 | |
| 		if(expression.operator == "+") {
 | |
| 			var left = this.parseCalculatedString(expression.left);
 | |
| 			var right = this.parseCalculatedString(expression.right);
 | |
| 			if(left.code) {
 | |
| 				return {range: left.range, value: left.value, code: true};
 | |
| 			} else if(right.code) {
 | |
| 				return {range: [left.range[0], right.range ? right.range[1] : left.range[1]], value: left.value + right.value, code: true};
 | |
| 			} else {
 | |
| 				return {range: [left.range[0], right.range[1]], value: left.value + right.value};
 | |
| 			}
 | |
| 		}
 | |
| 		break;
 | |
| 	case "ConditionalExpression":
 | |
| 		var consequent = this.parseCalculatedString(expression.consequent);
 | |
| 		var alternate = this.parseCalculatedString(expression.alternate);
 | |
| 		var items = [];
 | |
| 		if(consequent.conditional)
 | |
| 			Array.prototype.push.apply(items, consequent.conditional);
 | |
| 		else if(!consequent.code)
 | |
| 			items.push(consequent);
 | |
| 		else break;
 | |
| 		if(alternate.conditional)
 | |
| 			Array.prototype.push.apply(items, alternate.conditional);
 | |
| 		else if(!alternate.code)
 | |
| 			items.push(alternate);
 | |
| 		else break;
 | |
| 		return {value: "", code: true, conditional: items};
 | |
| 	case "Literal":
 | |
| 		return {range: expression.range, value: expression.value+""};
 | |
| 		break;
 | |
| 	}
 | |
| 	return {value: "", code: true};
 | |
| }
 | |
| 
 | |
| Parser.prototype.parseStringArray = function parseStringArray(expression) {
 | |
| 	switch(expression.type) {
 | |
| 	case "ArrayExpression":
 | |
| 		var arr = [];
 | |
| 		if(expression.elements)
 | |
| 			expression.elements.forEach(function(expr) {
 | |
| 				arr.push(this.parseString(expr));
 | |
| 			}, this);
 | |
| 		return arr;
 | |
| 	}
 | |
| 	return [this.parseString(expression)];
 | |
| }
 | |
| 
 | |
| Parser.prototype.parseCalculatedStringArray = function parseCalculatedStringArray(expression) {
 | |
| 	switch(expression.type) {
 | |
| 	case "ArrayExpression":
 | |
| 		var arr = [];
 | |
| 		if(expression.elements)
 | |
| 			expression.elements.forEach(function(expr) {
 | |
| 				arr.push(this.parseCalculatedString(expr));
 | |
| 			}, this);
 | |
| 		return arr;
 | |
| 	}
 | |
| 	return [this.parseCalculatedString(expression)];
 | |
| }
 | |
| 
 | |
| Parser.prototype.parse = function parse(source, initialState) {
 | |
| 	var ast = esprima.parse(source, {range: true, loc: true, raw: true});
 | |
| 	if(!ast || typeof ast != "object")
 | |
| 		throw new Error("Source couldn't be parsed");
 | |
| 	var oldScope = this.scope;
 | |
| 	var oldState = this.state;
 | |
| 	this.scope = {
 | |
| 		inTry: false,
 | |
| 		definitions: [],
 | |
| 		renames: {}
 | |
| 	};
 | |
| 	var state = this.state = initialState || {};
 | |
| 	if(this.applyPluginsBailResult("program", ast) === undefined)
 | |
| 		this.walkStatements(ast.body);
 | |
| 	this.scope = oldScope;
 | |
| 	this.state = oldState;
 | |
| 	return state;
 | |
| }
 | |
| 
 | |
| Parser.prototype.evaluate = function evaluate(source) {
 | |
| 	var ast = esprima.parse("("+source+")", {range: true, loc: true, raw: true});
 | |
| 	if(!ast || typeof ast != "object" || ast.type !== "Program")
 | |
| 		throw new Error("evaluate: Source couldn't be parsed");
 | |
| 	if(ast.body.length !== 1 || ast.body[0].type !== "ExpressionStatement")
 | |
| 		throw new Error("evaluate: Source is not a expression");
 | |
| 	return this.evaluateExpression(ast.body[0].expression);
 | |
| };
 |