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-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
|
---|