kokio-sdk 0.0.2 → 0.0.3

package/package.json

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

package/src/config.ts

@@ -14,6 +14,7 @@ export class Kokio {
     viemWalletClient: WalletClient;
     turnkeyClient: TurnkeyClient;
     organizationId: string;
+    gasPolicyId: string;
 
     constants: ConstantsSubPackage;
 
@@ -29,6 +30,7 @@ export class Kokio {
         viemWalletClient: WalletClient,
         turnkeyClient: TurnkeyClient,
         organizationId: string,
+        gasPolicyId: string,
         smartAccountClient?: SmartAccountClient,
         deviceWalletAddress?: Address,
         eSIMWalletAddress?: Address
@@ -36,10 +38,11 @@ export class Kokio {
         this.viemWalletClient = viemWalletClient;
         this.turnkeyClient = turnkeyClient;
         this.organizationId = organizationId;
+        this.gasPolicyId = gasPolicyId;
 
         this.constants = new ConstantsSubPackage(this.viemWalletClient);
 
-        this.smartAccount = new SmartAccountSubPackage(this.viemWalletClient, this.turnkeyClient, this.organizationId);
+        this.smartAccount = new SmartAccountSubPackage(this.viemWalletClient, this.turnkeyClient, this.organizationId, 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/smartAccountClass.ts

@@ -9,11 +9,13 @@ export class SmartAccountSubPackage {
     client;
     turnkeyClient;
     organizationId; 
+    gasPolicyId;
     
-    constructor(client: WalletClient, turnkeyClient: TurnkeyClient, organiationId: string) {
+    constructor(client: WalletClient, turnkeyClient: TurnkeyClient, organiationId: string, gasPolicyId: string) {
         this.client = client;
         this.turnkeyClient = turnkeyClient;
         this.organizationId = organiationId;
+        this.gasPolicyId = gasPolicyId;
     }
 
     getSmartWallet (deviceUniqueIdentifier: string, deviceWalletOwnerKey: P256Key, salt: bigint, sender?: Address) {
@@ -21,6 +23,6 @@ export class SmartAccountSubPackage {
     }
 
     getSmartWalletClient (account: SmartContractAccount) {
-        return _getSmartWalletClient(this.client, account)
+        return _getSmartWalletClient(this.client, this.gasPolicyId, account)
     }
 }

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

@@ -26,6 +26,7 @@ import { _add0x, _concatUint8Arrays, _remove0x, _shouldRemoveLeadingZero } from
 import { P256Key, WebAuthnSignature } from "../../types.js";
 import { DeviceWallet, DeviceWalletFactory } from "../../abis/index.js";
 import { _signMessageWithTurnkey, _signTypedDataWithTurnkey } from "../services/turnkeyClient.js";
+import { alchemyGasManagerMiddleware } from "@account-kit/infra";
 
 const _encodeExecute = async (tx: AccountOp) => {
 
@@ -227,7 +228,15 @@ const _signUserOperationHash = async (hash: Hex, turnkeyClient: TurnkeyClient, o
 	return _signMessage(message, turnkeyClient, organiationId, signWith);
 }
 
-export const _getSmartWallet = async (client: WalletClient, turnkeyClient: TurnkeyClient, organiationId: string, deviceUniqueIdentifier: string, deviceWalletOwnerKey: P256Key, salt: bigint, sender?: Address): Promise<SmartContractAccount> => {
+export const _getSmartWallet = async (
+	client: WalletClient,
+	turnkeyClient: TurnkeyClient,
+	organiationId: string,
+	deviceUniqueIdentifier: string,
+	deviceWalletOwnerKey: P256Key,
+	salt: bigint,
+	sender?: Address
+): Promise<SmartContractAccount> => {
 
 	const chainID = await client.getChainId();
 	const rpcURL = client.transport.url;
@@ -244,14 +253,14 @@ export const _getSmartWallet = async (client: WalletClient, turnkeyClient: Turnk
 		chain: values.chain,
 
 		// The EntryPointDef that your account is compatible with
-		entryPoint: getEntryPoint(values.chain, {addressOverride: values.factoryAddresses.ENTRY_POINT}), 
+        entryPoint: getEntryPoint(values.chain, {version: "0.7.0"}), 
 
 		getAccountInitCode: async (): Promise<Hex> => await _getAccountInitCode(client, deviceUniqueIdentifier, deviceWalletOwnerKey, salt),
 
 		// getAccountInitCodeHash: async (): Promise<BytesLike> => await getInitCodeHash(client, deviceUniqueIdentifier, deviceWalletOwnerKey),
 		
 		// an invalid signature that doesn't cause your account to revert during validation
-		getDummySignature: async (): Promise<Hash> => "0xfffffffffffffffffffffffffffffff0000000000000000000000000000000007aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa1c", //from Alchemy docs
+		getDummySignature: async (): Promise<Hash> => "0x",
 		
 		// given a UO in the form of {target, data, value} should output the calldata for calling your contract's execution method
 		encodeExecute: async (uo): Promise<Hash> => _encodeExecute(uo),
@@ -272,7 +281,7 @@ export const _getSmartWallet = async (client: WalletClient, turnkeyClient: Turnk
 	});
 }
 
-export const _getSmartWalletClient = async (client: WalletClient, account: SmartContractAccount) => {
+export const _getSmartWalletClient = async (client: WalletClient, gasPolicyId: string, account: SmartContractAccount) => {
 
 	const chainID = await client.getChainId();
 	const rpcURL = client.transport.url;
@@ -283,5 +292,6 @@ export const _getSmartWalletClient = async (client: WalletClient, account: Smart
 		account: account,
 		chain: values.chain,
 		transport: http(values.rpcURL),
+		...alchemyGasManagerMiddleware(gasPolicyId)
 	});
 }

package/src/logic/constants.ts

@@ -44,7 +44,7 @@ export const sepoliaFactoryAddresses: Record<string, Address> = {
     LAZY_WALLET_REGISTRY: '0x8a1E53b903efcc6b252CE4bD3b255202318505Ef',
     REGISTRY: '0xCa447f5C75C57f6C59027304A5Fb5A09F0E005c9',
     ENTRY_POINT: '0x0000000071727De22E5E9d8BAf0edAc6f37da032',
-    SENDER_CREATOR: '0x9406Cc6185a346906296840746125a0E44976454',
+    SENDER_CREATOR: '0xefc2c1444ebcc4db75e7613d20c6a62ff67a167c',
     P256VERIFIER: '0xF04f3b3935aD461D17d4a8a78E7ea21d4a61AEb1'
 }
 
@@ -73,8 +73,8 @@ export const optimismSepoliaFactoryAddresses: Record<string, Address> = {
     ESIM_WALLET_FACTORY: '0xaD2Ba6248D0e6990a844C94ACE78F8775A68b631',
     LAZY_WALLET_REGISTRY: '0x29b98C32D83604664fd0742b9112b0825a74849F',
     REGISTRY: '0xaeB98F0f092fCb4a476ea3a5EB6A4B9E3D63A686',
-    ENTRY_POINT: '0x5FF137D4b0FDCD49DcA30c7CF57E578a026d2789',
-    SENDER_CREATOR: '0x7fc98430eaedbb6070b35b39d798725049088348',
+    ENTRY_POINT: '0x0000000071727De22E5E9d8BAf0edAc6f37da032',
+    SENDER_CREATOR: '0xefc2c1444ebcc4db75e7613d20c6a62ff67a167c',
     P256VERIFIER: '0x5349aEA97Faa6D5d6D78A8847068300a4eC9D39E'
 }