index.js 5.4 KB

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