svgedit
Version:
Powerful SVG-Editor for your browser
46 lines (43 loc) • 1.44 kB
JavaScript
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