index.js 4.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120
  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. parentPath,
  25. scope
  26. } = path;
  27. let isDeleteOperation = false;
  28. const optionals = [];
  29. let optionalPath = path;
  30. while (optionalPath.isOptionalMemberExpression() || optionalPath.isOptionalCallExpression()) {
  31. const {
  32. node
  33. } = optionalPath;
  34. if (node.optional) {
  35. optionals.push(node);
  36. }
  37. if (optionalPath.isOptionalMemberExpression()) {
  38. optionalPath.node.type = "MemberExpression";
  39. optionalPath = optionalPath.get("object");
  40. } else if (optionalPath.isOptionalCallExpression()) {
  41. optionalPath.node.type = "CallExpression";
  42. optionalPath = optionalPath.get("callee");
  43. }
  44. }
  45. let replacementPath = path;
  46. if (parentPath.isUnaryExpression({
  47. operator: "delete"
  48. })) {
  49. replacementPath = parentPath;
  50. isDeleteOperation = true;
  51. }
  52. for (let i = optionals.length - 1; i >= 0; i--) {
  53. const node = optionals[i];
  54. const isCall = _core.types.isCallExpression(node);
  55. const replaceKey = isCall ? "callee" : "object";
  56. const chain = node[replaceKey];
  57. let ref;
  58. let check;
  59. if (loose && isCall && isSimpleMemberExpression(chain)) {
  60. check = ref = chain;
  61. } else {
  62. ref = scope.maybeGenerateMemoised(chain);
  63. if (ref) {
  64. check = _core.types.assignmentExpression("=", _core.types.cloneNode(ref), chain);
  65. node[replaceKey] = ref;
  66. } else {
  67. check = ref = chain;
  68. }
  69. }
  70. if (isCall && _core.types.isMemberExpression(chain)) {
  71. if (loose && isSimpleMemberExpression(chain)) {
  72. node.callee = chain;
  73. } else {
  74. const {
  75. object
  76. } = chain;
  77. let context = scope.maybeGenerateMemoised(object);
  78. if (context) {
  79. chain.object = _core.types.assignmentExpression("=", context, object);
  80. } else if (_core.types.isSuper(object)) {
  81. context = _core.types.thisExpression();
  82. } else {
  83. context = object;
  84. }
  85. node.arguments.unshift(_core.types.cloneNode(context));
  86. node.callee = _core.types.memberExpression(node.callee, _core.types.identifier("call"));
  87. }
  88. }
  89. 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(), replacementPath.node));
  90. replacementPath = replacementPath.get("alternate");
  91. }
  92. }
  93. }
  94. };
  95. });
  96. exports.default = _default;