index.js 19 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615
  1. 'use strict';
  2. Object.defineProperty(exports, '__esModule', { value: true });
  3. var helperPluginUtils = require('@babel/helper-plugin-utils');
  4. var core = require('@babel/core');
  5. function unshiftForXStatementBody(statementPath, newStatements) {
  6. statementPath.ensureBlock();
  7. const {
  8. scope,
  9. node
  10. } = statementPath;
  11. const bodyScopeBindings = statementPath.get("body").scope.bindings;
  12. const hasShadowedBlockScopedBindings = Object.keys(bodyScopeBindings).some(name => scope.hasBinding(name));
  13. if (hasShadowedBlockScopedBindings) {
  14. node.body = core.types.blockStatement([...newStatements, node.body]);
  15. } else {
  16. node.body.body.unshift(...newStatements);
  17. }
  18. }
  19. function hasArrayRest(pattern) {
  20. return pattern.elements.some(elem => core.types.isRestElement(elem));
  21. }
  22. function hasObjectRest(pattern) {
  23. return pattern.properties.some(prop => core.types.isRestElement(prop));
  24. }
  25. const STOP_TRAVERSAL = {};
  26. const arrayUnpackVisitor = (node, ancestors, state) => {
  27. if (!ancestors.length) {
  28. return;
  29. }
  30. if (core.types.isIdentifier(node) && core.types.isReferenced(node, ancestors[ancestors.length - 1].node) && state.bindings[node.name]) {
  31. state.deopt = true;
  32. throw STOP_TRAVERSAL;
  33. }
  34. };
  35. class DestructuringTransformer {
  36. constructor(opts) {
  37. this.blockHoist = void 0;
  38. this.operator = void 0;
  39. this.arrayRefSet = void 0;
  40. this.nodes = void 0;
  41. this.scope = void 0;
  42. this.kind = void 0;
  43. this.iterableIsArray = void 0;
  44. this.arrayLikeIsIterable = void 0;
  45. this.objectRestNoSymbols = void 0;
  46. this.useBuiltIns = void 0;
  47. this.addHelper = void 0;
  48. this.blockHoist = opts.blockHoist;
  49. this.operator = opts.operator;
  50. this.arrayRefSet = new Set();
  51. this.nodes = opts.nodes || [];
  52. this.scope = opts.scope;
  53. this.kind = opts.kind;
  54. this.iterableIsArray = opts.iterableIsArray;
  55. this.arrayLikeIsIterable = opts.arrayLikeIsIterable;
  56. this.objectRestNoSymbols = opts.objectRestNoSymbols;
  57. this.useBuiltIns = opts.useBuiltIns;
  58. this.addHelper = opts.addHelper;
  59. }
  60. getExtendsHelper() {
  61. return this.useBuiltIns ? core.types.memberExpression(core.types.identifier("Object"), core.types.identifier("assign")) : this.addHelper("extends");
  62. }
  63. buildVariableAssignment(id, init) {
  64. let op = this.operator;
  65. if (core.types.isMemberExpression(id)) op = "=";
  66. let node;
  67. if (op) {
  68. node = core.types.expressionStatement(core.types.assignmentExpression(op, id, core.types.cloneNode(init) || this.scope.buildUndefinedNode()));
  69. } else {
  70. let nodeInit;
  71. if (this.kind === "const" && init === null) {
  72. nodeInit = this.scope.buildUndefinedNode();
  73. } else {
  74. nodeInit = core.types.cloneNode(init);
  75. }
  76. node = core.types.variableDeclaration(this.kind, [core.types.variableDeclarator(id, nodeInit)]);
  77. }
  78. node._blockHoist = this.blockHoist;
  79. return node;
  80. }
  81. buildVariableDeclaration(id, init) {
  82. const declar = core.types.variableDeclaration("var", [core.types.variableDeclarator(core.types.cloneNode(id), core.types.cloneNode(init))]);
  83. declar._blockHoist = this.blockHoist;
  84. return declar;
  85. }
  86. push(id, _init) {
  87. const init = core.types.cloneNode(_init);
  88. if (core.types.isObjectPattern(id)) {
  89. this.pushObjectPattern(id, init);
  90. } else if (core.types.isArrayPattern(id)) {
  91. this.pushArrayPattern(id, init);
  92. } else if (core.types.isAssignmentPattern(id)) {
  93. this.pushAssignmentPattern(id, init);
  94. } else {
  95. this.nodes.push(this.buildVariableAssignment(id, init));
  96. }
  97. }
  98. toArray(node, count) {
  99. if (this.iterableIsArray || core.types.isIdentifier(node) && this.arrayRefSet.has(node.name)) {
  100. return node;
  101. } else {
  102. return this.scope.toArray(node, count, this.arrayLikeIsIterable);
  103. }
  104. }
  105. pushAssignmentPattern({
  106. left,
  107. right
  108. }, valueRef) {
  109. if (valueRef === null) {
  110. this.push(left, right);
  111. return;
  112. }
  113. const tempId = this.scope.generateUidIdentifierBasedOnNode(valueRef);
  114. this.nodes.push(this.buildVariableDeclaration(tempId, valueRef));
  115. const tempConditional = core.types.conditionalExpression(core.types.binaryExpression("===", core.types.cloneNode(tempId), this.scope.buildUndefinedNode()), right, core.types.cloneNode(tempId));
  116. if (core.types.isPattern(left)) {
  117. let patternId;
  118. let node;
  119. if (this.kind === "const" || this.kind === "let") {
  120. patternId = this.scope.generateUidIdentifier(tempId.name);
  121. node = this.buildVariableDeclaration(patternId, tempConditional);
  122. } else {
  123. patternId = tempId;
  124. node = core.types.expressionStatement(core.types.assignmentExpression("=", core.types.cloneNode(tempId), tempConditional));
  125. }
  126. this.nodes.push(node);
  127. this.push(left, patternId);
  128. } else {
  129. this.nodes.push(this.buildVariableAssignment(left, tempConditional));
  130. }
  131. }
  132. pushObjectRest(pattern, objRef, spreadProp, spreadPropIndex) {
  133. const value = buildObjectExcludingKeys(pattern.properties.slice(0, spreadPropIndex), objRef, this.scope, name => this.addHelper(name), this.objectRestNoSymbols, this.useBuiltIns);
  134. this.nodes.push(this.buildVariableAssignment(spreadProp.argument, value));
  135. }
  136. pushObjectProperty(prop, propRef) {
  137. if (core.types.isLiteral(prop.key)) prop.computed = true;
  138. const pattern = prop.value;
  139. const objRef = core.types.memberExpression(core.types.cloneNode(propRef), prop.key, prop.computed);
  140. if (core.types.isPattern(pattern)) {
  141. this.push(pattern, objRef);
  142. } else {
  143. this.nodes.push(this.buildVariableAssignment(pattern, objRef));
  144. }
  145. }
  146. pushObjectPattern(pattern, objRef) {
  147. if (!pattern.properties.length || objRef === null) {
  148. this.nodes.push(core.types.expressionStatement(core.types.callExpression(this.addHelper("objectDestructuringEmpty"), objRef !== null ? [objRef] : [])));
  149. return;
  150. }
  151. if (pattern.properties.length > 1 && !this.scope.isStatic(objRef)) {
  152. const temp = this.scope.generateUidIdentifierBasedOnNode(objRef);
  153. this.nodes.push(this.buildVariableDeclaration(temp, objRef));
  154. objRef = temp;
  155. }
  156. if (hasObjectRest(pattern)) {
  157. let copiedPattern;
  158. for (let i = 0; i < pattern.properties.length; i++) {
  159. const prop = pattern.properties[i];
  160. if (core.types.isRestElement(prop)) {
  161. break;
  162. }
  163. const key = prop.key;
  164. if (prop.computed && !this.scope.isPure(key)) {
  165. const name = this.scope.generateUidIdentifierBasedOnNode(key);
  166. this.nodes.push(this.buildVariableDeclaration(name, key));
  167. if (!copiedPattern) {
  168. copiedPattern = pattern = Object.assign({}, pattern, {
  169. properties: pattern.properties.slice()
  170. });
  171. }
  172. copiedPattern.properties[i] = Object.assign({}, prop, {
  173. key: name
  174. });
  175. }
  176. }
  177. }
  178. for (let i = 0; i < pattern.properties.length; i++) {
  179. const prop = pattern.properties[i];
  180. if (core.types.isRestElement(prop)) {
  181. this.pushObjectRest(pattern, objRef, prop, i);
  182. } else {
  183. this.pushObjectProperty(prop, objRef);
  184. }
  185. }
  186. }
  187. canUnpackArrayPattern(pattern, arr) {
  188. if (!core.types.isArrayExpression(arr)) return false;
  189. if (pattern.elements.length > arr.elements.length) return;
  190. if (pattern.elements.length < arr.elements.length && !hasArrayRest(pattern)) {
  191. return false;
  192. }
  193. for (const elem of pattern.elements) {
  194. if (!elem) return false;
  195. if (core.types.isMemberExpression(elem)) return false;
  196. }
  197. for (const elem of arr.elements) {
  198. if (core.types.isSpreadElement(elem)) return false;
  199. if (core.types.isCallExpression(elem)) return false;
  200. if (core.types.isMemberExpression(elem)) return false;
  201. }
  202. const bindings = core.types.getBindingIdentifiers(pattern);
  203. const state = {
  204. deopt: false,
  205. bindings
  206. };
  207. try {
  208. core.types.traverse(arr, arrayUnpackVisitor, state);
  209. } catch (e) {
  210. if (e !== STOP_TRAVERSAL) throw e;
  211. }
  212. return !state.deopt;
  213. }
  214. pushUnpackedArrayPattern(pattern, arr) {
  215. for (let i = 0; i < pattern.elements.length; i++) {
  216. const elem = pattern.elements[i];
  217. if (core.types.isRestElement(elem)) {
  218. this.push(elem.argument, core.types.arrayExpression(arr.elements.slice(i)));
  219. } else {
  220. this.push(elem, arr.elements[i]);
  221. }
  222. }
  223. }
  224. pushArrayPattern(pattern, arrayRef) {
  225. if (arrayRef === null) {
  226. this.nodes.push(core.types.expressionStatement(core.types.callExpression(this.addHelper("objectDestructuringEmpty"), [])));
  227. return;
  228. }
  229. if (!pattern.elements) return;
  230. if (this.canUnpackArrayPattern(pattern, arrayRef)) {
  231. return this.pushUnpackedArrayPattern(pattern, arrayRef);
  232. }
  233. const count = !hasArrayRest(pattern) && pattern.elements.length;
  234. const toArray = this.toArray(arrayRef, count);
  235. if (core.types.isIdentifier(toArray)) {
  236. arrayRef = toArray;
  237. } else {
  238. arrayRef = this.scope.generateUidIdentifierBasedOnNode(arrayRef);
  239. this.arrayRefSet.add(arrayRef.name);
  240. this.nodes.push(this.buildVariableDeclaration(arrayRef, toArray));
  241. }
  242. for (let i = 0; i < pattern.elements.length; i++) {
  243. const elem = pattern.elements[i];
  244. if (!elem) continue;
  245. let elemRef;
  246. if (core.types.isRestElement(elem)) {
  247. elemRef = this.toArray(arrayRef);
  248. elemRef = core.types.callExpression(core.types.memberExpression(elemRef, core.types.identifier("slice")), [core.types.numericLiteral(i)]);
  249. this.push(elem.argument, elemRef);
  250. } else {
  251. elemRef = core.types.memberExpression(arrayRef, core.types.numericLiteral(i), true);
  252. this.push(elem, elemRef);
  253. }
  254. }
  255. }
  256. init(pattern, ref) {
  257. if (!core.types.isArrayExpression(ref) && !core.types.isMemberExpression(ref)) {
  258. const memo = this.scope.maybeGenerateMemoised(ref, true);
  259. if (memo) {
  260. this.nodes.push(this.buildVariableDeclaration(memo, core.types.cloneNode(ref)));
  261. ref = memo;
  262. }
  263. }
  264. this.push(pattern, ref);
  265. return this.nodes;
  266. }
  267. }
  268. function buildObjectExcludingKeys(excludedKeys, objRef, scope, addHelper, objectRestNoSymbols, useBuiltIns) {
  269. const keys = [];
  270. let allLiteral = true;
  271. let hasTemplateLiteral = false;
  272. for (let i = 0; i < excludedKeys.length; i++) {
  273. const prop = excludedKeys[i];
  274. const key = prop.key;
  275. if (core.types.isIdentifier(key) && !prop.computed) {
  276. keys.push(core.types.stringLiteral(key.name));
  277. } else if (core.types.isTemplateLiteral(key)) {
  278. keys.push(core.types.cloneNode(key));
  279. hasTemplateLiteral = true;
  280. } else if (core.types.isLiteral(key)) {
  281. keys.push(core.types.stringLiteral(String(key.value)));
  282. } else if (core.types.isPrivateName(key)) ; else {
  283. keys.push(core.types.cloneNode(key));
  284. allLiteral = false;
  285. }
  286. }
  287. let value;
  288. if (keys.length === 0) {
  289. const extendsHelper = useBuiltIns ? core.types.memberExpression(core.types.identifier("Object"), core.types.identifier("assign")) : addHelper("extends");
  290. value = core.types.callExpression(extendsHelper, [core.types.objectExpression([]), core.types.cloneNode(objRef)]);
  291. } else {
  292. let keyExpression = core.types.arrayExpression(keys);
  293. if (!allLiteral) {
  294. keyExpression = core.types.callExpression(core.types.memberExpression(keyExpression, core.types.identifier("map")), [addHelper("toPropertyKey")]);
  295. } else if (!hasTemplateLiteral && !core.types.isProgram(scope.block)) {
  296. const programScope = scope.getProgramParent();
  297. const id = programScope.generateUidIdentifier("excluded");
  298. programScope.push({
  299. id,
  300. init: keyExpression,
  301. kind: "const"
  302. });
  303. keyExpression = core.types.cloneNode(id);
  304. }
  305. value = core.types.callExpression(addHelper(`objectWithoutProperties${objectRestNoSymbols ? "Loose" : ""}`), [core.types.cloneNode(objRef), keyExpression]);
  306. }
  307. return value;
  308. }
  309. function convertVariableDeclaration(path, addHelper, arrayLikeIsIterable, iterableIsArray, objectRestNoSymbols, useBuiltIns) {
  310. const {
  311. node,
  312. scope
  313. } = path;
  314. const nodeKind = node.kind;
  315. const nodeLoc = node.loc;
  316. const nodes = [];
  317. for (let i = 0; i < node.declarations.length; i++) {
  318. const declar = node.declarations[i];
  319. const patternId = declar.init;
  320. const pattern = declar.id;
  321. const destructuring = new DestructuringTransformer({
  322. blockHoist: node._blockHoist,
  323. nodes: nodes,
  324. scope: scope,
  325. kind: node.kind,
  326. iterableIsArray,
  327. arrayLikeIsIterable,
  328. useBuiltIns,
  329. objectRestNoSymbols,
  330. addHelper
  331. });
  332. if (core.types.isPattern(pattern)) {
  333. destructuring.init(pattern, patternId);
  334. if (+i !== node.declarations.length - 1) {
  335. core.types.inherits(nodes[nodes.length - 1], declar);
  336. }
  337. } else {
  338. nodes.push(core.types.inherits(destructuring.buildVariableAssignment(pattern, patternId), declar));
  339. }
  340. }
  341. let tail = null;
  342. const nodesOut = [];
  343. for (const node of nodes) {
  344. if (core.types.isVariableDeclaration(node)) {
  345. if (tail !== null) {
  346. tail.declarations.push(...node.declarations);
  347. continue;
  348. } else {
  349. node.kind = nodeKind;
  350. tail = node;
  351. }
  352. } else {
  353. tail = null;
  354. }
  355. if (!node.loc) {
  356. node.loc = nodeLoc;
  357. }
  358. nodesOut.push(node);
  359. }
  360. if (nodesOut.length === 1) {
  361. path.replaceWith(nodesOut[0]);
  362. } else {
  363. path.replaceWithMultiple(nodesOut);
  364. }
  365. scope.crawl();
  366. }
  367. function convertAssignmentExpression(path, addHelper, arrayLikeIsIterable, iterableIsArray, objectRestNoSymbols, useBuiltIns) {
  368. const {
  369. node,
  370. scope,
  371. parentPath
  372. } = path;
  373. const nodes = [];
  374. const destructuring = new DestructuringTransformer({
  375. operator: node.operator,
  376. scope: scope,
  377. nodes: nodes,
  378. arrayLikeIsIterable,
  379. iterableIsArray,
  380. objectRestNoSymbols,
  381. useBuiltIns,
  382. addHelper
  383. });
  384. let ref;
  385. if (!parentPath.isExpressionStatement() && !parentPath.isSequenceExpression() || path.isCompletionRecord()) {
  386. ref = scope.generateUidIdentifierBasedOnNode(node.right, "ref");
  387. nodes.push(core.types.variableDeclaration("var", [core.types.variableDeclarator(ref, node.right)]));
  388. if (core.types.isArrayExpression(node.right)) {
  389. destructuring.arrayRefSet.add(ref.name);
  390. }
  391. }
  392. destructuring.init(node.left, ref || node.right);
  393. if (ref) {
  394. if (parentPath.isArrowFunctionExpression()) {
  395. path.replaceWith(core.types.blockStatement([]));
  396. nodes.push(core.types.returnStatement(core.types.cloneNode(ref)));
  397. } else {
  398. nodes.push(core.types.expressionStatement(core.types.cloneNode(ref)));
  399. }
  400. }
  401. path.replaceWithMultiple(nodes);
  402. scope.crawl();
  403. }
  404. function variableDeclarationHasPattern(node) {
  405. for (const declar of node.declarations) {
  406. if (core.types.isPattern(declar.id)) {
  407. return true;
  408. }
  409. }
  410. return false;
  411. }
  412. var index = helperPluginUtils.declare((api, options) => {
  413. var _ref, _api$assumption, _ref2, _options$allowArrayLi, _ref3, _api$assumption2;
  414. api.assertVersion(7);
  415. const {
  416. useBuiltIns = false
  417. } = options;
  418. const iterableIsArray = (_ref = (_api$assumption = api.assumption("iterableIsArray")) != null ? _api$assumption : options.loose) != null ? _ref : false;
  419. const arrayLikeIsIterable = (_ref2 = (_options$allowArrayLi = options.allowArrayLike) != null ? _options$allowArrayLi : api.assumption("arrayLikeIsIterable")) != null ? _ref2 : false;
  420. const objectRestNoSymbols = (_ref3 = (_api$assumption2 = api.assumption("objectRestNoSymbols")) != null ? _api$assumption2 : options.loose) != null ? _ref3 : false;
  421. return {
  422. name: "transform-destructuring",
  423. visitor: {
  424. ExportNamedDeclaration(path) {
  425. const declaration = path.get("declaration");
  426. if (!declaration.isVariableDeclaration()) return;
  427. if (!variableDeclarationHasPattern(declaration.node)) return;
  428. const specifiers = [];
  429. for (const name of Object.keys(path.getOuterBindingIdentifiers())) {
  430. specifiers.push(core.types.exportSpecifier(core.types.identifier(name), core.types.identifier(name)));
  431. }
  432. path.replaceWith(declaration.node);
  433. path.insertAfter(core.types.exportNamedDeclaration(null, specifiers));
  434. path.scope.crawl();
  435. },
  436. ForXStatement(path) {
  437. const {
  438. node,
  439. scope
  440. } = path;
  441. const left = node.left;
  442. if (core.types.isPattern(left)) {
  443. const temp = scope.generateUidIdentifier("ref");
  444. node.left = core.types.variableDeclaration("var", [core.types.variableDeclarator(temp)]);
  445. path.ensureBlock();
  446. const statementBody = path.node.body.body;
  447. const nodes = [];
  448. if (statementBody.length === 0 && path.isCompletionRecord()) {
  449. nodes.unshift(core.types.expressionStatement(scope.buildUndefinedNode()));
  450. }
  451. nodes.unshift(core.types.expressionStatement(core.types.assignmentExpression("=", left, core.types.cloneNode(temp))));
  452. unshiftForXStatementBody(path, nodes);
  453. scope.crawl();
  454. return;
  455. }
  456. if (!core.types.isVariableDeclaration(left)) return;
  457. const pattern = left.declarations[0].id;
  458. if (!core.types.isPattern(pattern)) return;
  459. const key = scope.generateUidIdentifier("ref");
  460. node.left = core.types.variableDeclaration(left.kind, [core.types.variableDeclarator(key, null)]);
  461. const nodes = [];
  462. const destructuring = new DestructuringTransformer({
  463. kind: left.kind,
  464. scope: scope,
  465. nodes: nodes,
  466. arrayLikeIsIterable,
  467. iterableIsArray,
  468. objectRestNoSymbols,
  469. useBuiltIns,
  470. addHelper: name => this.addHelper(name)
  471. });
  472. destructuring.init(pattern, key);
  473. unshiftForXStatementBody(path, nodes);
  474. scope.crawl();
  475. },
  476. CatchClause({
  477. node,
  478. scope
  479. }) {
  480. const pattern = node.param;
  481. if (!core.types.isPattern(pattern)) return;
  482. const ref = scope.generateUidIdentifier("ref");
  483. node.param = ref;
  484. const nodes = [];
  485. const destructuring = new DestructuringTransformer({
  486. kind: "let",
  487. scope: scope,
  488. nodes: nodes,
  489. arrayLikeIsIterable,
  490. iterableIsArray,
  491. objectRestNoSymbols,
  492. useBuiltIns,
  493. addHelper: name => this.addHelper(name)
  494. });
  495. destructuring.init(pattern, ref);
  496. node.body.body = [...nodes, ...node.body.body];
  497. scope.crawl();
  498. },
  499. AssignmentExpression(path, state) {
  500. if (!core.types.isPattern(path.node.left)) return;
  501. convertAssignmentExpression(path, name => state.addHelper(name), arrayLikeIsIterable, iterableIsArray, objectRestNoSymbols, useBuiltIns);
  502. },
  503. VariableDeclaration(path, state) {
  504. const {
  505. node,
  506. parent
  507. } = path;
  508. if (core.types.isForXStatement(parent)) return;
  509. if (!parent || !path.container) return;
  510. if (!variableDeclarationHasPattern(node)) return;
  511. convertVariableDeclaration(path, name => state.addHelper(name), arrayLikeIsIterable, iterableIsArray, objectRestNoSymbols, useBuiltIns);
  512. }
  513. }
  514. };
  515. });
  516. exports.buildObjectExcludingKeys = buildObjectExcludingKeys;
  517. exports["default"] = index;
  518. exports.unshiftForXStatementBody = unshiftForXStatementBody;
  519. //# sourceMappingURL=index.js.map