index.js 18 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451
  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. var pluginTransformParameters = require('@babel/plugin-transform-parameters');
  6. var helperCompilationTargets = require('@babel/helper-compilation-targets');
  7. var compatData = require('@babel/compat-data/corejs2-built-ins');
  8. function _interopDefault (e) { return e && e.__esModule ? e : { default: e }; }
  9. var compatData__default = /*#__PURE__*/_interopDefault(compatData);
  10. const {
  11. isObjectProperty: isObjectProperty$1,
  12. isArrayPattern,
  13. isObjectPattern,
  14. isAssignmentPattern: isAssignmentPattern$1,
  15. isRestElement,
  16. isIdentifier
  17. } = core.types;
  18. function shouldStoreRHSInTemporaryVariable(node) {
  19. if (isArrayPattern(node)) {
  20. const nonNullElements = node.elements.filter(element => element !== null);
  21. if (nonNullElements.length > 1) return true;else return shouldStoreRHSInTemporaryVariable(nonNullElements[0]);
  22. } else if (isObjectPattern(node)) {
  23. const {
  24. properties
  25. } = node;
  26. if (properties.length > 1) return true;else if (properties.length === 0) return false;else {
  27. const firstProperty = properties[0];
  28. if (isObjectProperty$1(firstProperty)) {
  29. return shouldStoreRHSInTemporaryVariable(firstProperty.value);
  30. } else {
  31. return shouldStoreRHSInTemporaryVariable(firstProperty);
  32. }
  33. }
  34. } else if (isAssignmentPattern$1(node)) {
  35. return shouldStoreRHSInTemporaryVariable(node.left);
  36. } else if (isRestElement(node)) {
  37. if (isIdentifier(node.argument)) return true;
  38. return shouldStoreRHSInTemporaryVariable(node.argument);
  39. } else {
  40. return false;
  41. }
  42. }
  43. const {
  44. isAssignmentPattern,
  45. isObjectProperty
  46. } = core.types;
  47. {
  48. const node = core.types.identifier("a");
  49. const property = core.types.objectProperty(core.types.identifier("key"), node);
  50. const pattern = core.types.objectPattern([property]);
  51. var ZERO_REFS = core.types.isReferenced(node, property, pattern) ? 1 : 0;
  52. }
  53. var index = helperPluginUtils.declare((api, opts) => {
  54. var _api$assumption, _api$assumption2, _api$assumption3, _api$assumption4;
  55. api.assertVersion(7);
  56. const targets = api.targets();
  57. const supportsObjectAssign = !helperCompilationTargets.isRequired("es6.object.assign", targets, {
  58. compatData: compatData__default.default
  59. });
  60. const {
  61. useBuiltIns = supportsObjectAssign,
  62. loose = false
  63. } = opts;
  64. if (typeof loose !== "boolean") {
  65. throw new Error(".loose must be a boolean, or undefined");
  66. }
  67. const ignoreFunctionLength = (_api$assumption = api.assumption("ignoreFunctionLength")) != null ? _api$assumption : loose;
  68. const objectRestNoSymbols = (_api$assumption2 = api.assumption("objectRestNoSymbols")) != null ? _api$assumption2 : loose;
  69. const pureGetters = (_api$assumption3 = api.assumption("pureGetters")) != null ? _api$assumption3 : loose;
  70. const setSpreadProperties = (_api$assumption4 = api.assumption("setSpreadProperties")) != null ? _api$assumption4 : loose;
  71. function getExtendsHelper(file) {
  72. return useBuiltIns ? core.types.memberExpression(core.types.identifier("Object"), core.types.identifier("assign")) : file.addHelper("extends");
  73. }
  74. function hasRestElement(path) {
  75. let foundRestElement = false;
  76. visitRestElements(path, restElement => {
  77. foundRestElement = true;
  78. restElement.stop();
  79. });
  80. return foundRestElement;
  81. }
  82. function hasObjectPatternRestElement(path) {
  83. let foundRestElement = false;
  84. visitRestElements(path, restElement => {
  85. if (restElement.parentPath.isObjectPattern()) {
  86. foundRestElement = true;
  87. restElement.stop();
  88. }
  89. });
  90. return foundRestElement;
  91. }
  92. function visitRestElements(path, visitor) {
  93. path.traverse({
  94. Expression(path) {
  95. const {
  96. parent,
  97. key
  98. } = path;
  99. if (isAssignmentPattern(parent) && key === "right" || isObjectProperty(parent) && parent.computed && key === "key") {
  100. path.skip();
  101. }
  102. },
  103. RestElement: visitor
  104. });
  105. }
  106. function hasSpread(node) {
  107. for (const prop of node.properties) {
  108. if (core.types.isSpreadElement(prop)) {
  109. return true;
  110. }
  111. }
  112. return false;
  113. }
  114. function extractNormalizedKeys(node) {
  115. const props = node.properties;
  116. const keys = [];
  117. let allLiteral = true;
  118. let hasTemplateLiteral = false;
  119. for (const prop of props) {
  120. if (core.types.isIdentifier(prop.key) && !prop.computed) {
  121. keys.push(core.types.stringLiteral(prop.key.name));
  122. } else if (core.types.isTemplateLiteral(prop.key)) {
  123. keys.push(core.types.cloneNode(prop.key));
  124. hasTemplateLiteral = true;
  125. } else if (core.types.isLiteral(prop.key)) {
  126. keys.push(core.types.stringLiteral(String(prop.key.value)));
  127. } else {
  128. keys.push(core.types.cloneNode(prop.key));
  129. allLiteral = false;
  130. }
  131. }
  132. return {
  133. keys,
  134. allLiteral,
  135. hasTemplateLiteral
  136. };
  137. }
  138. function replaceImpureComputedKeys(properties, scope) {
  139. const impureComputedPropertyDeclarators = [];
  140. for (const propPath of properties) {
  141. const key = propPath.get("key");
  142. if (propPath.node.computed && !key.isPure()) {
  143. const name = scope.generateUidBasedOnNode(key.node);
  144. const declarator = core.types.variableDeclarator(core.types.identifier(name), key.node);
  145. impureComputedPropertyDeclarators.push(declarator);
  146. key.replaceWith(core.types.identifier(name));
  147. }
  148. }
  149. return impureComputedPropertyDeclarators;
  150. }
  151. function removeUnusedExcludedKeys(path) {
  152. const bindings = path.getOuterBindingIdentifierPaths();
  153. Object.keys(bindings).forEach(bindingName => {
  154. const bindingParentPath = bindings[bindingName].parentPath;
  155. if (path.scope.getBinding(bindingName).references > ZERO_REFS || !bindingParentPath.isObjectProperty()) {
  156. return;
  157. }
  158. bindingParentPath.remove();
  159. });
  160. }
  161. function createObjectRest(path, file, objRef) {
  162. const props = path.get("properties");
  163. const last = props[props.length - 1];
  164. core.types.assertRestElement(last.node);
  165. const restElement = core.types.cloneNode(last.node);
  166. last.remove();
  167. const impureComputedPropertyDeclarators = replaceImpureComputedKeys(path.get("properties"), path.scope);
  168. const {
  169. keys,
  170. allLiteral,
  171. hasTemplateLiteral
  172. } = extractNormalizedKeys(path.node);
  173. if (keys.length === 0) {
  174. return [impureComputedPropertyDeclarators, restElement.argument, core.types.callExpression(getExtendsHelper(file), [core.types.objectExpression([]), core.types.sequenceExpression([core.types.callExpression(file.addHelper("objectDestructuringEmpty"), [core.types.cloneNode(objRef)]), core.types.cloneNode(objRef)])])];
  175. }
  176. let keyExpression;
  177. if (!allLiteral) {
  178. keyExpression = core.types.callExpression(core.types.memberExpression(core.types.arrayExpression(keys), core.types.identifier("map")), [file.addHelper("toPropertyKey")]);
  179. } else {
  180. keyExpression = core.types.arrayExpression(keys);
  181. if (!hasTemplateLiteral && !core.types.isProgram(path.scope.block)) {
  182. const program = path.findParent(path => path.isProgram());
  183. const id = path.scope.generateUidIdentifier("excluded");
  184. program.scope.push({
  185. id,
  186. init: keyExpression,
  187. kind: "const"
  188. });
  189. keyExpression = core.types.cloneNode(id);
  190. }
  191. }
  192. return [impureComputedPropertyDeclarators, restElement.argument, core.types.callExpression(file.addHelper(`objectWithoutProperties${objectRestNoSymbols ? "Loose" : ""}`), [core.types.cloneNode(objRef), keyExpression])];
  193. }
  194. function replaceRestElement(parentPath, paramPath, container) {
  195. if (paramPath.isAssignmentPattern()) {
  196. replaceRestElement(parentPath, paramPath.get("left"), container);
  197. return;
  198. }
  199. if (paramPath.isArrayPattern() && hasRestElement(paramPath)) {
  200. const elements = paramPath.get("elements");
  201. for (let i = 0; i < elements.length; i++) {
  202. replaceRestElement(parentPath, elements[i], container);
  203. }
  204. }
  205. if (paramPath.isObjectPattern() && hasRestElement(paramPath)) {
  206. const uid = parentPath.scope.generateUidIdentifier("ref");
  207. const declar = core.types.variableDeclaration("let", [core.types.variableDeclarator(paramPath.node, uid)]);
  208. if (container) {
  209. container.push(declar);
  210. } else {
  211. parentPath.ensureBlock();
  212. parentPath.get("body").unshiftContainer("body", declar);
  213. }
  214. paramPath.replaceWith(core.types.cloneNode(uid));
  215. }
  216. }
  217. return {
  218. name: "transform-object-rest-spread",
  219. inherits: require("@babel/plugin-syntax-object-rest-spread").default,
  220. visitor: {
  221. Function(path) {
  222. const params = path.get("params");
  223. const paramsWithRestElement = new Set();
  224. const idsInRestParams = new Set();
  225. for (let i = 0; i < params.length; ++i) {
  226. const param = params[i];
  227. if (hasRestElement(param)) {
  228. paramsWithRestElement.add(i);
  229. for (const name of Object.keys(param.getBindingIdentifiers())) {
  230. idsInRestParams.add(name);
  231. }
  232. }
  233. }
  234. let idInRest = false;
  235. const IdentifierHandler = function (path, functionScope) {
  236. const name = path.node.name;
  237. if (path.scope.getBinding(name) === functionScope.getBinding(name) && idsInRestParams.has(name)) {
  238. idInRest = true;
  239. path.stop();
  240. }
  241. };
  242. let i;
  243. for (i = 0; i < params.length && !idInRest; ++i) {
  244. const param = params[i];
  245. if (!paramsWithRestElement.has(i)) {
  246. if (param.isReferencedIdentifier() || param.isBindingIdentifier()) {
  247. IdentifierHandler(param, path.scope);
  248. } else {
  249. param.traverse({
  250. "Scope|TypeAnnotation|TSTypeAnnotation": path => path.skip(),
  251. "ReferencedIdentifier|BindingIdentifier": IdentifierHandler
  252. }, path.scope);
  253. }
  254. }
  255. }
  256. if (!idInRest) {
  257. for (let i = 0; i < params.length; ++i) {
  258. const param = params[i];
  259. if (paramsWithRestElement.has(i)) {
  260. replaceRestElement(path, param);
  261. }
  262. }
  263. } else {
  264. const shouldTransformParam = idx => idx >= i - 1 || paramsWithRestElement.has(idx);
  265. pluginTransformParameters.convertFunctionParams(path, ignoreFunctionLength, shouldTransformParam, replaceRestElement);
  266. }
  267. },
  268. VariableDeclarator(path, file) {
  269. if (!path.get("id").isObjectPattern()) {
  270. return;
  271. }
  272. let insertionPath = path;
  273. const originalPath = path;
  274. visitRestElements(path.get("id"), path => {
  275. if (!path.parentPath.isObjectPattern()) {
  276. return;
  277. }
  278. if (shouldStoreRHSInTemporaryVariable(originalPath.node.id) && !core.types.isIdentifier(originalPath.node.init)) {
  279. const initRef = path.scope.generateUidIdentifierBasedOnNode(originalPath.node.init, "ref");
  280. originalPath.insertBefore(core.types.variableDeclarator(initRef, originalPath.node.init));
  281. originalPath.replaceWith(core.types.variableDeclarator(originalPath.node.id, core.types.cloneNode(initRef)));
  282. return;
  283. }
  284. let ref = originalPath.node.init;
  285. const refPropertyPath = [];
  286. let kind;
  287. path.findParent(path => {
  288. if (path.isObjectProperty()) {
  289. refPropertyPath.unshift(path);
  290. } else if (path.isVariableDeclarator()) {
  291. kind = path.parentPath.node.kind;
  292. return true;
  293. }
  294. });
  295. const impureObjRefComputedDeclarators = replaceImpureComputedKeys(refPropertyPath, path.scope);
  296. refPropertyPath.forEach(prop => {
  297. const {
  298. node
  299. } = prop;
  300. ref = core.types.memberExpression(ref, core.types.cloneNode(node.key), node.computed || core.types.isLiteral(node.key));
  301. });
  302. const objectPatternPath = path.findParent(path => path.isObjectPattern());
  303. const [impureComputedPropertyDeclarators, argument, callExpression] = createObjectRest(objectPatternPath, file, ref);
  304. if (pureGetters) {
  305. removeUnusedExcludedKeys(objectPatternPath);
  306. }
  307. core.types.assertIdentifier(argument);
  308. insertionPath.insertBefore(impureComputedPropertyDeclarators);
  309. insertionPath.insertBefore(impureObjRefComputedDeclarators);
  310. insertionPath = insertionPath.insertAfter(core.types.variableDeclarator(argument, callExpression))[0];
  311. path.scope.registerBinding(kind, insertionPath);
  312. if (objectPatternPath.node.properties.length === 0) {
  313. objectPatternPath.findParent(path => path.isObjectProperty() || path.isVariableDeclarator()).remove();
  314. }
  315. });
  316. },
  317. ExportNamedDeclaration(path) {
  318. const declaration = path.get("declaration");
  319. if (!declaration.isVariableDeclaration()) return;
  320. const hasRest = declaration.get("declarations").some(path => hasObjectPatternRestElement(path.get("id")));
  321. if (!hasRest) return;
  322. const specifiers = [];
  323. for (const name of Object.keys(path.getOuterBindingIdentifiers(true))) {
  324. specifiers.push(core.types.exportSpecifier(core.types.identifier(name), core.types.identifier(name)));
  325. }
  326. path.replaceWith(declaration.node);
  327. path.insertAfter(core.types.exportNamedDeclaration(null, specifiers));
  328. },
  329. CatchClause(path) {
  330. const paramPath = path.get("param");
  331. replaceRestElement(path, paramPath);
  332. },
  333. AssignmentExpression(path, file) {
  334. const leftPath = path.get("left");
  335. if (leftPath.isObjectPattern() && hasRestElement(leftPath)) {
  336. const nodes = [];
  337. const refName = path.scope.generateUidBasedOnNode(path.node.right, "ref");
  338. nodes.push(core.types.variableDeclaration("var", [core.types.variableDeclarator(core.types.identifier(refName), path.node.right)]));
  339. const [impureComputedPropertyDeclarators, argument, callExpression] = createObjectRest(leftPath, file, core.types.identifier(refName));
  340. if (impureComputedPropertyDeclarators.length > 0) {
  341. nodes.push(core.types.variableDeclaration("var", impureComputedPropertyDeclarators));
  342. }
  343. const nodeWithoutSpread = core.types.cloneNode(path.node);
  344. nodeWithoutSpread.right = core.types.identifier(refName);
  345. nodes.push(core.types.expressionStatement(nodeWithoutSpread));
  346. nodes.push(core.types.expressionStatement(core.types.assignmentExpression("=", argument, callExpression)));
  347. nodes.push(core.types.expressionStatement(core.types.identifier(refName)));
  348. path.replaceWithMultiple(nodes);
  349. }
  350. },
  351. ForXStatement(path) {
  352. const {
  353. node,
  354. scope
  355. } = path;
  356. const leftPath = path.get("left");
  357. const left = node.left;
  358. if (!hasObjectPatternRestElement(leftPath)) {
  359. return;
  360. }
  361. if (!core.types.isVariableDeclaration(left)) {
  362. const temp = scope.generateUidIdentifier("ref");
  363. node.left = core.types.variableDeclaration("var", [core.types.variableDeclarator(temp)]);
  364. path.ensureBlock();
  365. const body = path.node.body;
  366. if (body.body.length === 0 && path.isCompletionRecord()) {
  367. body.body.unshift(core.types.expressionStatement(scope.buildUndefinedNode()));
  368. }
  369. body.body.unshift(core.types.expressionStatement(core.types.assignmentExpression("=", left, core.types.cloneNode(temp))));
  370. } else {
  371. const pattern = left.declarations[0].id;
  372. const key = scope.generateUidIdentifier("ref");
  373. node.left = core.types.variableDeclaration(left.kind, [core.types.variableDeclarator(key, null)]);
  374. path.ensureBlock();
  375. const body = node.body;
  376. body.body.unshift(core.types.variableDeclaration(node.left.kind, [core.types.variableDeclarator(pattern, core.types.cloneNode(key))]));
  377. }
  378. },
  379. ArrayPattern(path) {
  380. const objectPatterns = [];
  381. visitRestElements(path, path => {
  382. if (!path.parentPath.isObjectPattern()) {
  383. return;
  384. }
  385. const objectPattern = path.parentPath;
  386. const uid = path.scope.generateUidIdentifier("ref");
  387. objectPatterns.push(core.types.variableDeclarator(objectPattern.node, uid));
  388. objectPattern.replaceWith(core.types.cloneNode(uid));
  389. path.skip();
  390. });
  391. if (objectPatterns.length > 0) {
  392. const statementPath = path.getStatementParent();
  393. const statementNode = statementPath.node;
  394. const kind = statementNode.type === "VariableDeclaration" ? statementNode.kind : "var";
  395. statementPath.insertAfter(core.types.variableDeclaration(kind, objectPatterns));
  396. }
  397. },
  398. ObjectExpression(path, file) {
  399. if (!hasSpread(path.node)) return;
  400. let helper;
  401. if (setSpreadProperties) {
  402. helper = getExtendsHelper(file);
  403. } else {
  404. try {
  405. helper = file.addHelper("objectSpread2");
  406. } catch (_unused) {
  407. this.file.declarations["objectSpread2"] = null;
  408. helper = file.addHelper("objectSpread");
  409. }
  410. }
  411. let exp = null;
  412. let props = [];
  413. function make() {
  414. const hadProps = props.length > 0;
  415. const obj = core.types.objectExpression(props);
  416. props = [];
  417. if (!exp) {
  418. exp = core.types.callExpression(helper, [obj]);
  419. return;
  420. }
  421. if (pureGetters) {
  422. if (hadProps) {
  423. exp.arguments.push(obj);
  424. }
  425. return;
  426. }
  427. exp = core.types.callExpression(core.types.cloneNode(helper), [exp, ...(hadProps ? [core.types.objectExpression([]), obj] : [])]);
  428. }
  429. for (const prop of path.node.properties) {
  430. if (core.types.isSpreadElement(prop)) {
  431. make();
  432. exp.arguments.push(prop.argument);
  433. } else {
  434. props.push(prop);
  435. }
  436. }
  437. if (props.length) make();
  438. path.replaceWith(exp);
  439. }
  440. }
  441. };
  442. });
  443. exports.default = index;
  444. //# sourceMappingURL=index.js.map