index.js 1.6 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849
  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 _helperAnnotateAsPure = require("@babel/helper-annotate-as-pure");
  8. var _core = require("@babel/core");
  9. const PURE_CALLS = [["react", new Set(["cloneElement", "createContext", "createElement", "createFactory", "createRef", "forwardRef", "isValidElement", "memo", "lazy"])], ["react-dom", new Set(["createPortal"])]];
  10. var _default = exports.default = (0, _helperPluginUtils.declare)(api => {
  11. api.assertVersion(7);
  12. return {
  13. name: "transform-react-pure-annotations",
  14. visitor: {
  15. CallExpression(path) {
  16. if (isReactCall(path)) {
  17. (0, _helperAnnotateAsPure.default)(path);
  18. }
  19. }
  20. }
  21. };
  22. });
  23. function isReactCall(path) {
  24. const calleePath = path.get("callee");
  25. if (!calleePath.isMemberExpression()) {
  26. for (const [module, methods] of PURE_CALLS) {
  27. for (const method of methods) {
  28. if (calleePath.referencesImport(module, method)) {
  29. return true;
  30. }
  31. }
  32. }
  33. return false;
  34. }
  35. const object = calleePath.get("object");
  36. const callee = calleePath.node;
  37. if (!callee.computed && _core.types.isIdentifier(callee.property)) {
  38. const propertyName = callee.property.name;
  39. for (const [module, methods] of PURE_CALLS) {
  40. if (object.referencesImport(module, "default") || object.referencesImport(module, "*")) {
  41. return methods.has(propertyName);
  42. }
  43. }
  44. }
  45. return false;
  46. }
  47. //# sourceMappingURL=index.js.map