index.js 5.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164
  1. "use strict";
  2. Object.defineProperty(exports, "__esModule", {
  3. value: true
  4. });
  5. exports.default = void 0;
  6. var _core = require("@babel/core");
  7. var _helperPluginUtils = require("@babel/helper-plugin-utils");
  8. var _template = require("@babel/template");
  9. var DefineAccessorHelper = _template.default.expression.ast`
  10. function (type, obj, key, fn) {
  11. var desc = { configurable: true, enumerable: true };
  12. desc[type] = fn;
  13. return Object.defineProperty(obj, key, desc);
  14. }
  15. `;
  16. DefineAccessorHelper._compact = true;
  17. var _default = exports.default = (0, _helperPluginUtils.declare)((api, options) => {
  18. var _api$assumption;
  19. api.assertVersion(7);
  20. const setComputedProperties = (_api$assumption = api.assumption("setComputedProperties")) != null ? _api$assumption : options.loose;
  21. const pushComputedProps = setComputedProperties ? pushComputedPropsLoose : pushComputedPropsSpec;
  22. function buildDefineAccessor(state, obj, prop) {
  23. const type = prop.kind;
  24. const key = !prop.computed && _core.types.isIdentifier(prop.key) ? _core.types.stringLiteral(prop.key.name) : prop.key;
  25. const fn = getValue(prop);
  26. let helper;
  27. if (state.availableHelper("defineAccessor")) {
  28. helper = state.addHelper("defineAccessor");
  29. } else {
  30. const file = state.file;
  31. helper = file.get("fallbackDefineAccessorHelper");
  32. if (!helper) {
  33. const id = file.scope.generateUidIdentifier("defineAccessor");
  34. file.scope.push({
  35. id,
  36. init: DefineAccessorHelper
  37. });
  38. file.set("fallbackDefineAccessorHelper", helper = id);
  39. }
  40. helper = _core.types.cloneNode(helper);
  41. }
  42. return _core.types.callExpression(helper, [_core.types.stringLiteral(type), obj, key, fn]);
  43. }
  44. function getValue(prop) {
  45. if (_core.types.isObjectProperty(prop)) {
  46. return prop.value;
  47. } else if (_core.types.isObjectMethod(prop)) {
  48. return _core.types.functionExpression(null, prop.params, prop.body, prop.generator, prop.async);
  49. }
  50. }
  51. function pushAssign(objId, prop, body) {
  52. body.push(_core.types.expressionStatement(_core.types.assignmentExpression("=", _core.types.memberExpression(_core.types.cloneNode(objId), prop.key, prop.computed || _core.types.isLiteral(prop.key)), getValue(prop))));
  53. }
  54. function pushComputedPropsLoose(info) {
  55. const {
  56. computedProps,
  57. state,
  58. initPropExpression,
  59. objId,
  60. body
  61. } = info;
  62. for (const prop of computedProps) {
  63. if (_core.types.isObjectMethod(prop) && (prop.kind === "get" || prop.kind === "set")) {
  64. if (computedProps.length === 1) {
  65. return buildDefineAccessor(state, initPropExpression, prop);
  66. } else {
  67. body.push(_core.types.expressionStatement(buildDefineAccessor(state, _core.types.cloneNode(objId), prop)));
  68. }
  69. } else {
  70. pushAssign(_core.types.cloneNode(objId), prop, body);
  71. }
  72. }
  73. }
  74. function pushComputedPropsSpec(info) {
  75. const {
  76. objId,
  77. body,
  78. computedProps,
  79. state
  80. } = info;
  81. const CHUNK_LENGTH_CAP = 10;
  82. let currentChunk = null;
  83. const computedPropsChunks = [];
  84. for (const prop of computedProps) {
  85. if (!currentChunk || currentChunk.length === CHUNK_LENGTH_CAP) {
  86. currentChunk = [];
  87. computedPropsChunks.push(currentChunk);
  88. }
  89. currentChunk.push(prop);
  90. }
  91. for (const chunk of computedPropsChunks) {
  92. const single = computedPropsChunks.length === 1;
  93. let node = single ? info.initPropExpression : _core.types.cloneNode(objId);
  94. for (const prop of chunk) {
  95. if (_core.types.isObjectMethod(prop) && (prop.kind === "get" || prop.kind === "set")) {
  96. node = buildDefineAccessor(info.state, node, prop);
  97. } else {
  98. node = _core.types.callExpression(state.addHelper("defineProperty"), [node, _core.types.toComputedKey(prop), getValue(prop)]);
  99. }
  100. }
  101. if (single) return node;
  102. body.push(_core.types.expressionStatement(node));
  103. }
  104. }
  105. return {
  106. name: "transform-computed-properties",
  107. visitor: {
  108. ObjectExpression: {
  109. exit(path, state) {
  110. const {
  111. node,
  112. parent,
  113. scope
  114. } = path;
  115. let hasComputed = false;
  116. for (const prop of node.properties) {
  117. hasComputed = prop.computed === true;
  118. if (hasComputed) break;
  119. }
  120. if (!hasComputed) return;
  121. const initProps = [];
  122. const computedProps = [];
  123. let foundComputed = false;
  124. for (const prop of node.properties) {
  125. if (_core.types.isSpreadElement(prop)) {
  126. continue;
  127. }
  128. if (prop.computed) {
  129. foundComputed = true;
  130. }
  131. if (foundComputed) {
  132. computedProps.push(prop);
  133. } else {
  134. initProps.push(prop);
  135. }
  136. }
  137. const objId = scope.generateUidIdentifierBasedOnNode(parent);
  138. const initPropExpression = _core.types.objectExpression(initProps);
  139. const body = [];
  140. body.push(_core.types.variableDeclaration("var", [_core.types.variableDeclarator(objId, initPropExpression)]));
  141. const single = pushComputedProps({
  142. scope,
  143. objId,
  144. body,
  145. computedProps,
  146. initPropExpression,
  147. state
  148. });
  149. if (single) {
  150. path.replaceWith(single);
  151. } else {
  152. if (setComputedProperties) {
  153. body.push(_core.types.expressionStatement(_core.types.cloneNode(objId)));
  154. }
  155. path.replaceWithMultiple(body);
  156. }
  157. }
  158. }
  159. }
  160. };
  161. });
  162. //# sourceMappingURL=index.js.map