@zerodev/wallet-react 0.0.1-alpha.1 → 0.0.1-alpha.11

package/src/store.ts

@@ -10,7 +10,12 @@ import type { LocalAccount } from 'viem'
 import type { SmartAccount } from 'viem/account-abstraction'
 import { create } from 'zustand'
 import { persist, subscribeWithSelector } from 'zustand/middleware'
-import type { OAuthConfig } from './oauth.js'
+
+// Internal OAuth config stored in the state (derived from connector params)
+type InternalOAuthConfig = {
+  backendUrl: string
+  projectId: string
+}
 
 export type ZeroDevWalletState = {
   // Core
@@ -19,15 +24,15 @@ export type ZeroDevWalletState = {
   session: ZeroDevWalletSession | null
 
   // Multi-chain support
-  chainIds: number[] // [activeChain, ...otherChains]
+  activeChainId: number | null
   kernelAccounts: Map<number, SmartAccount<KernelSmartAccountImplementation>>
   kernelClients: Map<number, KernelAccountClient>
 
   // Session expiry
   isExpiring: boolean
 
-  // OAuth config (optional)
-  oauthConfig: OAuthConfig | null
+  // OAuth config (derived from connector params)
+  oauthConfig: InternalOAuthConfig | null
 
   // Actions
   setWallet: (wallet: ZeroDevWalletSDK) => void
@@ -38,9 +43,9 @@ export type ZeroDevWalletState = {
   ) => void
   setKernelClient: (chainId: number, client: KernelAccountClient) => void
   setSession: (session: ZeroDevWalletSession | null) => void
-  setActiveChain: (chainId: number) => void
+  setActiveChainId: (chainId: number | null) => void
   setIsExpiring: (isExpiring: boolean) => void
-  setOAuthConfig: (config: OAuthConfig | null) => void
+  setOAuthConfig: (config: InternalOAuthConfig | null) => void
   clear: () => void
 }
 
@@ -53,7 +58,7 @@ export const createZeroDevWalletStore = () =>
           wallet: null,
           eoaAccount: null,
           session: null,
-          chainIds: [],
+          activeChainId: null,
           kernelAccounts: new Map(),
           kernelClients: new Map(),
           isExpiring: false,
@@ -78,13 +83,7 @@ export const createZeroDevWalletStore = () =>
 
           setSession: (session) => set({ session }),
 
-          setActiveChain: (chainId) => {
-            const { chainIds } = get()
-            // Move chainId to front, remove duplicates
-            set({
-              chainIds: [chainId, ...chainIds.filter((id) => id !== chainId)],
-            })
-          },
+          setActiveChainId: (chainId) => set({ activeChainId: chainId }),
 
           setIsExpiring: (isExpiring) => set({ isExpiring }),
 
@@ -97,7 +96,7 @@ export const createZeroDevWalletStore = () =>
               kernelAccounts: new Map(),
               kernelClients: new Map(),
               isExpiring: false,
-              chainIds: [],
+              activeChainId: null,
             }),
         }),
         {
@@ -105,7 +104,7 @@ export const createZeroDevWalletStore = () =>
           // Only persist session data, not clients or accounts
           partialize: (state) => ({
             session: state.session,
-            chainIds: state.chainIds,
+            activeChainId: state.activeChainId,
           }),
         },
       ),