| 1 | "use strict";
|
|---|
| 2 |
|
|---|
| 3 | Object.defineProperty(exports, "__esModule", {
|
|---|
| 4 | value: true
|
|---|
| 5 | });
|
|---|
| 6 | exports.default = void 0;
|
|---|
| 7 | var _helperPluginUtils = require("@babel/helper-plugin-utils");
|
|---|
| 8 | var _helperCreateClassFeaturesPlugin = require("@babel/helper-create-class-features-plugin");
|
|---|
| 9 | var _helperAnnotateAsPure = require("@babel/helper-annotate-as-pure");
|
|---|
| 10 | var _default = exports.default = (0, _helperPluginUtils.declare)((api, opt) => {
|
|---|
| 11 | api.assertVersion("^7.0.0-0 || ^8.0.0-0 || >8.0.0-alpha <8.0.0-beta");
|
|---|
| 12 | const {
|
|---|
| 13 | types: t,
|
|---|
| 14 | template
|
|---|
| 15 | } = api;
|
|---|
| 16 | const {
|
|---|
| 17 | loose
|
|---|
| 18 | } = opt;
|
|---|
| 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 | const value = fieldPath.get("value");
|
|---|
| 30 | if (before) {
|
|---|
| 31 | value.insertBefore(expr);
|
|---|
| 32 | } else {
|
|---|
| 33 | value.insertAfter(expr);
|
|---|
| 34 | }
|
|---|
| 35 | } else {
|
|---|
| 36 | fieldPath.set("value", t.unaryExpression("void", expr));
|
|---|
| 37 | }
|
|---|
| 38 | }
|
|---|
| 39 | function injectInitialization(classPath, init) {
|
|---|
| 40 | let firstFieldPath;
|
|---|
| 41 | let constructorPath;
|
|---|
| 42 | for (const el of classPath.get("body.body")) {
|
|---|
| 43 | if ((el.isClassProperty() || el.isClassPrivateProperty()) && !el.node.static) {
|
|---|
| 44 | firstFieldPath = el;
|
|---|
| 45 | break;
|
|---|
| 46 | }
|
|---|
| 47 | if (!constructorPath && el.isClassMethod({
|
|---|
| 48 | kind: "constructor"
|
|---|
| 49 | })) {
|
|---|
| 50 | constructorPath = el;
|
|---|
| 51 | }
|
|---|
| 52 | }
|
|---|
| 53 | if (firstFieldPath) {
|
|---|
| 54 | injectToFieldInit(firstFieldPath, init, true);
|
|---|
| 55 | } else {
|
|---|
| 56 | (0, _helperCreateClassFeaturesPlugin.injectInitialization)(classPath, constructorPath, [t.expressionStatement(init)]);
|
|---|
| 57 | }
|
|---|
| 58 | }
|
|---|
| 59 | function getWeakSetId(weakSets, outerClass, reference, name = "", inject) {
|
|---|
| 60 | let id = weakSets.get(reference.node);
|
|---|
| 61 | if (!id) {
|
|---|
| 62 | id = outerClass.scope.generateUidIdentifier(`${name || ""} brandCheck`);
|
|---|
| 63 | weakSets.set(reference.node, id);
|
|---|
| 64 | inject(reference, template.expression.ast`${t.cloneNode(id)}.add(this)`);
|
|---|
| 65 | const newExpr = t.newExpression(t.identifier("WeakSet"), []);
|
|---|
| 66 | (0, _helperAnnotateAsPure.default)(newExpr);
|
|---|
| 67 | outerClass.insertBefore(template.ast`var ${id} = ${newExpr}`);
|
|---|
| 68 | }
|
|---|
| 69 | return t.cloneNode(id);
|
|---|
| 70 | }
|
|---|
| 71 | return {
|
|---|
| 72 | name: "transform-private-property-in-object",
|
|---|
| 73 | manipulateOptions: (_, parser) => parser.plugins.push("privateIn"),
|
|---|
| 74 | pre() {
|
|---|
| 75 | (0, _helperCreateClassFeaturesPlugin.enableFeature)(this.file, _helperCreateClassFeaturesPlugin.FEATURES.privateIn, loose);
|
|---|
| 76 | },
|
|---|
| 77 | visitor: {
|
|---|
| 78 | BinaryExpression(path, state) {
|
|---|
| 79 | const {
|
|---|
| 80 | node
|
|---|
| 81 | } = path;
|
|---|
| 82 | const {
|
|---|
| 83 | file
|
|---|
| 84 | } = state;
|
|---|
| 85 | if (node.operator !== "in") return;
|
|---|
| 86 | if (!t.isPrivateName(node.left)) return;
|
|---|
| 87 | const {
|
|---|
| 88 | name
|
|---|
| 89 | } = node.left.id;
|
|---|
| 90 | let privateElement;
|
|---|
| 91 | const outerClass = path.findParent(path => {
|
|---|
| 92 | if (!path.isClass()) return false;
|
|---|
| 93 | privateElement = path.get("body.body").find(({
|
|---|
| 94 | node
|
|---|
| 95 | }) => t.isPrivate(node) && node.key.id.name === name);
|
|---|
| 96 | return !!privateElement;
|
|---|
| 97 | });
|
|---|
| 98 | if (outerClass.parentPath.scope.path.isPattern()) {
|
|---|
| 99 | outerClass.replaceWith(template.ast`(() => ${outerClass.node})()`);
|
|---|
| 100 | return;
|
|---|
| 101 | }
|
|---|
| 102 | if (privateElement.node.type === "ClassPrivateMethod") {
|
|---|
| 103 | if (privateElement.node.static) {
|
|---|
| 104 | if (outerClass.node.id) {
|
|---|
| 105 | unshadow(outerClass.node.id.name, outerClass.scope, path.scope);
|
|---|
| 106 | } else {
|
|---|
| 107 | outerClass.set("id", path.scope.generateUidIdentifier("class"));
|
|---|
| 108 | }
|
|---|
| 109 | path.replaceWith(template.expression.ast`
|
|---|
| 110 | ${t.cloneNode(outerClass.node.id)} === ${(0, _helperCreateClassFeaturesPlugin.buildCheckInRHS)(node.right, file)}
|
|---|
| 111 | `);
|
|---|
| 112 | } else {
|
|---|
| 113 | var _outerClass$node$id;
|
|---|
| 114 | const id = getWeakSetId(classWeakSets, outerClass, outerClass, (_outerClass$node$id = outerClass.node.id) == null ? void 0 : _outerClass$node$id.name, injectInitialization);
|
|---|
| 115 | path.replaceWith(template.expression.ast`${id}.has(${(0, _helperCreateClassFeaturesPlugin.buildCheckInRHS)(node.right, file)})`);
|
|---|
| 116 | }
|
|---|
| 117 | } else {
|
|---|
| 118 | const id = getWeakSetId(fieldsWeakSets, outerClass, privateElement, privateElement.node.key.id.name, injectToFieldInit);
|
|---|
| 119 | path.replaceWith(template.expression.ast`${id}.has(${(0, _helperCreateClassFeaturesPlugin.buildCheckInRHS)(node.right, file)})`);
|
|---|
| 120 | }
|
|---|
| 121 | }
|
|---|
| 122 | }
|
|---|
| 123 | };
|
|---|
| 124 | });
|
|---|
| 125 |
|
|---|
| 126 | //# sourceMappingURL=index.js.map
|
|---|