var traverse = require("../traverse");
var util = require("../util");
var b = require("ast-types").builders;
var _ = require("lodash");
var blockTypes = ["FunctionDeclaration", "FunctionExpression", "BlockStatement"];
var isLet = function (node) {
if (node && node.type === "VariableDeclaration" && node.kind === "let") {
node.kind = "var";
node._ignoreBlockBindingHoist = true;
return true;
}
};
var hasLet = function (nodes) {
var has = false;
_.each(nodes, function (node) {
if (isLet(node)) has = true;
});
return has;
};
exports.Program = function (node) {
if (hasLet(node.body)) node.body = buildNode(node.body).node;
};
exports.BlockStatement = function (node, parent, opts, generateUid) {
if (!hasLet(node.body)) return;
// ignore if we're the body of a closure already
if (parent.type === "FunctionExpression") return;
var body = node.body;
var built = buildNode(node.body, true);
node.body = built.node;
traverse.replace(built.body, function (node) {
if (node.type === "ContinueStatement") {
return b.returnStatement(null);
}
}, blockTypes);
if (traverse.hasType(body, "BreakStatement", blockTypes)) {
var id = b.identifier(generateUid("break"));
node.body.unshift(b.variableDeclaration("var", [
b.variableDeclarator(id, b.literal(false))
]));
traverse.replace(built.body, function (node) {
if (node.type === "BreakStatement") {
return b.returnStatement(b.assignmentExpression("=", id, b.literal(true)));
}
}, blockTypes);
node.body.push(b.ifStatement(id, b.breakStatement()));
}
};
var buildForStatement = function (key) {
return function (node, parent) {
if (isLet(node[key])) {
Iif (parent.type === "LabeledStatement") {
throw util.errorWithNode(parent, "Label statements not supported with block binding yet.");
}
return buildNode(node).node;
}
};
};
exports.ForOfStatement = exports.ForInStatement = buildForStatement("left");
exports.ForStatement = buildForStatement("init");
var buildNode = function (node) {
var nodes = [];
// hoist normal variable declarations
node = [].concat(node);
node = node.map(function (node) {
if (node._ignoreBlockBindingHoist) return node;
if (node.type === "VariableDeclaration" && node.kind === "var") {
var declars = node.declarations.map(function (declar) {
return b.variableDeclarator(declar.id, null);
});
nodes.push(b.variableDeclaration("var", declars));
return node.declarations.map(function (declar) {
return util.template("assign", {
VALUE: declar.init,
KEY: declar.id
}, true);
});
} else if (node.type === "ForInStatement" && node.left.type === "VariableDeclaration" && !node.left._ignoreBlockBindingHoist) {
var id = node.left.declarations[0].id;
node.left = id;
nodes.push(util.template("variable-declare", {
KEY: id
}));
}
return node;
});
//
var block = b.blockStatement([]);
block.body = node;
var func = b.functionExpression(null, [], block, false);
var templateName = "function-call";
if (traverse.hasType(node, "ThisExpression")) templateName += "-this";
//
nodes.push(util.template(templateName, {
FUNCTION: func
}, true));
return {
node: nodes,
body: block
};
};
|