35 lines
931 B
JavaScript
35 lines
931 B
JavaScript
var util = require("../util");
|
|
var b = require("ast-types").builders;
|
|
|
|
exports.ForOfStatement = function (node, parent, opts, generateUid) {
|
|
var left = node.left;
|
|
var declar;
|
|
|
|
var stepKey = b.identifier(generateUid("step"));
|
|
var stepValueId = b.memberExpression(stepKey, b.identifier("value"), false);
|
|
|
|
if (left.type === "Identifier") {
|
|
declar = b.expressionStatement(b.assignmentExpression("=", left, stepValueId));
|
|
} else if (left.type === "VariableDeclaration") {
|
|
declar = b.variableDeclaration(left.kind, [
|
|
b.variableDeclarator(left.declarations[0].id, stepValueId)
|
|
]);
|
|
} else {
|
|
return;
|
|
}
|
|
|
|
var node2 = util.template("for-of", {
|
|
ITERATOR_KEY: generateUid("iterator"),
|
|
STEP_KEY: stepKey,
|
|
OBJECT: node.right
|
|
});
|
|
|
|
util.ensureBlock(node);
|
|
|
|
var block = node2.body;
|
|
block.body.push(declar);
|
|
block.body = block.body.concat(node.body.body);
|
|
|
|
return node2;
|
|
};
|