feat: add multichain enable functionality

madjid80 · Oct 16, 2024 · fc47e6e · fc47e6e
1 parent 24f3625
commit fc47e6e
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 6 deletions.
diff --git a/src/module/smart-sessions/usage.ts b/src/module/smart-sessions/usage.ts
@@ -589,13 +589,15 @@ export const getDisableActionPoliciesAction = ({
 export const getEnableSessionDetails = async ({
   sessions,
   sessionIndex,
+  enableMode,
   account,
-  client,
+  clients,
 }: {
   sessions: Session[]
   sessionIndex?: number
+  enableMode?: SmartSessionModeType
   account: Account
-  client: PublicClient
+  clients: PublicClient[]
 }) => {
   const chainDigests = []
   const chainSessions: ChainSession[] = []
@@ -604,6 +606,14 @@ export const getEnableSessionDetails = async ({
       session,
     })
 
+    const client = clients.find(
+      (c) => BigInt(c.chain?.id ?? 0) === session.chainId,
+    )
+
+    if (!client) {
+      throw new Error(`Client not found for chainId ${session.chainId}`)
+    }
+
     const sessionNonce = await getSessionNonce({
       client,
       account,
@@ -614,7 +624,7 @@ export const getEnableSessionDetails = async ({
       client,
       account,
       session,
-      mode: SmartSessionMode.ENABLE,
+      mode: enableMode || SmartSessionMode.ENABLE,
       permissionId,
     })
 
@@ -629,7 +639,7 @@ export const getEnableSessionDetails = async ({
         ...session,
         account: account.address,
         smartSession: SMART_SESSIONS_ADDRESS,
-        mode: 1,
+        mode: Number(enableMode) || Number(SmartSessionMode.ENABLE),
         nonce: sessionNonce,
       },
     })
@@ -644,7 +654,7 @@ export const getEnableSessionDetails = async ({
 
   return {
     permissionEnableHash,
-    mode: SmartSessionMode.ENABLE,
+    mode: enableMode || SmartSessionMode.ENABLE,
     permissionId,
     signature: '0x' as Hex,
     enableSessionData: {

diff --git a/test/e2e/modules/smartSessionsValidator.ts b/test/e2e/modules/smartSessionsValidator.ts
@@ -151,7 +151,7 @@ export const testSmartSessionsValidator = async ({
     const sessionDetails = await getEnableSessionDetails({
       sessions: [session],
       account,
-      client: publicClient,
+      clients: [publicClient],
     })
 
     sessionDetails.enableSessionData.enableSession.permissionEnableSig =