index.js 17 KB

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