1054 lines
40 KiB
JavaScript
1054 lines
40 KiB
JavaScript
"use strict";
|
|
|
|
Object.defineProperty(exports, "__esModule", {
|
|
value: true
|
|
});
|
|
exports.buildCheckInRHS = buildCheckInRHS;
|
|
exports.buildFieldsInitNodes = buildFieldsInitNodes;
|
|
exports.buildPrivateNamesMap = buildPrivateNamesMap;
|
|
exports.buildPrivateNamesNodes = buildPrivateNamesNodes;
|
|
exports.privateNameVisitorFactory = privateNameVisitorFactory;
|
|
exports.transformPrivateNamesUsage = transformPrivateNamesUsage;
|
|
var _core = require("@babel/core");
|
|
var _helperReplaceSupers = require("@babel/helper-replace-supers");
|
|
var _helperEnvironmentVisitor = require("@babel/helper-environment-visitor");
|
|
var _helperMemberExpressionToFunctions = require("@babel/helper-member-expression-to-functions");
|
|
var _helperOptimiseCallExpression = require("@babel/helper-optimise-call-expression");
|
|
var _helperAnnotateAsPure = require("@babel/helper-annotate-as-pure");
|
|
var _helperSkipTransparentExpressionWrappers = require("@babel/helper-skip-transparent-expression-wrappers");
|
|
var ts = require("./typescript.js");
|
|
{
|
|
var newHelpers = file => {
|
|
;
|
|
return file.availableHelper("classPrivateFieldGet2");
|
|
};
|
|
}
|
|
function buildPrivateNamesMap(className, privateFieldsAsSymbolsOrProperties, props, file) {
|
|
const privateNamesMap = new Map();
|
|
let classBrandId;
|
|
for (const prop of props) {
|
|
if (prop.isPrivate()) {
|
|
const {
|
|
name
|
|
} = prop.node.key.id;
|
|
let update = privateNamesMap.get(name);
|
|
if (!update) {
|
|
const isMethod = !prop.isProperty();
|
|
const isStatic = prop.node.static;
|
|
let initAdded = false;
|
|
let id;
|
|
if (!privateFieldsAsSymbolsOrProperties && newHelpers(file) && isMethod && !isStatic) {
|
|
var _classBrandId;
|
|
initAdded = !!classBrandId;
|
|
(_classBrandId = classBrandId) != null ? _classBrandId : classBrandId = prop.scope.generateUidIdentifier(`${className}_brand`);
|
|
id = classBrandId;
|
|
} else {
|
|
id = prop.scope.generateUidIdentifier(name);
|
|
}
|
|
update = {
|
|
id,
|
|
static: isStatic,
|
|
method: isMethod,
|
|
initAdded
|
|
};
|
|
privateNamesMap.set(name, update);
|
|
}
|
|
if (prop.isClassPrivateMethod()) {
|
|
if (prop.node.kind === "get") {
|
|
const {
|
|
body
|
|
} = prop.node.body;
|
|
let $;
|
|
if (body.length === 1 && _core.types.isReturnStatement($ = body[0]) && _core.types.isCallExpression($ = $.argument) && $.arguments.length === 1 && _core.types.isThisExpression($.arguments[0]) && _core.types.isIdentifier($ = $.callee)) {
|
|
update.getId = _core.types.cloneNode($);
|
|
update.getterDeclared = true;
|
|
} else {
|
|
update.getId = prop.scope.generateUidIdentifier(`get_${name}`);
|
|
}
|
|
} else if (prop.node.kind === "set") {
|
|
const {
|
|
params
|
|
} = prop.node;
|
|
const {
|
|
body
|
|
} = prop.node.body;
|
|
let $;
|
|
if (body.length === 1 && _core.types.isExpressionStatement($ = body[0]) && _core.types.isCallExpression($ = $.expression) && $.arguments.length === 2 && _core.types.isThisExpression($.arguments[0]) && _core.types.isIdentifier($.arguments[1], {
|
|
name: params[0].name
|
|
}) && _core.types.isIdentifier($ = $.callee)) {
|
|
update.setId = _core.types.cloneNode($);
|
|
update.setterDeclared = true;
|
|
} else {
|
|
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, privateFieldsAsProperties, privateFieldsAsSymbols, state) {
|
|
const initNodes = [];
|
|
const injectedIds = new Set();
|
|
for (const [name, value] of privateNamesMap) {
|
|
const {
|
|
static: isStatic,
|
|
method: isMethod,
|
|
getId,
|
|
setId
|
|
} = value;
|
|
const isGetterOrSetter = getId || setId;
|
|
const id = _core.types.cloneNode(value.id);
|
|
let init;
|
|
if (privateFieldsAsProperties) {
|
|
init = _core.types.callExpression(state.addHelper("classPrivateFieldLooseKey"), [_core.types.stringLiteral(name)]);
|
|
} else if (privateFieldsAsSymbols) {
|
|
init = _core.types.callExpression(_core.types.identifier("Symbol"), [_core.types.stringLiteral(name)]);
|
|
} else if (!isStatic) {
|
|
if (injectedIds.has(id.name)) continue;
|
|
injectedIds.add(id.name);
|
|
init = _core.types.newExpression(_core.types.identifier(isMethod && (!isGetterOrSetter || newHelpers(state)) ? "WeakSet" : "WeakMap"), []);
|
|
}
|
|
if (init) {
|
|
if (!privateFieldsAsSymbols) {
|
|
(0, _helperAnnotateAsPure.default)(init);
|
|
}
|
|
initNodes.push(_core.template.statement.ast`var ${id} = ${init}`);
|
|
}
|
|
}
|
|
return initNodes;
|
|
}
|
|
function privateNameVisitorFactory(visitor) {
|
|
const nestedVisitor = _core.traverse.visitors.merge([Object.assign({}, visitor), _helperEnvironmentVisitor.default]);
|
|
const privateNameVisitor = Object.assign({}, visitor, {
|
|
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(nestedVisitor, Object.assign({}, this, {
|
|
redeclared
|
|
}));
|
|
path.traverse(privateNameVisitor, Object.assign({}, this, {
|
|
privateNamesMap: visiblePrivateNames
|
|
}));
|
|
path.skipKey("body");
|
|
}
|
|
});
|
|
return privateNameVisitor;
|
|
}
|
|
const privateNameVisitor = privateNameVisitorFactory({
|
|
PrivateName(path, {
|
|
noDocumentAll
|
|
}) {
|
|
const {
|
|
privateNamesMap,
|
|
redeclared
|
|
} = this;
|
|
const {
|
|
node,
|
|
parentPath
|
|
} = path;
|
|
if (!parentPath.isMemberExpression({
|
|
property: node
|
|
}) && !parentPath.isOptionalMemberExpression({
|
|
property: node
|
|
})) {
|
|
return;
|
|
}
|
|
const {
|
|
name
|
|
} = node.id;
|
|
if (!privateNamesMap.has(name)) return;
|
|
if (redeclared != null && redeclared.includes(name)) return;
|
|
this.handle(parentPath, noDocumentAll);
|
|
}
|
|
});
|
|
function unshadow(name, scope, innerBinding) {
|
|
while ((_scope = scope) != null && _scope.hasBinding(name) && !scope.bindingIdentifierEquals(name, innerBinding)) {
|
|
var _scope;
|
|
scope.rename(name);
|
|
scope = scope.parent;
|
|
}
|
|
}
|
|
function buildCheckInRHS(rhs, file, inRHSIsObject) {
|
|
if (inRHSIsObject || !(file.availableHelper != null && file.availableHelper("checkInRHS"))) return rhs;
|
|
return _core.types.callExpression(file.addHelper("checkInRHS"), [rhs]);
|
|
}
|
|
const privateInVisitor = privateNameVisitorFactory({
|
|
BinaryExpression(path, {
|
|
file
|
|
}) {
|
|
const {
|
|
operator,
|
|
left,
|
|
right
|
|
} = path.node;
|
|
if (operator !== "in") return;
|
|
if (!_core.types.isPrivateName(left)) return;
|
|
const {
|
|
privateFieldsAsProperties,
|
|
privateNamesMap,
|
|
redeclared
|
|
} = this;
|
|
const {
|
|
name
|
|
} = left.id;
|
|
if (!privateNamesMap.has(name)) return;
|
|
if (redeclared != null && redeclared.includes(name)) return;
|
|
unshadow(this.classRef.name, path.scope, this.innerBinding);
|
|
if (privateFieldsAsProperties) {
|
|
const {
|
|
id
|
|
} = privateNamesMap.get(name);
|
|
path.replaceWith(_core.template.expression.ast`
|
|
Object.prototype.hasOwnProperty.call(${buildCheckInRHS(right, file)}, ${_core.types.cloneNode(id)})
|
|
`);
|
|
return;
|
|
}
|
|
const {
|
|
id,
|
|
static: isStatic
|
|
} = privateNamesMap.get(name);
|
|
if (isStatic) {
|
|
path.replaceWith(_core.template.expression.ast`${buildCheckInRHS(right, file)} === ${_core.types.cloneNode(this.classRef)}`);
|
|
return;
|
|
}
|
|
path.replaceWith(_core.template.expression.ast`${_core.types.cloneNode(id)}.has(${buildCheckInRHS(right, file)})`);
|
|
}
|
|
});
|
|
function readOnlyError(file, name) {
|
|
return _core.types.callExpression(file.addHelper("readOnlyError"), [_core.types.stringLiteral(`#${name}`)]);
|
|
}
|
|
function writeOnlyError(file, name) {
|
|
if (!file.availableHelper("writeOnlyError")) {
|
|
console.warn(`@babel/helpers is outdated, update it to silence this warning.`);
|
|
return _core.types.buildUndefinedNode();
|
|
}
|
|
return _core.types.callExpression(file.addHelper("writeOnlyError"), [_core.types.stringLiteral(`#${name}`)]);
|
|
}
|
|
function buildStaticPrivateFieldAccess(expr, noUninitializedPrivateFieldAccess) {
|
|
if (noUninitializedPrivateFieldAccess) return expr;
|
|
return _core.types.memberExpression(expr, _core.types.identifier("_"));
|
|
}
|
|
function autoInherits(fn) {
|
|
return function (member) {
|
|
return _core.types.inherits(fn.apply(this, arguments), member.node);
|
|
};
|
|
}
|
|
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: autoInherits(function (member) {
|
|
const {
|
|
classRef,
|
|
privateNamesMap,
|
|
file,
|
|
innerBinding,
|
|
noUninitializedPrivateFieldAccess
|
|
} = this;
|
|
const privateName = member.node.property;
|
|
const {
|
|
name
|
|
} = privateName.id;
|
|
const {
|
|
id,
|
|
static: isStatic,
|
|
method: isMethod,
|
|
methodId,
|
|
getId,
|
|
setId
|
|
} = privateNamesMap.get(name);
|
|
const isGetterOrSetter = getId || setId;
|
|
const cloneId = id => _core.types.inherits(_core.types.cloneNode(id), privateName);
|
|
if (isStatic) {
|
|
unshadow(classRef.name, member.scope, innerBinding);
|
|
if (!newHelpers(file)) {
|
|
const helperName = isMethod && !isGetterOrSetter ? "classStaticPrivateMethodGet" : "classStaticPrivateFieldSpecGet";
|
|
return _core.types.callExpression(file.addHelper(helperName), [this.receiver(member), _core.types.cloneNode(classRef), cloneId(id)]);
|
|
}
|
|
const receiver = this.receiver(member);
|
|
const skipCheck = _core.types.isIdentifier(receiver) && receiver.name === classRef.name;
|
|
if (!isMethod) {
|
|
if (skipCheck) {
|
|
return buildStaticPrivateFieldAccess(cloneId(id), noUninitializedPrivateFieldAccess);
|
|
}
|
|
return buildStaticPrivateFieldAccess(_core.types.callExpression(file.addHelper("assertClassBrand"), [_core.types.cloneNode(classRef), receiver, cloneId(id)]), noUninitializedPrivateFieldAccess);
|
|
}
|
|
if (getId) {
|
|
if (skipCheck) {
|
|
return _core.types.callExpression(cloneId(getId), [receiver]);
|
|
}
|
|
return _core.types.callExpression(file.addHelper("classPrivateGetter"), [_core.types.cloneNode(classRef), receiver, cloneId(getId)]);
|
|
}
|
|
if (setId) {
|
|
const err = _core.types.buildUndefinedNode();
|
|
if (skipCheck) return err;
|
|
return _core.types.sequenceExpression([_core.types.callExpression(file.addHelper("assertClassBrand"), [_core.types.cloneNode(classRef), receiver]), err]);
|
|
}
|
|
if (skipCheck) return cloneId(id);
|
|
return _core.types.callExpression(file.addHelper("assertClassBrand"), [_core.types.cloneNode(classRef), receiver, cloneId(id)]);
|
|
}
|
|
if (isMethod) {
|
|
if (isGetterOrSetter) {
|
|
if (!getId) {
|
|
return _core.types.sequenceExpression([this.receiver(member), writeOnlyError(file, name)]);
|
|
}
|
|
if (!newHelpers(file)) {
|
|
return _core.types.callExpression(file.addHelper("classPrivateFieldGet"), [this.receiver(member), cloneId(id)]);
|
|
}
|
|
return _core.types.callExpression(file.addHelper("classPrivateGetter"), [_core.types.cloneNode(id), this.receiver(member), cloneId(getId)]);
|
|
}
|
|
if (!newHelpers(file)) {
|
|
return _core.types.callExpression(file.addHelper("classPrivateMethodGet"), [this.receiver(member), _core.types.cloneNode(id), cloneId(methodId)]);
|
|
}
|
|
return _core.types.callExpression(file.addHelper("assertClassBrand"), [_core.types.cloneNode(id), this.receiver(member), cloneId(methodId)]);
|
|
}
|
|
if (newHelpers(file)) {
|
|
return _core.types.callExpression(file.addHelper("classPrivateFieldGet2"), [cloneId(id), this.receiver(member)]);
|
|
}
|
|
return _core.types.callExpression(file.addHelper("classPrivateFieldGet"), [this.receiver(member), cloneId(id)]);
|
|
}),
|
|
boundGet(member) {
|
|
this.memoise(member, 1);
|
|
return _core.types.callExpression(_core.types.memberExpression(this.get(member), _core.types.identifier("bind")), [this.receiver(member)]);
|
|
},
|
|
set: autoInherits(function (member, value) {
|
|
const {
|
|
classRef,
|
|
privateNamesMap,
|
|
file,
|
|
noUninitializedPrivateFieldAccess
|
|
} = this;
|
|
const privateName = member.node.property;
|
|
const {
|
|
name
|
|
} = privateName.id;
|
|
const {
|
|
id,
|
|
static: isStatic,
|
|
method: isMethod,
|
|
setId,
|
|
getId
|
|
} = privateNamesMap.get(name);
|
|
const isGetterOrSetter = getId || setId;
|
|
const cloneId = id => _core.types.inherits(_core.types.cloneNode(id), privateName);
|
|
if (isStatic) {
|
|
if (!newHelpers(file)) {
|
|
const helperName = isMethod && !isGetterOrSetter ? "classStaticPrivateMethodSet" : "classStaticPrivateFieldSpecSet";
|
|
return _core.types.callExpression(file.addHelper(helperName), [this.receiver(member), _core.types.cloneNode(classRef), cloneId(id), value]);
|
|
}
|
|
const receiver = this.receiver(member);
|
|
const skipCheck = _core.types.isIdentifier(receiver) && receiver.name === classRef.name;
|
|
if (isMethod && !setId) {
|
|
const err = readOnlyError(file, name);
|
|
if (skipCheck) return _core.types.sequenceExpression([value, err]);
|
|
return _core.types.sequenceExpression([value, _core.types.callExpression(file.addHelper("assertClassBrand"), [_core.types.cloneNode(classRef), receiver]), readOnlyError(file, name)]);
|
|
}
|
|
if (setId) {
|
|
if (skipCheck) {
|
|
return _core.types.callExpression(_core.types.cloneNode(setId), [receiver, value]);
|
|
}
|
|
return _core.types.callExpression(file.addHelper("classPrivateSetter"), [_core.types.cloneNode(classRef), cloneId(setId), receiver, value]);
|
|
}
|
|
return _core.types.assignmentExpression("=", buildStaticPrivateFieldAccess(cloneId(id), noUninitializedPrivateFieldAccess), skipCheck ? value : _core.types.callExpression(file.addHelper("assertClassBrand"), [_core.types.cloneNode(classRef), receiver, value]));
|
|
}
|
|
if (isMethod) {
|
|
if (setId) {
|
|
if (!newHelpers(file)) {
|
|
return _core.types.callExpression(file.addHelper("classPrivateFieldSet"), [this.receiver(member), cloneId(id), value]);
|
|
}
|
|
return _core.types.callExpression(file.addHelper("classPrivateSetter"), [_core.types.cloneNode(id), cloneId(setId), this.receiver(member), value]);
|
|
}
|
|
return _core.types.sequenceExpression([this.receiver(member), value, readOnlyError(file, name)]);
|
|
}
|
|
if (newHelpers(file)) {
|
|
return _core.types.callExpression(file.addHelper("classPrivateFieldSet2"), [cloneId(id), this.receiver(member), value]);
|
|
}
|
|
return _core.types.callExpression(file.addHelper("classPrivateFieldSet"), [this.receiver(member), cloneId(id), value]);
|
|
}),
|
|
destructureSet(member) {
|
|
const {
|
|
classRef,
|
|
privateNamesMap,
|
|
file,
|
|
noUninitializedPrivateFieldAccess
|
|
} = this;
|
|
const privateName = member.node.property;
|
|
const {
|
|
name
|
|
} = privateName.id;
|
|
const {
|
|
id,
|
|
static: isStatic,
|
|
method: isMethod,
|
|
setId
|
|
} = privateNamesMap.get(name);
|
|
const cloneId = id => _core.types.inherits(_core.types.cloneNode(id), privateName);
|
|
if (!newHelpers(file)) {
|
|
if (isStatic) {
|
|
try {
|
|
var helper = file.addHelper("classStaticPrivateFieldDestructureSet");
|
|
} catch (_unused) {
|
|
throw new Error("Babel can not transpile `[C.#p] = [0]` with @babel/helpers < 7.13.10, \n" + "please update @babel/helpers to the latest version.");
|
|
}
|
|
return _core.types.memberExpression(_core.types.callExpression(helper, [this.receiver(member), _core.types.cloneNode(classRef), cloneId(id)]), _core.types.identifier("value"));
|
|
}
|
|
return _core.types.memberExpression(_core.types.callExpression(file.addHelper("classPrivateFieldDestructureSet"), [this.receiver(member), cloneId(id)]), _core.types.identifier("value"));
|
|
}
|
|
if (isMethod && !setId) {
|
|
return _core.types.memberExpression(_core.types.sequenceExpression([member.node.object, readOnlyError(file, name)]), _core.types.identifier("_"));
|
|
}
|
|
if (isStatic && !isMethod) {
|
|
const getCall = this.get(member);
|
|
if (!noUninitializedPrivateFieldAccess || !_core.types.isCallExpression(getCall)) {
|
|
return getCall;
|
|
}
|
|
const ref = getCall.arguments.pop();
|
|
getCall.arguments.push(_core.template.expression.ast`(_) => ${ref} = _`);
|
|
return _core.types.memberExpression(_core.types.callExpression(file.addHelper("toSetter"), [getCall]), _core.types.identifier("_"));
|
|
}
|
|
const setCall = this.set(member, _core.types.identifier("_"));
|
|
if (!_core.types.isCallExpression(setCall) || !_core.types.isIdentifier(setCall.arguments[setCall.arguments.length - 1], {
|
|
name: "_"
|
|
})) {
|
|
throw member.buildCodeFrameError("Internal Babel error while compiling this code. This is a Babel bug. " + "Please report it at https://github.com/babel/babel/issues.");
|
|
}
|
|
let args;
|
|
if (_core.types.isMemberExpression(setCall.callee, {
|
|
computed: false
|
|
}) && _core.types.isIdentifier(setCall.callee.property) && setCall.callee.property.name === "call") {
|
|
args = [setCall.callee.object, _core.types.arrayExpression(setCall.arguments.slice(1, -1)), setCall.arguments[0]];
|
|
} else {
|
|
args = [setCall.callee, _core.types.arrayExpression(setCall.arguments.slice(0, -1))];
|
|
}
|
|
return _core.types.memberExpression(_core.types.callExpression(file.addHelper("toSetter"), args), _core.types.identifier("_"));
|
|
},
|
|
call(member, args) {
|
|
this.memoise(member, 1);
|
|
return (0, _helperOptimiseCallExpression.default)(this.get(member), this.receiver(member), args, false);
|
|
},
|
|
optionalCall(member, args) {
|
|
this.memoise(member, 1);
|
|
return (0, _helperOptimiseCallExpression.default)(this.get(member), this.receiver(member), args, true);
|
|
},
|
|
delete() {
|
|
throw new Error("Internal Babel error: deleting private elements is a parsing error.");
|
|
}
|
|
};
|
|
const privateNameHandlerLoose = {
|
|
get(member) {
|
|
const {
|
|
privateNamesMap,
|
|
file
|
|
} = this;
|
|
const {
|
|
object
|
|
} = member.node;
|
|
const {
|
|
name
|
|
} = member.node.property.id;
|
|
return _core.template.expression`BASE(REF, PROP)[PROP]`({
|
|
BASE: file.addHelper("classPrivateFieldLooseBase"),
|
|
REF: _core.types.cloneNode(object),
|
|
PROP: _core.types.cloneNode(privateNamesMap.get(name).id)
|
|
});
|
|
},
|
|
set() {
|
|
throw new Error("private name handler with loose = true don't need set()");
|
|
},
|
|
boundGet(member) {
|
|
return _core.types.callExpression(_core.types.memberExpression(this.get(member), _core.types.identifier("bind")), [_core.types.cloneNode(member.node.object)]);
|
|
},
|
|
simpleSet(member) {
|
|
return this.get(member);
|
|
},
|
|
destructureSet(member) {
|
|
return this.get(member);
|
|
},
|
|
call(member, args) {
|
|
return _core.types.callExpression(this.get(member), args);
|
|
},
|
|
optionalCall(member, args) {
|
|
return _core.types.optionalCallExpression(this.get(member), args, true);
|
|
},
|
|
delete() {
|
|
throw new Error("Internal Babel error: deleting private elements is a parsing error.");
|
|
}
|
|
};
|
|
function transformPrivateNamesUsage(ref, path, privateNamesMap, {
|
|
privateFieldsAsProperties,
|
|
noUninitializedPrivateFieldAccess,
|
|
noDocumentAll,
|
|
innerBinding
|
|
}, state) {
|
|
if (!privateNamesMap.size) return;
|
|
const body = path.get("body");
|
|
const handler = privateFieldsAsProperties ? privateNameHandlerLoose : privateNameHandlerSpec;
|
|
(0, _helperMemberExpressionToFunctions.default)(body, privateNameVisitor, Object.assign({
|
|
privateNamesMap,
|
|
classRef: ref,
|
|
file: state
|
|
}, handler, {
|
|
noDocumentAll,
|
|
noUninitializedPrivateFieldAccess,
|
|
innerBinding
|
|
}));
|
|
body.traverse(privateInVisitor, {
|
|
privateNamesMap,
|
|
classRef: ref,
|
|
file: state,
|
|
privateFieldsAsProperties,
|
|
innerBinding
|
|
});
|
|
}
|
|
function buildPrivateFieldInitLoose(ref, prop, privateNamesMap) {
|
|
const {
|
|
id
|
|
} = privateNamesMap.get(prop.node.key.id.name);
|
|
const value = prop.node.value || prop.scope.buildUndefinedNode();
|
|
return inheritPropComments(_core.template.statement.ast`
|
|
Object.defineProperty(${ref}, ${_core.types.cloneNode(id)}, {
|
|
// configurable is false by default
|
|
// enumerable is false by default
|
|
writable: true,
|
|
value: ${value}
|
|
});
|
|
`, prop);
|
|
}
|
|
function buildPrivateInstanceFieldInitSpec(ref, prop, privateNamesMap, state) {
|
|
const {
|
|
id
|
|
} = privateNamesMap.get(prop.node.key.id.name);
|
|
const value = prop.node.value || prop.scope.buildUndefinedNode();
|
|
{
|
|
if (!state.availableHelper("classPrivateFieldInitSpec")) {
|
|
return inheritPropComments(_core.template.statement.ast`${_core.types.cloneNode(id)}.set(${ref}, {
|
|
// configurable is always false for private elements
|
|
// enumerable is always false for private elements
|
|
writable: true,
|
|
value: ${value},
|
|
})`, prop);
|
|
}
|
|
}
|
|
const helper = state.addHelper("classPrivateFieldInitSpec");
|
|
return inheritLoc(inheritPropComments(_core.types.expressionStatement(_core.types.callExpression(helper, [_core.types.thisExpression(), inheritLoc(_core.types.cloneNode(id), prop.node.key), newHelpers(state) ? value : _core.template.expression.ast`{ writable: true, value: ${value} }`])), prop), prop.node);
|
|
}
|
|
function buildPrivateStaticFieldInitSpec(prop, privateNamesMap, noUninitializedPrivateFieldAccess) {
|
|
const privateName = privateNamesMap.get(prop.node.key.id.name);
|
|
const value = noUninitializedPrivateFieldAccess ? prop.node.value : _core.template.expression.ast`{
|
|
_: ${prop.node.value || _core.types.buildUndefinedNode()}
|
|
}`;
|
|
return inheritPropComments(_core.types.variableDeclaration("var", [_core.types.variableDeclarator(_core.types.cloneNode(privateName.id), value)]), prop);
|
|
}
|
|
{
|
|
var buildPrivateStaticFieldInitSpecOld = function (prop, privateNamesMap) {
|
|
const privateName = privateNamesMap.get(prop.node.key.id.name);
|
|
const {
|
|
id,
|
|
getId,
|
|
setId,
|
|
initAdded
|
|
} = privateName;
|
|
const isGetterOrSetter = getId || setId;
|
|
if (!prop.isProperty() && (initAdded || !isGetterOrSetter)) return;
|
|
if (isGetterOrSetter) {
|
|
privateNamesMap.set(prop.node.key.id.name, Object.assign({}, privateName, {
|
|
initAdded: true
|
|
}));
|
|
return inheritPropComments(_core.template.statement.ast`
|
|
var ${_core.types.cloneNode(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()}
|
|
}
|
|
`, prop);
|
|
}
|
|
const value = prop.node.value || prop.scope.buildUndefinedNode();
|
|
return inheritPropComments(_core.template.statement.ast`
|
|
var ${_core.types.cloneNode(id)} = {
|
|
// configurable is false by default
|
|
// enumerable is false by default
|
|
writable: true,
|
|
value: ${value}
|
|
};
|
|
`, prop);
|
|
};
|
|
}
|
|
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 inheritPropComments(_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}
|
|
});
|
|
`, prop);
|
|
}
|
|
const isGetterOrSetter = getId || setId;
|
|
if (isGetterOrSetter) {
|
|
privateNamesMap.set(prop.node.key.id.name, Object.assign({}, privateName, {
|
|
initAdded: true
|
|
}));
|
|
return inheritPropComments(_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()}
|
|
});
|
|
`, prop);
|
|
}
|
|
}
|
|
function buildPrivateInstanceMethodInitSpec(ref, prop, privateNamesMap, state) {
|
|
const privateName = privateNamesMap.get(prop.node.key.id.name);
|
|
if (privateName.initAdded) return;
|
|
if (!newHelpers(state)) {
|
|
const isGetterOrSetter = privateName.getId || privateName.setId;
|
|
if (isGetterOrSetter) {
|
|
return buildPrivateAccessorInitialization(ref, prop, privateNamesMap, state);
|
|
}
|
|
}
|
|
return buildPrivateInstanceMethodInitialization(ref, prop, privateNamesMap, state);
|
|
}
|
|
function buildPrivateAccessorInitialization(ref, prop, privateNamesMap, state) {
|
|
const privateName = privateNamesMap.get(prop.node.key.id.name);
|
|
const {
|
|
id,
|
|
getId,
|
|
setId
|
|
} = privateName;
|
|
privateNamesMap.set(prop.node.key.id.name, Object.assign({}, privateName, {
|
|
initAdded: true
|
|
}));
|
|
{
|
|
if (!state.availableHelper("classPrivateFieldInitSpec")) {
|
|
return inheritPropComments(_core.template.statement.ast`
|
|
${id}.set(${ref}, {
|
|
get: ${getId ? getId.name : prop.scope.buildUndefinedNode()},
|
|
set: ${setId ? setId.name : prop.scope.buildUndefinedNode()}
|
|
});
|
|
`, prop);
|
|
}
|
|
}
|
|
const helper = state.addHelper("classPrivateFieldInitSpec");
|
|
return inheritLoc(inheritPropComments(_core.template.statement.ast`${helper}(
|
|
${_core.types.thisExpression()},
|
|
${_core.types.cloneNode(id)},
|
|
{
|
|
get: ${getId ? getId.name : prop.scope.buildUndefinedNode()},
|
|
set: ${setId ? setId.name : prop.scope.buildUndefinedNode()}
|
|
},
|
|
)`, prop), prop.node);
|
|
}
|
|
function buildPrivateInstanceMethodInitialization(ref, prop, privateNamesMap, state) {
|
|
const privateName = privateNamesMap.get(prop.node.key.id.name);
|
|
const {
|
|
id
|
|
} = privateName;
|
|
{
|
|
if (!state.availableHelper("classPrivateMethodInitSpec")) {
|
|
return inheritPropComments(_core.template.statement.ast`${id}.add(${ref})`, prop);
|
|
}
|
|
}
|
|
const helper = state.addHelper("classPrivateMethodInitSpec");
|
|
return inheritPropComments(_core.template.statement.ast`${helper}(
|
|
${_core.types.thisExpression()},
|
|
${_core.types.cloneNode(id)}
|
|
)`, prop);
|
|
}
|
|
function buildPublicFieldInitLoose(ref, prop) {
|
|
const {
|
|
key,
|
|
computed
|
|
} = prop.node;
|
|
const value = prop.node.value || prop.scope.buildUndefinedNode();
|
|
return inheritPropComments(_core.types.expressionStatement(_core.types.assignmentExpression("=", _core.types.memberExpression(ref, key, computed || _core.types.isLiteral(key)), value)), prop);
|
|
}
|
|
function buildPublicFieldInitSpec(ref, prop, state) {
|
|
const {
|
|
key,
|
|
computed
|
|
} = prop.node;
|
|
const value = prop.node.value || prop.scope.buildUndefinedNode();
|
|
return inheritPropComments(_core.types.expressionStatement(_core.types.callExpression(state.addHelper("defineProperty"), [ref, computed || _core.types.isLiteral(key) ? key : _core.types.stringLiteral(key.name), value])), prop);
|
|
}
|
|
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 isGetterOrSetter = getId || setId;
|
|
if (isGetterOrSetter) {
|
|
privateNamesMap.set(prop.node.key.id.name, Object.assign({}, privateName, {
|
|
initAdded: true
|
|
}));
|
|
return inheritPropComments(_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()}
|
|
})
|
|
`, prop);
|
|
}
|
|
return inheritPropComments(_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}
|
|
});
|
|
`, prop);
|
|
}
|
|
function buildPrivateMethodDeclaration(file, prop, privateNamesMap, privateFieldsAsSymbolsOrProperties = 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 isGetter = getId && params.length === 0;
|
|
const isSetter = setId && params.length > 0;
|
|
if (isGetter && getterDeclared || isSetter && setterDeclared) {
|
|
privateNamesMap.set(prop.node.key.id.name, Object.assign({}, privateName, {
|
|
initAdded: true
|
|
}));
|
|
return null;
|
|
}
|
|
if (newHelpers(file) && (isGetter || isSetter) && !privateFieldsAsSymbolsOrProperties) {
|
|
const scope = prop.get("body").scope;
|
|
const thisArg = scope.generateUidIdentifier("this");
|
|
const state = {
|
|
thisRef: thisArg,
|
|
argumentsPath: []
|
|
};
|
|
prop.traverse(thisContextVisitor, state);
|
|
if (state.argumentsPath.length) {
|
|
const argumentsId = scope.generateUidIdentifier("arguments");
|
|
scope.push({
|
|
id: argumentsId,
|
|
init: _core.template.expression.ast`[].slice.call(arguments, 1)`
|
|
});
|
|
for (const path of state.argumentsPath) {
|
|
path.replaceWith(_core.types.cloneNode(argumentsId));
|
|
}
|
|
}
|
|
params.unshift(_core.types.cloneNode(thisArg));
|
|
}
|
|
let declId = methodId;
|
|
if (isGetter) {
|
|
privateNamesMap.set(prop.node.key.id.name, Object.assign({}, privateName, {
|
|
getterDeclared: true,
|
|
initAdded: true
|
|
}));
|
|
declId = getId;
|
|
} else if (isSetter) {
|
|
privateNamesMap.set(prop.node.key.id.name, Object.assign({}, privateName, {
|
|
setterDeclared: true,
|
|
initAdded: true
|
|
}));
|
|
declId = setId;
|
|
} else if (isStatic && !privateFieldsAsSymbolsOrProperties) {
|
|
declId = id;
|
|
}
|
|
return inheritPropComments(_core.types.functionDeclaration(_core.types.cloneNode(declId), params, body, generator, async), prop);
|
|
}
|
|
const thisContextVisitor = _core.traverse.visitors.merge([{
|
|
Identifier(path, state) {
|
|
if (state.argumentsPath && path.node.name === "arguments") {
|
|
state.argumentsPath.push(path);
|
|
}
|
|
},
|
|
UnaryExpression(path) {
|
|
const {
|
|
node
|
|
} = path;
|
|
if (node.operator === "delete") {
|
|
const argument = (0, _helperSkipTransparentExpressionWrappers.skipTransparentExprWrapperNodes)(node.argument);
|
|
if (_core.types.isThisExpression(argument)) {
|
|
path.replaceWith(_core.types.booleanLiteral(true));
|
|
}
|
|
}
|
|
},
|
|
ThisExpression(path, state) {
|
|
state.needsClassRef = true;
|
|
path.replaceWith(_core.types.cloneNode(state.thisRef));
|
|
},
|
|
MetaProperty(path) {
|
|
const {
|
|
node,
|
|
scope
|
|
} = path;
|
|
if (node.meta.name === "new" && node.property.name === "target") {
|
|
path.replaceWith(scope.buildUndefinedNode());
|
|
}
|
|
}
|
|
}, _helperEnvironmentVisitor.default]);
|
|
const innerReferencesVisitor = {
|
|
ReferencedIdentifier(path, state) {
|
|
if (path.scope.bindingIdentifierEquals(path.node.name, state.innerBinding)) {
|
|
state.needsClassRef = true;
|
|
path.node.name = state.thisRef.name;
|
|
}
|
|
}
|
|
};
|
|
function replaceThisContext(path, ref, innerBindingRef) {
|
|
var _state$thisRef;
|
|
const state = {
|
|
thisRef: ref,
|
|
needsClassRef: false,
|
|
innerBinding: innerBindingRef
|
|
};
|
|
if (!path.isMethod()) {
|
|
path.traverse(thisContextVisitor, state);
|
|
}
|
|
if (innerBindingRef != null && (_state$thisRef = state.thisRef) != null && _state$thisRef.name && state.thisRef.name !== innerBindingRef.name) {
|
|
path.traverse(innerReferencesVisitor, state);
|
|
}
|
|
return state.needsClassRef;
|
|
}
|
|
function isNameOrLength({
|
|
key,
|
|
computed
|
|
}) {
|
|
if (key.type === "Identifier") {
|
|
return !computed && (key.name === "name" || key.name === "length");
|
|
}
|
|
if (key.type === "StringLiteral") {
|
|
return key.value === "name" || key.value === "length";
|
|
}
|
|
return false;
|
|
}
|
|
function inheritPropComments(node, prop) {
|
|
_core.types.inheritLeadingComments(node, prop.node);
|
|
_core.types.inheritInnerComments(node, prop.node);
|
|
return node;
|
|
}
|
|
function inheritLoc(node, original) {
|
|
node.start = original.start;
|
|
node.end = original.end;
|
|
node.loc = original.loc;
|
|
return node;
|
|
}
|
|
function buildFieldsInitNodes(ref, superRef, props, privateNamesMap, file, setPublicClassFields, privateFieldsAsSymbolsOrProperties, noUninitializedPrivateFieldAccess, constantSuper, innerBindingRef) {
|
|
var _ref, _ref2;
|
|
let classRefFlags = 0;
|
|
let injectSuperRef;
|
|
const staticNodes = [];
|
|
const instanceNodes = [];
|
|
let lastInstanceNodeReturnsThis = false;
|
|
const pureStaticNodes = [];
|
|
let classBindingNode = null;
|
|
const getSuperRef = _core.types.isIdentifier(superRef) ? () => superRef : () => {
|
|
var _injectSuperRef;
|
|
(_injectSuperRef = injectSuperRef) != null ? _injectSuperRef : injectSuperRef = props[0].scope.generateUidIdentifierBasedOnNode(superRef);
|
|
return injectSuperRef;
|
|
};
|
|
const classRefForInnerBinding = (_ref = ref) != null ? _ref : props[0].scope.generateUidIdentifier((innerBindingRef == null ? void 0 : innerBindingRef.name) || "Class");
|
|
(_ref2 = ref) != null ? _ref2 : ref = _core.types.cloneNode(innerBindingRef);
|
|
for (const prop of props) {
|
|
prop.isClassProperty() && ts.assertFieldTransformed(prop);
|
|
const isStatic = !(_core.types.isStaticBlock != null && _core.types.isStaticBlock(prop.node)) && prop.node.static;
|
|
const isInstance = !isStatic;
|
|
const isPrivate = prop.isPrivate();
|
|
const isPublic = !isPrivate;
|
|
const isField = prop.isProperty();
|
|
const isMethod = !isField;
|
|
const isStaticBlock = prop.isStaticBlock == null ? void 0 : prop.isStaticBlock();
|
|
if (isStatic) classRefFlags |= 1;
|
|
if (isStatic || isMethod && isPrivate || isStaticBlock) {
|
|
new _helperReplaceSupers.default({
|
|
methodPath: prop,
|
|
constantSuper,
|
|
file: file,
|
|
refToPreserve: innerBindingRef,
|
|
getSuperRef,
|
|
getObjectRef() {
|
|
classRefFlags |= 2;
|
|
if (isStatic || isStaticBlock) {
|
|
return classRefForInnerBinding;
|
|
} else {
|
|
return _core.types.memberExpression(classRefForInnerBinding, _core.types.identifier("prototype"));
|
|
}
|
|
}
|
|
}).replace();
|
|
const replaced = replaceThisContext(prop, classRefForInnerBinding, innerBindingRef);
|
|
if (replaced) {
|
|
classRefFlags |= 2;
|
|
}
|
|
}
|
|
lastInstanceNodeReturnsThis = false;
|
|
switch (true) {
|
|
case isStaticBlock:
|
|
{
|
|
const blockBody = prop.node.body;
|
|
if (blockBody.length === 1 && _core.types.isExpressionStatement(blockBody[0])) {
|
|
staticNodes.push(inheritPropComments(blockBody[0], prop));
|
|
} else {
|
|
staticNodes.push(_core.types.inheritsComments(_core.template.statement.ast`(() => { ${blockBody} })()`, prop.node));
|
|
}
|
|
break;
|
|
}
|
|
case isStatic && isPrivate && isField && privateFieldsAsSymbolsOrProperties:
|
|
staticNodes.push(buildPrivateFieldInitLoose(_core.types.cloneNode(ref), prop, privateNamesMap));
|
|
break;
|
|
case isStatic && isPrivate && isField && !privateFieldsAsSymbolsOrProperties:
|
|
if (!newHelpers(file)) {
|
|
staticNodes.push(buildPrivateStaticFieldInitSpecOld(prop, privateNamesMap));
|
|
} else {
|
|
staticNodes.push(buildPrivateStaticFieldInitSpec(prop, privateNamesMap, noUninitializedPrivateFieldAccess));
|
|
}
|
|
break;
|
|
case isStatic && isPublic && isField && setPublicClassFields:
|
|
if (!isNameOrLength(prop.node)) {
|
|
staticNodes.push(buildPublicFieldInitLoose(_core.types.cloneNode(ref), prop));
|
|
break;
|
|
}
|
|
case isStatic && isPublic && isField && !setPublicClassFields:
|
|
staticNodes.push(buildPublicFieldInitSpec(_core.types.cloneNode(ref), prop, file));
|
|
break;
|
|
case isInstance && isPrivate && isField && privateFieldsAsSymbolsOrProperties:
|
|
instanceNodes.push(buildPrivateFieldInitLoose(_core.types.thisExpression(), prop, privateNamesMap));
|
|
break;
|
|
case isInstance && isPrivate && isField && !privateFieldsAsSymbolsOrProperties:
|
|
instanceNodes.push(buildPrivateInstanceFieldInitSpec(_core.types.thisExpression(), prop, privateNamesMap, file));
|
|
break;
|
|
case isInstance && isPrivate && isMethod && privateFieldsAsSymbolsOrProperties:
|
|
instanceNodes.unshift(buildPrivateMethodInitLoose(_core.types.thisExpression(), prop, privateNamesMap));
|
|
pureStaticNodes.push(buildPrivateMethodDeclaration(file, prop, privateNamesMap, privateFieldsAsSymbolsOrProperties));
|
|
break;
|
|
case isInstance && isPrivate && isMethod && !privateFieldsAsSymbolsOrProperties:
|
|
instanceNodes.unshift(buildPrivateInstanceMethodInitSpec(_core.types.thisExpression(), prop, privateNamesMap, file));
|
|
pureStaticNodes.push(buildPrivateMethodDeclaration(file, prop, privateNamesMap, privateFieldsAsSymbolsOrProperties));
|
|
break;
|
|
case isStatic && isPrivate && isMethod && !privateFieldsAsSymbolsOrProperties:
|
|
if (!newHelpers(file)) {
|
|
staticNodes.unshift(buildPrivateStaticFieldInitSpecOld(prop, privateNamesMap));
|
|
}
|
|
pureStaticNodes.push(buildPrivateMethodDeclaration(file, prop, privateNamesMap, privateFieldsAsSymbolsOrProperties));
|
|
break;
|
|
case isStatic && isPrivate && isMethod && privateFieldsAsSymbolsOrProperties:
|
|
staticNodes.unshift(buildPrivateStaticMethodInitLoose(_core.types.cloneNode(ref), prop, file, privateNamesMap));
|
|
pureStaticNodes.push(buildPrivateMethodDeclaration(file, prop, privateNamesMap, privateFieldsAsSymbolsOrProperties));
|
|
break;
|
|
case isInstance && isPublic && isField && setPublicClassFields:
|
|
instanceNodes.push(buildPublicFieldInitLoose(_core.types.thisExpression(), prop));
|
|
break;
|
|
case isInstance && isPublic && isField && !setPublicClassFields:
|
|
lastInstanceNodeReturnsThis = true;
|
|
instanceNodes.push(buildPublicFieldInitSpec(_core.types.thisExpression(), prop, file));
|
|
break;
|
|
default:
|
|
throw new Error("Unreachable.");
|
|
}
|
|
}
|
|
if (classRefFlags & 2 && innerBindingRef != null) {
|
|
classBindingNode = _core.types.expressionStatement(_core.types.assignmentExpression("=", _core.types.cloneNode(classRefForInnerBinding), _core.types.cloneNode(innerBindingRef)));
|
|
}
|
|
return {
|
|
staticNodes: staticNodes.filter(Boolean),
|
|
instanceNodes: instanceNodes.filter(Boolean),
|
|
lastInstanceNodeReturnsThis,
|
|
pureStaticNodes: pureStaticNodes.filter(Boolean),
|
|
classBindingNode,
|
|
wrapClass(path) {
|
|
for (const prop of props) {
|
|
prop.node.leadingComments = null;
|
|
prop.remove();
|
|
}
|
|
if (injectSuperRef) {
|
|
path.scope.push({
|
|
id: _core.types.cloneNode(injectSuperRef)
|
|
});
|
|
path.set("superClass", _core.types.assignmentExpression("=", injectSuperRef, path.node.superClass));
|
|
}
|
|
if (classRefFlags !== 0) {
|
|
if (path.isClassExpression()) {
|
|
path.scope.push({
|
|
id: ref
|
|
});
|
|
path.replaceWith(_core.types.assignmentExpression("=", _core.types.cloneNode(ref), path.node));
|
|
} else {
|
|
if (innerBindingRef == null) {
|
|
path.node.id = ref;
|
|
}
|
|
if (classBindingNode != null) {
|
|
path.scope.push({
|
|
id: classRefForInnerBinding
|
|
});
|
|
}
|
|
}
|
|
}
|
|
return path;
|
|
}
|
|
};
|
|
}
|
|
|
|
//# sourceMappingURL=fields.js.map
|