|
- "use strict";
-
- Object.defineProperty(exports, "__esModule", {
- value: true
- });
- exports.buildPrivateNamesMap = buildPrivateNamesMap;
- exports.buildPrivateNamesNodes = buildPrivateNamesNodes;
- exports.transformPrivateNamesUsage = transformPrivateNamesUsage;
- exports.buildFieldsInitNodes = buildFieldsInitNodes;
-
- var _core = require("@babel/core");
-
- var _helperReplaceSupers = _interopRequireWildcard(require("@babel/helper-replace-supers"));
-
- var _helperMemberExpressionToFunctions = _interopRequireDefault(require("@babel/helper-member-expression-to-functions"));
-
- var _helperOptimiseCallExpression = _interopRequireDefault(require("@babel/helper-optimise-call-expression"));
-
- var ts = _interopRequireWildcard(require("./typescript"));
-
- function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
-
- function _getRequireWildcardCache() { if (typeof WeakMap !== "function") return null; var cache = new WeakMap(); _getRequireWildcardCache = function () { return cache; }; return cache; }
-
- function _interopRequireWildcard(obj) { if (obj && obj.__esModule) { return obj; } if (obj === null || typeof obj !== "object" && typeof obj !== "function") { return { default: obj }; } var cache = _getRequireWildcardCache(); if (cache && cache.has(obj)) { return cache.get(obj); } var newObj = {}; var hasPropertyDescriptor = Object.defineProperty && Object.getOwnPropertyDescriptor; for (var key in obj) { if (Object.prototype.hasOwnProperty.call(obj, key)) { var desc = hasPropertyDescriptor ? Object.getOwnPropertyDescriptor(obj, key) : null; if (desc && (desc.get || desc.set)) { Object.defineProperty(newObj, key, desc); } else { newObj[key] = obj[key]; } } } newObj.default = obj; if (cache) { cache.set(obj, newObj); } return newObj; }
-
- function buildPrivateNamesMap(props) {
- const privateNamesMap = new Map();
-
- for (const prop of props) {
- const isPrivate = prop.isPrivate();
- const isMethod = !prop.isProperty();
- const isInstance = !prop.node.static;
-
- if (isPrivate) {
- const {
- name
- } = prop.node.key.id;
- const update = privateNamesMap.has(name) ? privateNamesMap.get(name) : {
- id: prop.scope.generateUidIdentifier(name),
- static: !isInstance,
- method: isMethod
- };
-
- if (prop.node.kind === "get") {
- update.getId = prop.scope.generateUidIdentifier(`get_${name}`);
- } else if (prop.node.kind === "set") {
- update.setId = prop.scope.generateUidIdentifier(`set_${name}`);
- } else if (prop.node.kind === "method") {
- update.methodId = prop.scope.generateUidIdentifier(name);
- }
-
- privateNamesMap.set(name, update);
- }
- }
-
- return privateNamesMap;
- }
-
- function buildPrivateNamesNodes(privateNamesMap, loose, state) {
- const initNodes = [];
-
- for (const [name, value] of privateNamesMap) {
- const {
- id,
- static: isStatic,
- method: isMethod,
- getId,
- setId
- } = value;
- const isAccessor = getId || setId;
-
- if (loose) {
- initNodes.push(_core.template.statement.ast`
- var ${id} = ${state.addHelper("classPrivateFieldLooseKey")}("${name}")
- `);
- } else if (isMethod && !isStatic) {
- if (isAccessor) {
- initNodes.push(_core.template.statement.ast`var ${id} = new WeakMap();`);
- } else {
- initNodes.push(_core.template.statement.ast`var ${id} = new WeakSet();`);
- }
- } else if (!isStatic) {
- initNodes.push(_core.template.statement.ast`var ${id} = new WeakMap();`);
- }
- }
-
- return initNodes;
- }
-
- const privateNameVisitor = {
- PrivateName(path) {
- const {
- privateNamesMap
- } = this;
- const {
- node,
- parentPath
- } = path;
- if (!parentPath.isMemberExpression({
- property: node
- })) return;
- if (!privateNamesMap.has(node.id.name)) return;
- this.handle(parentPath);
- },
-
- Class(path) {
- const {
- privateNamesMap
- } = this;
- const body = path.get("body.body");
- const visiblePrivateNames = new Map(privateNamesMap);
- const redeclared = [];
-
- for (const prop of body) {
- if (!prop.isPrivate()) continue;
- const {
- name
- } = prop.node.key.id;
- visiblePrivateNames.delete(name);
- redeclared.push(name);
- }
-
- if (!redeclared.length) {
- return;
- }
-
- path.get("body").traverse(privateNameNestedVisitor, Object.assign({}, this, {
- redeclared
- }));
- path.traverse(privateNameVisitor, Object.assign({}, this, {
- privateNamesMap: visiblePrivateNames
- }));
- path.skipKey("body");
- }
-
- };
-
- const privateNameNestedVisitor = _core.traverse.visitors.merge([{
- PrivateName(path) {
- const {
- redeclared
- } = this;
- const {
- name
- } = path.node.id;
- if (redeclared.includes(name)) path.skip();
- }
-
- }, {
- PrivateName: privateNameVisitor.PrivateName
- }, _helperReplaceSupers.environmentVisitor]);
-
- const privateNameHandlerSpec = {
- memoise(member, count) {
- const {
- scope
- } = member;
- const {
- object
- } = member.node;
- const memo = scope.maybeGenerateMemoised(object);
-
- if (!memo) {
- return;
- }
-
- this.memoiser.set(object, memo, count);
- },
-
- receiver(member) {
- const {
- object
- } = member.node;
-
- if (this.memoiser.has(object)) {
- return _core.types.cloneNode(this.memoiser.get(object));
- }
-
- return _core.types.cloneNode(object);
- },
-
- get(member) {
- const {
- classRef,
- privateNamesMap,
- file
- } = this;
- const {
- name
- } = member.node.property.id;
- const {
- id,
- static: isStatic,
- method: isMethod,
- methodId,
- getId,
- setId
- } = privateNamesMap.get(name);
- const isAccessor = getId || setId;
-
- if (isStatic) {
- const helperName = isMethod && !isAccessor ? "classStaticPrivateMethodGet" : "classStaticPrivateFieldSpecGet";
- return _core.types.callExpression(file.addHelper(helperName), [this.receiver(member), _core.types.cloneNode(classRef), _core.types.cloneNode(id)]);
- }
-
- if (isMethod) {
- if (isAccessor) {
- return _core.types.callExpression(file.addHelper("classPrivateFieldGet"), [this.receiver(member), _core.types.cloneNode(id)]);
- }
-
- return _core.types.callExpression(file.addHelper("classPrivateMethodGet"), [this.receiver(member), _core.types.cloneNode(id), _core.types.cloneNode(methodId)]);
- }
-
- return _core.types.callExpression(file.addHelper("classPrivateFieldGet"), [this.receiver(member), _core.types.cloneNode(id)]);
- },
-
- set(member, value) {
- const {
- classRef,
- privateNamesMap,
- file
- } = this;
- const {
- name
- } = member.node.property.id;
- const {
- id,
- static: isStatic,
- method: isMethod,
- setId,
- getId
- } = privateNamesMap.get(name);
- const isAccessor = getId || setId;
-
- if (isStatic) {
- const helperName = isMethod && !isAccessor ? "classStaticPrivateMethodSet" : "classStaticPrivateFieldSpecSet";
- return _core.types.callExpression(file.addHelper(helperName), [this.receiver(member), _core.types.cloneNode(classRef), _core.types.cloneNode(id), value]);
- }
-
- if (isMethod) {
- if (setId) {
- return _core.types.callExpression(file.addHelper("classPrivateFieldSet"), [this.receiver(member), _core.types.cloneNode(id), value]);
- }
-
- return _core.types.callExpression(file.addHelper("classPrivateMethodSet"), []);
- }
-
- return _core.types.callExpression(file.addHelper("classPrivateFieldSet"), [this.receiver(member), _core.types.cloneNode(id), value]);
- },
-
- destructureSet(member) {
- const {
- privateNamesMap,
- file
- } = this;
- const {
- name
- } = member.node.property.id;
- const {
- id
- } = privateNamesMap.get(name);
- return _core.types.memberExpression(_core.types.callExpression(file.addHelper("classPrivateFieldDestructureSet"), [this.receiver(member), _core.types.cloneNode(id)]), _core.types.identifier("value"));
- },
-
- call(member, args) {
- this.memoise(member, 1);
- return (0, _helperOptimiseCallExpression.default)(this.get(member), this.receiver(member), args);
- }
-
- };
- const privateNameHandlerLoose = {
- handle(member) {
- const {
- privateNamesMap,
- file
- } = this;
- const {
- object
- } = member.node;
- const {
- name
- } = member.node.property.id;
- member.replaceWith(_core.template.expression`BASE(REF, PROP)[PROP]`({
- BASE: file.addHelper("classPrivateFieldLooseBase"),
- REF: object,
- PROP: privateNamesMap.get(name).id
- }));
- }
-
- };
-
- function transformPrivateNamesUsage(ref, path, privateNamesMap, loose, state) {
- if (!privateNamesMap.size) return;
- const body = path.get("body");
-
- if (loose) {
- body.traverse(privateNameVisitor, Object.assign({
- privateNamesMap,
- file: state
- }, privateNameHandlerLoose));
- } else {
- (0, _helperMemberExpressionToFunctions.default)(body, privateNameVisitor, Object.assign({
- privateNamesMap,
- classRef: ref,
- file: state
- }, privateNameHandlerSpec));
- }
- }
-
- function buildPrivateFieldInitLoose(ref, prop, privateNamesMap) {
- const {
- id
- } = privateNamesMap.get(prop.node.key.id.name);
- const value = prop.node.value || prop.scope.buildUndefinedNode();
- return _core.template.statement.ast`
- Object.defineProperty(${ref}, ${id}, {
- // configurable is false by default
- // enumerable is false by default
- writable: true,
- value: ${value}
- });
- `;
- }
-
- function buildPrivateInstanceFieldInitSpec(ref, prop, privateNamesMap) {
- const {
- id
- } = privateNamesMap.get(prop.node.key.id.name);
- const value = prop.node.value || prop.scope.buildUndefinedNode();
- return _core.template.statement.ast`${id}.set(${ref}, {
- // configurable is always false for private elements
- // enumerable is always false for private elements
- writable: true,
- value: ${value},
- })`;
- }
-
- function buildPrivateStaticFieldInitSpec(prop, privateNamesMap) {
- const privateName = privateNamesMap.get(prop.node.key.id.name);
- const {
- id,
- getId,
- setId,
- initAdded
- } = privateName;
- const isAccessor = getId || setId;
- if (!prop.isProperty() && (initAdded || !isAccessor)) return;
-
- if (isAccessor) {
- privateNamesMap.set(prop.node.key.id.name, Object.assign({}, privateName, {
- initAdded: true
- }));
- return _core.template.statement.ast`
- var ${id.name} = {
- // configurable is false by default
- // enumerable is false by default
- // writable is false by default
- get: ${getId ? getId.name : prop.scope.buildUndefinedNode()},
- set: ${setId ? setId.name : prop.scope.buildUndefinedNode()}
- }
- `;
- }
-
- const value = prop.node.value || prop.scope.buildUndefinedNode();
- return _core.template.statement.ast`
- var ${id} = {
- // configurable is false by default
- // enumerable is false by default
- writable: true,
- value: ${value}
- };
- `;
- }
-
- function buildPrivateMethodInitLoose(ref, prop, privateNamesMap) {
- const privateName = privateNamesMap.get(prop.node.key.id.name);
- const {
- methodId,
- id,
- getId,
- setId,
- initAdded
- } = privateName;
- if (initAdded) return;
-
- if (methodId) {
- return _core.template.statement.ast`
- Object.defineProperty(${ref}, ${id}, {
- // configurable is false by default
- // enumerable is false by default
- // writable is false by default
- value: ${methodId.name}
- });
- `;
- }
-
- const isAccessor = getId || setId;
-
- if (isAccessor) {
- privateNamesMap.set(prop.node.key.id.name, Object.assign({}, privateName, {
- initAdded: true
- }));
- return _core.template.statement.ast`
- Object.defineProperty(${ref}, ${id}, {
- // configurable is false by default
- // enumerable is false by default
- // writable is false by default
- get: ${getId ? getId.name : prop.scope.buildUndefinedNode()},
- set: ${setId ? setId.name : prop.scope.buildUndefinedNode()}
- });
- `;
- }
- }
-
- function buildPrivateInstanceMethodInitSpec(ref, prop, privateNamesMap) {
- const privateName = privateNamesMap.get(prop.node.key.id.name);
- const {
- id,
- getId,
- setId,
- initAdded
- } = privateName;
- if (initAdded) return;
- const isAccessor = getId || setId;
-
- if (isAccessor) {
- privateNamesMap.set(prop.node.key.id.name, Object.assign({}, privateName, {
- initAdded: true
- }));
- return _core.template.statement.ast`
- ${id}.set(${ref}, {
- get: ${getId ? getId.name : prop.scope.buildUndefinedNode()},
- set: ${setId ? setId.name : prop.scope.buildUndefinedNode()}
- });
- `;
- }
-
- return _core.template.statement.ast`${id}.add(${ref})`;
- }
-
- function buildPublicFieldInitLoose(ref, prop) {
- const {
- key,
- computed
- } = prop.node;
- const value = prop.node.value || prop.scope.buildUndefinedNode();
- return _core.types.expressionStatement(_core.types.assignmentExpression("=", _core.types.memberExpression(ref, key, computed || _core.types.isLiteral(key)), value));
- }
-
- function buildPublicFieldInitSpec(ref, prop, state) {
- const {
- key,
- computed
- } = prop.node;
- const value = prop.node.value || prop.scope.buildUndefinedNode();
- return _core.types.expressionStatement(_core.types.callExpression(state.addHelper("defineProperty"), [ref, computed || _core.types.isLiteral(key) ? key : _core.types.stringLiteral(key.name), value]));
- }
-
- function buildPrivateStaticMethodInitLoose(ref, prop, state, privateNamesMap) {
- const privateName = privateNamesMap.get(prop.node.key.id.name);
- const {
- id,
- methodId,
- getId,
- setId,
- initAdded
- } = privateName;
- if (initAdded) return;
- const isAccessor = getId || setId;
-
- if (isAccessor) {
- privateNamesMap.set(prop.node.key.id.name, Object.assign({}, privateName, {
- initAdded: true
- }));
- return _core.template.statement.ast`
- Object.defineProperty(${ref}, ${id}, {
- // configurable is false by default
- // enumerable is false by default
- // writable is false by default
- get: ${getId ? getId.name : prop.scope.buildUndefinedNode()},
- set: ${setId ? setId.name : prop.scope.buildUndefinedNode()}
- })
- `;
- }
-
- return _core.template.statement.ast`
- Object.defineProperty(${ref}, ${id}, {
- // configurable is false by default
- // enumerable is false by default
- // writable is false by default
- value: ${methodId.name}
- });
- `;
- }
-
- function buildPrivateMethodDeclaration(prop, privateNamesMap, loose = false) {
- const privateName = privateNamesMap.get(prop.node.key.id.name);
- const {
- id,
- methodId,
- getId,
- setId,
- getterDeclared,
- setterDeclared,
- static: isStatic
- } = privateName;
- const {
- params,
- body,
- generator,
- async
- } = prop.node;
-
- const methodValue = _core.types.functionExpression(methodId, params, body, generator, async);
-
- const isGetter = getId && !getterDeclared && params.length === 0;
- const isSetter = setId && !setterDeclared && params.length > 0;
-
- if (isGetter) {
- privateNamesMap.set(prop.node.key.id.name, Object.assign({}, privateName, {
- getterDeclared: true
- }));
- return _core.types.variableDeclaration("var", [_core.types.variableDeclarator(getId, methodValue)]);
- }
-
- if (isSetter) {
- privateNamesMap.set(prop.node.key.id.name, Object.assign({}, privateName, {
- setterDeclared: true
- }));
- return _core.types.variableDeclaration("var", [_core.types.variableDeclarator(setId, methodValue)]);
- }
-
- if (isStatic && !loose) {
- return _core.types.variableDeclaration("var", [_core.types.variableDeclarator(id, _core.types.functionExpression(id, params, body, generator, async))]);
- }
-
- return _core.types.variableDeclaration("var", [_core.types.variableDeclarator(methodId, methodValue)]);
- }
-
- const thisContextVisitor = _core.traverse.visitors.merge([{
- ThisExpression(path, state) {
- state.needsClassRef = true;
- path.replaceWith(_core.types.cloneNode(state.classRef));
- }
-
- }, _helperReplaceSupers.environmentVisitor]);
-
- function replaceThisContext(path, ref, superRef, file, loose) {
- const state = {
- classRef: ref,
- needsClassRef: false
- };
- const replacer = new _helperReplaceSupers.default({
- methodPath: path,
- isLoose: loose,
- superRef,
- file,
-
- getObjectRef() {
- state.needsClassRef = true;
- return path.node.static ? ref : _core.types.memberExpression(ref, _core.types.identifier("prototype"));
- }
-
- });
- replacer.replace();
-
- if (path.isProperty()) {
- path.traverse(thisContextVisitor, state);
- }
-
- return state.needsClassRef;
- }
-
- function buildFieldsInitNodes(ref, superRef, props, privateNamesMap, state, loose) {
- const staticNodes = [];
- const instanceNodes = [];
- let needsClassRef = false;
-
- for (const prop of props) {
- ts.assertFieldTransformed(prop);
- const isStatic = prop.node.static;
- const isInstance = !isStatic;
- const isPrivate = prop.isPrivate();
- const isPublic = !isPrivate;
- const isField = prop.isProperty();
- const isMethod = !isField;
-
- if (isStatic || isMethod && isPrivate) {
- const replaced = replaceThisContext(prop, ref, superRef, state, loose);
- needsClassRef = needsClassRef || replaced;
- }
-
- switch (true) {
- case isStatic && isPrivate && isField && loose:
- needsClassRef = true;
- staticNodes.push(buildPrivateFieldInitLoose(_core.types.cloneNode(ref), prop, privateNamesMap));
- break;
-
- case isStatic && isPrivate && isField && !loose:
- needsClassRef = true;
- staticNodes.push(buildPrivateStaticFieldInitSpec(prop, privateNamesMap));
- break;
-
- case isStatic && isPublic && isField && loose:
- needsClassRef = true;
- staticNodes.push(buildPublicFieldInitLoose(_core.types.cloneNode(ref), prop));
- break;
-
- case isStatic && isPublic && isField && !loose:
- needsClassRef = true;
- staticNodes.push(buildPublicFieldInitSpec(_core.types.cloneNode(ref), prop, state));
- break;
-
- case isInstance && isPrivate && isField && loose:
- instanceNodes.push(buildPrivateFieldInitLoose(_core.types.thisExpression(), prop, privateNamesMap));
- break;
-
- case isInstance && isPrivate && isField && !loose:
- instanceNodes.push(buildPrivateInstanceFieldInitSpec(_core.types.thisExpression(), prop, privateNamesMap));
- break;
-
- case isInstance && isPrivate && isMethod && loose:
- instanceNodes.unshift(buildPrivateMethodInitLoose(_core.types.thisExpression(), prop, privateNamesMap));
- staticNodes.push(buildPrivateMethodDeclaration(prop, privateNamesMap, loose));
- break;
-
- case isInstance && isPrivate && isMethod && !loose:
- instanceNodes.unshift(buildPrivateInstanceMethodInitSpec(_core.types.thisExpression(), prop, privateNamesMap));
- staticNodes.push(buildPrivateMethodDeclaration(prop, privateNamesMap, loose));
- break;
-
- case isStatic && isPrivate && isMethod && !loose:
- needsClassRef = true;
- staticNodes.push(buildPrivateStaticFieldInitSpec(prop, privateNamesMap));
- staticNodes.unshift(buildPrivateMethodDeclaration(prop, privateNamesMap, loose));
- break;
-
- case isStatic && isPrivate && isMethod && loose:
- needsClassRef = true;
- staticNodes.push(buildPrivateStaticMethodInitLoose(_core.types.cloneNode(ref), prop, state, privateNamesMap));
- staticNodes.unshift(buildPrivateMethodDeclaration(prop, privateNamesMap, loose));
- break;
-
- case isInstance && isPublic && isField && loose:
- instanceNodes.push(buildPublicFieldInitLoose(_core.types.thisExpression(), prop));
- break;
-
- case isInstance && isPublic && isField && !loose:
- instanceNodes.push(buildPublicFieldInitSpec(_core.types.thisExpression(), prop, state));
- break;
-
- default:
- throw new Error("Unreachable.");
- }
- }
-
- return {
- staticNodes: staticNodes.filter(Boolean),
- instanceNodes: instanceNodes.filter(Boolean),
-
- wrapClass(path) {
- for (const prop of props) {
- prop.remove();
- }
-
- if (!needsClassRef) return path;
-
- if (path.isClassExpression()) {
- path.scope.push({
- id: ref
- });
- path.replaceWith(_core.types.assignmentExpression("=", _core.types.cloneNode(ref), path.node));
- } else if (!path.node.id) {
- path.node.id = ref;
- }
-
- return path;
- }
-
- };
- }
|