default.js 4.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164
  1. "use strict";
  2. exports.__esModule = true;
  3. exports.visitor = undefined;
  4. var _getIterator2 = require("babel-runtime/core-js/get-iterator");
  5. var _getIterator3 = _interopRequireDefault(_getIterator2);
  6. var _babelHelperGetFunctionArity = require("babel-helper-get-function-arity");
  7. var _babelHelperGetFunctionArity2 = _interopRequireDefault(_babelHelperGetFunctionArity);
  8. var _babelHelperCallDelegate = require("babel-helper-call-delegate");
  9. var _babelHelperCallDelegate2 = _interopRequireDefault(_babelHelperCallDelegate);
  10. var _babelTemplate = require("babel-template");
  11. var _babelTemplate2 = _interopRequireDefault(_babelTemplate);
  12. var _babelTypes = require("babel-types");
  13. var t = _interopRequireWildcard(_babelTypes);
  14. 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; } }
  15. function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
  16. /* eslint max-len: 0 */
  17. var buildDefaultParam = (0, _babelTemplate2.default)("\n let VARIABLE_NAME =\n ARGUMENTS.length <= ARGUMENT_KEY || ARGUMENTS[ARGUMENT_KEY] === undefined ?\n DEFAULT_VALUE\n :\n ARGUMENTS[ARGUMENT_KEY];\n");
  18. var buildCutOff = (0, _babelTemplate2.default)("\n let $0 = $1[$2];\n");
  19. function hasDefaults(node) {
  20. for (var _iterator = node.params, _isArray = Array.isArray(_iterator), _i = 0, _iterator = _isArray ? _iterator : (0, _getIterator3.default)(_iterator);;) {
  21. var _ref;
  22. if (_isArray) {
  23. if (_i >= _iterator.length) break;
  24. _ref = _iterator[_i++];
  25. } else {
  26. _i = _iterator.next();
  27. if (_i.done) break;
  28. _ref = _i.value;
  29. }
  30. var param = _ref;
  31. if (!t.isIdentifier(param)) return true;
  32. }
  33. return false;
  34. }
  35. var iifeVisitor = {
  36. ReferencedIdentifier: function ReferencedIdentifier(path, state) {
  37. var name = path.node.name;
  38. if (name === "eval" || path.scope.hasOwnBinding(name) && path.scope.getOwnBinding(name).kind !== "param") {
  39. state.iife = true;
  40. path.stop();
  41. }
  42. },
  43. Scope: function Scope(path) {
  44. // different bindings
  45. path.skip();
  46. }
  47. };
  48. var visitor = exports.visitor = {
  49. Function: function Function(path) {
  50. var node = path.node;
  51. var scope = path.scope;
  52. if (!hasDefaults(node)) return;
  53. // ensure it's a block, useful for arrow functions
  54. path.ensureBlock();
  55. var state = {
  56. iife: false,
  57. scope: scope
  58. };
  59. var body = [];
  60. //
  61. var argsIdentifier = t.identifier("arguments");
  62. argsIdentifier._shadowedFunctionLiteral = path;
  63. // push a default parameter definition
  64. function pushDefNode(left, right, i) {
  65. var defNode = buildDefaultParam({
  66. VARIABLE_NAME: left,
  67. DEFAULT_VALUE: right,
  68. ARGUMENT_KEY: t.numericLiteral(i),
  69. ARGUMENTS: argsIdentifier
  70. });
  71. defNode._blockHoist = node.params.length - i;
  72. body.push(defNode);
  73. }
  74. //
  75. var lastNonDefaultParam = (0, _babelHelperGetFunctionArity2.default)(node);
  76. //
  77. var params = path.get("params");
  78. for (var i = 0; i < params.length; i++) {
  79. var param = params[i];
  80. if (!param.isAssignmentPattern()) {
  81. if (!state.iife && !param.isIdentifier()) {
  82. param.traverse(iifeVisitor, state);
  83. }
  84. continue;
  85. }
  86. var left = param.get("left");
  87. var right = param.get("right");
  88. //
  89. if (i >= lastNonDefaultParam || left.isPattern()) {
  90. var placeholder = scope.generateUidIdentifier("x");
  91. placeholder._isDefaultPlaceholder = true;
  92. node.params[i] = placeholder;
  93. } else {
  94. node.params[i] = left.node;
  95. }
  96. //
  97. if (!state.iife) {
  98. if (right.isIdentifier() && scope.hasOwnBinding(right.node.name) && scope.getOwnBinding(right.node.name).kind !== "param") {
  99. // the right hand side references a parameter
  100. state.iife = true;
  101. } else {
  102. right.traverse(iifeVisitor, state);
  103. }
  104. }
  105. pushDefNode(left.node, right.node, i);
  106. }
  107. // add declarations for trailing parameters
  108. for (var _i2 = lastNonDefaultParam + 1; _i2 < node.params.length; _i2++) {
  109. var _param = node.params[_i2];
  110. if (_param._isDefaultPlaceholder) continue;
  111. var declar = buildCutOff(_param, argsIdentifier, t.numericLiteral(_i2));
  112. declar._blockHoist = node.params.length - _i2;
  113. body.push(declar);
  114. }
  115. // we need to cut off all trailing parameters
  116. node.params = node.params.slice(0, lastNonDefaultParam);
  117. if (state.iife) {
  118. body.push((0, _babelHelperCallDelegate2.default)(path, scope));
  119. path.set("body", t.blockStatement(body));
  120. } else {
  121. path.get("body").unshiftContainer("body", body);
  122. }
  123. }
  124. };