index.js 5.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150
  1. "use strict";
  2. Object.defineProperty(exports, "__esModule", {
  3. value: true
  4. });
  5. exports.default = void 0;
  6. var _helperPluginUtils = require("@babel/helper-plugin-utils");
  7. var _pluginSyntaxOptionalChaining = _interopRequireDefault(require("@babel/plugin-syntax-optional-chaining"));
  8. var _core = require("@babel/core");
  9. function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
  10. var _default = (0, _helperPluginUtils.declare)((api, options) => {
  11. api.assertVersion(7);
  12. const {
  13. loose = false
  14. } = options;
  15. function isSimpleMemberExpression(expression) {
  16. return _core.types.isIdentifier(expression) || _core.types.isSuper(expression) || _core.types.isMemberExpression(expression) && !expression.computed && isSimpleMemberExpression(expression.object);
  17. }
  18. return {
  19. name: "proposal-optional-chaining",
  20. inherits: _pluginSyntaxOptionalChaining.default,
  21. visitor: {
  22. "OptionalCallExpression|OptionalMemberExpression"(path) {
  23. const {
  24. scope
  25. } = path;
  26. let maybeParenthesized = path;
  27. const parentPath = path.findParent(p => {
  28. if (!p.isParenthesizedExpression()) return true;
  29. maybeParenthesized = p;
  30. });
  31. let isDeleteOperation = false;
  32. const parentIsCall = parentPath.isCallExpression({
  33. callee: maybeParenthesized.node
  34. }) && path.isOptionalMemberExpression();
  35. const optionals = [];
  36. let optionalPath = path;
  37. while (optionalPath.isOptionalMemberExpression() || optionalPath.isOptionalCallExpression() || optionalPath.isParenthesizedExpression() || optionalPath.isTSNonNullExpression()) {
  38. const {
  39. node
  40. } = optionalPath;
  41. if (node.optional) {
  42. optionals.push(node);
  43. }
  44. if (optionalPath.isOptionalMemberExpression()) {
  45. optionalPath.node.type = "MemberExpression";
  46. optionalPath = optionalPath.get("object");
  47. } else if (optionalPath.isOptionalCallExpression()) {
  48. optionalPath.node.type = "CallExpression";
  49. optionalPath = optionalPath.get("callee");
  50. } else {
  51. optionalPath = optionalPath.get("expression");
  52. }
  53. }
  54. let replacementPath = path;
  55. if (parentPath.isUnaryExpression({
  56. operator: "delete"
  57. })) {
  58. replacementPath = parentPath;
  59. isDeleteOperation = true;
  60. }
  61. for (let i = optionals.length - 1; i >= 0; i--) {
  62. const node = optionals[i];
  63. const isCall = _core.types.isCallExpression(node);
  64. const replaceKey = isCall ? "callee" : "object";
  65. const chain = node[replaceKey];
  66. let ref;
  67. let check;
  68. if (loose && isCall && isSimpleMemberExpression(chain)) {
  69. check = ref = chain;
  70. } else {
  71. ref = scope.maybeGenerateMemoised(chain);
  72. if (ref) {
  73. check = _core.types.assignmentExpression("=", _core.types.cloneNode(ref), chain);
  74. node[replaceKey] = ref;
  75. } else {
  76. check = ref = chain;
  77. }
  78. }
  79. if (isCall && _core.types.isMemberExpression(chain)) {
  80. if (loose && isSimpleMemberExpression(chain)) {
  81. node.callee = chain;
  82. } else {
  83. const {
  84. object
  85. } = chain;
  86. let context = scope.maybeGenerateMemoised(object);
  87. if (context) {
  88. chain.object = _core.types.assignmentExpression("=", context, object);
  89. } else if (_core.types.isSuper(object)) {
  90. context = _core.types.thisExpression();
  91. } else {
  92. context = object;
  93. }
  94. node.arguments.unshift(_core.types.cloneNode(context));
  95. node.callee = _core.types.memberExpression(node.callee, _core.types.identifier("call"));
  96. }
  97. }
  98. let replacement = replacementPath.node;
  99. if (i === 0 && parentIsCall) {
  100. var _baseRef;
  101. const {
  102. object
  103. } = replacement;
  104. let baseRef;
  105. if (!loose || !isSimpleMemberExpression(object)) {
  106. baseRef = scope.maybeGenerateMemoised(object);
  107. if (baseRef) {
  108. replacement.object = _core.types.assignmentExpression("=", baseRef, object);
  109. }
  110. }
  111. replacement = _core.types.callExpression(_core.types.memberExpression(replacement, _core.types.identifier("bind")), [_core.types.cloneNode((_baseRef = baseRef) != null ? _baseRef : object)]);
  112. }
  113. replacementPath.replaceWith(_core.types.conditionalExpression(loose ? _core.types.binaryExpression("==", _core.types.cloneNode(check), _core.types.nullLiteral()) : _core.types.logicalExpression("||", _core.types.binaryExpression("===", _core.types.cloneNode(check), _core.types.nullLiteral()), _core.types.binaryExpression("===", _core.types.cloneNode(ref), scope.buildUndefinedNode())), isDeleteOperation ? _core.types.booleanLiteral(true) : scope.buildUndefinedNode(), replacement));
  114. replacementPath = replacementPath.get("alternate");
  115. }
  116. }
  117. }
  118. };
  119. });
  120. exports.default = _default;