index.js 17 KB

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