Revision 1f9e1a69c6919b53aaa7c4e27756b65280e59456 authored by Sean Larkin on 29 November 2016, 15:32:10 UTC, committed by Tobias Koppers on 07 December 2016, 16:22:19 UTC
1 parent 23a127b
DefinePlugin.js
/*
MIT License http://www.opensource.org/licenses/mit-license.php
Author Tobias Koppers @sokra
*/
var ConstDependency = require("./dependencies/ConstDependency");
var BasicEvaluatedExpression = require("./BasicEvaluatedExpression");
var NullFactory = require("./NullFactory");
function DefinePlugin(definitions) {
this.definitions = definitions;
}
module.exports = DefinePlugin;
DefinePlugin.prototype.apply = function(compiler) {
var definitions = this.definitions;
compiler.plugin("compilation", function(compilation, params) {
compilation.dependencyFactories.set(ConstDependency, new NullFactory());
compilation.dependencyTemplates.set(ConstDependency, new ConstDependency.Template());
params.normalModuleFactory.plugin("parser", function(parser) {
(function walkDefinitions(definitions, prefix) {
Object.keys(definitions).forEach(function(key) {
var code = definitions[key];
if(code && typeof code === "object" && !(code instanceof RegExp)) {
walkDefinitions(code, prefix + key + ".");
applyObjectDefine(prefix + key, code);
return;
}
applyDefineKey(prefix, key);
applyDefine(prefix + key, code);
});
}(definitions, ""));
function stringifyObj(obj) {
return "{" + Object.keys(obj).map(function(key) {
var code = obj[key];
return JSON.stringify(key) + ":" + toCode(code);
}).join(",") + "}";
}
function toCode(code) {
if(code === null) return "null";
else if(code === undefined) return "undefined";
else if(code instanceof RegExp && code.toString) return code.toString();
else if(typeof code === "function" && code.toString) return "(" + code.toString() + ")";
else if(typeof code === "object") return stringifyObj(code);
else return code + "";
}
function applyDefineKey(prefix, key) {
var splittedKey = key.split(".");
splittedKey.slice(1).forEach(function(_, i) {
var fullKey = prefix + splittedKey.slice(0, i + 1).join(".");
parser.plugin("can-rename " + fullKey, function() {
return true;
});
});
}
function applyDefine(key, code) {
var isTypeof = /^typeof\s+/.test(key);
if(isTypeof) key = key.replace(/^typeof\s+/, "");
var recurse = false;
var recurseTypeof = false;
code = toCode(code);
if(!isTypeof) {
parser.plugin("can-rename " + key, function() {
return true;
});
parser.plugin("evaluate Identifier " + key, function(expr) {
if(recurse) return;
recurse = true;
var res = this.evaluate(code);
recurse = false;
res.setRange(expr.range);
return res;
});
parser.plugin("expression " + key, function(expr) {
var dep = new ConstDependency(code, expr.range);
dep.loc = expr.loc;
this.state.current.addDependency(dep);
return true;
});
}
var typeofCode = isTypeof ? code : "typeof (" + code + ")";
parser.plugin("evaluate typeof " + key, function(expr) {
if(recurseTypeof) return;
recurseTypeof = true;
var res = this.evaluate(typeofCode);
recurseTypeof = false;
res.setRange(expr.range);
return res;
});
parser.plugin("typeof " + key, function(expr) {
var res = this.evaluate(typeofCode);
if(!res.isString()) return;
var dep = new ConstDependency(JSON.stringify(res.string), expr.range);
dep.loc = expr.loc;
this.state.current.addDependency(dep);
return true;
});
}
function applyObjectDefine(key, obj) {
var code = stringifyObj(obj);
parser.plugin("can-rename " + key, function() {
return true;
});
parser.plugin("evaluate Identifier " + key, function(expr) {
return new BasicEvaluatedExpression().setRange(expr.range);
});
parser.plugin("evaluate typeof " + key, function(expr) {
return new BasicEvaluatedExpression().setString("object").setRange(expr.range);
});
parser.plugin("expression " + key, function(expr) {
var dep = new ConstDependency(code, expr.range);
dep.loc = expr.loc;
this.state.current.addDependency(dep);
return true;
});
parser.plugin("typeof " + key, function(expr) {
var dep = new ConstDependency("\"object\"", expr.range);
dep.loc = expr.loc;
this.state.current.addDependency(dep);
return true;
});
}
});
});
};
Computing file changes ...