kokio-sdk 0.0.5 → 0.0.6

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "kokio-sdk",
-  "version": "0.0.5",
+  "version": "0.0.6",
   "description": "",
   "type": "module",
   "main": "./dist/esm/config.js",

package/src/config.ts

@@ -16,6 +16,7 @@ export class Kokio {
     credentialId: string;
     rpId: string;
     organizationId: string;
+    pimlicoAPIKey: string;
     gasPolicyId: string;
 
     constants: ConstantsSubPackage;
@@ -34,6 +35,7 @@ export class Kokio {
         credentialId: string,
         rpId: string,
         organizationId: string,
+        pimlicoAPIKey: string,
         gasPolicyId: string,
         smartAccountClient?: SmartAccountClient,
         deviceWalletAddress?: Address,
@@ -44,11 +46,12 @@ export class Kokio {
         this.credentialId = credentialId;
         this.rpId = rpId;
         this.organizationId = organizationId;
+        this.pimlicoAPIKey = pimlicoAPIKey;
         this.gasPolicyId = gasPolicyId;
 
-        this.constants = new ConstantsSubPackage(this.viemWalletClient);
+        this.constants = new ConstantsSubPackage(this.viemWalletClient, this.pimlicoAPIKey);
 
-        this.smartAccount = new SmartAccountSubPackage(this.viemWalletClient, this.turnkeyClient, this.credentialId, this.rpId, this.organizationId, this.gasPolicyId);
+        this.smartAccount = new SmartAccountSubPackage(this.viemWalletClient, this.turnkeyClient, this.credentialId, this.rpId, this.organizationId, this.pimlicoAPIKey, this.gasPolicyId);
         this.deviceWalletFactory = smartAccountClient? new DeviceWalletFactorySubPackage(viemWalletClient, smartAccountClient): undefined;
         this.eSIMWalletFactory = smartAccountClient? new ESIMWalletFactorySubPackage(viemWalletClient, smartAccountClient): undefined;
         this.lazyWalletRegistry = smartAccountClient? new LazyWalletRegistrySubPackage(smartAccountClient): undefined;

package/src/interface/constantsClass.ts

@@ -21,11 +21,11 @@ export class ConstantsSubPackage {
             typeof arbitrumSepoliaFactoryAddresses;
     customErrors!: typeof customErrors;
 
-    constructor(client: WalletClient) {
+    constructor(client: WalletClient, pimlicoAPIKey: string) {
         (async () => {
           const chainID = await _extractChainID(client);
           const rpcURL = client.transport.url;
-          const values = _getChainSpecificConstants(chainID, rpcURL);
+          const values = _getChainSpecificConstants(chainID, rpcURL, pimlicoAPIKey);
     
           this.factoryAddresses = values?.factoryAddresses;
           this.customErrors = values?.customErrors;

package/src/interface/smartAccountClass.ts

@@ -10,15 +10,17 @@ export class SmartAccountSubPackage {
     turnkeyClient;
     credentialId;
     rpId;
-    organizationId; 
+    organizationId;
+    pimlicoAPIKey;
     gasPolicyId;
     
-    constructor(client: WalletClient, turnkeyClient: TurnkeyClient, credentialId: string, rpId: string, organiationId: string, gasPolicyId: string) {
+    constructor(client: WalletClient, turnkeyClient: TurnkeyClient, credentialId: string, rpId: string, organiationId: string, pimlicoAPIKey: string, gasPolicyId: string) {
         this.client = client;
         this.turnkeyClient = turnkeyClient;
         this.credentialId = credentialId;
         this.rpId = rpId;
         this.organizationId = organiationId;
+        this.pimlicoAPIKey = pimlicoAPIKey;
         this.gasPolicyId = gasPolicyId;
     }
 
@@ -27,6 +29,6 @@ export class SmartAccountSubPackage {
     }
 
     getSmartWalletClient (account: SmartContractAccount) {
-        return _getSmartWalletClient(this.client, this.gasPolicyId, account);
+        return _getSmartWalletClient(this.client, this.pimlicoAPIKey, this.gasPolicyId, account);
     }
 }

package/src/logic/account-kit/createSmartAccount.ts

@@ -266,17 +266,35 @@ export const _getSmartWallet = async (
 	});
 }
 
-export const _getSmartWalletClient = async (client: WalletClient, gasPolicyId: string, account: SmartContractAccount) => {
+export const _getSmartWalletClient = async (client: WalletClient, pimlicoAPIKey: string, gasPolicyId: string, account: SmartContractAccount) => {
 
 	const chainID = await client.getChainId();
 	const rpcURL = client.transport.url;
-	const values = _getChainSpecificConstants(chainID, rpcURL);
+	const values = _getChainSpecificConstants(chainID, rpcURL, pimlicoAPIKey);
+
+	const bundlerMethods = [
+			"eth_sendUserOperation",
+			"eth_estimateUserOperationGas",
+			"eth_getUserOperationReceipt",
+			"eth_getUserOperationByHash",
+			"eth_supportedEntryPoints",
+	];
 
+	// Uses Pimlico paymaster by default for above defined bundler methods and Alchemy as fallback
 	return createSmartAccountClient({
 		// created above
 		account: account,
 		chain: values.chain,
-		transport: http(values.rpcURL),
+		transport: split({
+			overrides: [
+				{
+					methods: bundlerMethods,
+					transport: http(`${values.pimlicoRpcURL}`),
+				},
+			],
+			fallback: http(values.rpcURL),
+		}),
+		// transport: http(values.rpcURL),
 		...alchemyGasManagerMiddleware(gasPolicyId)
 	});
 }

package/src/logic/constants.ts

@@ -36,7 +36,8 @@ export interface chainSpecifcConstants {
         typeof optimismSepolia |
         typeof arbitrum |
         typeof arbitrumSepolia
-    rpcURL: string; 
+    rpcURL: string;
+    pimlicoRpcURL: string;
     customErrors: typeof customErrors;
 }
 
@@ -122,7 +123,8 @@ export const _getChainSpecificConstants = (
         CHAIN_ID.OPTIMISM_SEPOLIA |
         CHAIN_ID.ARBITRUM_ONE |
         CHAIN_ID.ARBITRUM_SEPOLIA,
-    rpcURL: string
+    rpcURL: string,
+    pimlicoAPIKey?: string
     ): chainSpecifcConstants => {
 
     if (chainID == CHAIN_ID.SEPOLIA) {
@@ -130,6 +132,7 @@ export const _getChainSpecificConstants = (
             factoryAddresses: sepoliaFactoryAddresses,
             chain: sepolia,
             rpcURL: rpcURL,
+            pimlicoRpcURL: pimlicoAPIKey ? `https://api.pimlico.io/v2/${chainID}/rpc?apikey=${pimlicoAPIKey}` : "",
             customErrors: customErrors
         }
     }
@@ -138,6 +141,7 @@ export const _getChainSpecificConstants = (
             factoryAddresses: mainnetFactoryAddresses,
             chain: mainnet,
             rpcURL: rpcURL,
+            pimlicoRpcURL: pimlicoAPIKey ? `https://api.pimlico.io/v2/${chainID}/rpc?apikey=${pimlicoAPIKey}` : "",
             customErrors: customErrors
         }
     }
@@ -146,6 +150,7 @@ export const _getChainSpecificConstants = (
             factoryAddresses: optimismMainnetFactoryAddresses,
             chain: optimism,
             rpcURL: rpcURL,
+            pimlicoRpcURL: pimlicoAPIKey ? `https://api.pimlico.io/v2/${chainID}/rpc?apikey=${pimlicoAPIKey}` : "",
             customErrors: customErrors
         }
     }
@@ -154,6 +159,7 @@ export const _getChainSpecificConstants = (
             factoryAddresses: optimismSepoliaFactoryAddresses,
             chain: optimismSepolia,
             rpcURL: rpcURL,
+            pimlicoRpcURL: pimlicoAPIKey ? `https://api.pimlico.io/v2/${chainID}/rpc?apikey=${pimlicoAPIKey}` : "",
             customErrors: customErrors
         }
     }
@@ -162,6 +168,7 @@ export const _getChainSpecificConstants = (
             factoryAddresses: arbitrumOneFactoryAddresses,
             chain: arbitrum,
             rpcURL: rpcURL,
+            pimlicoRpcURL: pimlicoAPIKey ? `https://api.pimlico.io/v2/${chainID}/rpc?apikey=${pimlicoAPIKey}` : "",
             customErrors: customErrors
         }
     }
@@ -170,6 +177,7 @@ export const _getChainSpecificConstants = (
             factoryAddresses: arbitrumSepoliaFactoryAddresses,
             chain: arbitrumSepolia,
             rpcURL: rpcURL,
+            pimlicoRpcURL: pimlicoAPIKey ? `https://api.pimlico.io/v2/${chainID}/rpc?apikey=${pimlicoAPIKey}` : "",
             customErrors: customErrors
         }
     }