fixed accountTemplate type

etherspot · Oct 25, 2023 · 2636226 · 2636226
1 parent f38974e
commit 2636226
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 4 deletions.
diff --git a/src/components/EtherspotTransactionKit.tsx b/src/components/EtherspotTransactionKit.tsx
@@ -1,15 +1,18 @@
 import React from 'react';
 import { WalletProviderLike, Factory } from '@etherspot/prime-sdk';
-import EtherspotContextProvider from '../providers/EtherspotContextProvider';
+
+// types
+import { AccountTemplate } from '../types/EtherspotTransactionKit';
 
 // providers
 import EtherspotTransactionKitContextProvider from '../providers/EtherspotTransactionKitContextProvider';
 import ProviderWalletContextProvider from '../providers/ProviderWalletContextProvider';
+import EtherspotContextProvider from '../providers/EtherspotContextProvider';
 
 interface EtherspotTransactionKitProps extends React.PropsWithChildren {
   provider: WalletProviderLike;
   chainId?: number;
-  accountTemplate?: Factory;
+  accountTemplate?: AccountTemplate;
 }
 
 const EtherspotTransactionKit = ({

diff --git a/src/providers/EtherspotContextProvider.tsx b/src/providers/EtherspotContextProvider.tsx
@@ -15,6 +15,9 @@ import React, {
 // contexts
 import EtherspotContext from '../contexts/EtherspotContext';
 
+// types
+import { AccountTemplate } from '../types/EtherspotTransactionKit';
+
 let sdkPerChain: { [chainId: number]: PrimeSdk } = {};
 
 const EtherspotContextProvider = ({
@@ -26,7 +29,7 @@ const EtherspotContextProvider = ({
   children: ReactNode;
   provider: WalletProviderLike;
   chainId: number;
-  accountTemplate: Factory;
+  accountTemplate: AccountTemplate;
 }) => {
   const context = useContext(EtherspotContext);
 
@@ -49,7 +52,7 @@ const EtherspotContextProvider = ({
     const sdkForChain = new PrimeSdk(provider, {
       chainId: sdkChainId,
       projectKey: '__ETHERSPOT_PROJECT_KEY__' || undefined,
-      factoryWallet: accountTemplate,
+      factoryWallet: accountTemplate as Factory,
     });
 
     sdkPerChain = {

diff --git a/src/types/EtherspotTransactionKit.ts b/src/types/EtherspotTransactionKit.ts
@@ -121,3 +121,5 @@ interface UserOp {
   paymasterAndData: EtherspotPromiseOrValue<BytesLike>;
   signature: EtherspotPromiseOrValue<BytesLike>;
 }
+
+export type AccountTemplate = 'etherspot' | 'zeroDev' | 'simpleAccount';