babel/lib/6to5/transformers/constants.js
Sebastian McKenzie d7ae3b506a fix linting errors
2014-11-04 15:53:11 +11:00

44 lines
1.1 KiB
JavaScript

var traverse = require("../traverse");
var t = require("../types");
var _ = require("lodash");
exports.Program =
exports.BlockStatement =
exports.ForInStatement =
exports.ForOfStatement =
exports.ForStatement = function (node, parent, file) {
var constants = [];
var check = function (node, name) {
if (constants.indexOf(name) >= 0) {
throw file.errorWithNode(node, name + " is read-only");
}
};
_.each(node.body, function (child) {
if (child && t.isVariableDeclaration(child) && child.kind === "const") {
_.each(child.declarations, function (declar) {
_.each(t.getIds(declar.id), function (name) {
check(declar, name);
constants.push(name);
});
declar._ignoreConstant = true;
});
child.kind = "let";
}
});
if (!constants.length) return;
traverse(node, function (child) {
if (child._ignoreConstant) return;
if (t.isVariableDeclarator(child) || t.isFunctionDeclaration(child) || t.isClassDeclaration(child)) {
check(child, child.id.name);
} else if (t.isAssignmentExpression(child)) {
check(child, child.left.name);
}
});
};