index.js 10 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303
  1. "use strict";
  2. Object.defineProperty(exports, "__esModule", {
  3. value: true
  4. });
  5. exports.default = void 0;
  6. var _helperMemberExpressionToFunctions = require("@babel/helper-member-expression-to-functions");
  7. var _helperOptimiseCallExpression = require("@babel/helper-optimise-call-expression");
  8. var _core = require("@babel/core");
  9. var _traverse = require("@babel/traverse");
  10. const {
  11. assignmentExpression,
  12. callExpression,
  13. cloneNode,
  14. identifier,
  15. memberExpression,
  16. sequenceExpression,
  17. stringLiteral,
  18. thisExpression
  19. } = _core.types;
  20. exports.environmentVisitor = _traverse.visitors.environmentVisitor({});
  21. exports.skipAllButComputedKey = function skipAllButComputedKey(path) {
  22. path.skip();
  23. if (path.node.computed) {
  24. path.context.maybeQueue(path.get("key"));
  25. }
  26. };
  27. const visitor = _traverse.visitors.environmentVisitor({
  28. Super(path, state) {
  29. const {
  30. node,
  31. parentPath
  32. } = path;
  33. if (!parentPath.isMemberExpression({
  34. object: node
  35. })) return;
  36. state.handle(parentPath);
  37. }
  38. });
  39. const unshadowSuperBindingVisitor = _traverse.visitors.environmentVisitor({
  40. Scopable(path, {
  41. refName
  42. }) {
  43. const binding = path.scope.getOwnBinding(refName);
  44. if ((binding == null ? void 0 : binding.identifier.name) === refName) {
  45. path.scope.rename(refName);
  46. }
  47. }
  48. });
  49. const specHandlers = {
  50. memoise(superMember, count) {
  51. const {
  52. scope,
  53. node
  54. } = superMember;
  55. const {
  56. computed,
  57. property
  58. } = node;
  59. if (!computed) {
  60. return;
  61. }
  62. const memo = scope.maybeGenerateMemoised(property);
  63. if (!memo) {
  64. return;
  65. }
  66. this.memoiser.set(property, memo, count);
  67. },
  68. prop(superMember) {
  69. const {
  70. computed,
  71. property
  72. } = superMember.node;
  73. if (this.memoiser.has(property)) {
  74. return cloneNode(this.memoiser.get(property));
  75. }
  76. if (computed) {
  77. return cloneNode(property);
  78. }
  79. return stringLiteral(property.name);
  80. },
  81. _getPrototypeOfExpression() {
  82. const objectRef = cloneNode(this.getObjectRef());
  83. const targetRef = this.isStatic || this.isPrivateMethod ? objectRef : memberExpression(objectRef, identifier("prototype"));
  84. return callExpression(this.file.addHelper("getPrototypeOf"), [targetRef]);
  85. },
  86. get(superMember) {
  87. const objectRef = cloneNode(this.getObjectRef());
  88. return callExpression(this.file.addHelper("superPropGet"), [this.isDerivedConstructor ? sequenceExpression([thisExpression(), objectRef]) : objectRef, this.prop(superMember), thisExpression(), ...(this.isStatic || this.isPrivateMethod ? [] : [_core.types.numericLiteral(1)])]);
  89. },
  90. _call(superMember, args, optional) {
  91. const objectRef = cloneNode(this.getObjectRef());
  92. let argsNode;
  93. if (args.length === 1 && _core.types.isSpreadElement(args[0]) && (_core.types.isIdentifier(args[0].argument) || _core.types.isArrayExpression(args[0].argument))) {
  94. argsNode = args[0].argument;
  95. } else {
  96. argsNode = _core.types.arrayExpression(args);
  97. }
  98. const call = _core.types.callExpression(this.file.addHelper("superPropGet"), [this.isDerivedConstructor ? sequenceExpression([thisExpression(), objectRef]) : objectRef, this.prop(superMember), thisExpression(), _core.types.numericLiteral(2 | (this.isStatic || this.isPrivateMethod ? 0 : 1))]);
  99. if (optional) {
  100. return _core.types.optionalCallExpression(call, [argsNode], true);
  101. }
  102. return callExpression(call, [argsNode]);
  103. },
  104. set(superMember, value) {
  105. const objectRef = cloneNode(this.getObjectRef());
  106. return callExpression(this.file.addHelper("superPropSet"), [this.isDerivedConstructor ? sequenceExpression([thisExpression(), objectRef]) : objectRef, this.prop(superMember), value, thisExpression(), _core.types.numericLiteral(superMember.isInStrictMode() ? 1 : 0), ...(this.isStatic || this.isPrivateMethod ? [] : [_core.types.numericLiteral(1)])]);
  107. },
  108. destructureSet(superMember) {
  109. throw superMember.buildCodeFrameError(`Destructuring to a super field is not supported yet.`);
  110. },
  111. call(superMember, args) {
  112. return this._call(superMember, args, false);
  113. },
  114. optionalCall(superMember, args) {
  115. return this._call(superMember, args, true);
  116. },
  117. delete(superMember) {
  118. if (superMember.node.computed) {
  119. return sequenceExpression([callExpression(this.file.addHelper("toPropertyKey"), [cloneNode(superMember.node.property)]), _core.template.expression.ast`
  120. function () { throw new ReferenceError("'delete super[expr]' is invalid"); }()
  121. `]);
  122. } else {
  123. return _core.template.expression.ast`
  124. function () { throw new ReferenceError("'delete super.prop' is invalid"); }()
  125. `;
  126. }
  127. }
  128. };
  129. const specHandlers_old = {
  130. memoise(superMember, count) {
  131. const {
  132. scope,
  133. node
  134. } = superMember;
  135. const {
  136. computed,
  137. property
  138. } = node;
  139. if (!computed) {
  140. return;
  141. }
  142. const memo = scope.maybeGenerateMemoised(property);
  143. if (!memo) {
  144. return;
  145. }
  146. this.memoiser.set(property, memo, count);
  147. },
  148. prop(superMember) {
  149. const {
  150. computed,
  151. property
  152. } = superMember.node;
  153. if (this.memoiser.has(property)) {
  154. return cloneNode(this.memoiser.get(property));
  155. }
  156. if (computed) {
  157. return cloneNode(property);
  158. }
  159. return stringLiteral(property.name);
  160. },
  161. _getPrototypeOfExpression() {
  162. const objectRef = cloneNode(this.getObjectRef());
  163. const targetRef = this.isStatic || this.isPrivateMethod ? objectRef : memberExpression(objectRef, identifier("prototype"));
  164. return callExpression(this.file.addHelper("getPrototypeOf"), [targetRef]);
  165. },
  166. get(superMember) {
  167. return this._get(superMember);
  168. },
  169. _get(superMember) {
  170. const proto = this._getPrototypeOfExpression();
  171. return callExpression(this.file.addHelper("get"), [this.isDerivedConstructor ? sequenceExpression([thisExpression(), proto]) : proto, this.prop(superMember), thisExpression()]);
  172. },
  173. set(superMember, value) {
  174. const proto = this._getPrototypeOfExpression();
  175. return callExpression(this.file.addHelper("set"), [this.isDerivedConstructor ? sequenceExpression([thisExpression(), proto]) : proto, this.prop(superMember), value, thisExpression(), _core.types.booleanLiteral(superMember.isInStrictMode())]);
  176. },
  177. destructureSet(superMember) {
  178. throw superMember.buildCodeFrameError(`Destructuring to a super field is not supported yet.`);
  179. },
  180. call(superMember, args) {
  181. return (0, _helperOptimiseCallExpression.default)(this._get(superMember), thisExpression(), args, false);
  182. },
  183. optionalCall(superMember, args) {
  184. return (0, _helperOptimiseCallExpression.default)(this._get(superMember), cloneNode(thisExpression()), args, true);
  185. },
  186. delete(superMember) {
  187. if (superMember.node.computed) {
  188. return sequenceExpression([callExpression(this.file.addHelper("toPropertyKey"), [cloneNode(superMember.node.property)]), _core.template.expression.ast`
  189. function () { throw new ReferenceError("'delete super[expr]' is invalid"); }()
  190. `]);
  191. } else {
  192. return _core.template.expression.ast`
  193. function () { throw new ReferenceError("'delete super.prop' is invalid"); }()
  194. `;
  195. }
  196. }
  197. };
  198. const looseHandlers = Object.assign({}, specHandlers, {
  199. prop(superMember) {
  200. const {
  201. property
  202. } = superMember.node;
  203. if (this.memoiser.has(property)) {
  204. return cloneNode(this.memoiser.get(property));
  205. }
  206. return cloneNode(property);
  207. },
  208. get(superMember) {
  209. const {
  210. isStatic,
  211. getSuperRef
  212. } = this;
  213. const {
  214. computed
  215. } = superMember.node;
  216. const prop = this.prop(superMember);
  217. let object;
  218. if (isStatic) {
  219. var _getSuperRef;
  220. object = (_getSuperRef = getSuperRef()) != null ? _getSuperRef : memberExpression(identifier("Function"), identifier("prototype"));
  221. } else {
  222. var _getSuperRef2;
  223. object = memberExpression((_getSuperRef2 = getSuperRef()) != null ? _getSuperRef2 : identifier("Object"), identifier("prototype"));
  224. }
  225. return memberExpression(object, prop, computed);
  226. },
  227. set(superMember, value) {
  228. const {
  229. computed
  230. } = superMember.node;
  231. const prop = this.prop(superMember);
  232. return assignmentExpression("=", memberExpression(thisExpression(), prop, computed), value);
  233. },
  234. destructureSet(superMember) {
  235. const {
  236. computed
  237. } = superMember.node;
  238. const prop = this.prop(superMember);
  239. return memberExpression(thisExpression(), prop, computed);
  240. },
  241. call(superMember, args) {
  242. return (0, _helperOptimiseCallExpression.default)(this.get(superMember), thisExpression(), args, false);
  243. },
  244. optionalCall(superMember, args) {
  245. return (0, _helperOptimiseCallExpression.default)(this.get(superMember), thisExpression(), args, true);
  246. }
  247. });
  248. class ReplaceSupers {
  249. constructor(opts) {
  250. var _opts$constantSuper;
  251. const path = opts.methodPath;
  252. this.methodPath = path;
  253. this.isDerivedConstructor = path.isClassMethod({
  254. kind: "constructor"
  255. }) && !!opts.superRef;
  256. this.isStatic = path.isObjectMethod() || path.node.static || (path.isStaticBlock == null ? void 0 : path.isStaticBlock());
  257. this.isPrivateMethod = path.isPrivate() && path.isMethod();
  258. this.file = opts.file;
  259. this.constantSuper = (_opts$constantSuper = opts.constantSuper) != null ? _opts$constantSuper : opts.isLoose;
  260. this.opts = opts;
  261. }
  262. getObjectRef() {
  263. return cloneNode(this.opts.objectRef || this.opts.getObjectRef());
  264. }
  265. getSuperRef() {
  266. if (this.opts.superRef) return cloneNode(this.opts.superRef);
  267. if (this.opts.getSuperRef) {
  268. return cloneNode(this.opts.getSuperRef());
  269. }
  270. }
  271. replace() {
  272. const {
  273. methodPath
  274. } = this;
  275. if (this.opts.refToPreserve) {
  276. methodPath.traverse(unshadowSuperBindingVisitor, {
  277. refName: this.opts.refToPreserve.name
  278. });
  279. }
  280. const handler = this.constantSuper ? looseHandlers : this.file.availableHelper("superPropSet") ? specHandlers : specHandlers_old;
  281. visitor.shouldSkip = path => {
  282. if (path.parentPath === methodPath) {
  283. if (path.parentKey === "decorators" || path.parentKey === "key") {
  284. return true;
  285. }
  286. }
  287. };
  288. (0, _helperMemberExpressionToFunctions.default)(methodPath, visitor, Object.assign({
  289. file: this.file,
  290. scope: this.methodPath.scope,
  291. isDerivedConstructor: this.isDerivedConstructor,
  292. isStatic: this.isStatic,
  293. isPrivateMethod: this.isPrivateMethod,
  294. getObjectRef: this.getObjectRef.bind(this),
  295. getSuperRef: this.getSuperRef.bind(this),
  296. boundGet: handler.get
  297. }, handler));
  298. }
  299. }
  300. exports.default = ReplaceSupers;
  301. //# sourceMappingURL=index.js.map