123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397 |
- 'use strict';
- Object.defineProperty(exports, '__esModule', { value: true });
- var _t = require('@babel/types');
- function _interopNamespace(e) {
- if (e && e.__esModule) return e;
- var n = Object.create(null);
- if (e) {
- Object.keys(e).forEach(function (k) {
- if (k !== 'default') {
- var d = Object.getOwnPropertyDescriptor(e, k);
- Object.defineProperty(n, k, d.get ? d : {
- enumerable: true,
- get: function () { return e[k]; }
- });
- }
- });
- }
- n.default = e;
- return Object.freeze(n);
- }
- var _t__namespace = /*#__PURE__*/_interopNamespace(_t);
- function willPathCastToBoolean(path) {
- const maybeWrapped = path;
- const {
- node,
- parentPath
- } = maybeWrapped;
- if (parentPath.isLogicalExpression()) {
- const {
- operator,
- right
- } = parentPath.node;
- if (operator === "&&" || operator === "||" || operator === "??" && node === right) {
- return willPathCastToBoolean(parentPath);
- }
- }
- if (parentPath.isSequenceExpression()) {
- const {
- expressions
- } = parentPath.node;
- if (expressions[expressions.length - 1] === node) {
- return willPathCastToBoolean(parentPath);
- } else {
- return true;
- }
- }
- return parentPath.isConditional({
- test: node
- }) || parentPath.isUnaryExpression({
- operator: "!"
- }) || parentPath.isLoop({
- test: node
- });
- }
- const {
- LOGICAL_OPERATORS,
- arrowFunctionExpression,
- assignmentExpression,
- binaryExpression,
- booleanLiteral,
- callExpression,
- cloneNode,
- conditionalExpression,
- identifier,
- isMemberExpression,
- isOptionalCallExpression,
- isOptionalMemberExpression,
- isUpdateExpression,
- logicalExpression,
- memberExpression,
- nullLiteral,
- optionalCallExpression,
- optionalMemberExpression,
- sequenceExpression,
- updateExpression
- } = _t__namespace;
- class AssignmentMemoiser {
- constructor() {
- this._map = void 0;
- this._map = new WeakMap();
- }
- has(key) {
- return this._map.has(key);
- }
- get(key) {
- if (!this.has(key)) return;
- const record = this._map.get(key);
- const {
- value
- } = record;
- record.count--;
- if (record.count === 0) {
- return assignmentExpression("=", value, key);
- }
- return value;
- }
- set(key, value, count) {
- return this._map.set(key, {
- count,
- value
- });
- }
- }
- function toNonOptional(path, base) {
- const {
- node
- } = path;
- if (isOptionalMemberExpression(node)) {
- return memberExpression(base, node.property, node.computed);
- }
- if (path.isOptionalCallExpression()) {
- const callee = path.get("callee");
- if (path.node.optional && callee.isOptionalMemberExpression()) {
- const object = callee.node.object;
- const context = path.scope.maybeGenerateMemoised(object);
- callee.get("object").replaceWith(assignmentExpression("=", context, object));
- return callExpression(memberExpression(base, identifier("call")), [context, ...path.node.arguments]);
- }
- return callExpression(base, path.node.arguments);
- }
- return path.node;
- }
- function isInDetachedTree(path) {
- while (path) {
- if (path.isProgram()) break;
- const {
- parentPath,
- container,
- listKey
- } = path;
- const parentNode = parentPath.node;
- if (listKey) {
- if (container !== parentNode[listKey]) {
- return true;
- }
- } else {
- if (container !== parentNode) return true;
- }
- path = parentPath;
- }
- return false;
- }
- const handle = {
- memoise() {},
- handle(member, noDocumentAll) {
- const {
- node,
- parent,
- parentPath,
- scope
- } = member;
- if (member.isOptionalMemberExpression()) {
- if (isInDetachedTree(member)) return;
- const endPath = member.find(({
- node,
- parent
- }) => {
- if (isOptionalMemberExpression(parent)) {
- return parent.optional || parent.object !== node;
- }
- if (isOptionalCallExpression(parent)) {
- return node !== member.node && parent.optional || parent.callee !== node;
- }
- return true;
- });
- if (scope.path.isPattern()) {
- endPath.replaceWith(callExpression(arrowFunctionExpression([], endPath.node), []));
- return;
- }
- const willEndPathCastToBoolean = willPathCastToBoolean(endPath);
- const rootParentPath = endPath.parentPath;
- if (rootParentPath.isUpdateExpression({
- argument: node
- })) {
- throw member.buildCodeFrameError(`can't handle update expression`);
- }
- const isAssignment = rootParentPath.isAssignmentExpression({
- left: endPath.node
- });
- const isDeleteOperation = rootParentPath.isUnaryExpression({
- operator: "delete"
- });
- if (isDeleteOperation && endPath.isOptionalMemberExpression() && endPath.get("property").isPrivateName()) {
- throw member.buildCodeFrameError(`can't delete a private class element`);
- }
- let startingOptional = member;
- for (;;) {
- if (startingOptional.isOptionalMemberExpression()) {
- if (startingOptional.node.optional) break;
- startingOptional = startingOptional.get("object");
- continue;
- } else if (startingOptional.isOptionalCallExpression()) {
- if (startingOptional.node.optional) break;
- startingOptional = startingOptional.get("callee");
- continue;
- }
- throw new Error(`Internal error: unexpected ${startingOptional.node.type}`);
- }
- const startingNode = startingOptional.isOptionalMemberExpression() ? startingOptional.node.object : startingOptional.node.callee;
- const baseNeedsMemoised = scope.maybeGenerateMemoised(startingNode);
- const baseRef = baseNeedsMemoised != null ? baseNeedsMemoised : startingNode;
- const parentIsOptionalCall = parentPath.isOptionalCallExpression({
- callee: node
- });
- const isOptionalCall = parent => parentIsOptionalCall;
- const parentIsCall = parentPath.isCallExpression({
- callee: node
- });
- startingOptional.replaceWith(toNonOptional(startingOptional, baseRef));
- if (isOptionalCall()) {
- if (parent.optional) {
- parentPath.replaceWith(this.optionalCall(member, parent.arguments));
- } else {
- parentPath.replaceWith(this.call(member, parent.arguments));
- }
- } else if (parentIsCall) {
- member.replaceWith(this.boundGet(member));
- } else if (this.delete && parentPath.isUnaryExpression({
- operator: "delete"
- })) {
- parentPath.replaceWith(this.delete(member));
- } else if (parentPath.isAssignmentExpression()) {
- handleAssignment(this, member, parentPath);
- } else {
- member.replaceWith(this.get(member));
- }
- let regular = member.node;
- for (let current = member; current !== endPath;) {
- const parentPath = current.parentPath;
- if (parentPath === endPath && isOptionalCall() && parent.optional) {
- regular = parentPath.node;
- break;
- }
- regular = toNonOptional(parentPath, regular);
- current = parentPath;
- }
- let context;
- const endParentPath = endPath.parentPath;
- if (isMemberExpression(regular) && endParentPath.isOptionalCallExpression({
- callee: endPath.node,
- optional: true
- })) {
- const {
- object
- } = regular;
- context = member.scope.maybeGenerateMemoised(object);
- if (context) {
- regular.object = assignmentExpression("=", context, object);
- }
- }
- let replacementPath = endPath;
- if (isDeleteOperation || isAssignment) {
- replacementPath = endParentPath;
- regular = endParentPath.node;
- }
- const baseMemoised = baseNeedsMemoised ? assignmentExpression("=", cloneNode(baseRef), cloneNode(startingNode)) : cloneNode(baseRef);
- if (willEndPathCastToBoolean) {
- let nonNullishCheck;
- if (noDocumentAll) {
- nonNullishCheck = binaryExpression("!=", baseMemoised, nullLiteral());
- } else {
- nonNullishCheck = logicalExpression("&&", binaryExpression("!==", baseMemoised, nullLiteral()), binaryExpression("!==", cloneNode(baseRef), scope.buildUndefinedNode()));
- }
- replacementPath.replaceWith(logicalExpression("&&", nonNullishCheck, regular));
- } else {
- let nullishCheck;
- if (noDocumentAll) {
- nullishCheck = binaryExpression("==", baseMemoised, nullLiteral());
- } else {
- nullishCheck = logicalExpression("||", binaryExpression("===", baseMemoised, nullLiteral()), binaryExpression("===", cloneNode(baseRef), scope.buildUndefinedNode()));
- }
- replacementPath.replaceWith(conditionalExpression(nullishCheck, isDeleteOperation ? booleanLiteral(true) : scope.buildUndefinedNode(), regular));
- }
- if (context) {
- const endParent = endParentPath.node;
- endParentPath.replaceWith(optionalCallExpression(optionalMemberExpression(endParent.callee, identifier("call"), false, true), [cloneNode(context), ...endParent.arguments], false));
- }
- return;
- }
- if (isUpdateExpression(parent, {
- argument: node
- })) {
- if (this.simpleSet) {
- member.replaceWith(this.simpleSet(member));
- return;
- }
- const {
- operator,
- prefix
- } = parent;
- this.memoise(member, 2);
- const ref = scope.generateUidIdentifierBasedOnNode(node);
- scope.push({
- id: ref
- });
- const seq = [assignmentExpression("=", cloneNode(ref), this.get(member))];
- if (prefix) {
- seq.push(updateExpression(operator, cloneNode(ref), prefix));
- const value = sequenceExpression(seq);
- parentPath.replaceWith(this.set(member, value));
- return;
- } else {
- const ref2 = scope.generateUidIdentifierBasedOnNode(node);
- scope.push({
- id: ref2
- });
- seq.push(assignmentExpression("=", cloneNode(ref2), updateExpression(operator, cloneNode(ref), prefix)), cloneNode(ref));
- const value = sequenceExpression(seq);
- parentPath.replaceWith(sequenceExpression([this.set(member, value), cloneNode(ref2)]));
- return;
- }
- }
- if (parentPath.isAssignmentExpression({
- left: node
- })) {
- handleAssignment(this, member, parentPath);
- return;
- }
- if (parentPath.isCallExpression({
- callee: node
- })) {
- parentPath.replaceWith(this.call(member, parentPath.node.arguments));
- return;
- }
- if (parentPath.isOptionalCallExpression({
- callee: node
- })) {
- if (scope.path.isPattern()) {
- parentPath.replaceWith(callExpression(arrowFunctionExpression([], parentPath.node), []));
- return;
- }
- parentPath.replaceWith(this.optionalCall(member, parentPath.node.arguments));
- return;
- }
- if (this.delete && parentPath.isUnaryExpression({
- operator: "delete"
- })) {
- parentPath.replaceWith(this.delete(member));
- return;
- }
- if (parentPath.isForXStatement({
- left: node
- }) || parentPath.isObjectProperty({
- value: node
- }) && parentPath.parentPath.isObjectPattern() || parentPath.isAssignmentPattern({
- left: node
- }) && parentPath.parentPath.isObjectProperty({
- value: parent
- }) && parentPath.parentPath.parentPath.isObjectPattern() || parentPath.isArrayPattern() || parentPath.isAssignmentPattern({
- left: node
- }) && parentPath.parentPath.isArrayPattern() || parentPath.isRestElement()) {
- member.replaceWith(this.destructureSet(member));
- return;
- }
- if (parentPath.isTaggedTemplateExpression()) {
- member.replaceWith(this.boundGet(member));
- } else {
- member.replaceWith(this.get(member));
- }
- }
- };
- function handleAssignment(state, member, parentPath) {
- if (state.simpleSet) {
- member.replaceWith(state.simpleSet(member));
- return;
- }
- const {
- operator,
- right: value
- } = parentPath.node;
- if (operator === "=") {
- parentPath.replaceWith(state.set(member, value));
- } else {
- const operatorTrunc = operator.slice(0, -1);
- if (LOGICAL_OPERATORS.includes(operatorTrunc)) {
- state.memoise(member, 1);
- parentPath.replaceWith(logicalExpression(operatorTrunc, state.get(member), state.set(member, value)));
- } else {
- state.memoise(member, 2);
- parentPath.replaceWith(state.set(member, binaryExpression(operatorTrunc, state.get(member), value)));
- }
- }
- }
- function memberExpressionToFunctions(path, visitor, state) {
- path.traverse(visitor, Object.assign({}, handle, state, {
- memoiser: new AssignmentMemoiser()
- }));
- }
- exports.default = memberExpressionToFunctions;
- //# sourceMappingURL=index.js.map
|