feat: init smart sessions simplification

madjid80 · Oct 14, 2024 · 06fe448 · 06fe448
1 parent 6cdb06f
commit 06fe448
Show file tree

Hide file tree

Showing 7 changed files with 82 additions and 8 deletions.
diff --git a/src/module/index.ts b/src/module/index.ts
@@ -122,8 +122,8 @@ export {
   getDisableUserOpPoliciesAction,
   getEnableERC1271PoliciesAction,
   getDisableERC1271PoliciesAction,
-  getEnableActionPolicies,
-  getDisableActionPolicies,
+  getEnableActionPoliciesAction,
+  getDisableActionPoliciesAction,
   getPermissionId,
   getActionId,
   getSessionDigest,

diff --git a/src/module/smart-sessions/policies/spending-limits-policy/installation.ts b/src/module/smart-sessions/policies/spending-limits-policy/installation.ts
@@ -9,11 +9,11 @@ type Params = {
 
 export const getSpendingLimitsPolicy = (params: Params): Policy => {
   return {
+    policy: SPENDING_LIMITS_POLICY_ADDRESS,
     address: SPENDING_LIMITS_POLICY_ADDRESS,
     initData: encodeAbiParameters(
       [{ type: 'address[]' }, { type: 'uint256[]' }],
       [params.map(({ token }) => token), params.map(({ limit }) => limit)],
     ),
-    deInitData: '0x',
   }
 }
diff --git a/src/module/smart-sessions/policies/sudo-policy/installation.ts b/src/module/smart-sessions/policies/sudo-policy/installation.ts
@@ -3,8 +3,8 @@ import { Policy } from '../types'
 
 export const getSudoPolicy = (): Policy => {
   return {
+    policy: SUDO_POLICY_ADDRESS,
     address: SUDO_POLICY_ADDRESS,
     initData: '0x',
-    deInitData: '0x',
   }
 }
diff --git a/src/module/smart-sessions/policies/types.ts b/src/module/smart-sessions/policies/types.ts
@@ -1,7 +1,7 @@
 import { Hex, Address } from 'viem'
 
 export type Policy = {
+  policy: Address
   address: Address
   initData: Hex
-  deInitData: Hex
 }
diff --git a/src/module/smart-sessions/policies/universal-action-policy/installation.ts b/src/module/smart-sessions/policies/universal-action-policy/installation.ts
@@ -12,6 +12,7 @@ export const getUniversalActionPolicy = (
   }
 
   return {
+    policy: UNIVERSAL_ACTION_POLICY_ADDRESS,
     address: UNIVERSAL_ACTION_POLICY_ADDRESS,
     initData: encodeAbiParameters(abi, [
       {
@@ -22,6 +23,5 @@ export const getUniversalActionPolicy = (
         },
       },
     ]),
-    deInitData: '0x',
   }
 }
diff --git a/src/module/smart-sessions/types.ts b/src/module/smart-sessions/types.ts
@@ -4,7 +4,9 @@ import { Address, Hex } from 'viem'
 export type Session = {
   sessionValidator: Address
   sessionValidatorInitData: Hex
+  // todo: make this optional with default 0
   salt: Hex
+  // todo: make the below optional but require one of them to be defined
   userOpPolicies: PolicyData[]
   erc7739Policies: ERC7739Data
   actions: ActionData[]

diff --git a/src/module/smart-sessions/usage.ts b/src/module/smart-sessions/usage.ts
@@ -541,7 +541,7 @@ export const getDisableERC1271PoliciesAction = ({
   }
 }
 
-export const getEnableActionPolicies = ({
+export const getEnableActionPoliciesAction = ({
   permissionId,
   actionPolicies,
 }: {
@@ -563,7 +563,7 @@ export const getEnableActionPolicies = ({
   }
 }
 
-export const getDisableActionPolicies = ({
+export const getDisableActionPoliciesAction = ({
   permissionId,
   actionId,
   policies,
@@ -586,3 +586,75 @@ export const getDisableActionPolicies = ({
     data,
   }
 }
+
+// todo: make session an array
+export const getSessionDetails = async ({
+  session,
+  chainId,
+  mode,
+  account,
+  client,
+}: {
+  session: Session
+  chainId: bigint
+  mode: SmartSessionModeType
+  account: Account
+  client: PublicClient
+}) => {
+  const permissionId = getPermissionId({
+    session,
+  })
+
+  const sessionNonce = await getSessionNonce({
+    client,
+    account,
+    permissionId,
+  })
+
+  const sessionDigest = await getSessionDigest({
+    client,
+    account,
+    session,
+    mode,
+    permissionId,
+  })
+
+  const chainDigests = [
+    {
+      chainId: chainId,
+      sessionDigest,
+    },
+  ]
+
+  const chainSessions: ChainSession[] = [
+    {
+      chainId: chainId,
+      session: {
+        ...session,
+        account: account.address,
+        smartSession: SMART_SESSIONS_ADDRESS,
+        mode,
+        nonce: sessionNonce,
+      },
+    },
+  ]
+
+  const permissionEnableHash = hashChainSessions(chainSessions)
+
+  return {
+    permissionEnableHash,
+    mode,
+    permissionId,
+    signature: '0x' as Hex,
+    enableSessionData: {
+      enableSession: {
+        chainDigestIndex: 0,
+        hashesAndChainIds: chainDigests,
+        sessionToEnable: session,
+        permissionEnableSig: '0x' as Hex,
+      },
+      validator: session.sessionValidator,
+      accountType: account.type,
+    },
+  }
+}