cerxes/babel
3
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

refer to parent arguments inside arrow functions - fixes #39

Browse Source
This commit is contained in:
Sebastian McKenzie
2014-10-11 09:29:00 +11:00
parent 35522c2ef3
commit 9b82c5b99c
3 changed files with 64 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
lib/6to5/transformers/arrow-functions.js
View File

@@ -24,3 +24,32 @@ exports.ArrowFunctionExpression = function (node) {
return node;
}
};
exports.FunctionDeclaration =
exports.FunctionExpression = function (node, parent, opts, generateUid) {
var hasArguments = false;
var id;
// traverse the function and find all arrow functions
traverse(node, function (node) {
if (node.type !== "ArrowFunctionExpression") return;
traverse(node, function (node, parent) {
if (node.type === "Identifier" && node.name === "arguments" &&
parent.type !== "MemberExpression") {
hasArguments = true;
id = id || b.identifier(generateUid("arguments"));
return id;
}
}, ["FunctionDeclaration", "FunctionExpression", "ArrowFunctionExpression"]);
return false;
}, ["FunctionDeclaration", "FunctionExpression"]);
if (hasArguments) {
util.ensureBlock(node);
node.body.body.unshift(b.variableDeclaration("var", [
b.variableDeclarator(id, b.identifier("arguments"))
]));
}
};