more wip

Author: pcarranzav

packages/hypergraph/src/identity/smart-account.ts

@@ -74,6 +74,14 @@ export const GEOGENESIS = {
   },
 };
 
+type Action = {
+  actionTarget: Address;
+  actionTargetSelector: Hex;
+  actionPolicies: { policy: Address; address: Address; initData: Hex }[];
+};
+
+// Gets the legacy Geo smart account wallet client. If the smart account returned
+// by this function is deployed, it means it might need to be updated to have the 7579 module installed
 export const getLegacySmartAccountWalletClient = async (
   walletClient: WalletClient,
   chain: Chain = GEOGENESIS,
@@ -204,6 +212,24 @@ export const isSmartAccountDeployed = async (smartAccountClient: SmartAccountCli
   return smartAccountClient.account.isDeployed();
 };
 
+export const legacySmartAccountNeedsUpdate = async (
+  smartAccountClient: SmartAccountClient,
+  chain: Chain,
+  rpcUrl: string,
+): Promise<boolean> => {
+  if (!smartAccountClient.account) {
+    throw new Error('Invalid smart account');
+  }
+  // We assume the smart account is deployed, so we just need to check if it has the 7579 module and smart sesions validator installed
+  // TODO: call the isModuleInstalled function from the Safe7579 ABI on the
+  // smart account, checking if the smart sessions validator is installed. This would fail
+  // if the smart account doesn't have the 7579 module installed.
+  return false;
+};
+
+// Legacy Geo smart accounts (i.e. the ones that don't have the 7579 module installed)
+// need to be updated to have the 7579 module installed
+// with the ownable and smart sessions validators.
 export const updateLegacySmartAccount = async (smartAccountClient: SmartAccountClient) => {
   if (!smartAccountClient.account?.address) {
     throw new Error('Invalid smart account');
@@ -281,12 +307,11 @@ export const updateLegacySmartAccount = async (smartAccountClient: SmartAccountC
   return receipt;
 };
 
-type Action = {
-  actionTarget: Address;
-  actionTargetSelector: Hex;
-  actionPolicies: { policy: Address; address: Address; initData: Hex }[];
-};
-
+// This is the function that the Connect app uses to create a smart session and
+// enable it on the smart account.
+// It will prompt the user to sign the message to enable the session, and then
+// execute the transaction to enable the session.
+// It will return the permissionId.
 export const createSmartSession = async (
   walletClient: WalletClient,
   smartAccountClient: SmartAccountClient,
@@ -457,3 +482,70 @@ export const createSmartSession = async (
   }
   return getPermissionId({ session });
 };
+
+// This is the function that we use on the end user app to create a smart session client that can send transactions to the smart account.
+// The session must have previously been created by the createSmartSession function.
+export const getSmartSessionClient = async (
+  smartAccountClient: SmartAccountClient,
+  sessionPrivateKey: `0x${string}`,
+  permissionId: Hex,
+  chain: Chain,
+  rpcUrl: string,
+) => {
+  if (!smartAccountClient.account) {
+    throw new Error('Invalid smart account');
+  }
+  const sessionDetails = {
+    mode: SmartSessionMode.USE,
+    permissionId,
+    signature: getOwnableValidatorMockSignature({
+      threshold: 1,
+    }),
+  };
+  const smartSessions = getSmartSessionsValidator({});
+  const publicClient = createPublicClient({
+    transport: http(rpcUrl),
+    chain,
+  });
+  const sessionKeyAccount = privateKeyToAccount(sessionPrivateKey);
+  return {
+    sendTransaction: async <const calls extends readonly unknown[]>({ calls }: { calls: calls }) => {
+      if (!smartAccountClient.account) {
+        throw new Error('Invalid smart account');
+      }
+      const account = getAccount({
+        address: smartAccountClient.account.address,
+        type: 'safe',
+      });
+      const nonce = await getAccountNonce(publicClient, {
+        address: smartAccountClient.account.address,
+        entryPointAddress: entryPoint07Address,
+        key: encodeValidatorNonce({
+          account,
+          validator: smartSessions,
+        }),
+      });
+      const userOperation = await smartAccountClient.prepareUserOperation({
+        account: smartAccountClient.account,
+        calls,
+        nonce,
+        signature: encodeSmartSessionSignature(sessionDetails),
+      });
+
+      const userOpHashToSign = getUserOperationHash({
+        chainId: chain.id,
+        entryPointAddress: entryPoint07Address,
+        entryPointVersion: '0.7',
+        userOperation,
+      });
+
+      sessionDetails.signature = await sessionKeyAccount.signMessage({
+        message: { raw: userOpHashToSign },
+      });
+
+      userOperation.signature = encodeSmartSessionSignature(sessionDetails);
+
+      return smartAccountClient.sendUserOperation(userOperation as UserOperation);
+    },
+  };
+};