UNPKG

svgedit

Version:

Powerful SVG-Editor for your browser

46 lines (43 loc) 1.44 kB
import { declare } from '@babel/helper-plugin-utils'; import annotateAsPure from '@babel/helper-annotate-as-pure'; import { types } from '@babel/core'; const PURE_CALLS = [["react", new Set(["cloneElement", "createContext", "createElement", "createFactory", "createRef", "forwardRef", "isValidElement", "memo", "lazy"])], ["react-dom", new Set(["createPortal"])]]; const index = declare(api => { api.assertVersion("^7.0.0-0 || ^8.0.0"); return { name: "transform-react-pure-annotations", visitor: { CallExpression(path) { if (isReactCall(path)) { annotateAsPure(path); } } } }; }); function isReactCall(path) { const calleePath = path.get("callee"); if (!calleePath.isMemberExpression()) { for (const [module, methods] of PURE_CALLS) { for (const method of methods) { if (calleePath.referencesImport(module, method)) { return true; } } } return false; } const object = calleePath.get("object"); const callee = calleePath.node; if (!callee.computed && types.isIdentifier(callee.property)) { const propertyName = callee.property.name; for (const [module, methods] of PURE_CALLS) { if (object.referencesImport(module, "default") || object.referencesImport(module, "*")) { return methods.has(propertyName); } } } return false; } export { index as default }; //# sourceMappingURL=index.js.map