use shouldUpdateSessionKey in login flow

Author: joaquim-verges

apps/login/src/components/login-form.tsx

@@ -6,15 +6,17 @@ import { toast } from "sonner";
 import { getContract, sendAndConfirmTransaction } from "thirdweb";
 import type { Chain, ThirdwebClient } from "thirdweb";
 import { defineChain } from "thirdweb/chains";
-import { addSessionKey, isActiveSigner } from "thirdweb/extensions/erc4337";
+import {
+  addSessionKey,
+  shouldUpdateSessionKey,
+} from "thirdweb/extensions/erc4337";
 import {
   ConnectButton,
   ConnectEmbed,
   useActiveAccount,
   useActiveWallet,
   useSiweAuth,
 } from "thirdweb/react";
-import { isContractDeployed } from "thirdweb/utils";
 import { type Account, inAppWallet } from "thirdweb/wallets";
 import { createCode } from "../actions/create-code";
 import {
@@ -163,31 +165,28 @@ async function ensureSessionKey(options: {
     client: options.client,
   });
 
+  const shouldUpdateSessionkey = await shouldUpdateSessionKey({
+    account: options.account,
+    client: options.client,
+    chain: options.chain,
+    sessionKeySignerAddress: options.sessionKeySignerAddress,
+  });
+
   // check if already added
-  const accountDeployed = await isContractDeployed(accountContract);
-  if (accountDeployed) {
-    if (
-      await isActiveSigner({
-        contract: accountContract,
-        signer: options.sessionKeySignerAddress,
-      })
-    ) {
-      return {
-        success: true,
-        message: "Session key already added",
-        transaction: null,
-      };
-    }
+  if (!shouldUpdateSessionkey) {
+    return {
+      success: true,
+      message: "Session key already added",
+      transaction: null,
+    };
   }
-  // if not added, send tx to add the session key
-
+  // if not added or needs to be updated, send tx to add the session key
   const tx = await sendAndConfirmTransaction({
     account: options.account,
     transaction: addSessionKey({
       account: options.account,
       contract: accountContract,
       sessionKeyAddress: options.sessionKeySignerAddress,
-      // hard coded for now
       permissions: {
         approvedTargets:
           typeof options.permissions["contracts:write"] === "boolean"