index.js 18 KB

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