jotai
Version:
👻 Next gen state management that will spook you
112 lines (107 loc) • 4.04 kB
JavaScript
System.register(['@babel/template'], (function (exports) {
'use strict';
var _templateBuilder;
return {
setters: [function (module) {
_templateBuilder = module.default;
}],
execute: (function () {
exports('default', reactRefreshPlugin);
function isAtom(t, callee, customAtomNames = []) {
const atomNames = [...atomFunctionNames, ...customAtomNames];
if (t.isIdentifier(callee) && atomNames.includes(callee.name)) {
return true;
}
if (t.isMemberExpression(callee)) {
const { property } = callee;
if (t.isIdentifier(property) && atomNames.includes(property.name)) {
return true;
}
}
return false;
}
const atomFunctionNames = [
"abortableAtom",
"atom",
"atomFamily",
"atomWithDefault",
"atomWithHash",
"atomWithImmer",
"atomWithInfiniteQuery",
"atomWithMachine",
"atomWithMutation",
"atomWithObservable",
"atomWithProxy",
"atomWithQuery",
"atomWithReducer",
"atomWithReset",
"atomWithSubscription",
"atomWithStorage",
"atomWithStore",
"freezeAtom",
"loadable",
"selectAtom",
"splitAtom"
];
const templateBuilder = _templateBuilder.default || _templateBuilder;
function reactRefreshPlugin({ types: t }, options) {
return {
pre({ opts }) {
if (!opts.filename) {
throw new Error("Filename must be available");
}
},
visitor: {
Program: {
exit(path) {
const jotaiAtomCache = templateBuilder(`
globalThis.jotaiAtomCache = globalThis.jotaiAtomCache || {
cache: new Map(),
get(name, inst) {
if (this.cache.has(name)) {
return this.cache.get(name)
}
this.cache.set(name, inst)
return inst
},
}`)();
path.unshiftContainer("body", jotaiAtomCache);
}
},
ExportDefaultDeclaration(nodePath, state) {
const { node } = nodePath;
if (t.isCallExpression(node.declaration) && isAtom(t, node.declaration.callee, options == null ? void 0 : options.customAtomNames)) {
const filename = state.filename || "unknown";
const atomKey = `${filename}/defaultExport`;
const buildExport = templateBuilder(
`export default globalThis.jotaiAtomCache.get(%%atomKey%%, %%atom%%)`
);
const ast = buildExport({
atomKey: t.stringLiteral(atomKey),
atom: node.declaration
});
nodePath.replaceWith(ast);
}
},
VariableDeclarator(nodePath, state) {
var _a, _b;
if (t.isIdentifier(nodePath.node.id) && t.isCallExpression(nodePath.node.init) && isAtom(t, nodePath.node.init.callee, options == null ? void 0 : options.customAtomNames) && (((_a = nodePath.parentPath.parentPath) == null ? void 0 : _a.isProgram()) || ((_b = nodePath.parentPath.parentPath) == null ? void 0 : _b.isExportNamedDeclaration()))) {
const filename = state.filename || "unknown";
const atomKey = `${filename}/${nodePath.node.id.name}`;
const buildAtomDeclaration = templateBuilder(
`const %%atomIdentifier%% = globalThis.jotaiAtomCache.get(%%atomKey%%, %%atom%%)`
);
const ast = buildAtomDeclaration({
atomIdentifier: t.identifier(nodePath.node.id.name),
atomKey: t.stringLiteral(atomKey),
atom: nodePath.node.init
});
nodePath.parentPath.replaceWith(ast);
}
}
}
};
}
})
};
}));