mirror of
https://github.com/facebook/react.git
synced 2025-11-01 09:12:30 +00:00
92 lines
2.9 KiB
JavaScript
92 lines
2.9 KiB
JavaScript
/**
|
|
* Copyright 2013 Facebook, Inc.
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
'use strict';
|
|
|
|
var recast = require('recast');
|
|
var types = recast.types;
|
|
var namedTypes = types.namedTypes;
|
|
var builders = types.builders;
|
|
var hasOwn = Object.prototype.hasOwnProperty;
|
|
|
|
function propagate(constants, source) {
|
|
return recast.print(transform(recast.parse(source), constants));
|
|
}
|
|
|
|
function transform(ast, constants) {
|
|
constants = constants || {};
|
|
|
|
return types.traverse(ast, function(node, traverse) {
|
|
if (namedTypes.Identifier.check(node)) {
|
|
// If the identifier is the property of a member expression
|
|
// (e.g. object.property), then it definitely is not a constant
|
|
// expression that we want to replace.
|
|
if (namedTypes.MemberExpression.check(this.parent.node) &&
|
|
this.name === 'property' &&
|
|
!this.parent.node.computed) {
|
|
return false;
|
|
}
|
|
|
|
// There could in principle be a constant called "hasOwnProperty",
|
|
// so be careful always to use Object.prototype.hasOwnProperty.
|
|
if (hasOwn.call(constants, node.name)) {
|
|
this.replace(builders.literal(constants[node.name]));
|
|
return false;
|
|
}
|
|
|
|
} else if (namedTypes.CallExpression.check(node)) {
|
|
if (!constants.__DEV__) {
|
|
if (namedTypes.Identifier.check(node.callee) &&
|
|
node.callee.name === 'invariant') {
|
|
// Truncate the arguments of invariant(condition, ...)
|
|
// statements to just the condition.
|
|
node.arguments.length = 1;
|
|
}
|
|
}
|
|
|
|
} else if (namedTypes.IfStatement.check(node) &&
|
|
namedTypes.Literal.check(node.test)) {
|
|
if (node.test.value) {
|
|
// If the alternate (then) branch is dead code, remove it.
|
|
this.get("alternate").replace();
|
|
|
|
// This is what happens when you replace a node with nothing and
|
|
// it can't be removed from a list of statements.
|
|
assert.strictEqual(node.alternate, null);
|
|
|
|
} else if (node.alternate) {
|
|
// Replace the whole if-statement with just the alternate clause.
|
|
this.replace(node.alternate);
|
|
return false;
|
|
|
|
} else {
|
|
// Remove the entire if-statement.
|
|
this.replace();
|
|
return false;
|
|
}
|
|
}
|
|
});
|
|
}
|
|
|
|
if (!module.parent) {
|
|
var constants = JSON.parse(process.argv[3]);
|
|
recast.run(function(ast, callback) {
|
|
callback(transform(ast, constants));
|
|
});
|
|
}
|
|
|
|
exports.propagate = propagate;
|
|
exports.transform = transform;
|