1 | "use strict";
|
2 |
|
3 | exports.__esModule = true;
|
4 |
|
5 | exports.default = function (path) {
|
6 | var scope = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : path.scope;
|
7 | var node = path.node;
|
8 |
|
9 | var container = t.functionExpression(null, [], node.body, node.generator, node.async);
|
10 |
|
11 | var callee = container;
|
12 | var args = [];
|
13 |
|
14 | (0, _babelHelperHoistVariables2.default)(path, function (id) {
|
15 | return scope.push({ id: id });
|
16 | });
|
17 |
|
18 | var state = {
|
19 | foundThis: false,
|
20 | foundArguments: false
|
21 | };
|
22 |
|
23 | path.traverse(visitor, state);
|
24 |
|
25 | if (state.foundArguments) {
|
26 | callee = t.memberExpression(container, t.identifier("apply"));
|
27 | args = [];
|
28 |
|
29 | if (state.foundThis) {
|
30 | args.push(t.thisExpression());
|
31 | }
|
32 |
|
33 | if (state.foundArguments) {
|
34 | if (!state.foundThis) args.push(t.nullLiteral());
|
35 | args.push(t.identifier("arguments"));
|
36 | }
|
37 | }
|
38 |
|
39 | var call = t.callExpression(callee, args);
|
40 | if (node.generator) call = t.yieldExpression(call, true);
|
41 |
|
42 | return t.returnStatement(call);
|
43 | };
|
44 |
|
45 | var _babelHelperHoistVariables = require("babel-helper-hoist-variables");
|
46 |
|
47 | var _babelHelperHoistVariables2 = _interopRequireDefault(_babelHelperHoistVariables);
|
48 |
|
49 | var _babelTypes = require("babel-types");
|
50 |
|
51 | var t = _interopRequireWildcard(_babelTypes);
|
52 |
|
53 | 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; } }
|
54 |
|
55 | function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
|
56 |
|
57 | var visitor = {
|
58 | enter: function enter(path, state) {
|
59 | if (path.isThisExpression()) {
|
60 | state.foundThis = true;
|
61 | }
|
62 |
|
63 | if (path.isReferencedIdentifier({ name: "arguments" })) {
|
64 | state.foundArguments = true;
|
65 | }
|
66 | },
|
67 | Function: function Function(path) {
|
68 | path.skip();
|
69 | }
|
70 | };
|
71 |
|
72 | module.exports = exports["default"]; |
\ | No newline at end of file |