emit.js 26 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636
  1. "use strict";
  2. Object.defineProperty(exports, "__esModule", {
  3. value: true
  4. });
  5. exports.Emitter = void 0;
  6. var _assert = require("assert");
  7. var leap = require("./leap.js");
  8. var meta = require("./meta.js");
  9. var util = require("./util.js");
  10. var _core = require("@babel/core");
  11. const PENDING_LOCATION = Number.MAX_VALUE;
  12. function getDeclError(node) {
  13. return new Error("all declarations should have been transformed into " + "assignments before the Exploder began its work: " + JSON.stringify(node));
  14. }
  15. const catchParamVisitor = {
  16. Identifier: function (path, state) {
  17. if (path.node.name === state.catchParamName && util.isReference(path)) {
  18. util.replaceWithOrRemove(path, state.getSafeParam());
  19. }
  20. },
  21. Scope: function (path, state) {
  22. if (path.scope.hasOwnBinding(state.catchParamName)) {
  23. path.skip();
  24. }
  25. }
  26. };
  27. class Emitter {
  28. constructor(contextId, scope, vars, pluginPass) {
  29. this.nextTempId = void 0;
  30. this.contextId = void 0;
  31. this.index = void 0;
  32. this.indexMap = void 0;
  33. this.listing = void 0;
  34. this.returns = void 0;
  35. this.lastReferenceIndex = 0;
  36. this.marked = void 0;
  37. this.insertedLocs = void 0;
  38. this.finalLoc = void 0;
  39. this.tryEntries = void 0;
  40. this.leapManager = void 0;
  41. this.scope = void 0;
  42. this.vars = void 0;
  43. this.pluginPass = void 0;
  44. this.pluginPass = pluginPass;
  45. this.scope = scope;
  46. this.vars = vars;
  47. this.nextTempId = 0;
  48. this.contextId = contextId;
  49. this.listing = [];
  50. this.index = 0;
  51. this.indexMap = new Map([[0, 0]]);
  52. this.returns = new Set();
  53. this.lastReferenceIndex = 0;
  54. this.marked = [true];
  55. this.insertedLocs = new Set();
  56. this.finalLoc = this.loc();
  57. this.tryEntries = [];
  58. this.leapManager = new leap.LeapManager(this);
  59. }
  60. loc() {
  61. const l = _core.types.numericLiteral(PENDING_LOCATION);
  62. this.insertedLocs.add(l);
  63. return l;
  64. }
  65. getInsertedLocs() {
  66. return this.insertedLocs;
  67. }
  68. getContextId() {
  69. return _core.types.cloneNode(this.contextId);
  70. }
  71. getIndex() {
  72. if (!this.indexMap.has(this.listing.length)) {
  73. this.indexMap.set(this.listing.length, ++this.index);
  74. }
  75. return this.index;
  76. }
  77. mark(loc) {
  78. if (loc.value === PENDING_LOCATION) {
  79. loc.value = this.getIndex();
  80. } else {
  81. _assert.strictEqual(loc.value, this.index);
  82. }
  83. this.marked[this.listing.length] = true;
  84. if (loc.value > this.lastReferenceIndex) {
  85. this.lastReferenceIndex = loc.value;
  86. }
  87. return loc;
  88. }
  89. emit(node) {
  90. if (_core.types.isExpression(node)) {
  91. node = _core.types.expressionStatement(node);
  92. }
  93. _core.types.assertStatement(node);
  94. this.listing.push(node);
  95. }
  96. emitAssign(lhs, rhs) {
  97. this.emit(this.assign(lhs, rhs));
  98. return lhs;
  99. }
  100. assign(lhs, rhs) {
  101. return _core.types.expressionStatement(_core.types.assignmentExpression("=", _core.types.cloneNode(lhs), rhs));
  102. }
  103. contextProperty(name) {
  104. const computed = name === "catch";
  105. return _core.types.memberExpression(this.getContextId(), computed ? _core.types.stringLiteral(name) : _core.types.identifier(name), !!computed);
  106. }
  107. clearPendingException(tryLoc, assignee) {
  108. const catchCall = _core.types.callExpression(this.contextProperty("catch"), [_core.types.cloneNode(tryLoc)]);
  109. if (assignee) {
  110. this.emitAssign(assignee, catchCall);
  111. } else {
  112. this.emit(catchCall);
  113. }
  114. }
  115. jump(toLoc) {
  116. this.emitAssign(this.contextProperty(util.newHelpersAvailable(this.pluginPass) ? "n" : "next"), toLoc);
  117. this.emit(_core.types.breakStatement());
  118. }
  119. jumpIf(test, toLoc) {
  120. this.emit(_core.types.ifStatement(test, _core.types.blockStatement([this.assign(this.contextProperty(util.newHelpersAvailable(this.pluginPass) ? "n" : "next"), toLoc), _core.types.breakStatement()])));
  121. }
  122. jumpIfNot(test, toLoc) {
  123. let negatedTest;
  124. if (_core.types.isUnaryExpression(test) && test.operator === "!") {
  125. negatedTest = test.argument;
  126. } else {
  127. negatedTest = _core.types.unaryExpression("!", test);
  128. }
  129. this.emit(_core.types.ifStatement(negatedTest, _core.types.blockStatement([this.assign(this.contextProperty(util.newHelpersAvailable(this.pluginPass) ? "n" : "next"), toLoc), _core.types.breakStatement()])));
  130. }
  131. makeContextTempVar() {
  132. return this.contextProperty("t" + this.nextTempId++);
  133. }
  134. makeTempVar() {
  135. const id = this.scope.generateUidIdentifier("t");
  136. this.vars.push(_core.types.variableDeclarator(id));
  137. return _core.types.cloneNode(id);
  138. }
  139. getContextFunction() {
  140. return _core.types.functionExpression(null, [this.getContextId()], _core.types.blockStatement([this.getDispatchLoop()]), false, false);
  141. }
  142. getDispatchLoop() {
  143. const self = this;
  144. const cases = [];
  145. let current;
  146. let alreadyEnded = false;
  147. self.listing.forEach(function (stmt, i) {
  148. if (self.marked[i]) {
  149. cases.push(_core.types.switchCase(_core.types.numericLiteral(self.indexMap.get(i)), current = []));
  150. alreadyEnded = false;
  151. }
  152. if (!alreadyEnded) {
  153. current.push(stmt);
  154. if (_core.types.isCompletionStatement(stmt)) alreadyEnded = true;
  155. }
  156. });
  157. this.finalLoc.value = this.getIndex();
  158. if (util.newHelpersAvailable(this.pluginPass)) {
  159. if (this.lastReferenceIndex === this.index || !this.returns.has(this.listing.length)) {
  160. cases.push(_core.types.switchCase(this.finalLoc, [_core.types.returnStatement(_core.types.callExpression(this.contextProperty("a"), [_core.types.numericLiteral(2)]))]));
  161. }
  162. } else {
  163. cases.push(_core.types.switchCase(this.finalLoc, []), _core.types.switchCase(_core.types.stringLiteral("end"), [_core.types.returnStatement(_core.types.callExpression(this.contextProperty("stop"), []))]));
  164. }
  165. return _core.types.whileStatement(_core.types.numericLiteral(1), _core.types.switchStatement(util.newHelpersAvailable(this.pluginPass) ? this.tryEntries.length === 0 ? this.contextProperty("n") : _core.types.assignmentExpression("=", this.contextProperty("p"), this.contextProperty("n")) : _core.types.assignmentExpression("=", this.contextProperty("prev"), this.contextProperty("next")), cases));
  166. }
  167. getTryLocsList() {
  168. if (this.tryEntries.length === 0) {
  169. return null;
  170. }
  171. let lastLocValue = 0;
  172. const arrayExpression = _core.types.arrayExpression(this.tryEntries.map(function (tryEntry) {
  173. const thisLocValue = tryEntry.firstLoc.value;
  174. _assert.ok(thisLocValue >= lastLocValue, "try entries out of order");
  175. lastLocValue = thisLocValue;
  176. const ce = tryEntry.catchEntry;
  177. const fe = tryEntry.finallyEntry;
  178. const locs = [tryEntry.firstLoc, ce ? ce.firstLoc : null];
  179. if (fe) {
  180. locs[2] = fe.firstLoc;
  181. locs[3] = fe.afterLoc;
  182. }
  183. return _core.types.arrayExpression(locs.map(loc => loc && _core.types.cloneNode(loc)));
  184. }));
  185. if (util.newHelpersAvailable(this.pluginPass)) {
  186. arrayExpression.elements.reverse();
  187. }
  188. return arrayExpression;
  189. }
  190. explode(path, ignoreResult) {
  191. const node = path.node;
  192. const self = this;
  193. if (_core.types.isDeclaration(node)) throw getDeclError(node);
  194. if (path.isStatement()) return self.explodeStatement(path);
  195. if (path.isExpression()) return self.explodeExpression(path, ignoreResult);
  196. switch (node.type) {
  197. case "VariableDeclarator":
  198. throw getDeclError(node);
  199. case "ObjectProperty":
  200. case "SwitchCase":
  201. case "CatchClause":
  202. throw new Error(node.type + " nodes should be handled by their parents");
  203. default:
  204. throw new Error("unknown Node of type " + JSON.stringify(node.type));
  205. }
  206. }
  207. explodeStatement(path, labelId = null) {
  208. const stmt = path.node;
  209. const self = this;
  210. let before, after, head;
  211. if (path.isBlockStatement()) {
  212. path.get("body").forEach(function (path) {
  213. self.explodeStatement(path);
  214. });
  215. return;
  216. }
  217. if (!meta.containsLeap(stmt)) {
  218. self.emit(stmt);
  219. return;
  220. }
  221. switch (path.type) {
  222. case "ExpressionStatement":
  223. self.explodeExpression(path.get("expression"), true);
  224. break;
  225. case "LabeledStatement":
  226. after = this.loc();
  227. self.leapManager.withEntry(new leap.LabeledEntry(after, path.node.label), function () {
  228. self.explodeStatement(path.get("body"), path.node.label);
  229. });
  230. self.mark(after);
  231. break;
  232. case "WhileStatement":
  233. before = this.loc();
  234. after = this.loc();
  235. self.mark(before);
  236. self.jumpIfNot(self.explodeExpression(path.get("test")), after);
  237. self.leapManager.withEntry(new leap.LoopEntry(after, before, labelId), function () {
  238. self.explodeStatement(path.get("body"));
  239. });
  240. self.jump(before);
  241. self.mark(after);
  242. break;
  243. case "DoWhileStatement":
  244. const first = this.loc();
  245. const test = this.loc();
  246. after = this.loc();
  247. self.mark(first);
  248. self.leapManager.withEntry(new leap.LoopEntry(after, test, labelId), function () {
  249. self.explode(path.get("body"));
  250. });
  251. self.mark(test);
  252. self.jumpIf(self.explodeExpression(path.get("test")), first);
  253. self.mark(after);
  254. break;
  255. case "ForStatement":
  256. head = this.loc();
  257. const update = this.loc();
  258. after = this.loc();
  259. if (path.node.init) {
  260. self.explode(path.get("init"), true);
  261. }
  262. self.mark(head);
  263. if (path.node.test) {
  264. self.jumpIfNot(self.explodeExpression(path.get("test")), after);
  265. } else {}
  266. self.leapManager.withEntry(new leap.LoopEntry(after, update, labelId), function () {
  267. self.explodeStatement(path.get("body"));
  268. });
  269. self.mark(update);
  270. if (path.node.update) {
  271. self.explode(path.get("update"), true);
  272. }
  273. self.jump(head);
  274. self.mark(after);
  275. break;
  276. case "TypeCastExpression":
  277. return self.explodeExpression(path.get("expression"));
  278. case "ForInStatement":
  279. head = this.loc();
  280. after = this.loc();
  281. const keyIterNextFn = self.makeTempVar();
  282. const helper = util.newHelpersAvailable(this.pluginPass) ? this.pluginPass.addHelper("regeneratorKeys") : util.runtimeProperty(this.pluginPass, "keys");
  283. self.emitAssign(keyIterNextFn, _core.types.callExpression(helper, [self.explodeExpression(path.get("right"))]));
  284. self.mark(head);
  285. const keyInfoTmpVar = self.makeTempVar();
  286. self.jumpIf(_core.types.memberExpression(_core.types.assignmentExpression("=", keyInfoTmpVar, _core.types.callExpression(_core.types.cloneNode(keyIterNextFn), [])), _core.types.identifier("done"), false), after);
  287. self.emitAssign(path.node.left, _core.types.memberExpression(_core.types.cloneNode(keyInfoTmpVar), _core.types.identifier("value"), false));
  288. self.leapManager.withEntry(new leap.LoopEntry(after, head, labelId), function () {
  289. self.explodeStatement(path.get("body"));
  290. });
  291. self.jump(head);
  292. self.mark(after);
  293. break;
  294. case "BreakStatement":
  295. self.emitAbruptCompletion({
  296. type: 3,
  297. target: self.leapManager.getBreakLoc(path.node.label)
  298. });
  299. break;
  300. case "ContinueStatement":
  301. self.emitAbruptCompletion({
  302. type: 3,
  303. target: self.leapManager.getContinueLoc(path.node.label)
  304. });
  305. break;
  306. case "SwitchStatement":
  307. const disc = self.emitAssign(self.makeTempVar(), self.explodeExpression(path.get("discriminant")));
  308. after = this.loc();
  309. const defaultLoc = this.loc();
  310. let condition = defaultLoc;
  311. const caseLocs = [];
  312. const cases = path.node.cases || [];
  313. for (let i = cases.length - 1; i >= 0; --i) {
  314. const c = cases[i];
  315. if (c.test) {
  316. condition = _core.types.conditionalExpression(_core.types.binaryExpression("===", _core.types.cloneNode(disc), c.test), caseLocs[i] = this.loc(), condition);
  317. } else {
  318. caseLocs[i] = defaultLoc;
  319. }
  320. }
  321. const discriminant = path.get("discriminant");
  322. util.replaceWithOrRemove(discriminant, condition);
  323. self.jump(self.explodeExpression(discriminant));
  324. self.leapManager.withEntry(new leap.SwitchEntry(after), function () {
  325. path.get("cases").forEach(function (casePath) {
  326. const i = casePath.key;
  327. self.mark(caseLocs[i]);
  328. casePath.get("consequent").forEach(function (path) {
  329. self.explodeStatement(path);
  330. });
  331. });
  332. });
  333. self.mark(after);
  334. if (defaultLoc.value === PENDING_LOCATION) {
  335. self.mark(defaultLoc);
  336. _assert.strictEqual(after.value, defaultLoc.value);
  337. }
  338. break;
  339. case "IfStatement":
  340. const elseLoc = path.node.alternate && this.loc();
  341. after = this.loc();
  342. self.jumpIfNot(self.explodeExpression(path.get("test")), elseLoc || after);
  343. self.explodeStatement(path.get("consequent"));
  344. if (elseLoc) {
  345. self.jump(after);
  346. self.mark(elseLoc);
  347. self.explodeStatement(path.get("alternate"));
  348. }
  349. self.mark(after);
  350. break;
  351. case "ReturnStatement":
  352. self.emitAbruptCompletion({
  353. type: 2,
  354. value: self.explodeExpression(path.get("argument"))
  355. });
  356. break;
  357. case "WithStatement":
  358. throw new Error("WithStatement not supported in generator functions.");
  359. case "TryStatement":
  360. after = this.loc();
  361. const handler = path.node.handler;
  362. const catchLoc = handler && this.loc();
  363. const catchEntry = catchLoc && new leap.CatchEntry(catchLoc, handler.param);
  364. const finallyLoc = path.node.finalizer && this.loc();
  365. const finallyEntry = finallyLoc && new leap.FinallyEntry(finallyLoc, after);
  366. const tryEntry = new leap.TryEntry(self.getUnmarkedCurrentLoc(), catchEntry, finallyEntry);
  367. self.tryEntries.push(tryEntry);
  368. self.updateContextPrevLoc(tryEntry.firstLoc);
  369. self.leapManager.withEntry(tryEntry, () => {
  370. self.explodeStatement(path.get("block"));
  371. if (catchLoc) {
  372. const body = path.node.block.body;
  373. if (finallyLoc) {
  374. self.jump(finallyLoc);
  375. } else if (body.length && body[body.length - 1].type === "ReturnStatement") {
  376. after = null;
  377. } else {
  378. self.jump(after);
  379. }
  380. self.updateContextPrevLoc(self.mark(catchLoc));
  381. const bodyPath = path.get("handler.body");
  382. const safeParam = self.makeTempVar();
  383. if (util.newHelpersAvailable(this.pluginPass)) {
  384. this.emitAssign(safeParam, self.contextProperty("v"));
  385. } else {
  386. self.clearPendingException(tryEntry.firstLoc, safeParam);
  387. }
  388. bodyPath.traverse(catchParamVisitor, {
  389. getSafeParam: () => _core.types.cloneNode(safeParam),
  390. catchParamName: handler.param.name
  391. });
  392. self.leapManager.withEntry(catchEntry, function () {
  393. self.explodeStatement(bodyPath);
  394. });
  395. }
  396. if (finallyLoc) {
  397. self.updateContextPrevLoc(self.mark(finallyLoc));
  398. self.leapManager.withEntry(finallyEntry, function () {
  399. self.explodeStatement(path.get("finalizer"));
  400. });
  401. self.emit(_core.types.returnStatement(_core.types.callExpression(self.contextProperty(util.newHelpersAvailable(this.pluginPass) ? "f" : "finish"), [finallyEntry.firstLoc])));
  402. }
  403. });
  404. if (after) self.mark(after);
  405. break;
  406. case "ThrowStatement":
  407. self.emit(_core.types.throwStatement(self.explodeExpression(path.get("argument"))));
  408. break;
  409. case "ClassDeclaration":
  410. self.emit(self.explodeClass(path));
  411. break;
  412. default:
  413. throw new Error("unknown Statement of type " + JSON.stringify(stmt.type));
  414. }
  415. }
  416. emitAbruptCompletion(record) {
  417. const abruptArgs = [util.newHelpersAvailable(this.pluginPass) ? _core.types.numericLiteral(record.type) : _core.types.stringLiteral(record.type === 3 ? "continue" : "return")];
  418. if (record.type === 3) {
  419. abruptArgs[1] = this.insertedLocs.has(record.target) ? record.target : _core.types.cloneNode(record.target);
  420. } else if (record.type === 2) {
  421. if (record.value) {
  422. abruptArgs[1] = _core.types.cloneNode(record.value);
  423. }
  424. }
  425. this.emit(_core.types.returnStatement(_core.types.callExpression(this.contextProperty(util.newHelpersAvailable(this.pluginPass) ? "a" : "abrupt"), abruptArgs)));
  426. if (record.type === 2) {
  427. this.returns.add(this.listing.length);
  428. }
  429. }
  430. getUnmarkedCurrentLoc() {
  431. return _core.types.numericLiteral(this.getIndex());
  432. }
  433. updateContextPrevLoc(loc) {
  434. if (loc) {
  435. if (loc.value === PENDING_LOCATION) {
  436. loc.value = this.getIndex();
  437. } else {
  438. _assert.strictEqual(loc.value, this.index);
  439. }
  440. } else {
  441. loc = this.getUnmarkedCurrentLoc();
  442. }
  443. this.emitAssign(this.contextProperty(util.newHelpersAvailable(this.pluginPass) ? "p" : "prev"), loc);
  444. }
  445. explodeViaTempVar(tempVar, childPath, hasLeapingChildren, ignoreChildResult) {
  446. _assert.ok(!ignoreChildResult || !tempVar, "Ignoring the result of a child expression but forcing it to " + "be assigned to a temporary variable?");
  447. let result = this.explodeExpression(childPath, ignoreChildResult);
  448. if (ignoreChildResult) {} else if (tempVar || hasLeapingChildren && !_core.types.isLiteral(result)) {
  449. result = this.emitAssign(tempVar || this.makeTempVar(), result);
  450. }
  451. return result;
  452. }
  453. explodeExpression(path, ignoreResult) {
  454. const expr = path.node;
  455. if (!expr) {
  456. return expr;
  457. }
  458. const self = this;
  459. let result;
  460. let after;
  461. function finish(expr) {
  462. if (ignoreResult) {
  463. self.emit(expr);
  464. }
  465. return expr;
  466. }
  467. if (!meta.containsLeap(expr)) {
  468. return finish(expr);
  469. }
  470. const hasLeapingChildren = meta.containsLeap.onlyChildren(expr);
  471. switch (path.type) {
  472. case "MemberExpression":
  473. return finish(_core.types.memberExpression(self.explodeExpression(path.get("object")), path.node.computed ? self.explodeViaTempVar(null, path.get("property"), hasLeapingChildren) : path.node.property, path.node.computed));
  474. case "CallExpression":
  475. const calleePath = path.get("callee");
  476. const argsPath = path.get("arguments");
  477. let newCallee;
  478. let newArgs;
  479. const hasLeapingArgs = argsPath.some(argPath => meta.containsLeap(argPath.node));
  480. let injectFirstArg = null;
  481. if (_core.types.isMemberExpression(calleePath.node)) {
  482. if (hasLeapingArgs) {
  483. const newObject = self.explodeViaTempVar(self.makeTempVar(), calleePath.get("object"), hasLeapingChildren);
  484. const newProperty = calleePath.node.computed ? self.explodeViaTempVar(null, calleePath.get("property"), hasLeapingChildren) : calleePath.node.property;
  485. injectFirstArg = newObject;
  486. newCallee = _core.types.memberExpression(_core.types.memberExpression(_core.types.cloneNode(newObject), newProperty, calleePath.node.computed), _core.types.identifier("call"), false);
  487. } else {
  488. newCallee = self.explodeExpression(calleePath);
  489. }
  490. } else {
  491. newCallee = self.explodeViaTempVar(null, calleePath, hasLeapingChildren);
  492. if (_core.types.isMemberExpression(newCallee)) {
  493. newCallee = _core.types.sequenceExpression([_core.types.numericLiteral(0), _core.types.cloneNode(newCallee)]);
  494. }
  495. }
  496. if (hasLeapingArgs) {
  497. newArgs = argsPath.map(argPath => self.explodeViaTempVar(null, argPath, hasLeapingChildren));
  498. if (injectFirstArg) newArgs.unshift(injectFirstArg);
  499. newArgs = newArgs.map(arg => _core.types.cloneNode(arg));
  500. } else {
  501. newArgs = path.node.arguments;
  502. }
  503. return finish(_core.types.callExpression(newCallee, newArgs));
  504. case "NewExpression":
  505. return finish(_core.types.newExpression(self.explodeViaTempVar(null, path.get("callee"), hasLeapingChildren), path.get("arguments").map(function (argPath) {
  506. return self.explodeViaTempVar(null, argPath, hasLeapingChildren);
  507. })));
  508. case "ObjectExpression":
  509. return finish(_core.types.objectExpression(path.get("properties").map(function (propPath) {
  510. if (propPath.isObjectProperty()) {
  511. return _core.types.objectProperty(propPath.node.key, self.explodeViaTempVar(null, propPath.get("value"), hasLeapingChildren), propPath.node.computed);
  512. } else {
  513. return propPath.node;
  514. }
  515. })));
  516. case "ArrayExpression":
  517. return finish(_core.types.arrayExpression(path.get("elements").map(function (elemPath) {
  518. if (!elemPath.node) {
  519. return null;
  520. }
  521. if (elemPath.isSpreadElement()) {
  522. return _core.types.spreadElement(self.explodeViaTempVar(null, elemPath.get("argument"), hasLeapingChildren));
  523. } else {
  524. return self.explodeViaTempVar(null, elemPath, hasLeapingChildren);
  525. }
  526. })));
  527. case "SequenceExpression":
  528. const lastIndex = path.node.expressions.length - 1;
  529. path.get("expressions").forEach(function (exprPath) {
  530. if (exprPath.key === lastIndex) {
  531. result = self.explodeExpression(exprPath, ignoreResult);
  532. } else {
  533. self.explodeExpression(exprPath, true);
  534. }
  535. });
  536. return result;
  537. case "LogicalExpression":
  538. after = this.loc();
  539. if (!ignoreResult) {
  540. result = self.makeTempVar();
  541. }
  542. const left = self.explodeViaTempVar(result, path.get("left"), hasLeapingChildren);
  543. if (path.node.operator === "&&") {
  544. self.jumpIfNot(left, after);
  545. } else {
  546. _assert.strictEqual(path.node.operator, "||");
  547. self.jumpIf(left, after);
  548. }
  549. self.explodeViaTempVar(result, path.get("right"), hasLeapingChildren, ignoreResult);
  550. self.mark(after);
  551. return result;
  552. case "ConditionalExpression":
  553. const elseLoc = this.loc();
  554. after = this.loc();
  555. const test = self.explodeExpression(path.get("test"));
  556. self.jumpIfNot(test, elseLoc);
  557. if (!ignoreResult) {
  558. result = self.makeTempVar();
  559. }
  560. self.explodeViaTempVar(result, path.get("consequent"), hasLeapingChildren, ignoreResult);
  561. self.jump(after);
  562. self.mark(elseLoc);
  563. self.explodeViaTempVar(result, path.get("alternate"), hasLeapingChildren, ignoreResult);
  564. self.mark(after);
  565. return result;
  566. case "UnaryExpression":
  567. return finish(_core.types.unaryExpression(path.node.operator, self.explodeExpression(path.get("argument")), !!path.node.prefix));
  568. case "BinaryExpression":
  569. return finish(_core.types.binaryExpression(path.node.operator, self.explodeViaTempVar(null, path.get("left"), hasLeapingChildren), self.explodeViaTempVar(null, path.get("right"), hasLeapingChildren)));
  570. case "AssignmentExpression":
  571. if (path.node.operator === "=") {
  572. return finish(_core.types.assignmentExpression(path.node.operator, self.explodeExpression(path.get("left")), self.explodeExpression(path.get("right"))));
  573. }
  574. const lhs = self.explodeExpression(path.get("left"));
  575. const temp = self.emitAssign(self.makeTempVar(), lhs);
  576. return finish(_core.types.assignmentExpression("=", _core.types.cloneNode(lhs), _core.types.assignmentExpression(path.node.operator, _core.types.cloneNode(temp), self.explodeExpression(path.get("right")))));
  577. case "UpdateExpression":
  578. return finish(_core.types.updateExpression(path.node.operator, self.explodeExpression(path.get("argument")), path.node.prefix));
  579. case "YieldExpression":
  580. after = this.loc();
  581. const arg = path.node.argument && self.explodeExpression(path.get("argument"));
  582. if (arg && path.node.delegate) {
  583. if (util.newHelpersAvailable(this.pluginPass)) {
  584. const ret = _core.types.returnStatement(_core.types.callExpression(self.contextProperty("d"), [_core.types.callExpression(this.pluginPass.addHelper("regeneratorValues"), [arg]), after]));
  585. ret.loc = expr.loc;
  586. self.emit(ret);
  587. self.mark(after);
  588. return self.contextProperty("v");
  589. } else {
  590. const result = self.makeContextTempVar();
  591. const ret = _core.types.returnStatement(_core.types.callExpression(self.contextProperty("delegateYield"), [arg, _core.types.stringLiteral(result.property.name), after]));
  592. ret.loc = expr.loc;
  593. self.emit(ret);
  594. self.mark(after);
  595. return result;
  596. }
  597. }
  598. self.emitAssign(self.contextProperty(util.newHelpersAvailable(this.pluginPass) ? "n" : "next"), after);
  599. const ret = _core.types.returnStatement(_core.types.cloneNode(arg) || null);
  600. ret.loc = expr.loc;
  601. self.emit(ret);
  602. self.mark(after);
  603. return self.contextProperty(util.newHelpersAvailable(self.pluginPass) ? "v" : "sent");
  604. case "ClassExpression":
  605. return finish(self.explodeClass(path));
  606. default:
  607. throw new Error("unknown Expression of type " + JSON.stringify(expr.type));
  608. }
  609. }
  610. explodeClass(path) {
  611. const explodingChildren = [];
  612. if (path.node.superClass) {
  613. explodingChildren.push(path.get("superClass"));
  614. }
  615. path.get("body.body").forEach(member => {
  616. if (member.node.computed) {
  617. explodingChildren.push(member.get("key"));
  618. }
  619. });
  620. const hasLeapingChildren = explodingChildren.some(child => meta.containsLeap(child));
  621. for (let i = 0; i < explodingChildren.length; i++) {
  622. const child = explodingChildren[i];
  623. const isLast = i === explodingChildren.length - 1;
  624. if (isLast) {
  625. child.replaceWith(this.explodeExpression(child));
  626. } else {
  627. child.replaceWith(this.explodeViaTempVar(null, child, hasLeapingChildren));
  628. }
  629. }
  630. return path.node;
  631. }
  632. }
  633. exports.Emitter = Emitter;
  634. //# sourceMappingURL=emit.js.map