index.js 4.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146
  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 _pluginSyntaxPrivatePropertyInObject = _interopRequireDefault(require("@babel/plugin-syntax-private-property-in-object"));
  8. var _helperCreateClassFeaturesPlugin = require("@babel/helper-create-class-features-plugin");
  9. var _helperAnnotateAsPure = _interopRequireDefault(require("@babel/helper-annotate-as-pure"));
  10. function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
  11. var _default = (0, _helperPluginUtils.declare)(({
  12. assertVersion,
  13. types: t,
  14. template
  15. }, {
  16. loose
  17. }) => {
  18. assertVersion(7);
  19. const classWeakSets = new WeakMap();
  20. const fieldsWeakSets = new WeakMap();
  21. function unshadow(name, targetScope, scope) {
  22. while (scope !== targetScope) {
  23. if (scope.hasOwnBinding(name)) scope.rename(name);
  24. scope = scope.parent;
  25. }
  26. }
  27. function injectToFieldInit(fieldPath, expr, before = false) {
  28. if (fieldPath.node.value) {
  29. if (before) {
  30. fieldPath.get("value").insertBefore(expr);
  31. } else {
  32. fieldPath.get("value").insertAfter(expr);
  33. }
  34. } else {
  35. fieldPath.set("value", t.unaryExpression("void", expr));
  36. }
  37. }
  38. function injectInitialization(classPath, init) {
  39. let firstFieldPath;
  40. let consturctorPath;
  41. for (const el of classPath.get("body.body")) {
  42. if ((el.isClassProperty() || el.isClassPrivateProperty()) && !el.node.static) {
  43. firstFieldPath = el;
  44. break;
  45. }
  46. if (!consturctorPath && el.isClassMethod({
  47. kind: "constructor"
  48. })) {
  49. consturctorPath = el;
  50. }
  51. }
  52. if (firstFieldPath) {
  53. injectToFieldInit(firstFieldPath, init, true);
  54. } else {
  55. (0, _helperCreateClassFeaturesPlugin.injectInitialization)(classPath, consturctorPath, [t.expressionStatement(init)]);
  56. }
  57. }
  58. function getWeakSetId(weakSets, outerClass, reference, name = "", inject) {
  59. let id = classWeakSets.get(reference.node);
  60. if (!id) {
  61. id = outerClass.scope.generateUidIdentifier(`${name || ""} brandCheck`);
  62. classWeakSets.set(reference.node, id);
  63. inject(reference, template.expression.ast`${t.cloneNode(id)}.add(this)`);
  64. const newExpr = t.newExpression(t.identifier("WeakSet"), []);
  65. (0, _helperAnnotateAsPure.default)(newExpr);
  66. outerClass.insertBefore(template.ast`var ${id} = ${newExpr}`);
  67. }
  68. return t.cloneNode(id);
  69. }
  70. return {
  71. name: "proposal-private-property-in-object",
  72. inherits: _pluginSyntaxPrivatePropertyInObject.default,
  73. pre() {
  74. (0, _helperCreateClassFeaturesPlugin.enableFeature)(this.file, _helperCreateClassFeaturesPlugin.FEATURES.privateIn, loose);
  75. },
  76. visitor: {
  77. BinaryExpression(path) {
  78. const {
  79. node
  80. } = path;
  81. if (node.operator !== "in") return;
  82. if (!t.isPrivateName(node.left)) return;
  83. const {
  84. name
  85. } = node.left.id;
  86. let privateElement;
  87. const outerClass = path.findParent(path => {
  88. if (!path.isClass()) return false;
  89. privateElement = path.get("body.body").find(({
  90. node
  91. }) => t.isPrivate(node) && node.key.id.name === name);
  92. return !!privateElement;
  93. });
  94. if (outerClass.parentPath.scope.path.isPattern()) {
  95. outerClass.replaceWith(template.ast`(() => ${outerClass.node})()`);
  96. return;
  97. }
  98. if (privateElement.isMethod()) {
  99. if (privateElement.node.static) {
  100. if (outerClass.node.id) {
  101. unshadow(outerClass.node.id.name, outerClass.scope, path.scope);
  102. } else {
  103. outerClass.set("id", path.scope.generateUidIdentifier("class"));
  104. }
  105. path.replaceWith(template.expression.ast`
  106. ${t.cloneNode(outerClass.node.id)} === ${path.node.right}
  107. `);
  108. } else {
  109. var _outerClass$node$id;
  110. const id = getWeakSetId(classWeakSets, outerClass, outerClass, (_outerClass$node$id = outerClass.node.id) == null ? void 0 : _outerClass$node$id.name, injectInitialization);
  111. path.replaceWith(template.expression.ast`${id}.has(${path.node.right})`);
  112. }
  113. } else {
  114. const id = getWeakSetId(fieldsWeakSets, outerClass, privateElement, privateElement.node.key.id.name, injectToFieldInit);
  115. path.replaceWith(template.expression.ast`${id}.has(${path.node.right})`);
  116. }
  117. }
  118. }
  119. };
  120. });
  121. exports.default = _default;