modality-ts 0.0.14 → 0.0.15

package/dist/extract/sources/jotai/discover.js

@@ -1,65 +1,182 @@
 import * as ts from "typescript";
-import { inferAtomDomain, initialValueForAtom, typeAliasDeclarations, } from "./domains.js";
+import { atomCreatorName, isAtomCreatorCall, resolveJotaiImports, } from "./imports.js";
+import { classifyAtomCall, classifyFamilyInstance, staticFamilyParam, typeAliasDeclarations, } from "./domains.js";
+import { atomVarId, familyVarId } from "./ids.js";
+import { metadataToRecord } from "./types.js";
+import { applyHydrationOverrides, discoverHydrationOverrides, } from "./hydration.js";
+import { discoverComponentStoreScopes } from "./stores.js";
 export function discoverJotaiAtoms(sourceText, fileName = "state.ts") {
+    return discoverJotaiAtomsDetailed(sourceText, fileName).decls;
+}
+export function discoverJotaiAtomsDetailed(sourceText, fileName = "state.ts") {
     const source = ts.createSourceFile(fileName, sourceText, ts.ScriptTarget.Latest, true, ts.ScriptKind.TSX);
-    const importedAtomNames = atomImportNames(source);
-    if (importedAtomNames.size === 0)
-        return [];
+    const imports = resolveJotaiImports(source);
+    if (imports.atomCreators.size === 0)
+        return emptyDiscoverResult();
     const typeAliases = typeAliasDeclarations(source);
+    const warnings = [];
+    const atomMetadata = new Map();
+    const atomNames = new Set();
+    const familyFactories = new Map();
     const decls = [];
     const visit = (node) => {
         if (ts.isVariableDeclaration(node) &&
             ts.isIdentifier(node.name) &&
             node.initializer &&
-            isAtomCall(node.initializer, importedAtomNames)) {
-            const origin = { file: fileName, ...lineAndColumn(source, node) };
-            const domain = inferAtomDomain(node.initializer, typeAliases);
-            const variable = {
-                id: `atom:${node.name.text}`,
-                domain,
-                origin,
-                scope: { kind: "global" },
-                initial: initialValueForAtom(node.initializer, domain),
-            };
-            decls.push({
-                id: variable.id,
-                kind: "jotai/atom",
-                var: variable,
-                origin,
-                metadata: { atomName: node.name.text },
-            });
+            isAtomCreatorCall(node.initializer, imports.atomCreators)) {
+            const atomName = node.name.text;
+            const creator = atomCreatorName(node.initializer, imports.atomCreators);
+            if (creator === "atomFamily") {
+                familyFactories.set(atomName, node.initializer);
+                const classification = classifyAtomCall(node.initializer, atomName, imports, typeAliases);
+                atomMetadata.set(atomName, classification.metadata);
+                if (classification.warning) {
+                    warnings.push({
+                        message: classification.warning,
+                        source: anchor(source, fileName, node),
+                    });
+                }
+            }
+            else {
+                const classification = classifyAtomCall(node.initializer, atomName, imports, typeAliases);
+                atomNames.add(atomName);
+                atomMetadata.set(atomName, classification.metadata);
+                if (classification.warning) {
+                    warnings.push({
+                        message: classification.warning,
+                        source: anchor(source, fileName, node),
+                    });
+                }
+                if (classification.emitVar) {
+                    decls.push(atomDecl(node, atomName, classification, fileName, source));
+                }
+            }
+        }
+        if (ts.isCallExpression(node) &&
+            ts.isIdentifier(node.expression) &&
+            familyFactories.has(node.expression.text)) {
+            const familyName = node.expression.text;
+            const paramArg = node.arguments[0];
+            if (!paramArg)
+                return;
+            const staticParam = staticFamilyParam(paramArg);
+            if (!staticParam) {
+                warnings.push({
+                    message: `Jotai dynamic atom family param unsupported for ${familyName}`,
+                    source: anchor(source, fileName, node),
+                });
+            }
+            else {
+                const familyCall = familyFactories.get(familyName);
+                const initializer = familyCall?.arguments[0];
+                let innerCall;
+                if (initializer &&
+                    (ts.isArrowFunction(initializer) ||
+                        ts.isFunctionExpression(initializer))) {
+                    const body = initializer.body;
+                    if (ts.isCallExpression(body) &&
+                        isAtomCreatorCall(body, imports.atomCreators)) {
+                        innerCall = body;
+                    }
+                    else if (ts.isBlock(body)) {
+                        for (const stmt of body.statements) {
+                            if (ts.isReturnStatement(stmt) &&
+                                stmt.expression &&
+                                ts.isCallExpression(stmt.expression) &&
+                                isAtomCreatorCall(stmt.expression, imports.atomCreators)) {
+                                innerCall = stmt.expression;
+                                break;
+                            }
+                        }
+                    }
+                }
+                if (innerCall) {
+                    const classification = classifyFamilyInstance(familyName, staticParam, innerCall, typeAliases);
+                    const varId = familyVarId(familyName, staticParam);
+                    atomNames.add(varId);
+                    atomMetadata.set(varId, classification.metadata);
+                    decls.push({
+                        id: varId,
+                        kind: "jotai/atom-family",
+                        var: {
+                            id: varId,
+                            domain: classification.domain,
+                            origin: anchor(source, fileName, node),
+                            scope: { kind: "global" },
+                            initial: classification.initial,
+                        },
+                        origin: anchor(source, fileName, node),
+                        metadata: metadataToRecord(classification.metadata),
+                    });
+                }
+            }
         }
         ts.forEachChild(node, visit);
     };
     visit(source);
-    return decls;
-}
-function atomImportNames(source) {
-    const names = new Set();
-    for (const statement of source.statements) {
-        if (!ts.isImportDeclaration(statement) ||
-            !isJotaiModule(statement.moduleSpecifier))
-            continue;
-        const bindings = statement.importClause?.namedBindings;
-        if (!bindings || !ts.isNamedImports(bindings))
-            continue;
-        for (const specifier of bindings.elements) {
-            if ((specifier.propertyName?.text ?? specifier.name.text) === "atom")
-                names.add(specifier.name.text);
-        }
+    const hydration = discoverHydrationOverrides(source, fileName, imports);
+    for (const warning of hydration.warnings)
+        warnings.push(warning);
+    const hydratedDecls = applyHydrationOverrides(decls, hydration.overrides);
+    const componentStoreScopes = discoverComponentStoreScopes(source, imports);
+    const storeScopedDecls = [];
+    for (const storeScope of new Set(componentStoreScopes.values())) {
+        storeScopedDecls.push(...duplicateAtomsForStore(hydratedDecls, storeScope));
     }
-    return names;
-}
-function isJotaiModule(moduleSpecifier) {
-    return (ts.isStringLiteral(moduleSpecifier) && moduleSpecifier.text === "jotai");
+    return {
+        decls: [...hydratedDecls, ...storeScopedDecls],
+        warnings,
+        atomNames,
+        atomMetadata,
+    };
 }
-function isAtomCall(node, atomNames) {
-    return (ts.isCallExpression(node) &&
-        ts.isIdentifier(node.expression) &&
-        atomNames.has(node.expression.text));
+function atomDecl(node, atomName, classification, fileName, source) {
+    const origin = anchor(source, fileName, node);
+    const variable = {
+        id: atomVarId(atomName),
+        domain: classification.domain,
+        origin,
+        scope: { kind: "global" },
+        initial: classification.initial,
+    };
+    return {
+        id: variable.id,
+        kind: "jotai/atom",
+        var: variable,
+        origin,
+        metadata: metadataToRecord(classification.metadata),
+    };
 }
-function lineAndColumn(source, node) {
+function anchor(source, fileName, node) {
     const pos = source.getLineAndCharacterOfPosition(node.getStart(source));
-    return { line: pos.line + 1, column: pos.character + 1 };
+    return { file: fileName, line: pos.line + 1, column: pos.character + 1 };
+}
+function emptyDiscoverResult() {
+    return {
+        decls: [],
+        warnings: [],
+        atomNames: new Set(),
+        atomMetadata: new Map(),
+    };
+}
+export function duplicateAtomsForStore(decls, storeScope) {
+    return decls
+        .filter((decl) => decl.var && decl.kind.startsWith("jotai/"))
+        .map((decl) => {
+        const atomName = typeof decl.metadata?.atomName === "string"
+            ? decl.metadata.atomName
+            : decl.var.id.replace(/^atom:/, "").replace(/@store:.+$/, "");
+        const varId = atomVarId(atomName, storeScope);
+        return {
+            ...decl,
+            id: varId,
+            var: { ...decl.var, id: varId },
+            metadata: { ...decl.metadata, storeScope },
+        };
+    });
+}
+export function isAtomCall(node, atomCreators) {
+    return isAtomCreatorCall(node, atomCreators);
 }
+export { resolveJotaiImports } from "./imports.js";
 //# sourceMappingURL=discover.js.map

package/dist/extract/sources/jotai/discover.js.map

@@ -1 +1 @@
-{"version":3,"file":"discover.js","sourceRoot":"","sources":["../../../../src/extract/sources/jotai/discover.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,MAAM,YAAY,CAAC;AAGjC,OAAO,EACL,eAAe,EACf,mBAAmB,EACnB,qBAAqB,GACtB,MAAM,cAAc,CAAC;AAEtB,MAAM,UAAU,kBAAkB,CAChC,UAAkB,EAClB,QAAQ,GAAG,UAAU;IAErB,MAAM,MAAM,GAAG,EAAE,CAAC,gBAAgB,CAChC,QAAQ,EACR,UAAU,EACV,EAAE,CAAC,YAAY,CAAC,MAAM,EACtB,IAAI,EACJ,EAAE,CAAC,UAAU,CAAC,GAAG,CAClB,CAAC;IACF,MAAM,iBAAiB,GAAG,eAAe,CAAC,MAAM,CAAC,CAAC;IAClD,IAAI,iBAAiB,CAAC,IAAI,KAAK,CAAC;QAAE,OAAO,EAAE,CAAC;IAC5C,MAAM,WAAW,GAAG,qBAAqB,CAAC,MAAM,CAAC,CAAC;IAElD,MAAM,KAAK,GAAiB,EAAE,CAAC;IAC/B,MAAM,KAAK,GAAG,CAAC,IAAa,EAAQ,EAAE;QACpC,IACE,EAAE,CAAC,qBAAqB,CAAC,IAAI,CAAC;YAC9B,EAAE,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC;YAC1B,IAAI,CAAC,WAAW;YAChB,UAAU,CAAC,IAAI,CAAC,WAAW,EAAE,iBAAiB,CAAC,EAC/C,CAAC;YACD,MAAM,MAAM,GAAG,EAAE,IAAI,EAAE,QAAQ,EAAE,GAAG,aAAa,CAAC,MAAM,EAAE,IAAI,CAAC,EAAE,CAAC;YAClE,MAAM,MAAM,GAAG,eAAe,CAAC,IAAI,CAAC,WAAW,EAAE,WAAW,CAAC,CAAC;YAC9D,MAAM,QAAQ,GAAiB;gBAC7B,EAAE,EAAE,QAAQ,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE;gBAC5B,MAAM;gBACN,MAAM;gBACN,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;gBACzB,OAAO,EAAE,mBAAmB,CAAC,IAAI,CAAC,WAAW,EAAE,MAAM,CAAC;aACvD,CAAC;YACF,KAAK,CAAC,IAAI,CAAC;gBACT,EAAE,EAAE,QAAQ,CAAC,EAAE;gBACf,IAAI,EAAE,YAAY;gBAClB,GAAG,EAAE,QAAQ;gBACb,MAAM;gBACN,QAAQ,EAAE,EAAE,QAAQ,EAAE,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE;aACvC,CAAC,CAAC;QACL,CAAC;QACD,EAAE,CAAC,YAAY,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;IAC/B,CAAC,CAAC;IACF,KAAK,CAAC,MAAM,CAAC,CAAC;IACd,OAAO,KAAK,CAAC;AACf,CAAC;AAED,SAAS,eAAe,CAAC,MAAqB;IAC5C,MAAM,KAAK,GAAG,IAAI,GAAG,EAAU,CAAC;IAChC,KAAK,MAAM,SAAS,IAAI,MAAM,CAAC,UAAU,EAAE,CAAC;QAC1C,IACE,CAAC,EAAE,CAAC,mBAAmB,CAAC,SAAS,CAAC;YAClC,CAAC,aAAa,CAAC,SAAS,CAAC,eAAe,CAAC;YAEzC,SAAS;QACX,MAAM,QAAQ,GAAG,SAAS,CAAC,YAAY,EAAE,aAAa,CAAC;QACvD,IAAI,CAAC,QAAQ,IAAI,CAAC,EAAE,CAAC,cAAc,CAAC,QAAQ,CAAC;YAAE,SAAS;QACxD,KAAK,MAAM,SAAS,IAAI,QAAQ,CAAC,QAAQ,EAAE,CAAC;YAC1C,IAAI,CAAC,SAAS,CAAC,YAAY,EAAE,IAAI,IAAI,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,MAAM;gBAClE,KAAK,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACnC,CAAC;IACH,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC;AAED,SAAS,aAAa,CAAC,eAA8B;IACnD,OAAO,CACL,EAAE,CAAC,eAAe,CAAC,eAAe,CAAC,IAAI,eAAe,CAAC,IAAI,KAAK,OAAO,CACxE,CAAC;AACJ,CAAC;AAED,SAAS,UAAU,CACjB,IAAmB,EACnB,SAAsB;IAEtB,OAAO,CACL,EAAE,CAAC,gBAAgB,CAAC,IAAI,CAAC;QACzB,EAAE,CAAC,YAAY,CAAC,IAAI,CAAC,UAAU,CAAC;QAChC,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,CACpC,CAAC;AACJ,CAAC;AAED,SAAS,aAAa,CACpB,MAAqB,EACrB,IAAa;IAEb,MAAM,GAAG,GAAG,MAAM,CAAC,6BAA6B,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC;IACxE,OAAO,EAAE,IAAI,EAAE,GAAG,CAAC,IAAI,GAAG,CAAC,EAAE,MAAM,EAAE,GAAG,CAAC,SAAS,GAAG,CAAC,EAAE,CAAC;AAC3D,CAAC"}
+{"version":3,"file":"discover.js","sourceRoot":"","sources":["../../../../src/extract/sources/jotai/discover.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,MAAM,YAAY,CAAC;AAGjC,OAAO,EACL,eAAe,EACf,iBAAiB,EACjB,mBAAmB,GACpB,MAAM,cAAc,CAAC;AACtB,OAAO,EACL,gBAAgB,EAChB,sBAAsB,EACtB,iBAAiB,EACjB,qBAAqB,GACtB,MAAM,cAAc,CAAC;AACtB,OAAO,EAAE,SAAS,EAAE,WAAW,EAAE,MAAM,UAAU,CAAC;AAClD,OAAO,EAAE,gBAAgB,EAAE,MAAM,YAAY,CAAC;AAC9C,OAAO,EACL,uBAAuB,EACvB,0BAA0B,GAC3B,MAAM,gBAAgB,CAAC;AACxB,OAAO,EAAE,4BAA4B,EAAE,MAAM,aAAa,CAAC;AAS3D,MAAM,UAAU,kBAAkB,CAChC,UAAkB,EAClB,QAAQ,GAAG,UAAU;IAErB,OAAO,0BAA0B,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC,KAAK,CAAC;AAChE,CAAC;AAED,MAAM,UAAU,0BAA0B,CACxC,UAAkB,EAClB,QAAQ,GAAG,UAAU;IAErB,MAAM,MAAM,GAAG,EAAE,CAAC,gBAAgB,CAChC,QAAQ,EACR,UAAU,EACV,EAAE,CAAC,YAAY,CAAC,MAAM,EACtB,IAAI,EACJ,EAAE,CAAC,UAAU,CAAC,GAAG,CAClB,CAAC;IACF,MAAM,OAAO,GAAG,mBAAmB,CAAC,MAAM,CAAC,CAAC;IAC5C,IAAI,OAAO,CAAC,YAAY,CAAC,IAAI,KAAK,CAAC;QAAE,OAAO,mBAAmB,EAAE,CAAC;IAElE,MAAM,WAAW,GAAG,qBAAqB,CAAC,MAAM,CAAC,CAAC;IAClD,MAAM,QAAQ,GAAiD,EAAE,CAAC;IAClE,MAAM,YAAY,GAAG,IAAI,GAAG,EAGzB,CAAC;IACJ,MAAM,SAAS,GAAG,IAAI,GAAG,EAAU,CAAC;IACpC,MAAM,eAAe,GAAG,IAAI,GAAG,EAA6B,CAAC;IAC7D,MAAM,KAAK,GAAiB,EAAE,CAAC;IAE/B,MAAM,KAAK,GAAG,CAAC,IAAa,EAAQ,EAAE;QACpC,IACE,EAAE,CAAC,qBAAqB,CAAC,IAAI,CAAC;YAC9B,EAAE,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC;YAC1B,IAAI,CAAC,WAAW;YAChB,iBAAiB,CAAC,IAAI,CAAC,WAAW,EAAE,OAAO,CAAC,YAAY,CAAC,EACzD,CAAC;YACD,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC;YAChC,MAAM,OAAO,GAAG,eAAe,CAAC,IAAI,CAAC,WAAW,EAAE,OAAO,CAAC,YAAY,CAAC,CAAC;YACxE,IAAI,OAAO,KAAK,YAAY,EAAE,CAAC;gBAC7B,eAAe,CAAC,GAAG,CAAC,QAAQ,EAAE,IAAI,CAAC,WAAW,CAAC,CAAC;gBAChD,MAAM,cAAc,GAAG,gBAAgB,CACrC,IAAI,CAAC,WAAW,EAChB,QAAQ,EACR,OAAO,EACP,WAAW,CACZ,CAAC;gBACF,YAAY,CAAC,GAAG,CAAC,QAAQ,EAAE,cAAc,CAAC,QAAQ,CAAC,CAAC;gBACpD,IAAI,cAAc,CAAC,OAAO,EAAE,CAAC;oBAC3B,QAAQ,CAAC,IAAI,CAAC;wBACZ,OAAO,EAAE,cAAc,CAAC,OAAO;wBAC/B,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,QAAQ,EAAE,IAAI,CAAC;qBACvC,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;iBAAM,CAAC;gBACN,MAAM,cAAc,GAAG,gBAAgB,CACrC,IAAI,CAAC,WAAW,EAChB,QAAQ,EACR,OAAO,EACP,WAAW,CACZ,CAAC;gBACF,SAAS,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;gBACxB,YAAY,CAAC,GAAG,CAAC,QAAQ,EAAE,cAAc,CAAC,QAAQ,CAAC,CAAC;gBACpD,IAAI,cAAc,CAAC,OAAO,EAAE,CAAC;oBAC3B,QAAQ,CAAC,IAAI,CAAC;wBACZ,OAAO,EAAE,cAAc,CAAC,OAAO;wBAC/B,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,QAAQ,EAAE,IAAI,CAAC;qBACvC,CAAC,CAAC;gBACL,CAAC;gBACD,IAAI,cAAc,CAAC,OAAO,EAAE,CAAC;oBAC3B,KAAK,CAAC,IAAI,CACR,QAAQ,CAAC,IAAI,EAAE,QAAQ,EAAE,cAAc,EAAE,QAAQ,EAAE,MAAM,CAAC,CAC3D,CAAC;gBACJ,CAAC;YACH,CAAC;QACH,CAAC;QACD,IACE,EAAE,CAAC,gBAAgB,CAAC,IAAI,CAAC;YACzB,EAAE,CAAC,YAAY,CAAC,IAAI,CAAC,UAAU,CAAC;YAChC,eAAe,CAAC,GAAG,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,EACzC,CAAC;YACD,MAAM,UAAU,GAAG,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC;YACxC,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;YACnC,IAAI,CAAC,QAAQ;gBAAE,OAAO;YACtB,MAAM,WAAW,GAAG,iBAAiB,CAAC,QAAQ,CAAC,CAAC;YAChD,IAAI,CAAC,WAAW,EAAE,CAAC;gBACjB,QAAQ,CAAC,IAAI,CAAC;oBACZ,OAAO,EAAE,mDAAmD,UAAU,EAAE;oBACxE,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,QAAQ,EAAE,IAAI,CAAC;iBACvC,CAAC,CAAC;YACL,CAAC;iBAAM,CAAC;gBACN,MAAM,UAAU,GAAG,eAAe,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;gBACnD,MAAM,WAAW,GAAG,UAAU,EAAE,SAAS,CAAC,CAAC,CAAC,CAAC;gBAC7C,IAAI,SAAwC,CAAC;gBAC7C,IACE,WAAW;oBACX,CAAC,EAAE,CAAC,eAAe,CAAC,WAAW,CAAC;wBAC9B,EAAE,CAAC,oBAAoB,CAAC,WAAW,CAAC,CAAC,EACvC,CAAC;oBACD,MAAM,IAAI,GAAG,WAAW,CAAC,IAAI,CAAC;oBAC9B,IACE,EAAE,CAAC,gBAAgB,CAAC,IAAI,CAAC;wBACzB,iBAAiB,CAAC,IAAI,EAAE,OAAO,CAAC,YAAY,CAAC,EAC7C,CAAC;wBACD,SAAS,GAAG,IAAI,CAAC;oBACnB,CAAC;yBAAM,IAAI,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC;wBAC5B,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;4BACnC,IACE,EAAE,CAAC,iBAAiB,CAAC,IAAI,CAAC;gCAC1B,IAAI,CAAC,UAAU;gCACf,EAAE,CAAC,gBAAgB,CAAC,IAAI,CAAC,UAAU,CAAC;gCACpC,iBAAiB,CAAC,IAAI,CAAC,UAAU,EAAE,OAAO,CAAC,YAAY,CAAC,EACxD,CAAC;gCACD,SAAS,GAAG,IAAI,CAAC,UAAU,CAAC;gCAC5B,MAAM;4BACR,CAAC;wBACH,CAAC;oBACH,CAAC;gBACH,CAAC;gBACD,IAAI,SAAS,EAAE,CAAC;oBACd,MAAM,cAAc,GAAG,sBAAsB,CAC3C,UAAU,EACV,WAAW,EACX,SAAS,EACT,WAAW,CACZ,CAAC;oBACF,MAAM,KAAK,GAAG,WAAW,CAAC,UAAU,EAAE,WAAW,CAAC,CAAC;oBACnD,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;oBACrB,YAAY,CAAC,GAAG,CAAC,KAAK,EAAE,cAAc,CAAC,QAAQ,CAAC,CAAC;oBACjD,KAAK,CAAC,IAAI,CAAC;wBACT,EAAE,EAAE,KAAK;wBACT,IAAI,EAAE,mBAAmB;wBACzB,GAAG,EAAE;4BACH,EAAE,EAAE,KAAK;4BACT,MAAM,EAAE,cAAc,CAAC,MAAM;4BAC7B,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,QAAQ,EAAE,IAAI,CAAC;4BACtC,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;4BACzB,OAAO,EAAE,cAAc,CAAC,OAAO;yBAChC;wBACD,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,QAAQ,EAAE,IAAI,CAAC;wBACtC,QAAQ,EAAE,gBAAgB,CAAC,cAAc,CAAC,QAAQ,CAAC;qBACpD,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;QACH,CAAC;QACD,EAAE,CAAC,YAAY,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;IAC/B,CAAC,CAAC;IACF,KAAK,CAAC,MAAM,CAAC,CAAC;IAEd,MAAM,SAAS,GAAG,0BAA0B,CAAC,MAAM,EAAE,QAAQ,EAAE,OAAO,CAAC,CAAC;IACxE,KAAK,MAAM,OAAO,IAAI,SAAS,CAAC,QAAQ;QAAE,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACjE,MAAM,aAAa,GAAG,uBAAuB,CAAC,KAAK,EAAE,SAAS,CAAC,SAAS,CAAC,CAAC;IAE1E,MAAM,oBAAoB,GAAG,4BAA4B,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAC3E,MAAM,gBAAgB,GAAiB,EAAE,CAAC;IAC1C,KAAK,MAAM,UAAU,IAAI,IAAI,GAAG,CAAC,oBAAoB,CAAC,MAAM,EAAE,CAAC,EAAE,CAAC;QAChE,gBAAgB,CAAC,IAAI,CAAC,GAAG,sBAAsB,CAAC,aAAa,EAAE,UAAU,CAAC,CAAC,CAAC;IAC9E,CAAC;IAED,OAAO;QACL,KAAK,EAAE,CAAC,GAAG,aAAa,EAAE,GAAG,gBAAgB,CAAC;QAC9C,QAAQ;QACR,SAAS;QACT,YAAY;KACb,CAAC;AACJ,CAAC;AAED,SAAS,QAAQ,CACf,IAA4B,EAC5B,QAAgB,EAChB,cAAmD,EACnD,QAAgB,EAChB,MAAqB;IAErB,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,EAAE,QAAQ,EAAE,IAAI,CAAC,CAAC;IAC9C,MAAM,QAAQ,GAAiB;QAC7B,EAAE,EAAE,SAAS,CAAC,QAAQ,CAAC;QACvB,MAAM,EAAE,cAAc,CAAC,MAAM;QAC7B,MAAM;QACN,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;QACzB,OAAO,EAAE,cAAc,CAAC,OAAO;KAChC,CAAC;IACF,OAAO;QACL,EAAE,EAAE,QAAQ,CAAC,EAAE;QACf,IAAI,EAAE,YAAY;QAClB,GAAG,EAAE,QAAQ;QACb,MAAM;QACN,QAAQ,EAAE,gBAAgB,CAAC,cAAc,CAAC,QAAQ,CAAC;KACpD,CAAC;AACJ,CAAC;AAED,SAAS,MAAM,CACb,MAAqB,EACrB,QAAgB,EAChB,IAAa;IAEb,MAAM,GAAG,GAAG,MAAM,CAAC,6BAA6B,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC;IACxE,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,GAAG,CAAC,IAAI,GAAG,CAAC,EAAE,MAAM,EAAE,GAAG,CAAC,SAAS,GAAG,CAAC,EAAE,CAAC;AAC3E,CAAC;AAED,SAAS,mBAAmB;IAC1B,OAAO;QACL,KAAK,EAAE,EAAE;QACT,QAAQ,EAAE,EAAE;QACZ,SAAS,EAAE,IAAI,GAAG,EAAE;QACpB,YAAY,EAAE,IAAI,GAAG,EAAE;KACxB,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,sBAAsB,CACpC,KAA4B,EAC5B,UAAkB;IAElB,OAAO,KAAK;SACT,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,GAAG,IAAI,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;SAC5D,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;QACZ,MAAM,QAAQ,GACZ,OAAO,IAAI,CAAC,QAAQ,EAAE,QAAQ,KAAK,QAAQ;YACzC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,QAAQ;YACxB,CAAC,CAAC,IAAI,CAAC,GAAI,CAAC,EAAE,CAAC,OAAO,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,YAAY,EAAE,EAAE,CAAC,CAAC;QACnE,MAAM,KAAK,GAAG,SAAS,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;QAC9C,OAAO;YACL,GAAG,IAAI;YACP,EAAE,EAAE,KAAK;YACT,GAAG,EAAE,EAAE,GAAG,IAAI,CAAC,GAAI,EAAE,EAAE,EAAE,KAAK,EAAE;YAChC,QAAQ,EAAE,EAAE,GAAG,IAAI,CAAC,QAAQ,EAAE,UAAU,EAAE;SAC3C,CAAC;IACJ,CAAC,CAAC,CAAC;AACP,CAAC;AAED,MAAM,UAAU,UAAU,CACxB,IAAmB,EACnB,YAAyC;IAEzC,OAAO,iBAAiB,CAAC,IAAI,EAAE,YAAY,CAAC,CAAC;AAC/C,CAAC;AAED,OAAO,EAAE,mBAAmB,EAAE,MAAM,cAAc,CAAC"}

package/dist/extract/sources/jotai/domains.d.ts

@@ -1,7 +1,20 @@
 import * as ts from "typescript";
+import type { AbstractDomain, Value } from "modality-ts/core";
 import { typeAliasDeclarations } from "modality-ts/extract/engine/spi";
-import { type AbstractDomain, type Value } from "modality-ts/core";
+import { type JotaiResolvedImports } from "./imports.js";
+import type { AtomConfigKind, JotaiAtomMetadata } from "./types.js";
 export { typeAliasDeclarations };
+export interface AtomClassification {
+    configKind: AtomConfigKind;
+    domain: AbstractDomain;
+    initial: Value;
+    metadata: JotaiAtomMetadata;
+    emitVar: boolean;
+    warning?: string;
+}
+export declare function classifyAtomCall(call: ts.CallExpression, atomName: string, imports: JotaiResolvedImports, typeAliases: ReadonlyMap<string, ts.TypeNode>): AtomClassification;
+export declare function classifyFamilyInstance(familyName: string, param: string, innerCall: ts.CallExpression, typeAliases: ReadonlyMap<string, ts.TypeNode>): AtomClassification;
 export declare function inferAtomDomain(call: ts.CallExpression, typeAliases?: ReadonlyMap<string, ts.TypeNode>): AbstractDomain;
 export declare function initialValueForAtom(call: ts.CallExpression, domain: AbstractDomain): Value;
+export declare function staticFamilyParam(expression: ts.Expression): string | undefined;
 //# sourceMappingURL=domains.d.ts.map

package/dist/extract/sources/jotai/domains.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"domains.d.ts","sourceRoot":"","sources":["../../../../src/extract/sources/jotai/domains.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,MAAM,YAAY,CAAC;AACjC,OAAO,EAGL,qBAAqB,EACtB,MAAM,gCAAgC,CAAC;AACxC,OAAO,EAEL,KAAK,cAAc,EACnB,KAAK,KAAK,EACX,MAAM,kBAAkB,CAAC;AAE1B,OAAO,EAAE,qBAAqB,EAAE,CAAC;AAEjC,wBAAgB,eAAe,CAC7B,IAAI,EAAE,EAAE,CAAC,cAAc,EACvB,WAAW,GAAE,WAAW,CAAC,MAAM,EAAE,EAAE,CAAC,QAAQ,CAAa,GACxD,cAAc,CAwBhB;AAED,wBAAgB,mBAAmB,CACjC,IAAI,EAAE,EAAE,CAAC,cAAc,EACvB,MAAM,EAAE,cAAc,GACrB,KAAK,CAyBP"}
+{"version":3,"file":"domains.d.ts","sourceRoot":"","sources":["../../../../src/extract/sources/jotai/domains.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,MAAM,YAAY,CAAC;AACjC,OAAO,KAAK,EAAE,cAAc,EAAE,KAAK,EAAE,MAAM,kBAAkB,CAAC;AAC9D,OAAO,EAGL,qBAAqB,EACtB,MAAM,gCAAgC,CAAC;AAMxC,OAAO,EAAmB,KAAK,oBAAoB,EAAE,MAAM,cAAc,CAAC;AAM1E,OAAO,KAAK,EAAE,cAAc,EAAE,iBAAiB,EAAE,MAAM,YAAY,CAAC;AAEpE,OAAO,EAAE,qBAAqB,EAAE,CAAC;AAEjC,MAAM,WAAW,kBAAkB;IACjC,UAAU,EAAE,cAAc,CAAC;IAC3B,MAAM,EAAE,cAAc,CAAC;IACvB,OAAO,EAAE,KAAK,CAAC;IACf,QAAQ,EAAE,iBAAiB,CAAC;IAC5B,OAAO,EAAE,OAAO,CAAC;IACjB,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB;AAED,wBAAgB,gBAAgB,CAC9B,IAAI,EAAE,EAAE,CAAC,cAAc,EACvB,QAAQ,EAAE,MAAM,EAChB,OAAO,EAAE,oBAAoB,EAC7B,WAAW,EAAE,WAAW,CAAC,MAAM,EAAE,EAAE,CAAC,QAAQ,CAAC,GAC5C,kBAAkB,CAwBpB;AAoZD,wBAAgB,sBAAsB,CACpC,UAAU,EAAE,MAAM,EAClB,KAAK,EAAE,MAAM,EACb,SAAS,EAAE,EAAE,CAAC,cAAc,EAC5B,WAAW,EAAE,WAAW,CAAC,MAAM,EAAE,EAAE,CAAC,QAAQ,CAAC,GAC5C,kBAAkB,CAgBpB;AAED,wBAAgB,eAAe,CAC7B,IAAI,EAAE,EAAE,CAAC,cAAc,EACvB,WAAW,GAAE,WAAW,CAAC,MAAM,EAAE,EAAE,CAAC,QAAQ,CAAa,GACxD,cAAc,CAMhB;AAED,wBAAgB,mBAAmB,CACjC,IAAI,EAAE,EAAE,CAAC,cAAc,EACvB,MAAM,EAAE,cAAc,GACrB,KAAK,CAIP;AAuHD,wBAAgB,iBAAiB,CAC/B,UAAU,EAAE,EAAE,CAAC,UAAU,GACxB,MAAM,GAAG,SAAS,CAiBpB"}