index.js 5.4 KB

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