56 lines
1.5 KiB
JavaScript
56 lines
1.5 KiB
JavaScript
var traverse = require("../traverse");
|
|
var util = require("../util");
|
|
var b = require("ast-types").builders;
|
|
|
|
exports.ArrowFunctionExpression = function (node) {
|
|
var body = node.body;
|
|
|
|
// expression body
|
|
if (body.type !== "BlockStatement") {
|
|
body = b.blockStatement([
|
|
b.returnStatement(body)
|
|
]);
|
|
}
|
|
|
|
node.expression = false;
|
|
node.body = body;
|
|
node.type = "FunctionExpression";
|
|
|
|
if (traverse.hasType(node, "ThisExpression")) {
|
|
return util.template("function-bind-this", {
|
|
FUNCTION: node
|
|
});
|
|
} else {
|
|
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"))
|
|
]));
|
|
}
|
|
};
|