You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

72 lines
1.9 KiB
JavaScript

6 years ago
"use strict";
exports.__esModule = true;
exports.default = function (path) {
var scope = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : path.scope;
var node = path.node;
var container = t.functionExpression(null, [], node.body, node.generator, node.async);
var callee = container;
var args = [];
(0, _babelHelperHoistVariables2.default)(path, function (id) {
return scope.push({ id: id });
});
var state = {
foundThis: false,
foundArguments: false
};
path.traverse(visitor, state);
if (state.foundArguments) {
callee = t.memberExpression(container, t.identifier("apply"));
args = [];
if (state.foundThis) {
args.push(t.thisExpression());
}
if (state.foundArguments) {
if (!state.foundThis) args.push(t.nullLiteral());
args.push(t.identifier("arguments"));
}
}
var call = t.callExpression(callee, args);
if (node.generator) call = t.yieldExpression(call, true);
return t.returnStatement(call);
};
var _babelHelperHoistVariables = require("babel-helper-hoist-variables");
var _babelHelperHoistVariables2 = _interopRequireDefault(_babelHelperHoistVariables);
var _babelTypes = require("babel-types");
var t = _interopRequireWildcard(_babelTypes);
function _interopRequireWildcard(obj) { if (obj && obj.__esModule) { return obj; } else { var newObj = {}; if (obj != null) { for (var key in obj) { if (Object.prototype.hasOwnProperty.call(obj, key)) newObj[key] = obj[key]; } } newObj.default = obj; return newObj; } }
function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
var visitor = {
enter: function enter(path, state) {
if (path.isThisExpression()) {
state.foundThis = true;
}
if (path.isReferencedIdentifier({ name: "arguments" })) {
state.foundArguments = true;
}
},
Function: function Function(path) {
path.skip();
}
};
module.exports = exports["default"];