@zerodev/wallet-react 0.0.1-alpha.4 → 0.0.1-alpha.6

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@zerodev/wallet-react",
-  "version": "0.0.1-alpha.4",
+  "version": "0.0.1-alpha.6",
   "description": "React hooks for ZeroDev Wallet SDK",
   "sideEffects": false,
   "main": "./dist/_cjs/index.js",
@@ -30,7 +30,7 @@
     "wagmi": "^3.0.0",
     "zustand": "^5.0.3",
     "ox": "^0.3.0",
-    "@zerodev/wallet-core": "0.0.1-alpha.5"
+    "@zerodev/wallet-core": "0.0.1-alpha.6"
   },
   "devDependencies": {
     "@types/react": "^19",

package/src/actions.ts

@@ -1,5 +1,10 @@
 import type { Config, Connector } from '@wagmi/core'
 import { connect as wagmiConnect } from '@wagmi/core/actions'
+import {
+  createIframeStamper,
+  exportPrivateKey as exportPrivateKeySdk,
+  exportWallet as exportWalletSdk,
+} from '@zerodev/wallet-core'
 import type { OAuthProvider } from './oauth.js'
 import {
   buildOAuthUrl,
@@ -362,10 +367,6 @@ export async function exportWallet(
 
   if (!wallet) throw new Error('Wallet not initialized')
 
-  const { exportWallet: exportWalletSdk, createIframeStamper } = await import(
-    '@zerodev/wallet-core'
-  )
-
   const iframeContainer = document.getElementById(parameters.iframeContainerId)
   if (!iframeContainer) {
     throw new Error('Iframe container not found')
@@ -400,3 +401,62 @@ export declare namespace exportWallet {
   type ReturnType = void
   type ErrorType = Error
 }
+
+/**
+ * Export private key
+ */
+export async function exportPrivateKey(
+  config: Config,
+  parameters: {
+    iframeContainerId: string
+    address?: string
+    keyFormat?: 'Hexadecimal' | 'Solana'
+    connector?: Connector
+  },
+): Promise<void> {
+  const connector = parameters.connector ?? getZeroDevConnector(config)
+
+  // @ts-expect-error - getStore is a custom method
+  const store = await connector.getStore()
+  const wallet = store.getState().wallet
+
+  if (!wallet) throw new Error('Wallet not initialized')
+
+  const iframeContainer = document.getElementById(parameters.iframeContainerId)
+  if (!iframeContainer) {
+    throw new Error('Iframe container not found')
+  }
+
+  const iframeStamper = await createIframeStamper({
+    iframeUrl: 'https://export.turnkey.com',
+    iframeContainer,
+    iframeElementId: 'export-private-key-iframe',
+  })
+
+  const publicKey = await iframeStamper.init()
+  const { exportBundle, organizationId } = await exportPrivateKeySdk({
+    wallet,
+    targetPublicKey: publicKey,
+    ...(parameters.address && { address: parameters.address }),
+  })
+
+  const success = await iframeStamper.injectKeyExportBundle(
+    exportBundle,
+    organizationId,
+    parameters.keyFormat ?? 'Hexadecimal',
+  )
+  if (success !== true) {
+    throw new Error('Failed to inject export bundle')
+  }
+}
+
+export declare namespace exportPrivateKey {
+  type Parameters = {
+    iframeContainerId: string
+    address?: string
+    keyFormat?: 'Hexadecimal' | 'Solana'
+    connector?: Connector
+  }
+  type ReturnType = void
+  type ErrorType = Error
+}

package/src/connector.ts

@@ -75,10 +75,6 @@ export function zeroDevWallet(
       store = createZeroDevWalletStore()
       store.getState().setWallet(wallet)
 
-      // Initialize chainIds
-      const chainIds = params.chains.map((c) => c.id)
-      store.setState({ chainIds })
-
       // Store OAuth config if provided
       if (params.oauthConfig) {
         store.getState().setOAuthConfig(params.oauthConfig)
@@ -132,10 +128,8 @@ export function zeroDevWallet(
         const state = store.getState()
 
         // Determine active chain
-        const activeChainId = chainId || state.chainIds[0]
-        if (!activeChainId) {
-          throw new Error('No chain configured')
-        }
+        const activeChainId =
+          chainId ?? state.activeChainId ?? params.chains[0].id
 
         // If reconnecting and already have kernel account, return immediately
         if (isReconnecting && state.kernelAccounts.has(activeChainId)) {
@@ -198,7 +192,7 @@ export function zeroDevWallet(
         }
 
         // Set as active chain
-        store.getState().setActiveChain(activeChainId)
+        store.getState().setActiveChainId(activeChainId)
 
         // Get fresh state after updates
         const freshState = store.getState()
@@ -229,7 +223,7 @@ export function zeroDevWallet(
 
       async getAccounts() {
         if (!store) return []
-        const { eoaAccount, kernelAccounts, chainIds } = store.getState()
+        const { eoaAccount, kernelAccounts, activeChainId } = store.getState()
 
         // Return EOA address if we have it (EIP-7702: EOA address = kernel address)
         if (eoaAccount) {
@@ -237,15 +231,15 @@ export function zeroDevWallet(
         }
 
         // Fallback: check kernel accounts
-        const activeAccount = chainIds[0]
-          ? kernelAccounts.get(chainIds[0])
+        const activeAccount = activeChainId
+          ? kernelAccounts.get(activeChainId)
           : null
         return activeAccount ? [activeAccount.address] : []
       },
 
       async getChainId() {
         if (!store) return params.chains[0].id
-        return store.getState().chainIds[0] || params.chains[0].id
+        return store.getState().activeChainId ?? params.chains[0].id
       },
 
       async getProvider() {
@@ -264,7 +258,7 @@ export function zeroDevWallet(
         }
 
         // Update active chain
-        store.getState().setActiveChain(chainId)
+        store.getState().setActiveChainId(chainId)
 
         // Create kernel account for new chain if doesn't exist
         if (!state.kernelAccounts.has(chainId)) {

package/src/hooks/useExportPrivateKey.ts

@@ -0,0 +1,57 @@
+'use client'
+
+import {
+  type UseMutationOptions,
+  type UseMutationResult,
+  useMutation,
+} from '@tanstack/react-query'
+import { type Config, type ResolvedRegister, useConfig } from 'wagmi'
+import { exportPrivateKey } from '../actions.js'
+
+type ConfigParameter<config extends Config = Config> = {
+  config?: Config | config | undefined
+}
+
+/**
+ * Hook to export private key
+ */
+export function useExportPrivateKey<
+  config extends Config = ResolvedRegister['config'],
+  context = unknown,
+>(
+  parameters: useExportPrivateKey.Parameters<config, context> = {},
+): useExportPrivateKey.ReturnType<context> {
+  const { mutation } = parameters
+  const config = useConfig(parameters)
+
+  return useMutation({
+    ...mutation,
+    async mutationFn(variables: exportPrivateKey.Parameters) {
+      return exportPrivateKey(config, variables)
+    },
+    mutationKey: ['exportPrivateKey'],
+  })
+}
+
+export declare namespace useExportPrivateKey {
+  type Parameters<
+    config extends Config = Config,
+    context = unknown,
+  > = ConfigParameter<config> & {
+    mutation?:
+      | UseMutationOptions<
+          exportPrivateKey.ReturnType,
+          exportPrivateKey.ErrorType,
+          exportPrivateKey.Parameters,
+          context
+        >
+      | undefined
+  }
+
+  type ReturnType<context = unknown> = UseMutationResult<
+    exportPrivateKey.ReturnType,
+    exportPrivateKey.ErrorType,
+    exportPrivateKey.Parameters,
+    context
+  >
+}

package/src/index.ts

@@ -1,6 +1,7 @@
 export type { ZeroDevWalletConnectorParams } from './connector.js'
 export { zeroDevWallet } from './connector.js'
 export { useAuthenticateOAuth } from './hooks/useAuthenticateOAuth.js'
+export { useExportPrivateKey } from './hooks/useExportPrivateKey.js'
 export { useExportWallet } from './hooks/useExportWallet.js'
 export { useLoginPasskey } from './hooks/useLoginPasskey.js'
 export { useRefreshSession } from './hooks/useRefreshSession.js'

package/src/provider.ts

@@ -115,7 +115,10 @@ export function createProvider({
 
     async request({ method, params }: { method: string; params?: any[] }) {
       const state = store.getState()
-      const activeChainId = state.chainIds[0]
+      const activeChainId = state.activeChainId
+      if (!activeChainId) {
+        throw new Error('No active chain')
+      }
 
       switch (method) {
         case 'eth_accounts': {
@@ -219,7 +222,7 @@ export function createProvider({
           const chainId_number = parseInt(chainId, 16)
 
           // Update active chain
-          store.getState().setActiveChain(chainId_number)
+          store.getState().setActiveChainId(chainId_number)
 
           // Emit chainChanged event
           emitter.emit('chainChanged', chainId)

package/src/store.ts

@@ -19,7 +19,7 @@ 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>
 
@@ -38,7 +38,7 @@ 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
   clear: () => void
@@ -53,7 +53,7 @@ export const createZeroDevWalletStore = () =>
           wallet: null,
           eoaAccount: null,
           session: null,
-          chainIds: [],
+          activeChainId: null,
           kernelAccounts: new Map(),
           kernelClients: new Map(),
           isExpiring: false,
@@ -78,13 +78,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 +91,7 @@ export const createZeroDevWalletStore = () =>
               kernelAccounts: new Map(),
               kernelClients: new Map(),
               isExpiring: false,
-              chainIds: [],
+              activeChainId: null,
             }),
         }),
         {
@@ -105,7 +99,7 @@ export const createZeroDevWalletStore = () =>
           // Only persist session data, not clients or accounts
           partialize: (state) => ({
             session: state.session,
-            chainIds: state.chainIds,
+            activeChainId: state.activeChainId,
           }),
         },
       ),