[SDK]: move autoConnect function into it's own standalone function (#5889)

---
title: "[SDK/Dashboard/Portal] Feature/Fix: Concise title for the changes"
---

https://linear.app/thirdweb/issue/TOOL-2697/add-js-autoconnect

## Notes for the reviewer
Anything important to call out? Be sure to also clarify these in your comments.

## How to test
Unit tests, playground, etc.

<!-- start pr-codex -->

---

## PR-Codex overview
This PR focuses on refactoring the auto-connect functionality for wallets in the `thirdweb` library, enhancing the separation of concerns and improving the structure of the code. It introduces a standalone `autoConnect` function and updates various components to utilize it.

### Detailed summary
- Introduced a standalone `autoConnect` function in `packages/thirdweb/src/wallets/connection/autoConnect.ts`.
- Updated `useAutoConnectCore` to utilize `autoConnectCore`.
- Refactored types and imports for `AutoConnectProps` to point to the new structure.
- Modified tests to accommodate the new `autoConnect` implementation.
- Changed wallet connection logic to improve clarity and maintainability.
- Updated multiple components to use the new wallet connection logic.

> ✨ Ask PR-Codex anything about this PR by commenting with `/codex {your question}`

<!-- end pr-codex -->

Author: ElasticBottle

.changeset/green-rockets-lie.md

@@ -0,0 +1,17 @@
+---
+"thirdweb": minor
+---
+
+Exposes autoConnect as a standalone function for use outside of react.
+
+```tsx
+import { autoConnect } from "thirdweb/wallets";
+ 
+const autoConnected = await autoConnect({
+  client,
+  onConnect: (wallet) => {
+    console.log("wallet", wallet); /// wallet that is have been auto connected.
+  },
+});
+console.log('isAutoConnected', isAutoConnected) // true or false
+```

packages/thirdweb/src/exports/react.native.ts

@@ -86,7 +86,7 @@ export {
 
 // Components
 export { AutoConnect } from "../react/native/ui/AutoConnect/AutoConnect.js";
-export type { AutoConnectProps } from "../react/core/hooks/connection/types.js";
+export type { AutoConnectProps } from "../wallets/connection/types.js";
 
 export { TransactionButton } from "../react/native/ui/transaction/TransactionButton.js";
 export type { TransactionButtonProps } from "../react/core/hooks/transaction/transaction-button-utils.js";

packages/thirdweb/src/exports/react.ts

@@ -121,7 +121,7 @@ export {
 } from "../react/core/hooks/pay/usePostOnrampQuote.js";
 
 export { AutoConnect } from "../react/web/ui/AutoConnect/AutoConnect.js";
-export type { AutoConnectProps } from "../react/core/hooks/connection/types.js";
+export type { AutoConnectProps } from "../wallets/connection/types.js";
 
 // auth
 export type { SiweAuthOptions } from "../react/core/hooks/auth/useSiweAuth.js";

packages/thirdweb/src/exports/wallets.ts

@@ -161,4 +161,6 @@ export { injectedProvider } from "../wallets/injected/mipdStore.js";
 
 export type { ConnectionManager } from "../wallets/manager/index.js";
 
+export type { AutoConnectProps } from "../wallets/connection/types.js";
+export { autoConnect } from "../wallets/connection/autoConnect.js";
 export { deploySmartAccount } from "../wallets/smart/lib/signing.js";

packages/thirdweb/src/react/core/hooks/wallets/useAutoConnect.ts

@@ -1,26 +1,14 @@
 "use client";
 
 import { useQuery } from "@tanstack/react-query";
-import type { Chain } from "../../../../chains/types.js";
-import type { ThirdwebClient } from "../../../../client/client.js";
 import type { AsyncStorage } from "../../../../utils/storage/AsyncStorage.js";
-import { isEcosystemWallet } from "../../../../wallets/ecosystem/is-ecosystem-wallet.js";
-import { ClientScopedStorage } from "../../../../wallets/in-app/core/authentication/client-scoped-storage.js";
-import type { AuthStoredTokenWithCookieReturnType } from "../../../../wallets/in-app/core/authentication/types.js";
-import { getUrlToken } from "../../../../wallets/in-app/web/lib/get-url-token.js";
+import { autoConnectCore } from "../../../../wallets/connection/autoConnectCore.js";
+import type { AutoConnectProps } from "../../../../wallets/connection/types.js";
 import type { Wallet } from "../../../../wallets/interfaces/wallet.js";
-import {
-  getLastConnectedChain,
-  getStoredActiveWalletId,
-  getStoredConnectedWalletIds,
-} from "../../../../wallets/manager/index.js";
 import type { WalletId } from "../../../../wallets/wallet-types.js";
 import { useConnectionManagerCtx } from "../../providers/connection-manager.js";
 import { setLastAuthProvider } from "../../utils/storage.js";
-import { timeoutPromise } from "../../utils/timeoutPromise.js";
-import type { AutoConnectProps } from "../connection/types.js";
 import { useConnect } from "./useConnect.js";
-import { useSetActiveWalletConnectionStatus } from "./useSetActiveWalletConnectionStatus.js";
 
 export function useAutoConnectCore(
   storage: AsyncStorage,
@@ -29,161 +17,27 @@ export function useAutoConnectCore(
   getInstalledWallets?: () => Wallet[],
 ) {
   const manager = useConnectionManagerCtx("useAutoConnect");
-  const setConnectionStatus = useSetActiveWalletConnectionStatus();
   const { connect } = useConnect({
     client: props.client,
     accountAbstraction: props.accountAbstraction,
   });
-  const { isAutoConnecting } = manager;
-  const { wallets, onConnect } = props;
-  const timeout = props.timeout ?? 15000;
-  // get the supported wallets from thirdweb provider
-  // check the storage for last connected wallets and connect them all
-  // check the storage for last active wallet and set it as active
-  const autoConnect = async (): Promise<boolean> => {
-    let autoConnected = false;
-    isAutoConnecting.setValue(true);
-    let [lastConnectedWalletIds, lastActiveWalletId] = await Promise.all([
-      getStoredConnectedWalletIds(storage),
-      getStoredActiveWalletId(storage),
-    ]);
-
-    const { authResult, walletId, authProvider, authCookie } = getUrlToken();
-    const wallet = wallets.find((w) => w.id === walletId);
-
-    // If an auth cookie is found and this site supports the wallet, we'll set the auth cookie in the client storage
-    if (authCookie && wallet) {
-      const clientStorage = new ClientScopedStorage({
-        storage,
-        clientId: props.client.clientId,
-        ecosystem: isEcosystemWallet(wallet)
-          ? {
-              id: wallet.id,
-              partnerId: wallet.getConfig()?.partnerId,
-            }
-          : undefined,
-      });
-      await clientStorage.saveAuthCookie(authCookie);
-    }
-
-    if (walletId) {
-      lastActiveWalletId = walletId;
-      lastConnectedWalletIds = lastConnectedWalletIds?.includes(walletId)
-        ? lastConnectedWalletIds
-        : [walletId, ...(lastConnectedWalletIds || [])];
-    }
-    if (authProvider) {
-      await setLastAuthProvider(authProvider, storage);
-    }
-
-    // if no wallets were last connected or we didn't receive an auth token
-    if (!lastConnectedWalletIds) {
-      return autoConnected;
-    }
-
-    // this flow can actually be used for a first connection in the case of a redirect
-    // in that case, we default to the passed chain to connect to
-    const lastConnectedChain =
-      (await getLastConnectedChain(storage)) || props.chain;
-
-    const availableWallets = [...wallets, ...(getInstalledWallets?.() ?? [])];
-    const activeWallet =
-      lastActiveWalletId &&
-      (availableWallets.find((w) => w.id === lastActiveWalletId) ||
-        createWalletFn(lastActiveWalletId));
-
-    if (activeWallet) {
-      try {
-        setConnectionStatus("connecting"); // only set connecting status if we are connecting the last active EOA
-        await timeoutPromise(
-          handleWalletConnection({
-            wallet: activeWallet,
-            client: props.client,
-            lastConnectedChain,
-            authResult,
-          }),
-          {
-            ms: timeout,
-            message: `AutoConnect timeout: ${timeout}ms limit exceeded.`,
-          },
-        ).catch((err) => {
-          console.warn(err.message);
-          if (props.onTimeout) {
-            props.onTimeout();
-          }
-        });
-
-        // connected wallet could be activeWallet or smart wallet
-        const connectedWallet = await connect(activeWallet);
-
-        if (connectedWallet) {
-          if (onConnect) {
-            try {
-              onConnect(connectedWallet);
-              autoConnected = true;
-            } catch {
-              // ignore
-            }
-          }
-        } else {
-          setConnectionStatus("disconnected");
-        }
-      } catch (e) {
-        if (e instanceof Error) {
-          console.warn("Error auto connecting wallet:", e.message);
-        }
-        setConnectionStatus("disconnected");
-      }
-    } else {
-      setConnectionStatus("disconnected");
-    }
-
-    // then connect wallets that were last connected but were not set as active
-    const otherWallets = availableWallets.filter(
-      (w) =>
-        w.id !== lastActiveWalletId && lastConnectedWalletIds.includes(w.id),
-    );
-
-    for (const wallet of otherWallets) {
-      try {
-        await handleWalletConnection({
-          wallet,
-          client: props.client,
-          lastConnectedChain,
-          authResult,
-        });
-        manager.addConnectedWallet(wallet);
-      } catch {
-        // no-op
-      }
-    }
-    isAutoConnecting.setValue(false);
-    return autoConnected; // useQuery needs a return value
-  };
 
   // trigger the auto connect on first mount only
   const query = useQuery({
     queryKey: ["autoConnect", props.client.clientId],
-    queryFn: autoConnect,
+    queryFn: () =>
+      autoConnectCore({
+        createWalletFn,
+        manager,
+        props,
+        storage,
+        connectOverride: connect,
+        getInstalledWallets,
+        setLastAuthProvider,
+      }),
     refetchOnMount: false,
     refetchOnWindowFocus: false,
   });
 
   return query;
 }
-
-/**
- * @internal
- */
-export async function handleWalletConnection(props: {
-  wallet: Wallet;
-  client: ThirdwebClient;
-  authResult: AuthStoredTokenWithCookieReturnType | undefined;
-  lastConnectedChain: Chain | undefined;
-}) {
-  return props.wallet.autoConnect({
-    client: props.client,
-    chain: props.lastConnectedChain,
-    authResult: props.authResult,
-  });
-}

packages/thirdweb/src/react/core/hooks/wallets/useAutoConnectCore.test.tsx

@@ -6,15 +6,11 @@ import { TEST_CLIENT } from "~test/test-clients.js";
 import { TEST_ACCOUNT_A } from "~test/test-wallets.js";
 import { createWalletAdapter } from "../../../../adapters/wallet-adapter.js";
 import { ethereum } from "../../../../chains/chain-definitions/ethereum.js";
-import { isAddress } from "../../../../utils/address.js";
 import { createConnectionManager } from "../../../../wallets/manager/index.js";
 import type { WalletId } from "../../../../wallets/wallet-types.js";
 import { ThirdwebProvider } from "../../../web/providers/thirdweb-provider.js";
 import { ConnectionManagerCtx } from "../../providers/connection-manager.js";
-import {
-  handleWalletConnection,
-  useAutoConnectCore,
-} from "./useAutoConnect.js";
+import { useAutoConnectCore } from "./useAutoConnect.js";
 
 describe("useAutoConnectCore", () => {
   const mockStorage = new MockStorage();
@@ -159,29 +155,3 @@ describe("useAutoConnectCore", () => {
     );
   });
 });
-
-describe("handleWalletConnection", () => {
-  const wallet = createWalletAdapter({
-    adaptedAccount: TEST_ACCOUNT_A,
-    client: TEST_CLIENT,
-    chain: ethereum,
-    onDisconnect: () => {},
-    switchChain: () => {},
-  });
-  it("should return the correct result", async () => {
-    const result = await handleWalletConnection({
-      client: TEST_CLIENT,
-      lastConnectedChain: ethereum,
-      authResult: undefined,
-      wallet,
-    });
-
-    expect("address" in result).toBe(true);
-    expect(isAddress(result.address)).toBe(true);
-    expect("sendTransaction" in result).toBe(true);
-    expect(typeof result.sendTransaction).toBe("function");
-    expect("signMessage" in result).toBe(true);
-    expect("signTypedData" in result).toBe(true);
-    expect("signTransaction" in result).toBe(true);
-  });
-});

packages/thirdweb/src/react/native/hooks/wallets/useAutoConnect.ts

@@ -1,6 +1,6 @@
 import { nativeLocalStorage } from "../../../../utils/storage/nativeStorage.js";
+import type { AutoConnectProps } from "../../../../wallets/connection/types.js";
 import { createWallet } from "../../../../wallets/native/create-wallet.js";
-import type { AutoConnectProps } from "../../../core/hooks/connection/types.js";
 import { useAutoConnectCore } from "../../../core/hooks/wallets/useAutoConnect.js";
 import { getDefaultWallets } from "../../wallets/defaultWallets.js";
 

packages/thirdweb/src/react/native/ui/AutoConnect/AutoConnect.tsx

@@ -1,5 +1,5 @@
 "use client";
-import type { AutoConnectProps } from "../../../core/hooks/connection/types.js";
+import type { AutoConnectProps } from "../../../../wallets/connection/types.js";
 import { useAutoConnect } from "../../hooks/wallets/useAutoConnect.js";
 
 /**

packages/thirdweb/src/react/web/hooks/wallets/useAutoConnect.ts

@@ -1,9 +1,9 @@
 import { webLocalStorage } from "../../../../utils/storage/webStorage.js";
+import type { AutoConnectProps } from "../../../../wallets/connection/types.js";
 import { createWallet } from "../../../../wallets/create-wallet.js";
+import { getDefaultWallets } from "../../../../wallets/defaultWallets.js";
 import { getInstalledWalletProviders } from "../../../../wallets/injected/mipdStore.js";
-import type { AutoConnectProps } from "../../../core/hooks/connection/types.js";
 import { useAutoConnectCore } from "../../../core/hooks/wallets/useAutoConnect.js";
-import { getDefaultWallets } from "../../wallets/defaultWallets.js";
 
 /**
  * Autoconnect the last previously connected wallet.

packages/thirdweb/src/react/web/ui/AutoConnect/AutoConnect.tsx

@@ -1,5 +1,5 @@
 "use client";
-import type { AutoConnectProps } from "../../../core/hooks/connection/types.js";
+import type { AutoConnectProps } from "../../../../wallets/connection/types.js";
 import { useAutoConnect } from "../../hooks/wallets/useAutoConnect.js";
 
 /**