123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950 |
- "use strict";
- Object.defineProperty(exports, "__esModule", {
- value: true
- });
- exports.default = _default;
- var _core = require("@babel/core");
- var _helperReplaceSupers = require("@babel/helper-replace-supers");
- var _helperSplitExportDeclaration = require("@babel/helper-split-export-declaration");
- var _helperSkipTransparentExpressionWrappers = require("@babel/helper-skip-transparent-expression-wrappers");
- var _fields = require("./fields.js");
- function incrementId(id, idx = id.length - 1) {
- if (idx === -1) {
- id.unshift(65);
- return;
- }
- const current = id[idx];
- if (current === 90) {
- id[idx] = 97;
- } else if (current === 122) {
- id[idx] = 65;
- incrementId(id, idx - 1);
- } else {
- id[idx] = current + 1;
- }
- }
- function createPrivateUidGeneratorForClass(classPath) {
- const currentPrivateId = [];
- const privateNames = new Set();
- classPath.traverse({
- PrivateName(path) {
- privateNames.add(path.node.id.name);
- }
- });
- return () => {
- let reifiedId;
- do {
- incrementId(currentPrivateId);
- reifiedId = String.fromCharCode(...currentPrivateId);
- } while (privateNames.has(reifiedId));
- return _core.types.privateName(_core.types.identifier(reifiedId));
- };
- }
- function createLazyPrivateUidGeneratorForClass(classPath) {
- let generator;
- return () => {
- if (!generator) {
- generator = createPrivateUidGeneratorForClass(classPath);
- }
- return generator();
- };
- }
- function replaceClassWithVar(path, className) {
- if (path.type === "ClassDeclaration") {
- const id = path.node.id;
- const className = id.name;
- const varId = path.scope.generateUidIdentifierBasedOnNode(id);
- const classId = _core.types.identifier(className);
- path.scope.rename(className, varId.name);
- path.get("id").replaceWith(classId);
- return {
- id: _core.types.cloneNode(varId),
- path
- };
- } else {
- let varId;
- if (path.node.id) {
- className = path.node.id.name;
- varId = path.scope.parent.generateDeclaredUidIdentifier(className);
- path.scope.rename(className, varId.name);
- } else {
- varId = path.scope.parent.generateDeclaredUidIdentifier(typeof className === "string" ? className : "decorated_class");
- }
- const newClassExpr = _core.types.classExpression(typeof className === "string" ? _core.types.identifier(className) : null, path.node.superClass, path.node.body);
- const [newPath] = path.replaceWith(_core.types.sequenceExpression([newClassExpr, varId]));
- return {
- id: _core.types.cloneNode(varId),
- path: newPath.get("expressions.0")
- };
- }
- }
- function generateClassProperty(key, value, isStatic) {
- if (key.type === "PrivateName") {
- return _core.types.classPrivateProperty(key, value, undefined, isStatic);
- } else {
- return _core.types.classProperty(key, value, undefined, undefined, isStatic);
- }
- }
- function addProxyAccessorsFor(className, element, originalKey, targetKey, version, isComputed = false) {
- const {
- static: isStatic
- } = element.node;
- const thisArg = version === "2023-05" && isStatic ? className : _core.types.thisExpression();
- const getterBody = _core.types.blockStatement([_core.types.returnStatement(_core.types.memberExpression(_core.types.cloneNode(thisArg), _core.types.cloneNode(targetKey)))]);
- const setterBody = _core.types.blockStatement([_core.types.expressionStatement(_core.types.assignmentExpression("=", _core.types.memberExpression(_core.types.cloneNode(thisArg), _core.types.cloneNode(targetKey)), _core.types.identifier("v")))]);
- let getter, setter;
- if (originalKey.type === "PrivateName") {
- getter = _core.types.classPrivateMethod("get", _core.types.cloneNode(originalKey), [], getterBody, isStatic);
- setter = _core.types.classPrivateMethod("set", _core.types.cloneNode(originalKey), [_core.types.identifier("v")], setterBody, isStatic);
- } else {
- getter = _core.types.classMethod("get", _core.types.cloneNode(originalKey), [], getterBody, isComputed, isStatic);
- setter = _core.types.classMethod("set", _core.types.cloneNode(originalKey), [_core.types.identifier("v")], setterBody, isComputed, isStatic);
- }
- element.insertAfter(setter);
- element.insertAfter(getter);
- }
- function extractProxyAccessorsFor(targetKey, version) {
- if (version !== "2023-05" && version !== "2023-01") {
- return [_core.template.expression.ast`
- function () {
- return this.${_core.types.cloneNode(targetKey)};
- }
- `, _core.template.expression.ast`
- function (value) {
- this.${_core.types.cloneNode(targetKey)} = value;
- }
- `];
- }
- return [_core.template.expression.ast`
- o => o.${_core.types.cloneNode(targetKey)}
- `, _core.template.expression.ast`
- (o, v) => o.${_core.types.cloneNode(targetKey)} = v
- `];
- }
- function prependExpressionsToFieldInitializer(expressions, fieldPath) {
- const initializer = fieldPath.get("value");
- if (initializer.node) {
- expressions.push(initializer.node);
- } else if (expressions.length > 0) {
- expressions[expressions.length - 1] = _core.types.unaryExpression("void", expressions[expressions.length - 1]);
- }
- initializer.replaceWith(maybeSequenceExpression(expressions));
- }
- function prependExpressionsToConstructor(expressions, constructorPath) {
- constructorPath.node.body.body.unshift(_core.types.expressionStatement(maybeSequenceExpression(expressions)));
- }
- function isProtoInitCallExpression(expression, protoInitCall) {
- return _core.types.isCallExpression(expression) && _core.types.isIdentifier(expression.callee, {
- name: protoInitCall.name
- });
- }
- function optimizeSuperCallAndExpressions(expressions, protoInitLocal) {
- if (expressions.length >= 2 && isProtoInitCallExpression(expressions[1], protoInitLocal)) {
- const mergedSuperCall = _core.types.callExpression(_core.types.cloneNode(protoInitLocal), [expressions[0]]);
- expressions.splice(0, 2, mergedSuperCall);
- }
- if (expressions.length >= 2 && _core.types.isThisExpression(expressions[expressions.length - 1]) && isProtoInitCallExpression(expressions[expressions.length - 2], protoInitLocal)) {
- expressions.splice(expressions.length - 1, 1);
- }
- return maybeSequenceExpression(expressions);
- }
- function insertExpressionsAfterSuperCallAndOptimize(expressions, constructorPath, protoInitLocal) {
- constructorPath.traverse({
- CallExpression: {
- exit(path) {
- if (!path.get("callee").isSuper()) return;
- const newNodes = [path.node, ...expressions.map(expr => _core.types.cloneNode(expr))];
- if (path.isCompletionRecord()) {
- newNodes.push(_core.types.thisExpression());
- }
- path.replaceWith(optimizeSuperCallAndExpressions(newNodes, protoInitLocal));
- path.skip();
- }
- },
- ClassMethod(path) {
- if (path.node.kind === "constructor") {
- path.skip();
- }
- }
- });
- }
- function createConstructorFromExpressions(expressions, isDerivedClass) {
- const body = [_core.types.expressionStatement(maybeSequenceExpression(expressions))];
- if (isDerivedClass) {
- body.unshift(_core.types.expressionStatement(_core.types.callExpression(_core.types.super(), [_core.types.spreadElement(_core.types.identifier("args"))])));
- }
- return _core.types.classMethod("constructor", _core.types.identifier("constructor"), isDerivedClass ? [_core.types.restElement(_core.types.identifier("args"))] : [], _core.types.blockStatement(body));
- }
- const FIELD = 0;
- const ACCESSOR = 1;
- const METHOD = 2;
- const GETTER = 3;
- const SETTER = 4;
- const STATIC_OLD_VERSION = 5;
- const STATIC = 8;
- const DECORATORS_HAVE_THIS = 16;
- function getElementKind(element) {
- switch (element.node.type) {
- case "ClassProperty":
- case "ClassPrivateProperty":
- return FIELD;
- case "ClassAccessorProperty":
- return ACCESSOR;
- case "ClassMethod":
- case "ClassPrivateMethod":
- if (element.node.kind === "get") {
- return GETTER;
- } else if (element.node.kind === "set") {
- return SETTER;
- } else {
- return METHOD;
- }
- }
- }
- function isDecoratorInfo(info) {
- return "decorators" in info;
- }
- function filteredOrderedDecoratorInfo(info) {
- const filtered = info.filter(isDecoratorInfo);
- return [...filtered.filter(el => el.isStatic && el.kind >= ACCESSOR && el.kind <= SETTER), ...filtered.filter(el => !el.isStatic && el.kind >= ACCESSOR && el.kind <= SETTER), ...filtered.filter(el => el.isStatic && el.kind === FIELD), ...filtered.filter(el => !el.isStatic && el.kind === FIELD)];
- }
- function generateDecorationList(decorators, decoratorsThis, version) {
- const decsCount = decorators.length;
- const hasOneThis = decoratorsThis.some(Boolean);
- const decs = [];
- for (let i = 0; i < decsCount; i++) {
- if (version === "2023-05" && hasOneThis) {
- decs.push(decoratorsThis[i] || _core.types.unaryExpression("void", _core.types.numericLiteral(0)));
- }
- decs.push(decorators[i]);
- }
- return {
- hasThis: hasOneThis,
- decs
- };
- }
- function generateDecorationExprs(info, version) {
- return _core.types.arrayExpression(filteredOrderedDecoratorInfo(info).map(el => {
- const {
- decs,
- hasThis
- } = generateDecorationList(el.decorators, el.decoratorsThis, version);
- let flag = el.kind;
- if (el.isStatic) {
- flag += version === "2023-05" ? STATIC : STATIC_OLD_VERSION;
- }
- if (hasThis) flag += DECORATORS_HAVE_THIS;
- return _core.types.arrayExpression([decs.length === 1 ? decs[0] : _core.types.arrayExpression(decs), _core.types.numericLiteral(flag), el.name, ...(el.privateMethods || [])]);
- }));
- }
- function extractElementLocalAssignments(decorationInfo) {
- const localIds = [];
- for (const el of filteredOrderedDecoratorInfo(decorationInfo)) {
- const {
- locals
- } = el;
- if (Array.isArray(locals)) {
- localIds.push(...locals);
- } else if (locals !== undefined) {
- localIds.push(locals);
- }
- }
- return localIds;
- }
- function addCallAccessorsFor(element, key, getId, setId) {
- element.insertAfter(_core.types.classPrivateMethod("get", _core.types.cloneNode(key), [], _core.types.blockStatement([_core.types.returnStatement(_core.types.callExpression(_core.types.cloneNode(getId), [_core.types.thisExpression()]))])));
- element.insertAfter(_core.types.classPrivateMethod("set", _core.types.cloneNode(key), [_core.types.identifier("v")], _core.types.blockStatement([_core.types.expressionStatement(_core.types.callExpression(_core.types.cloneNode(setId), [_core.types.thisExpression(), _core.types.identifier("v")]))])));
- }
- function isNotTsParameter(node) {
- return node.type !== "TSParameterProperty";
- }
- function movePrivateAccessor(element, key, methodLocalVar, isStatic) {
- let params;
- let block;
- if (element.node.kind === "set") {
- params = [_core.types.identifier("v")];
- block = [_core.types.expressionStatement(_core.types.callExpression(methodLocalVar, [_core.types.thisExpression(), _core.types.identifier("v")]))];
- } else {
- params = [];
- block = [_core.types.returnStatement(_core.types.callExpression(methodLocalVar, [_core.types.thisExpression()]))];
- }
- element.replaceWith(_core.types.classPrivateMethod(element.node.kind, _core.types.cloneNode(key), params, _core.types.blockStatement(block), isStatic));
- }
- function isClassDecoratableElementPath(path) {
- const {
- type
- } = path;
- return type !== "TSDeclareMethod" && type !== "TSIndexSignature" && type !== "StaticBlock";
- }
- function staticBlockToIIFE(block) {
- return _core.types.callExpression(_core.types.arrowFunctionExpression([], _core.types.blockStatement(block.body)), []);
- }
- function maybeSequenceExpression(exprs) {
- if (exprs.length === 0) return _core.types.unaryExpression("void", _core.types.numericLiteral(0));
- if (exprs.length === 1) return exprs[0];
- return _core.types.sequenceExpression(exprs);
- }
- function createSetFunctionNameCall(state, className) {
- return _core.types.callExpression(state.addHelper("setFunctionName"), [_core.types.thisExpression(), className]);
- }
- function createToPropertyKeyCall(state, propertyKey) {
- return _core.types.callExpression(state.addHelper("toPropertyKey"), [propertyKey]);
- }
- function checkPrivateMethodUpdateError(path, decoratedPrivateMethods) {
- const privateNameVisitor = (0, _fields.privateNameVisitorFactory)({
- PrivateName(path, state) {
- if (!state.privateNamesMap.has(path.node.id.name)) return;
- const parentPath = path.parentPath;
- const parentParentPath = parentPath.parentPath;
- if (parentParentPath.node.type === "AssignmentExpression" && parentParentPath.node.left === parentPath.node || parentParentPath.node.type === "UpdateExpression" || parentParentPath.node.type === "RestElement" || parentParentPath.node.type === "ArrayPattern" || parentParentPath.node.type === "ObjectProperty" && parentParentPath.node.value === parentPath.node && parentParentPath.parentPath.type === "ObjectPattern" || parentParentPath.node.type === "ForOfStatement" && parentParentPath.node.left === parentPath.node) {
- throw path.buildCodeFrameError(`Decorated private methods are read-only, but "#${path.node.id.name}" is updated via this expression.`);
- }
- }
- });
- const privateNamesMap = new Map();
- for (const name of decoratedPrivateMethods) {
- privateNamesMap.set(name, null);
- }
- path.traverse(privateNameVisitor, {
- privateNamesMap: privateNamesMap
- });
- }
- function transformClass(path, state, constantSuper, version, className, propertyVisitor) {
- const body = path.get("body.body");
- const classDecorators = path.node.decorators;
- let hasElementDecorators = false;
- const generateClassPrivateUid = createLazyPrivateUidGeneratorForClass(path);
- const assignments = [];
- const scopeParent = path.scope.parent;
- const memoiseExpression = (expression, hint) => {
- const localEvaluatedId = scopeParent.generateDeclaredUidIdentifier(hint);
- assignments.push(_core.types.assignmentExpression("=", localEvaluatedId, expression));
- return _core.types.cloneNode(localEvaluatedId);
- };
- let protoInitLocal;
- let staticInitLocal;
- for (const element of body) {
- if (!isClassDecoratableElementPath(element)) {
- continue;
- }
- if (isDecorated(element.node)) {
- switch (element.node.type) {
- case "ClassProperty":
- propertyVisitor.ClassProperty(element, state);
- break;
- case "ClassPrivateProperty":
- propertyVisitor.ClassPrivateProperty(element, state);
- break;
- case "ClassAccessorProperty":
- propertyVisitor.ClassAccessorProperty(element, state);
- default:
- if (element.node.static) {
- var _staticInitLocal;
- (_staticInitLocal = staticInitLocal) != null ? _staticInitLocal : staticInitLocal = scopeParent.generateDeclaredUidIdentifier("initStatic");
- } else {
- var _protoInitLocal;
- (_protoInitLocal = protoInitLocal) != null ? _protoInitLocal : protoInitLocal = scopeParent.generateDeclaredUidIdentifier("initProto");
- }
- break;
- }
- hasElementDecorators = true;
- } else if (element.node.type === "ClassAccessorProperty") {
- propertyVisitor.ClassAccessorProperty(element, state);
- const {
- key,
- value,
- static: isStatic,
- computed
- } = element.node;
- const newId = generateClassPrivateUid();
- const newField = generateClassProperty(newId, value, isStatic);
- const keyPath = element.get("key");
- const [newPath] = element.replaceWith(newField);
- addProxyAccessorsFor(path.node.id, newPath, computed && !keyPath.isConstantExpression() ? memoiseExpression(createToPropertyKeyCall(state, key), "computedKey") : key, newId, version, computed);
- }
- }
- if (!classDecorators && !hasElementDecorators) {
- if (assignments.length > 0) {
- path.insertBefore(assignments.map(expr => _core.types.expressionStatement(expr)));
- path.scope.crawl();
- }
- return;
- }
- const elementDecoratorInfo = [];
- let constructorPath;
- const decoratedPrivateMethods = new Set();
- let classInitLocal, classIdLocal;
- const decoratorsThis = new Map();
- const maybeExtractDecorators = (decorators, memoiseInPlace) => {
- let needMemoise = false;
- for (const decorator of decorators) {
- const {
- expression
- } = decorator;
- if (version === "2023-05" && _core.types.isMemberExpression(expression)) {
- let object;
- if (_core.types.isSuper(expression.object) || _core.types.isThisExpression(expression.object)) {
- needMemoise = true;
- if (memoiseInPlace) {
- object = memoiseExpression(_core.types.thisExpression(), "obj");
- } else {
- object = _core.types.thisExpression();
- }
- } else {
- if (!scopeParent.isStatic(expression.object)) {
- needMemoise = true;
- if (memoiseInPlace) {
- expression.object = memoiseExpression(expression.object, "obj");
- }
- }
- object = _core.types.cloneNode(expression.object);
- }
- decoratorsThis.set(decorator, object);
- }
- if (!scopeParent.isStatic(expression)) {
- needMemoise = true;
- if (memoiseInPlace) {
- decorator.expression = memoiseExpression(expression, "dec");
- }
- }
- }
- return needMemoise && !memoiseInPlace;
- };
- let needsDeclaraionForClassBinding = false;
- let classDecorationsFlag = 0;
- let classDecorations = [];
- let classDecorationsId;
- if (classDecorators) {
- classInitLocal = scopeParent.generateDeclaredUidIdentifier("initClass");
- needsDeclaraionForClassBinding = path.isClassDeclaration();
- ({
- id: classIdLocal,
- path
- } = replaceClassWithVar(path, className));
- path.node.decorators = null;
- const needMemoise = maybeExtractDecorators(classDecorators, false);
- const {
- hasThis,
- decs
- } = generateDecorationList(classDecorators.map(el => el.expression), classDecorators.map(dec => decoratorsThis.get(dec)), version);
- classDecorationsFlag = hasThis ? 1 : 0;
- classDecorations = decs;
- if (needMemoise) {
- classDecorationsId = memoiseExpression(_core.types.arrayExpression(classDecorations), "classDecs");
- }
- } else {
- if (!path.node.id) {
- path.node.id = path.scope.generateUidIdentifier("Class");
- }
- classIdLocal = _core.types.cloneNode(path.node.id);
- }
- let lastInstancePrivateName;
- let needsInstancePrivateBrandCheck = false;
- let fieldInitializerAssignments = [];
- if (hasElementDecorators) {
- if (protoInitLocal) {
- const protoInitCall = _core.types.callExpression(_core.types.cloneNode(protoInitLocal), [_core.types.thisExpression()]);
- fieldInitializerAssignments.push(protoInitCall);
- }
- for (const element of body) {
- if (!isClassDecoratableElementPath(element)) {
- continue;
- }
- const {
- node
- } = element;
- const decorators = element.node.decorators;
- const hasDecorators = !!(decorators != null && decorators.length);
- if (hasDecorators) {
- maybeExtractDecorators(decorators, true);
- }
- const isComputed = "computed" in element.node && element.node.computed;
- if (isComputed) {
- if (!element.get("key").isConstantExpression()) {
- node.key = memoiseExpression(createToPropertyKeyCall(state, node.key), "computedKey");
- }
- }
- const kind = getElementKind(element);
- const {
- key
- } = node;
- const isPrivate = key.type === "PrivateName";
- const isStatic = element.node.static;
- let name = "computedKey";
- if (isPrivate) {
- name = key.id.name;
- } else if (!isComputed && key.type === "Identifier") {
- name = key.name;
- }
- if (isPrivate && !isStatic) {
- if (hasDecorators) {
- needsInstancePrivateBrandCheck = true;
- }
- if (_core.types.isClassPrivateProperty(node) || !lastInstancePrivateName) {
- lastInstancePrivateName = key;
- }
- }
- if (element.isClassMethod({
- kind: "constructor"
- })) {
- constructorPath = element;
- }
- if (hasDecorators) {
- let locals;
- let privateMethods;
- if (kind === ACCESSOR) {
- const {
- value
- } = element.node;
- const params = [_core.types.thisExpression()];
- if (value) {
- params.push(_core.types.cloneNode(value));
- }
- const newId = generateClassPrivateUid();
- const newFieldInitId = element.scope.parent.generateDeclaredUidIdentifier(`init_${name}`);
- const newValue = _core.types.callExpression(_core.types.cloneNode(newFieldInitId), params);
- const newField = generateClassProperty(newId, newValue, isStatic);
- const [newPath] = element.replaceWith(newField);
- if (isPrivate) {
- privateMethods = extractProxyAccessorsFor(newId, version);
- const getId = newPath.scope.parent.generateDeclaredUidIdentifier(`get_${name}`);
- const setId = newPath.scope.parent.generateDeclaredUidIdentifier(`set_${name}`);
- addCallAccessorsFor(newPath, key, getId, setId);
- locals = [newFieldInitId, getId, setId];
- } else {
- addProxyAccessorsFor(path.node.id, newPath, key, newId, version, isComputed);
- locals = newFieldInitId;
- }
- } else if (kind === FIELD) {
- const initId = element.scope.parent.generateDeclaredUidIdentifier(`init_${name}`);
- const valuePath = element.get("value");
- valuePath.replaceWith(_core.types.callExpression(_core.types.cloneNode(initId), [_core.types.thisExpression(), valuePath.node].filter(v => v)));
- locals = initId;
- if (isPrivate) {
- privateMethods = extractProxyAccessorsFor(key, version);
- }
- } else if (isPrivate) {
- locals = element.scope.parent.generateDeclaredUidIdentifier(`call_${name}`);
- const replaceSupers = new _helperReplaceSupers.default({
- constantSuper,
- methodPath: element,
- objectRef: classIdLocal,
- superRef: path.node.superClass,
- file: state.file,
- refToPreserve: classIdLocal
- });
- replaceSupers.replace();
- const {
- params,
- body,
- async: isAsync
- } = element.node;
- privateMethods = [_core.types.functionExpression(undefined, params.filter(isNotTsParameter), body, isAsync)];
- if (kind === GETTER || kind === SETTER) {
- movePrivateAccessor(element, _core.types.cloneNode(key), _core.types.cloneNode(locals), isStatic);
- } else {
- const node = element.node;
- path.node.body.body.unshift(_core.types.classPrivateProperty(key, _core.types.cloneNode(locals), [], node.static));
- decoratedPrivateMethods.add(key.id.name);
- element.remove();
- }
- }
- let nameExpr;
- if (isComputed) {
- nameExpr = _core.types.cloneNode(key);
- } else if (key.type === "PrivateName") {
- nameExpr = _core.types.stringLiteral(key.id.name);
- } else if (key.type === "Identifier") {
- nameExpr = _core.types.stringLiteral(key.name);
- } else {
- nameExpr = _core.types.cloneNode(key);
- }
- elementDecoratorInfo.push({
- kind,
- decorators: decorators.map(d => d.expression),
- decoratorsThis: decorators.map(d => decoratorsThis.get(d)),
- name: nameExpr,
- isStatic,
- privateMethods,
- locals
- });
- if (element.node) {
- element.node.decorators = null;
- }
- }
- if (fieldInitializerAssignments.length > 0 && !isStatic && (kind === FIELD || kind === ACCESSOR)) {
- prependExpressionsToFieldInitializer(fieldInitializerAssignments, element);
- fieldInitializerAssignments = [];
- }
- }
- }
- if (fieldInitializerAssignments.length > 0) {
- const isDerivedClass = !!path.node.superClass;
- if (constructorPath) {
- if (isDerivedClass) {
- insertExpressionsAfterSuperCallAndOptimize(fieldInitializerAssignments, constructorPath, protoInitLocal);
- } else {
- prependExpressionsToConstructor(fieldInitializerAssignments, constructorPath);
- }
- } else {
- path.node.body.body.unshift(createConstructorFromExpressions(fieldInitializerAssignments, isDerivedClass));
- }
- fieldInitializerAssignments = [];
- }
- const elementDecorations = generateDecorationExprs(elementDecoratorInfo, version);
- const elementLocals = extractElementLocalAssignments(elementDecoratorInfo);
- if (protoInitLocal) {
- elementLocals.push(protoInitLocal);
- }
- if (staticInitLocal) {
- elementLocals.push(staticInitLocal);
- }
- const classLocals = [];
- let classInitInjected = false;
- const classInitCall = classInitLocal && _core.types.callExpression(_core.types.cloneNode(classInitLocal), []);
- const originalClass = path.node;
- if (classDecorators) {
- classLocals.push(classIdLocal, classInitLocal);
- const statics = [];
- let staticBlocks = [];
- path.get("body.body").forEach(element => {
- if (element.isStaticBlock()) {
- staticBlocks.push(element.node);
- element.remove();
- return;
- }
- const isProperty = element.isClassProperty() || element.isClassPrivateProperty();
- if ((isProperty || element.isClassPrivateMethod()) && element.node.static) {
- if (isProperty && staticBlocks.length > 0) {
- const allValues = staticBlocks.map(staticBlockToIIFE);
- if (element.node.value) allValues.push(element.node.value);
- element.node.value = maybeSequenceExpression(allValues);
- staticBlocks = [];
- }
- element.node.static = false;
- statics.push(element.node);
- element.remove();
- }
- });
- if (statics.length > 0 || staticBlocks.length > 0) {
- const staticsClass = _core.template.expression.ast`
- class extends ${state.addHelper("identity")} {}
- `;
- staticsClass.body.body = [_core.types.staticBlock([_core.types.toStatement(originalClass, true) || _core.types.expressionStatement(originalClass)]), ...statics];
- const constructorBody = [];
- const newExpr = _core.types.newExpression(staticsClass, []);
- if (staticBlocks.length > 0) {
- constructorBody.push(...staticBlocks.map(staticBlockToIIFE));
- }
- if (classInitCall) {
- classInitInjected = true;
- constructorBody.push(classInitCall);
- }
- if (constructorBody.length > 0) {
- constructorBody.unshift(_core.types.callExpression(_core.types.super(), [_core.types.cloneNode(classIdLocal)]));
- staticsClass.body.body.push(_core.types.classMethod("constructor", _core.types.identifier("constructor"), [], _core.types.blockStatement([_core.types.expressionStatement(_core.types.sequenceExpression(constructorBody))])));
- } else {
- newExpr.arguments.push(_core.types.cloneNode(classIdLocal));
- }
- path.replaceWith(newExpr);
- }
- }
- if (!classInitInjected && classInitCall) {
- path.node.body.body.push(_core.types.staticBlock([_core.types.expressionStatement(classInitCall)]));
- }
- let {
- superClass
- } = originalClass;
- if (superClass && version === "2023-05") {
- const id = path.scope.maybeGenerateMemoised(superClass);
- if (id) {
- originalClass.superClass = _core.types.assignmentExpression("=", id, superClass);
- superClass = id;
- }
- }
- originalClass.body.body.unshift(_core.types.staticBlock([_core.types.expressionStatement(createLocalsAssignment(elementLocals, classLocals, elementDecorations, classDecorationsId ? _core.types.cloneNode(classDecorationsId) : _core.types.arrayExpression(classDecorations), _core.types.numericLiteral(classDecorationsFlag), needsInstancePrivateBrandCheck ? lastInstancePrivateName : null, typeof className === "object" ? className : undefined, _core.types.cloneNode(superClass), state, version)), staticInitLocal && _core.types.expressionStatement(_core.types.callExpression(_core.types.cloneNode(staticInitLocal), [_core.types.thisExpression()]))].filter(Boolean)));
- path.insertBefore(assignments.map(expr => _core.types.expressionStatement(expr)));
- if (needsDeclaraionForClassBinding) {
- path.insertBefore(_core.types.variableDeclaration("let", [_core.types.variableDeclarator(_core.types.cloneNode(classIdLocal))]));
- }
- if (decoratedPrivateMethods.size > 0) {
- checkPrivateMethodUpdateError(path, decoratedPrivateMethods);
- }
- path.scope.crawl();
- return path;
- }
- function createLocalsAssignment(elementLocals, classLocals, elementDecorations, classDecorations, classDecorationsFlag, maybePrivateBranName, setClassName, superClass, state, version) {
- let lhs, rhs;
- const args = [setClassName ? createSetFunctionNameCall(state, setClassName) : _core.types.thisExpression(), elementDecorations, classDecorations];
- {
- if (version === "2021-12" || version === "2022-03" && !state.availableHelper("applyDecs2203R")) {
- const lhs = _core.types.arrayPattern([...elementLocals, ...classLocals]);
- const rhs = _core.types.callExpression(state.addHelper(version === "2021-12" ? "applyDecs" : "applyDecs2203"), args);
- return _core.types.assignmentExpression("=", lhs, rhs);
- }
- }
- if (version === "2023-05") {
- if (maybePrivateBranName || superClass || classDecorationsFlag.value !== 0) {
- args.push(classDecorationsFlag);
- }
- if (maybePrivateBranName) {
- args.push(_core.template.expression.ast`
- _ => ${_core.types.cloneNode(maybePrivateBranName)} in _
- `);
- } else if (superClass) {
- args.push(_core.types.unaryExpression("void", _core.types.numericLiteral(0)));
- }
- if (superClass) args.push(superClass);
- rhs = _core.types.callExpression(state.addHelper("applyDecs2305"), args);
- } else if (version === "2023-01") {
- if (maybePrivateBranName) {
- args.push(_core.template.expression.ast`
- _ => ${_core.types.cloneNode(maybePrivateBranName)} in _
- `);
- }
- rhs = _core.types.callExpression(state.addHelper("applyDecs2301"), args);
- } else {
- rhs = _core.types.callExpression(state.addHelper("applyDecs2203R"), args);
- }
- if (elementLocals.length > 0) {
- if (classLocals.length > 0) {
- lhs = _core.types.objectPattern([_core.types.objectProperty(_core.types.identifier("e"), _core.types.arrayPattern(elementLocals)), _core.types.objectProperty(_core.types.identifier("c"), _core.types.arrayPattern(classLocals))]);
- } else {
- lhs = _core.types.arrayPattern(elementLocals);
- rhs = _core.types.memberExpression(rhs, _core.types.identifier("e"), false, false);
- }
- } else {
- lhs = _core.types.arrayPattern(classLocals);
- rhs = _core.types.memberExpression(rhs, _core.types.identifier("c"), false, false);
- }
- return _core.types.assignmentExpression("=", lhs, rhs);
- }
- function isProtoKey(node) {
- return node.type === "Identifier" ? node.name === "__proto__" : node.value === "__proto__";
- }
- function isDecorated(node) {
- return node.decorators && node.decorators.length > 0;
- }
- function shouldTransformElement(node) {
- switch (node.type) {
- case "ClassAccessorProperty":
- return true;
- case "ClassMethod":
- case "ClassProperty":
- case "ClassPrivateMethod":
- case "ClassPrivateProperty":
- return isDecorated(node);
- default:
- return false;
- }
- }
- function shouldTransformClass(node) {
- return isDecorated(node) || node.body.body.some(shouldTransformElement);
- }
- function NamedEvaluationVisitoryFactory(isAnonymous, visitor) {
- function handleComputedProperty(propertyPath, key, state) {
- switch (key.type) {
- case "StringLiteral":
- return _core.types.stringLiteral(key.value);
- case "NumericLiteral":
- case "BigIntLiteral":
- {
- const keyValue = key.value + "";
- propertyPath.get("key").replaceWith(_core.types.stringLiteral(keyValue));
- return _core.types.stringLiteral(keyValue);
- }
- default:
- {
- const ref = propertyPath.scope.maybeGenerateMemoised(key);
- propertyPath.get("key").replaceWith(_core.types.assignmentExpression("=", ref, createToPropertyKeyCall(state, key)));
- return _core.types.cloneNode(ref);
- }
- }
- }
- return {
- VariableDeclarator(path, state) {
- const id = path.node.id;
- if (id.type === "Identifier") {
- const initializer = (0, _helperSkipTransparentExpressionWrappers.skipTransparentExprWrappers)(path.get("init"));
- if (isAnonymous(initializer)) {
- const name = id.name;
- visitor(initializer, state, name);
- }
- }
- },
- AssignmentExpression(path, state) {
- const id = path.node.left;
- if (id.type === "Identifier") {
- const initializer = (0, _helperSkipTransparentExpressionWrappers.skipTransparentExprWrappers)(path.get("right"));
- if (isAnonymous(initializer)) {
- switch (path.node.operator) {
- case "=":
- case "&&=":
- case "||=":
- case "??=":
- visitor(initializer, state, id.name);
- }
- }
- }
- },
- AssignmentPattern(path, state) {
- const id = path.node.left;
- if (id.type === "Identifier") {
- const initializer = (0, _helperSkipTransparentExpressionWrappers.skipTransparentExprWrappers)(path.get("right"));
- if (isAnonymous(initializer)) {
- const name = id.name;
- visitor(initializer, state, name);
- }
- }
- },
- ObjectExpression(path, state) {
- for (const propertyPath of path.get("properties")) {
- const {
- node
- } = propertyPath;
- if (node.type !== "ObjectProperty") continue;
- const id = node.key;
- const initializer = (0, _helperSkipTransparentExpressionWrappers.skipTransparentExprWrappers)(propertyPath.get("value"));
- if (isAnonymous(initializer)) {
- if (!node.computed) {
- if (!isProtoKey(id)) {
- if (id.type === "Identifier") {
- visitor(initializer, state, id.name);
- } else {
- const className = _core.types.stringLiteral(id.value + "");
- visitor(initializer, state, className);
- }
- }
- } else {
- const ref = handleComputedProperty(propertyPath, id, state);
- visitor(initializer, state, ref);
- }
- }
- }
- },
- ClassPrivateProperty(path, state) {
- const {
- node
- } = path;
- const initializer = (0, _helperSkipTransparentExpressionWrappers.skipTransparentExprWrappers)(path.get("value"));
- if (isAnonymous(initializer)) {
- const className = _core.types.stringLiteral("#" + node.key.id.name);
- visitor(initializer, state, className);
- }
- },
- ClassAccessorProperty(path, state) {
- const {
- node
- } = path;
- const id = node.key;
- const initializer = (0, _helperSkipTransparentExpressionWrappers.skipTransparentExprWrappers)(path.get("value"));
- if (isAnonymous(initializer)) {
- if (!node.computed) {
- if (id.type === "Identifier") {
- visitor(initializer, state, id.name);
- } else if (id.type === "PrivateName") {
- const className = _core.types.stringLiteral("#" + id.id.name);
- visitor(initializer, state, className);
- } else {
- const className = _core.types.stringLiteral(id.value + "");
- visitor(initializer, state, className);
- }
- } else {
- const ref = handleComputedProperty(path, id, state);
- visitor(initializer, state, ref);
- }
- }
- },
- ClassProperty(path, state) {
- const {
- node
- } = path;
- const id = node.key;
- const initializer = (0, _helperSkipTransparentExpressionWrappers.skipTransparentExprWrappers)(path.get("value"));
- if (isAnonymous(initializer)) {
- if (!node.computed) {
- if (id.type === "Identifier") {
- visitor(initializer, state, id.name);
- } else {
- const className = _core.types.stringLiteral(id.value + "");
- visitor(initializer, state, className);
- }
- } else {
- const ref = handleComputedProperty(path, id, state);
- visitor(initializer, state, ref);
- }
- }
- }
- };
- }
- function isDecoratedAnonymousClassExpression(path) {
- return path.isClassExpression({
- id: null
- }) && shouldTransformClass(path.node);
- }
- function _default({
- assertVersion,
- assumption
- }, {
- loose
- }, version, inherits) {
- var _assumption;
- {
- if (version === "2023-05" || version === "2023-01") {
- assertVersion("^7.21.0");
- } else if (version === "2021-12") {
- assertVersion("^7.16.0");
- } else {
- assertVersion("^7.19.0");
- }
- }
- const VISITED = new WeakSet();
- const constantSuper = (_assumption = assumption("constantSuper")) != null ? _assumption : loose;
- const namedEvaluationVisitor = NamedEvaluationVisitoryFactory(isDecoratedAnonymousClassExpression, visitClass);
- function visitClass(path, state, className) {
- var _className, _node$id;
- if (VISITED.has(path)) return;
- const {
- node
- } = path;
- (_className = className) != null ? _className : className = (_node$id = node.id) == null ? void 0 : _node$id.name;
- const newPath = transformClass(path, state, constantSuper, version, className, namedEvaluationVisitor);
- if (newPath) {
- VISITED.add(newPath);
- return;
- }
- VISITED.add(path);
- }
- return {
- name: "proposal-decorators",
- inherits: inherits,
- visitor: Object.assign({
- ExportDefaultDeclaration(path, state) {
- const {
- declaration
- } = path.node;
- if ((declaration == null ? void 0 : declaration.type) === "ClassDeclaration" && isDecorated(declaration)) {
- const isAnonymous = !declaration.id;
- const updatedVarDeclarationPath = (0, _helperSplitExportDeclaration.default)(path);
- if (isAnonymous) {
- visitClass(updatedVarDeclarationPath, state, _core.types.stringLiteral("default"));
- }
- }
- },
- ExportNamedDeclaration(path) {
- const {
- declaration
- } = path.node;
- if ((declaration == null ? void 0 : declaration.type) === "ClassDeclaration" && isDecorated(declaration)) {
- (0, _helperSplitExportDeclaration.default)(path);
- }
- },
- Class(path, state) {
- visitClass(path, state, undefined);
- }
- }, namedEvaluationVisitor)
- };
- }
- //# sourceMappingURL=decorators.js.map
|