index.js 5.5 KB

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