You can not 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
72 lines
1.9 KiB
"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"]; |