UNPKG

jotai

Version:

👻 Next gen state management that will spook you

122 lines (103 loc) • 5.56 kB
(function (global, factory) { typeof exports === 'object' && typeof module !== 'undefined' ? factory(exports, require('path'), require('@babel/template')) : typeof define === 'function' && define.amd ? define(['exports', 'path', '@babel/template'], factory) : (global = typeof globalThis !== 'undefined' ? globalThis : global || self, factory(global.jotaiPreset = {}, global.path, global.templateBuilder)); })(this, (function (exports, path, templateBuilder) { 'use strict'; function _interopDefaultLegacy (e) { return e && typeof e === 'object' && 'default' in e ? e : { 'default': e }; } var path__default = /*#__PURE__*/_interopDefaultLegacy(path); var templateBuilder__default = /*#__PURE__*/_interopDefaultLegacy(templateBuilder); function isAtom(t, callee) { if (t.isIdentifier(callee) && atomFunctionNames.includes(callee.name)) { return true; } if (t.isMemberExpression(callee)) { var property = callee.property; if (t.isIdentifier(property) && atomFunctionNames.includes(property.name)) { return true; } } return false; } var atomFunctionNames = ['atom', 'atomFamily', 'atomWithDefault', 'atomWithObservable', 'atomWithReducer', 'atomWithReset', 'atomWithStorage', 'freezeAtom', 'loadable', 'selectAtom', 'splitAtom']; function debugLabelPlugin(_ref) { var t = _ref.types; return { visitor: { ExportDefaultDeclaration: function ExportDefaultDeclaration(nodePath, state) { var node = nodePath.node; if (t.isCallExpression(node.declaration) && isAtom(t, node.declaration.callee)) { var filename = state.filename || 'unknown'; var displayName = path__default["default"].basename(filename, path__default["default"].extname(filename)); if (displayName === 'index') { displayName = path__default["default"].basename(path__default["default"].dirname(filename)); } var buildExport = templateBuilder__default["default"]("\n const %%atomIdentifier%% = %%atom%%;\n export default %%atomIdentifier%%\n "); var ast = buildExport({ atomIdentifier: t.identifier(displayName), atom: node.declaration }); nodePath.replaceWithMultiple(ast); } }, VariableDeclarator: function VariableDeclarator(path) { if (t.isIdentifier(path.node.id) && t.isCallExpression(path.node.init) && isAtom(t, path.node.init.callee)) { path.parentPath.insertAfter(t.expressionStatement(t.assignmentExpression('=', t.memberExpression(t.identifier(path.node.id.name), t.identifier('debugLabel')), t.stringLiteral(path.node.id.name)))); } } } }; } function reactRefreshPlugin(_ref) { var t = _ref.types; return { pre: function pre(_ref2) { var opts = _ref2.opts; if (!opts.filename) { throw new Error('Filename must be available'); } }, visitor: { Program: { exit: function exit(path) { var jotaiAtomCache = templateBuilder__default["default"]("\n globalThis.jotaiAtomCache = globalThis.jotaiAtomCache || {\n cache: new Map(),\n get(name, inst) { \n if (this.cache.has(name)) {\n return this.cache.get(name)\n }\n this.cache.set(name, inst)\n return inst\n },\n }")(); path.unshiftContainer('body', jotaiAtomCache); } }, ExportDefaultDeclaration: function ExportDefaultDeclaration(nodePath, state) { var node = nodePath.node; if (t.isCallExpression(node.declaration) && isAtom(t, node.declaration.callee)) { var filename = state.filename || 'unknown'; var atomKey = filename + "/defaultExport"; var buildExport = templateBuilder__default["default"]("export default globalThis.jotaiAtomCache.get(%%atomKey%%, %%atom%%)"); var ast = buildExport({ atomKey: t.stringLiteral(atomKey), atom: node.declaration }); nodePath.replaceWith(ast); } }, VariableDeclarator: function VariableDeclarator(nodePath, state) { var _nodePath$parentPath$, _nodePath$parentPath$2; if (t.isIdentifier(nodePath.node.id) && t.isCallExpression(nodePath.node.init) && isAtom(t, nodePath.node.init.callee) && ((_nodePath$parentPath$ = nodePath.parentPath.parentPath) != null && _nodePath$parentPath$.isProgram() || (_nodePath$parentPath$2 = nodePath.parentPath.parentPath) != null && _nodePath$parentPath$2.isExportNamedDeclaration())) { var filename = state.filename || 'unknown'; var atomKey = filename + "/" + nodePath.node.id.name; var buildAtomDeclaration = templateBuilder__default["default"]("const %%atomIdentifier%% = globalThis.jotaiAtomCache.get(%%atomKey%%, %%atom%%)"); var ast = buildAtomDeclaration({ atomIdentifier: t.identifier(nodePath.node.id.name), atomKey: t.stringLiteral(atomKey), atom: nodePath.node.init }); nodePath.parentPath.replaceWith(ast); } } } }; } function jotaiPreset() { return { plugins: [debugLabelPlugin, reactRefreshPlugin] }; } exports["default"] = jotaiPreset; Object.defineProperty(exports, '__esModule', { value: true }); }));