fields.js 26 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902
  1. "use strict";
  2. Object.defineProperty(exports, "__esModule", {
  3. value: true
  4. });
  5. exports.buildPrivateNamesMap = buildPrivateNamesMap;
  6. exports.buildPrivateNamesNodes = buildPrivateNamesNodes;
  7. exports.transformPrivateNamesUsage = transformPrivateNamesUsage;
  8. exports.buildFieldsInitNodes = buildFieldsInitNodes;
  9. var _core = require("@babel/core");
  10. var _helperReplaceSupers = require("@babel/helper-replace-supers");
  11. var _helperMemberExpressionToFunctions = require("@babel/helper-member-expression-to-functions");
  12. var _helperOptimiseCallExpression = require("@babel/helper-optimise-call-expression");
  13. var _helperAnnotateAsPure = require("@babel/helper-annotate-as-pure");
  14. var ts = require("./typescript");
  15. function buildPrivateNamesMap(props) {
  16. const privateNamesMap = new Map();
  17. for (const prop of props) {
  18. if (prop.isPrivate()) {
  19. const {
  20. name
  21. } = prop.node.key.id;
  22. const update = privateNamesMap.has(name) ? privateNamesMap.get(name) : {
  23. id: prop.scope.generateUidIdentifier(name),
  24. static: prop.node.static,
  25. method: !prop.isProperty()
  26. };
  27. if (prop.isClassPrivateMethod()) {
  28. if (prop.node.kind === "get") {
  29. update.getId = prop.scope.generateUidIdentifier(`get_${name}`);
  30. } else if (prop.node.kind === "set") {
  31. update.setId = prop.scope.generateUidIdentifier(`set_${name}`);
  32. } else if (prop.node.kind === "method") {
  33. update.methodId = prop.scope.generateUidIdentifier(name);
  34. }
  35. }
  36. privateNamesMap.set(name, update);
  37. }
  38. }
  39. return privateNamesMap;
  40. }
  41. function buildPrivateNamesNodes(privateNamesMap, privateFieldsAsProperties, state) {
  42. const initNodes = [];
  43. for (const [name, value] of privateNamesMap) {
  44. const {
  45. static: isStatic,
  46. method: isMethod,
  47. getId,
  48. setId
  49. } = value;
  50. const isAccessor = getId || setId;
  51. const id = _core.types.cloneNode(value.id);
  52. let init;
  53. if (privateFieldsAsProperties) {
  54. init = _core.types.callExpression(state.addHelper("classPrivateFieldLooseKey"), [_core.types.stringLiteral(name)]);
  55. } else if (!isStatic) {
  56. init = _core.types.newExpression(_core.types.identifier(!isMethod || isAccessor ? "WeakMap" : "WeakSet"), []);
  57. }
  58. if (init) {
  59. (0, _helperAnnotateAsPure.default)(init);
  60. initNodes.push(_core.template.statement.ast`var ${id} = ${init}`);
  61. }
  62. }
  63. return initNodes;
  64. }
  65. function privateNameVisitorFactory(visitor) {
  66. const privateNameVisitor = Object.assign({}, visitor, {
  67. Class(path) {
  68. const {
  69. privateNamesMap
  70. } = this;
  71. const body = path.get("body.body");
  72. const visiblePrivateNames = new Map(privateNamesMap);
  73. const redeclared = [];
  74. for (const prop of body) {
  75. if (!prop.isPrivate()) continue;
  76. const {
  77. name
  78. } = prop.node.key.id;
  79. visiblePrivateNames.delete(name);
  80. redeclared.push(name);
  81. }
  82. if (!redeclared.length) {
  83. return;
  84. }
  85. path.get("body").traverse(nestedVisitor, Object.assign({}, this, {
  86. redeclared
  87. }));
  88. path.traverse(privateNameVisitor, Object.assign({}, this, {
  89. privateNamesMap: visiblePrivateNames
  90. }));
  91. path.skipKey("body");
  92. }
  93. });
  94. const nestedVisitor = _core.traverse.visitors.merge([Object.assign({}, visitor), _helperReplaceSupers.environmentVisitor]);
  95. return privateNameVisitor;
  96. }
  97. const privateNameVisitor = privateNameVisitorFactory({
  98. PrivateName(path, {
  99. noDocumentAll
  100. }) {
  101. const {
  102. privateNamesMap,
  103. redeclared
  104. } = this;
  105. const {
  106. node,
  107. parentPath
  108. } = path;
  109. if (!parentPath.isMemberExpression({
  110. property: node
  111. }) && !parentPath.isOptionalMemberExpression({
  112. property: node
  113. })) {
  114. return;
  115. }
  116. const {
  117. name
  118. } = node.id;
  119. if (!privateNamesMap.has(name)) return;
  120. if (redeclared && redeclared.includes(name)) return;
  121. this.handle(parentPath, noDocumentAll);
  122. }
  123. });
  124. function unshadow(name, scope, innerBinding) {
  125. while ((_scope = scope) != null && _scope.hasBinding(name) && !scope.bindingIdentifierEquals(name, innerBinding)) {
  126. var _scope;
  127. scope.rename(name);
  128. scope = scope.parent;
  129. }
  130. }
  131. const privateInVisitor = privateNameVisitorFactory({
  132. BinaryExpression(path) {
  133. const {
  134. operator,
  135. left,
  136. right
  137. } = path.node;
  138. if (operator !== "in") return;
  139. if (!_core.types.isPrivateName(left)) return;
  140. const {
  141. privateFieldsAsProperties,
  142. privateNamesMap,
  143. redeclared
  144. } = this;
  145. const {
  146. name
  147. } = left.id;
  148. if (!privateNamesMap.has(name)) return;
  149. if (redeclared && redeclared.includes(name)) return;
  150. unshadow(this.classRef.name, path.scope, this.innerBinding);
  151. if (privateFieldsAsProperties) {
  152. const {
  153. id
  154. } = privateNamesMap.get(name);
  155. path.replaceWith(_core.template.expression.ast`
  156. Object.prototype.hasOwnProperty.call(${right}, ${_core.types.cloneNode(id)})
  157. `);
  158. return;
  159. }
  160. const {
  161. id,
  162. static: isStatic
  163. } = privateNamesMap.get(name);
  164. if (isStatic) {
  165. path.replaceWith(_core.template.expression.ast`${right} === ${this.classRef}`);
  166. return;
  167. }
  168. path.replaceWith(_core.template.expression.ast`${_core.types.cloneNode(id)}.has(${right})`);
  169. }
  170. });
  171. const privateNameHandlerSpec = {
  172. memoise(member, count) {
  173. const {
  174. scope
  175. } = member;
  176. const {
  177. object
  178. } = member.node;
  179. const memo = scope.maybeGenerateMemoised(object);
  180. if (!memo) {
  181. return;
  182. }
  183. this.memoiser.set(object, memo, count);
  184. },
  185. receiver(member) {
  186. const {
  187. object
  188. } = member.node;
  189. if (this.memoiser.has(object)) {
  190. return _core.types.cloneNode(this.memoiser.get(object));
  191. }
  192. return _core.types.cloneNode(object);
  193. },
  194. get(member) {
  195. const {
  196. classRef,
  197. privateNamesMap,
  198. file,
  199. innerBinding
  200. } = this;
  201. const {
  202. name
  203. } = member.node.property.id;
  204. const {
  205. id,
  206. static: isStatic,
  207. method: isMethod,
  208. methodId,
  209. getId,
  210. setId
  211. } = privateNamesMap.get(name);
  212. const isAccessor = getId || setId;
  213. if (isStatic) {
  214. const helperName = isMethod && !isAccessor ? "classStaticPrivateMethodGet" : "classStaticPrivateFieldSpecGet";
  215. unshadow(classRef.name, member.scope, innerBinding);
  216. return _core.types.callExpression(file.addHelper(helperName), [this.receiver(member), _core.types.cloneNode(classRef), _core.types.cloneNode(id)]);
  217. }
  218. if (isMethod) {
  219. if (isAccessor) {
  220. if (!getId && setId) {
  221. if (file.availableHelper("writeOnlyError")) {
  222. return _core.types.sequenceExpression([this.receiver(member), _core.types.callExpression(file.addHelper("writeOnlyError"), [_core.types.stringLiteral(`#${name}`)])]);
  223. }
  224. console.warn(`@babel/helpers is outdated, update it to silence this warning.`);
  225. }
  226. return _core.types.callExpression(file.addHelper("classPrivateFieldGet"), [this.receiver(member), _core.types.cloneNode(id)]);
  227. }
  228. return _core.types.callExpression(file.addHelper("classPrivateMethodGet"), [this.receiver(member), _core.types.cloneNode(id), _core.types.cloneNode(methodId)]);
  229. }
  230. return _core.types.callExpression(file.addHelper("classPrivateFieldGet"), [this.receiver(member), _core.types.cloneNode(id)]);
  231. },
  232. boundGet(member) {
  233. this.memoise(member, 1);
  234. return _core.types.callExpression(_core.types.memberExpression(this.get(member), _core.types.identifier("bind")), [this.receiver(member)]);
  235. },
  236. set(member, value) {
  237. const {
  238. classRef,
  239. privateNamesMap,
  240. file
  241. } = this;
  242. const {
  243. name
  244. } = member.node.property.id;
  245. const {
  246. id,
  247. static: isStatic,
  248. method: isMethod,
  249. setId,
  250. getId
  251. } = privateNamesMap.get(name);
  252. const isAccessor = getId || setId;
  253. if (isStatic) {
  254. const helperName = isMethod && !isAccessor ? "classStaticPrivateMethodSet" : "classStaticPrivateFieldSpecSet";
  255. return _core.types.callExpression(file.addHelper(helperName), [this.receiver(member), _core.types.cloneNode(classRef), _core.types.cloneNode(id), value]);
  256. }
  257. if (isMethod) {
  258. if (setId) {
  259. return _core.types.callExpression(file.addHelper("classPrivateFieldSet"), [this.receiver(member), _core.types.cloneNode(id), value]);
  260. }
  261. return _core.types.sequenceExpression([this.receiver(member), value, _core.types.callExpression(file.addHelper("readOnlyError"), [_core.types.stringLiteral(`#${name}`)])]);
  262. }
  263. return _core.types.callExpression(file.addHelper("classPrivateFieldSet"), [this.receiver(member), _core.types.cloneNode(id), value]);
  264. },
  265. destructureSet(member) {
  266. const {
  267. classRef,
  268. privateNamesMap,
  269. file
  270. } = this;
  271. const {
  272. name
  273. } = member.node.property.id;
  274. const {
  275. id,
  276. static: isStatic
  277. } = privateNamesMap.get(name);
  278. if (isStatic) {
  279. try {
  280. var helper = file.addHelper("classStaticPrivateFieldDestructureSet");
  281. } catch (_unused) {
  282. 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.");
  283. }
  284. return _core.types.memberExpression(_core.types.callExpression(helper, [this.receiver(member), _core.types.cloneNode(classRef), _core.types.cloneNode(id)]), _core.types.identifier("value"));
  285. }
  286. return _core.types.memberExpression(_core.types.callExpression(file.addHelper("classPrivateFieldDestructureSet"), [this.receiver(member), _core.types.cloneNode(id)]), _core.types.identifier("value"));
  287. },
  288. call(member, args) {
  289. this.memoise(member, 1);
  290. return (0, _helperOptimiseCallExpression.default)(this.get(member), this.receiver(member), args, false);
  291. },
  292. optionalCall(member, args) {
  293. this.memoise(member, 1);
  294. return (0, _helperOptimiseCallExpression.default)(this.get(member), this.receiver(member), args, true);
  295. }
  296. };
  297. const privateNameHandlerLoose = {
  298. get(member) {
  299. const {
  300. privateNamesMap,
  301. file
  302. } = this;
  303. const {
  304. object
  305. } = member.node;
  306. const {
  307. name
  308. } = member.node.property.id;
  309. return _core.template.expression`BASE(REF, PROP)[PROP]`({
  310. BASE: file.addHelper("classPrivateFieldLooseBase"),
  311. REF: _core.types.cloneNode(object),
  312. PROP: _core.types.cloneNode(privateNamesMap.get(name).id)
  313. });
  314. },
  315. set() {
  316. throw new Error("private name handler with loose = true don't need set()");
  317. },
  318. boundGet(member) {
  319. return _core.types.callExpression(_core.types.memberExpression(this.get(member), _core.types.identifier("bind")), [_core.types.cloneNode(member.node.object)]);
  320. },
  321. simpleSet(member) {
  322. return this.get(member);
  323. },
  324. destructureSet(member) {
  325. return this.get(member);
  326. },
  327. call(member, args) {
  328. return _core.types.callExpression(this.get(member), args);
  329. },
  330. optionalCall(member, args) {
  331. return _core.types.optionalCallExpression(this.get(member), args, true);
  332. }
  333. };
  334. function transformPrivateNamesUsage(ref, path, privateNamesMap, {
  335. privateFieldsAsProperties,
  336. noDocumentAll,
  337. innerBinding
  338. }, state) {
  339. if (!privateNamesMap.size) return;
  340. const body = path.get("body");
  341. const handler = privateFieldsAsProperties ? privateNameHandlerLoose : privateNameHandlerSpec;
  342. (0, _helperMemberExpressionToFunctions.default)(body, privateNameVisitor, Object.assign({
  343. privateNamesMap,
  344. classRef: ref,
  345. file: state
  346. }, handler, {
  347. noDocumentAll,
  348. innerBinding
  349. }));
  350. body.traverse(privateInVisitor, {
  351. privateNamesMap,
  352. classRef: ref,
  353. file: state,
  354. privateFieldsAsProperties,
  355. innerBinding
  356. });
  357. }
  358. function buildPrivateFieldInitLoose(ref, prop, privateNamesMap) {
  359. const {
  360. id
  361. } = privateNamesMap.get(prop.node.key.id.name);
  362. const value = prop.node.value || prop.scope.buildUndefinedNode();
  363. return _core.template.statement.ast`
  364. Object.defineProperty(${ref}, ${_core.types.cloneNode(id)}, {
  365. // configurable is false by default
  366. // enumerable is false by default
  367. writable: true,
  368. value: ${value}
  369. });
  370. `;
  371. }
  372. function buildPrivateInstanceFieldInitSpec(ref, prop, privateNamesMap, state) {
  373. const {
  374. id
  375. } = privateNamesMap.get(prop.node.key.id.name);
  376. const value = prop.node.value || prop.scope.buildUndefinedNode();
  377. {
  378. if (!state.availableHelper("classPrivateFieldInitSpec")) {
  379. return _core.template.statement.ast`${_core.types.cloneNode(id)}.set(${ref}, {
  380. // configurable is always false for private elements
  381. // enumerable is always false for private elements
  382. writable: true,
  383. value: ${value},
  384. })`;
  385. }
  386. }
  387. const helper = state.addHelper("classPrivateFieldInitSpec");
  388. return _core.template.statement.ast`${helper}(
  389. ${_core.types.thisExpression()},
  390. ${_core.types.cloneNode(id)},
  391. {
  392. writable: true,
  393. value: ${value}
  394. },
  395. )`;
  396. }
  397. function buildPrivateStaticFieldInitSpec(prop, privateNamesMap) {
  398. const privateName = privateNamesMap.get(prop.node.key.id.name);
  399. const {
  400. id,
  401. getId,
  402. setId,
  403. initAdded
  404. } = privateName;
  405. const isAccessor = getId || setId;
  406. if (!prop.isProperty() && (initAdded || !isAccessor)) return;
  407. if (isAccessor) {
  408. privateNamesMap.set(prop.node.key.id.name, Object.assign({}, privateName, {
  409. initAdded: true
  410. }));
  411. return _core.template.statement.ast`
  412. var ${_core.types.cloneNode(id)} = {
  413. // configurable is false by default
  414. // enumerable is false by default
  415. // writable is false by default
  416. get: ${getId ? getId.name : prop.scope.buildUndefinedNode()},
  417. set: ${setId ? setId.name : prop.scope.buildUndefinedNode()}
  418. }
  419. `;
  420. }
  421. const value = prop.node.value || prop.scope.buildUndefinedNode();
  422. return _core.template.statement.ast`
  423. var ${_core.types.cloneNode(id)} = {
  424. // configurable is false by default
  425. // enumerable is false by default
  426. writable: true,
  427. value: ${value}
  428. };
  429. `;
  430. }
  431. function buildPrivateMethodInitLoose(ref, prop, privateNamesMap) {
  432. const privateName = privateNamesMap.get(prop.node.key.id.name);
  433. const {
  434. methodId,
  435. id,
  436. getId,
  437. setId,
  438. initAdded
  439. } = privateName;
  440. if (initAdded) return;
  441. if (methodId) {
  442. return _core.template.statement.ast`
  443. Object.defineProperty(${ref}, ${id}, {
  444. // configurable is false by default
  445. // enumerable is false by default
  446. // writable is false by default
  447. value: ${methodId.name}
  448. });
  449. `;
  450. }
  451. const isAccessor = getId || setId;
  452. if (isAccessor) {
  453. privateNamesMap.set(prop.node.key.id.name, Object.assign({}, privateName, {
  454. initAdded: true
  455. }));
  456. return _core.template.statement.ast`
  457. Object.defineProperty(${ref}, ${id}, {
  458. // configurable is false by default
  459. // enumerable is false by default
  460. // writable is false by default
  461. get: ${getId ? getId.name : prop.scope.buildUndefinedNode()},
  462. set: ${setId ? setId.name : prop.scope.buildUndefinedNode()}
  463. });
  464. `;
  465. }
  466. }
  467. function buildPrivateInstanceMethodInitSpec(ref, prop, privateNamesMap, state) {
  468. const privateName = privateNamesMap.get(prop.node.key.id.name);
  469. const {
  470. getId,
  471. setId,
  472. initAdded
  473. } = privateName;
  474. if (initAdded) return;
  475. const isAccessor = getId || setId;
  476. if (isAccessor) {
  477. return buildPrivateAccessorInitialization(ref, prop, privateNamesMap, state);
  478. }
  479. return buildPrivateInstanceMethodInitalization(ref, prop, privateNamesMap, state);
  480. }
  481. function buildPrivateAccessorInitialization(ref, prop, privateNamesMap, state) {
  482. const privateName = privateNamesMap.get(prop.node.key.id.name);
  483. const {
  484. id,
  485. getId,
  486. setId
  487. } = privateName;
  488. privateNamesMap.set(prop.node.key.id.name, Object.assign({}, privateName, {
  489. initAdded: true
  490. }));
  491. {
  492. if (!state.availableHelper("classPrivateFieldInitSpec")) {
  493. return _core.template.statement.ast`
  494. ${id}.set(${ref}, {
  495. get: ${getId ? getId.name : prop.scope.buildUndefinedNode()},
  496. set: ${setId ? setId.name : prop.scope.buildUndefinedNode()}
  497. });
  498. `;
  499. }
  500. }
  501. const helper = state.addHelper("classPrivateFieldInitSpec");
  502. return _core.template.statement.ast`${helper}(
  503. ${_core.types.thisExpression()},
  504. ${_core.types.cloneNode(id)},
  505. {
  506. get: ${getId ? getId.name : prop.scope.buildUndefinedNode()},
  507. set: ${setId ? setId.name : prop.scope.buildUndefinedNode()}
  508. },
  509. )`;
  510. }
  511. function buildPrivateInstanceMethodInitalization(ref, prop, privateNamesMap, state) {
  512. const privateName = privateNamesMap.get(prop.node.key.id.name);
  513. const {
  514. id
  515. } = privateName;
  516. {
  517. if (!state.availableHelper("classPrivateMethodInitSpec")) {
  518. return _core.template.statement.ast`${id}.add(${ref})`;
  519. }
  520. }
  521. const helper = state.addHelper("classPrivateMethodInitSpec");
  522. return _core.template.statement.ast`${helper}(
  523. ${_core.types.thisExpression()},
  524. ${_core.types.cloneNode(id)}
  525. )`;
  526. }
  527. function buildPublicFieldInitLoose(ref, prop) {
  528. const {
  529. key,
  530. computed
  531. } = prop.node;
  532. const value = prop.node.value || prop.scope.buildUndefinedNode();
  533. return _core.types.expressionStatement(_core.types.assignmentExpression("=", _core.types.memberExpression(ref, key, computed || _core.types.isLiteral(key)), value));
  534. }
  535. function buildPublicFieldInitSpec(ref, prop, state) {
  536. const {
  537. key,
  538. computed
  539. } = prop.node;
  540. const value = prop.node.value || prop.scope.buildUndefinedNode();
  541. return _core.types.expressionStatement(_core.types.callExpression(state.addHelper("defineProperty"), [ref, computed || _core.types.isLiteral(key) ? key : _core.types.stringLiteral(key.name), value]));
  542. }
  543. function buildPrivateStaticMethodInitLoose(ref, prop, state, privateNamesMap) {
  544. const privateName = privateNamesMap.get(prop.node.key.id.name);
  545. const {
  546. id,
  547. methodId,
  548. getId,
  549. setId,
  550. initAdded
  551. } = privateName;
  552. if (initAdded) return;
  553. const isAccessor = getId || setId;
  554. if (isAccessor) {
  555. privateNamesMap.set(prop.node.key.id.name, Object.assign({}, privateName, {
  556. initAdded: true
  557. }));
  558. return _core.template.statement.ast`
  559. Object.defineProperty(${ref}, ${id}, {
  560. // configurable is false by default
  561. // enumerable is false by default
  562. // writable is false by default
  563. get: ${getId ? getId.name : prop.scope.buildUndefinedNode()},
  564. set: ${setId ? setId.name : prop.scope.buildUndefinedNode()}
  565. })
  566. `;
  567. }
  568. return _core.template.statement.ast`
  569. Object.defineProperty(${ref}, ${id}, {
  570. // configurable is false by default
  571. // enumerable is false by default
  572. // writable is false by default
  573. value: ${methodId.name}
  574. });
  575. `;
  576. }
  577. function buildPrivateMethodDeclaration(prop, privateNamesMap, privateFieldsAsProperties = false) {
  578. const privateName = privateNamesMap.get(prop.node.key.id.name);
  579. const {
  580. id,
  581. methodId,
  582. getId,
  583. setId,
  584. getterDeclared,
  585. setterDeclared,
  586. static: isStatic
  587. } = privateName;
  588. const {
  589. params,
  590. body,
  591. generator,
  592. async
  593. } = prop.node;
  594. const isGetter = getId && !getterDeclared && params.length === 0;
  595. const isSetter = setId && !setterDeclared && params.length > 0;
  596. let declId = methodId;
  597. if (isGetter) {
  598. privateNamesMap.set(prop.node.key.id.name, Object.assign({}, privateName, {
  599. getterDeclared: true
  600. }));
  601. declId = getId;
  602. } else if (isSetter) {
  603. privateNamesMap.set(prop.node.key.id.name, Object.assign({}, privateName, {
  604. setterDeclared: true
  605. }));
  606. declId = setId;
  607. } else if (isStatic && !privateFieldsAsProperties) {
  608. declId = id;
  609. }
  610. return _core.types.functionDeclaration(_core.types.cloneNode(declId), params, body, generator, async);
  611. }
  612. const thisContextVisitor = _core.traverse.visitors.merge([{
  613. ThisExpression(path, state) {
  614. state.needsClassRef = true;
  615. path.replaceWith(_core.types.cloneNode(state.classRef));
  616. },
  617. MetaProperty(path) {
  618. const meta = path.get("meta");
  619. const property = path.get("property");
  620. const {
  621. scope
  622. } = path;
  623. if (meta.isIdentifier({
  624. name: "new"
  625. }) && property.isIdentifier({
  626. name: "target"
  627. })) {
  628. path.replaceWith(scope.buildUndefinedNode());
  629. }
  630. }
  631. }, _helperReplaceSupers.environmentVisitor]);
  632. const innerReferencesVisitor = {
  633. ReferencedIdentifier(path, state) {
  634. if (path.scope.bindingIdentifierEquals(path.node.name, state.innerBinding)) {
  635. state.needsClassRef = true;
  636. path.node.name = state.classRef.name;
  637. }
  638. }
  639. };
  640. function replaceThisContext(path, ref, getSuperRef, file, isStaticBlock, constantSuper, innerBindingRef) {
  641. var _state$classRef;
  642. const state = {
  643. classRef: ref,
  644. needsClassRef: false,
  645. innerBinding: innerBindingRef
  646. };
  647. const replacer = new _helperReplaceSupers.default({
  648. methodPath: path,
  649. constantSuper,
  650. file,
  651. refToPreserve: ref,
  652. getSuperRef,
  653. getObjectRef() {
  654. state.needsClassRef = true;
  655. return isStaticBlock || path.node.static ? ref : _core.types.memberExpression(ref, _core.types.identifier("prototype"));
  656. }
  657. });
  658. replacer.replace();
  659. if (isStaticBlock || path.isProperty()) {
  660. path.traverse(thisContextVisitor, state);
  661. }
  662. if ((_state$classRef = state.classRef) != null && _state$classRef.name && state.classRef.name !== (innerBindingRef == null ? void 0 : innerBindingRef.name)) {
  663. path.traverse(innerReferencesVisitor, state);
  664. }
  665. return state.needsClassRef;
  666. }
  667. function buildFieldsInitNodes(ref, superRef, props, privateNamesMap, state, setPublicClassFields, privateFieldsAsProperties, constantSuper, innerBindingRef) {
  668. let needsClassRef = false;
  669. let injectSuperRef;
  670. const staticNodes = [];
  671. const instanceNodes = [];
  672. const pureStaticNodes = [];
  673. const getSuperRef = _core.types.isIdentifier(superRef) ? () => superRef : () => {
  674. var _injectSuperRef;
  675. (_injectSuperRef = injectSuperRef) != null ? _injectSuperRef : injectSuperRef = props[0].scope.generateUidIdentifierBasedOnNode(superRef);
  676. return injectSuperRef;
  677. };
  678. for (const prop of props) {
  679. prop.isClassProperty() && ts.assertFieldTransformed(prop);
  680. const isStatic = prop.node.static;
  681. const isInstance = !isStatic;
  682. const isPrivate = prop.isPrivate();
  683. const isPublic = !isPrivate;
  684. const isField = prop.isProperty();
  685. const isMethod = !isField;
  686. const isStaticBlock = prop.isStaticBlock == null ? void 0 : prop.isStaticBlock();
  687. if (isStatic || isMethod && isPrivate || isStaticBlock) {
  688. const replaced = replaceThisContext(prop, ref, getSuperRef, state, isStaticBlock, constantSuper, innerBindingRef);
  689. needsClassRef = needsClassRef || replaced;
  690. }
  691. switch (true) {
  692. case isStaticBlock:
  693. staticNodes.push(_core.template.statement.ast`(() => ${_core.types.blockStatement(prop.node.body)})()`);
  694. break;
  695. case isStatic && isPrivate && isField && privateFieldsAsProperties:
  696. needsClassRef = true;
  697. staticNodes.push(buildPrivateFieldInitLoose(_core.types.cloneNode(ref), prop, privateNamesMap));
  698. break;
  699. case isStatic && isPrivate && isField && !privateFieldsAsProperties:
  700. needsClassRef = true;
  701. staticNodes.push(buildPrivateStaticFieldInitSpec(prop, privateNamesMap));
  702. break;
  703. case isStatic && isPublic && isField && setPublicClassFields:
  704. needsClassRef = true;
  705. staticNodes.push(buildPublicFieldInitLoose(_core.types.cloneNode(ref), prop));
  706. break;
  707. case isStatic && isPublic && isField && !setPublicClassFields:
  708. needsClassRef = true;
  709. staticNodes.push(buildPublicFieldInitSpec(_core.types.cloneNode(ref), prop, state));
  710. break;
  711. case isInstance && isPrivate && isField && privateFieldsAsProperties:
  712. instanceNodes.push(buildPrivateFieldInitLoose(_core.types.thisExpression(), prop, privateNamesMap));
  713. break;
  714. case isInstance && isPrivate && isField && !privateFieldsAsProperties:
  715. instanceNodes.push(buildPrivateInstanceFieldInitSpec(_core.types.thisExpression(), prop, privateNamesMap, state));
  716. break;
  717. case isInstance && isPrivate && isMethod && privateFieldsAsProperties:
  718. instanceNodes.unshift(buildPrivateMethodInitLoose(_core.types.thisExpression(), prop, privateNamesMap));
  719. pureStaticNodes.push(buildPrivateMethodDeclaration(prop, privateNamesMap, privateFieldsAsProperties));
  720. break;
  721. case isInstance && isPrivate && isMethod && !privateFieldsAsProperties:
  722. instanceNodes.unshift(buildPrivateInstanceMethodInitSpec(_core.types.thisExpression(), prop, privateNamesMap, state));
  723. pureStaticNodes.push(buildPrivateMethodDeclaration(prop, privateNamesMap, privateFieldsAsProperties));
  724. break;
  725. case isStatic && isPrivate && isMethod && !privateFieldsAsProperties:
  726. needsClassRef = true;
  727. staticNodes.unshift(buildPrivateStaticFieldInitSpec(prop, privateNamesMap));
  728. pureStaticNodes.push(buildPrivateMethodDeclaration(prop, privateNamesMap, privateFieldsAsProperties));
  729. break;
  730. case isStatic && isPrivate && isMethod && privateFieldsAsProperties:
  731. needsClassRef = true;
  732. staticNodes.unshift(buildPrivateStaticMethodInitLoose(_core.types.cloneNode(ref), prop, state, privateNamesMap));
  733. pureStaticNodes.push(buildPrivateMethodDeclaration(prop, privateNamesMap, privateFieldsAsProperties));
  734. break;
  735. case isInstance && isPublic && isField && setPublicClassFields:
  736. instanceNodes.push(buildPublicFieldInitLoose(_core.types.thisExpression(), prop));
  737. break;
  738. case isInstance && isPublic && isField && !setPublicClassFields:
  739. instanceNodes.push(buildPublicFieldInitSpec(_core.types.thisExpression(), prop, state));
  740. break;
  741. default:
  742. throw new Error("Unreachable.");
  743. }
  744. }
  745. return {
  746. staticNodes: staticNodes.filter(Boolean),
  747. instanceNodes: instanceNodes.filter(Boolean),
  748. pureStaticNodes: pureStaticNodes.filter(Boolean),
  749. wrapClass(path) {
  750. for (const prop of props) {
  751. prop.remove();
  752. }
  753. if (injectSuperRef) {
  754. path.scope.push({
  755. id: _core.types.cloneNode(injectSuperRef)
  756. });
  757. path.set("superClass", _core.types.assignmentExpression("=", injectSuperRef, path.node.superClass));
  758. }
  759. if (!needsClassRef) return path;
  760. if (path.isClassExpression()) {
  761. path.scope.push({
  762. id: ref
  763. });
  764. path.replaceWith(_core.types.assignmentExpression("=", _core.types.cloneNode(ref), path.node));
  765. } else if (!path.node.id) {
  766. path.node.id = ref;
  767. }
  768. return path;
  769. }
  770. };
  771. }