@turnkey/react-wallet-kit 1.0.0-beta.5 → 1.0.0-beta.6

package/dist/providers/client/Provider.js

@@ -4,7 +4,8 @@
 var jsxRuntime = require('react/jsx-runtime');
 var sha2 = require('@noble/hashes/sha2');
 var utils$1 = require('@noble/hashes/utils');
-var utils = require('../../utils.js');
+var utils = require('../../utils/utils.js');
+var timers = require('../../utils/timers.js');
 var core = require('@turnkey/core');
 var react = require('react');
 var sdkTypes = require('@turnkey/sdk-types');
@@ -23,7 +24,7 @@ var UpdatePhoneNumber = require('../../components/user/UpdatePhoneNumber.js');
 var UpdateUserName = require('../../components/user/UpdateUserName.js');
 var RemoveOAuthProvider = require('../../components/user/RemoveOAuthProvider.js');
 var RemovePasskey = require('../../components/user/RemovePasskey.js');
-var LinkWallet = require('../../components/user/LinkWallet.js');
+var ConnectWallet = require('../../components/user/ConnectWallet.js');
 var Types = require('./Types.js');
 var OTP = require('../../components/auth/OTP.js');
 var RemoveEmail = require('../../components/user/RemoveEmail.js');
@@ -32,7 +33,7 @@ var RemovePhoneNumber = require('../../components/user/RemovePhoneNumber.js');
 /**
  * Provides Turnkey client authentication, session management, wallet operations, and user profile management
  * for the React Wallet Kit SDK. This context provider encapsulates all core authentication flows (Passkey, Wallet, OTP, OAuth),
- * session lifecycle (creation, expiration, refresh), wallet linking/import/export, and user profile updates (email, phone, name).
+ * session lifecycle (creation, expiration, refresh), wallet connecting/import/export, and user profile updates (email, phone, name).
  *
  * The provider automatically initializes the Turnkey client, fetches configuration (including proxy auth config if needed),
  * and synchronizes session and authentication state. It exposes a comprehensive set of methods for authentication flows,
@@ -41,9 +42,9 @@ var RemovePhoneNumber = require('../../components/user/RemovePhoneNumber.js');
  * Features:
  * - Passkey, Wallet, OTP (Email/SMS), and OAuth (Google, Apple, Facebook) authentication and sign-up flows.
  * - Session management: creation, expiration scheduling, refresh, and clearing.
- * - Wallet management: fetch, link, import, export, account management.
+ * - Wallet management: fetch, connect, import, export, account management.
  * - User profile management: email, phone, name, OAuth provider, and passkey linking/removal.
- * - Modal-driven UI flows for authentication, wallet linking, and profile updates.
+ * - Modal-driven UI flows for authentication, wallet connecting, and profile updates.
  * - Error handling and callback integration for custom error and event responses.
  *
  * Usage:
@@ -92,6 +93,7 @@ const ClientProvider = ({
           const flow = stateParams.get("flow");
           const publicKey = stateParams.get("publicKey");
           const openModal = stateParams.get("openModal");
+          const sessionKey = stateParams.get("sessionKey");
           if (provider === "facebook" && flow === "redirect" && publicKey) {
             // We have all the required parameters for a Facebook PKCE flow
             const clientId = masterConfig?.auth?.oauthConfig?.facebookClientId;
@@ -116,8 +118,13 @@ const ClientProvider = ({
                             await completeOauth({
                               oidcToken,
                               publicKey,
-                              providerName: "facebook"
+                              providerName: "facebook",
+                              ...(sessionKey && {
+                                sessionKey
+                              })
                             });
+                            // Clean up the URL after processing
+                            window.history.replaceState(null, document.title, window.location.pathname);
                             resolve();
                           } catch (err) {
                             reject(err);
@@ -140,6 +147,114 @@ const ClientProvider = ({
               });
             }
           }
+          if (provider === "discord" && flow === "redirect" && publicKey) {
+            const clientId = masterConfig?.auth?.oauthConfig?.discordClientId;
+            const redirectURI = masterConfig?.auth?.oauthConfig?.oauthRedirectUri;
+            const verifier = sessionStorage.getItem("discord_verifier");
+            const nonce = stateParams.get("nonce");
+            const sessionKey = stateParams.get("sessionKey");
+            if (clientId && redirectURI && verifier && nonce) {
+              await new Promise((resolve, reject) => {
+                pushPage({
+                  key: `Discord OAuth`,
+                  content: jsxRuntime.jsx(Action.ActionPage, {
+                    title: `Authenticating with Discord...`,
+                    action: async () => {
+                      try {
+                        const resp = await client?.httpClient.proxyOAuth2Authenticate({
+                          provider: "OAUTH2_PROVIDER_DISCORD",
+                          authCode: code,
+                          redirectUri: redirectURI,
+                          codeVerifier: verifier,
+                          nonce: nonce
+                        });
+                        sessionStorage.removeItem("discord_verifier");
+                        const oidcToken = resp?.oidcToken;
+                        if (!oidcToken) {
+                          throw new sdkTypes.TurnkeyError("Missing OIDC token", sdkTypes.TurnkeyErrorCodes.OAUTH_LOGIN_ERROR);
+                        }
+                        await completeOauth({
+                          oidcToken,
+                          publicKey,
+                          providerName: "discord",
+                          ...(sessionKey && {
+                            sessionKey
+                          })
+                        });
+                        // Clean up the URL after processing
+                        window.history.replaceState(null, document.title, window.location.pathname);
+                        resolve(null);
+                      } catch (err) {
+                        reject(err);
+                        if (callbacks?.onError) {
+                          callbacks.onError(err instanceof sdkTypes.TurnkeyError ? err : new sdkTypes.TurnkeyError("Discord authentication failed", sdkTypes.TurnkeyErrorCodes.OAUTH_SIGNUP_ERROR, err));
+                        }
+                      }
+                    },
+                    icon: jsxRuntime.jsx(reactFontawesome.FontAwesomeIcon, {
+                      size: "3x",
+                      icon: freeBrandsSvgIcons.faDiscord
+                    })
+                  }),
+                  showTitle: false
+                });
+              });
+            }
+          }
+          if (provider === "twitter" && flow === "redirect" && publicKey) {
+            const clientId = masterConfig?.auth?.oauthConfig?.xClientId;
+            const redirectURI = masterConfig?.auth?.oauthConfig?.oauthRedirectUri;
+            const verifier = sessionStorage.getItem("twitter_verifier");
+            const nonce = stateParams.get("nonce");
+            const sessionKey = stateParams.get("sessionKey");
+            if (clientId && redirectURI && verifier && nonce) {
+              await new Promise((resolve, reject) => {
+                pushPage({
+                  key: `Twitter OAuth`,
+                  content: jsxRuntime.jsx(Action.ActionPage, {
+                    title: `Authenticating with Twitter...`,
+                    action: async () => {
+                      try {
+                        const resp = await client?.httpClient.proxyOAuth2Authenticate({
+                          provider: "OAUTH2_PROVIDER_X",
+                          authCode: code,
+                          redirectUri: redirectURI,
+                          codeVerifier: verifier,
+                          nonce: nonce
+                        });
+                        sessionStorage.removeItem("twitter_verifier");
+                        const oidcToken = resp?.oidcToken;
+                        if (!oidcToken) {
+                          throw new sdkTypes.TurnkeyError("Missing OIDC token", sdkTypes.TurnkeyErrorCodes.OAUTH_LOGIN_ERROR);
+                        }
+                        await completeOauth({
+                          oidcToken,
+                          publicKey,
+                          providerName: "twitter",
+                          ...(sessionKey && {
+                            sessionKey
+                          })
+                        });
+                        // Clean up the URL after processing
+                        window.history.replaceState(null, document.title, window.location.pathname);
+                        resolve(null);
+                      } catch (err) {
+                        reject(err);
+                        if (callbacks?.onError) {
+                          callbacks.onError(err instanceof sdkTypes.TurnkeyError ? err : new sdkTypes.TurnkeyError("Twitter authentication failed", sdkTypes.TurnkeyErrorCodes.OAUTH_SIGNUP_ERROR, err));
+                        }
+                      }
+                    },
+                    icon: jsxRuntime.jsx(reactFontawesome.FontAwesomeIcon, {
+                      size: "3x",
+                      icon: freeBrandsSvgIcons.faXTwitter
+                    })
+                  }),
+                  showTitle: false
+                });
+              });
+            }
+          }
         }
       }
       // Handle Google/Apple redirects (uses hash with id_token)
@@ -151,7 +266,8 @@ const ClientProvider = ({
           provider,
           flow,
           publicKey,
-          openModal
+          openModal,
+          sessionKey
         } = utils.parseOAuthRedirect(hash);
         if (idToken && flow === "redirect" && publicKey) {
           if (openModal === "true") {
@@ -183,7 +299,10 @@ const ClientProvider = ({
                         publicKey,
                         ...(provider ? {
                           providerName: provider
-                        } : {})
+                        } : {}),
+                        ...(sessionKey && {
+                          sessionKey
+                        })
                       });
                       resolve(null);
                     } catch (err) {
@@ -223,11 +342,13 @@ const ClientProvider = ({
       passkeyAuthEnabled: config.auth?.methods?.passkeyAuthEnabled ?? proxyAuthConfig?.enabledProviders.includes("passkey"),
       walletAuthEnabled: config.auth?.methods?.walletAuthEnabled ?? proxyAuthConfig?.enabledProviders.includes("wallet"),
       googleOauthEnabled: config.auth?.methods?.googleOauthEnabled ?? proxyAuthConfig?.enabledProviders.includes("google"),
+      xOauthEnabled: config.auth?.methods?.xOauthEnabled ?? proxyAuthConfig?.enabledProviders.includes("x"),
+      discordOauthEnabled: config.auth?.methods?.discordOauthEnabled ?? proxyAuthConfig?.enabledProviders.includes("discord"),
       appleOauthEnabled: config.auth?.methods?.appleOauthEnabled ?? proxyAuthConfig?.enabledProviders.includes("apple"),
       facebookOauthEnabled: config.auth?.methods?.facebookOauthEnabled ?? proxyAuthConfig?.enabledProviders.includes("facebook")
     };
     // Set a default ordering for the oAuth methods
-    const oauthOrder = config.auth?.oauthOrder ?? ["google", "apple", "facebook"].filter(provider => resolvedMethods[`${provider}OauthEnabled`]);
+    const oauthOrder = config.auth?.oauthOrder ?? ["google", "apple", "x", "discord", "facebook"].filter(provider => resolvedMethods[`${provider}OauthEnabled`]);
     // Set a default ordering for the overall auth methods
     const methodOrder = config.auth?.methodOrder ?? [oauthOrder.length > 0 ? "socials" : null, resolvedMethods.emailOtpAuthEnabled ? "email" : null, resolvedMethods.smsOtpAuthEnabled ? "sms" : null, resolvedMethods.passkeyAuthEnabled ? "passkey" : null, resolvedMethods.walletAuthEnabled ? "wallet" : null].filter(Boolean);
     return {
@@ -442,14 +563,13 @@ const ClientProvider = ({
   }
   /**
    * @internal
-   * Schedules a session expiration and warning timeout for the given session key.
+   * Schedules a session expiration and warning timer for the given session key.
    *
-   * - This function sets up two timeouts: one for warning before the session expires and another to expire the session.
-   * - The warning timeout is set to trigger before the session expires, allowing for actions like refreshing the session.
-   * - The expiration timeout clears the session and triggers any necessary callbacks.
+   * - Sets up two timers: one for warning before expiry and one for actual expiry.
+   * - Uses capped timeouts under the hood so delays > 24.8 days are safe (see utils/timers.ts).
    *
    * @param params.sessionKey - The key of the session to schedule expiration for.
-   * @param params.expiry - The expiration time in seconds for the session.
+   * @param params.expiry - The expiration time in seconds since epoch.
    * @throws {TurnkeyError} If an error occurs while scheduling the session expiration.
    */
   async function scheduleSessionExpiration(params) {
@@ -458,21 +578,18 @@ const ClientProvider = ({
       expiry
     } = params;
     try {
-      // Clear any existing timeout for this session key
-      if (expiryTimeoutsRef.current[sessionKey]) {
-        clearTimeout(expiryTimeoutsRef.current[sessionKey]);
-        delete expiryTimeoutsRef.current[sessionKey];
-      }
-      if (expiryTimeoutsRef.current[`${sessionKey}-warning`]) {
-        clearTimeout(expiryTimeoutsRef.current[`${sessionKey}-warning`]);
-        delete expiryTimeoutsRef.current[`${sessionKey}-warning`];
-      }
-      const timeUntilExpiry = expiry * 1000 - Date.now();
+      const warnKey = `${sessionKey}-warning`;
+      // Replace any prior timers for this session
+      timers.clearKey(expiryTimeoutsRef.current, sessionKey);
+      timers.clearKey(expiryTimeoutsRef.current, warnKey);
+      const now = Date.now();
+      const expiryMs = expiry * 1000;
+      const timeUntilExpiry = expiryMs - now;
       const beforeExpiry = async () => {
         const activeSession = await getSession();
-        if (!activeSession && expiryTimeoutsRef.current[sessionKey]) {
-          clearTimeout(expiryTimeoutsRef.current[`${sessionKey}-warning`]);
-          expiryTimeoutsRef.current[`${sessionKey}-warning`] = setTimeout(beforeExpiry, 10000);
+        if (!activeSession && expiryTimeoutsRef.current[warnKey]) {
+          // Keep nudging until session materializes (short 10s timer is fine)
+          timers.setTimeoutInMap(expiryTimeoutsRef.current, warnKey, beforeExpiry, 10_000);
           return;
         }
         const session = await getSession({
@@ -500,27 +617,36 @@ const ClientProvider = ({
         if ((await getActiveSessionKey()) === sessionKey) {
           setSession(undefined);
         }
-        setAllSessions(prevSessions => {
-          if (!prevSessions) return prevSessions;
-          const newSessions = {
-            ...prevSessions
+        setAllSessions(prev => {
+          if (!prev) return prev;
+          const next = {
+            ...prev
           };
-          delete newSessions[sessionKey];
-          return newSessions;
+          delete next[sessionKey];
+          return next;
         });
         await clearSession({
           sessionKey
         });
-        delete expiryTimeoutsRef.current[sessionKey];
-        delete expiryTimeoutsRef.current[`${sessionKey}-warning`];
+        // Remove timers for this session
+        timers.clearKey(expiryTimeoutsRef.current, sessionKey);
+        timers.clearKey(expiryTimeoutsRef.current, warnKey);
         await logout();
       };
-      if (timeUntilExpiry <= utils.SESSION_WARNING_THRESHOLD_MS) {
-        beforeExpiry();
+      // Already expired → expire immediately
+      if (timeUntilExpiry <= 0) {
+        await expireSession();
+        return;
+      }
+      // Warning timer (if threshold is in the future)
+      const warnAt = expiryMs - utils.SESSION_WARNING_THRESHOLD_MS;
+      if (warnAt <= now) {
+        void beforeExpiry(); // fire-and-forget is fine
       } else {
-        expiryTimeoutsRef.current[`${sessionKey}-warning`] = setTimeout(beforeExpiry, timeUntilExpiry - utils.SESSION_WARNING_THRESHOLD_MS);
+        timers.setCappedTimeoutInMap(expiryTimeoutsRef.current, warnKey, beforeExpiry, warnAt - now);
       }
-      expiryTimeoutsRef.current[sessionKey] = setTimeout(expireSession, timeUntilExpiry);
+      // Actual expiry timer (safe for long delays)
+      timers.setCappedTimeoutInMap(expiryTimeoutsRef.current, sessionKey, expireSession, timeUntilExpiry);
     } catch (error) {
       if (error instanceof sdkTypes.TurnkeyError || error instanceof sdkTypes.TurnkeyNetworkError) {
         callbacks?.onError?.(error);
@@ -530,25 +656,25 @@ const ClientProvider = ({
     }
   }
   /**
-   * Clears all scheduled session expiration and warning timeouts for the client.
+   * Clears all scheduled session timers (warning + expiry).
    *
-   * - This function removes all active session expiration and warning timeouts managed by the provider.
-   * - It is called automatically when sessions are re-initialized or on logout to prevent memory leaks and ensure no stale timeouts remain.
-   * - All timeouts stored in `expiryTimeoutsRef` are cleared and the reference is reset.
+   * - Removes all active timers managed by this client.
+   * - Useful on re-init or logout to avoid stale timers.
    *
-   * @throws {TurnkeyError} If an error occurs while clearing the timeouts.
+   * @throws {TurnkeyError} If an error occurs while clearing the timers.
    */
-  function clearSessionTimeouts() {
+  function clearSessionTimeouts(sessionKeys) {
     try {
-      Object.values(expiryTimeoutsRef.current).forEach(timeout => {
-        clearTimeout(timeout);
-      });
-      expiryTimeoutsRef.current = {};
+      if (sessionKeys) {
+        timers.clearKeys(expiryTimeoutsRef.current, sessionKeys);
+      } else {
+        timers.clearAll(expiryTimeoutsRef.current); // clears & deletes everything
+      }
     } catch (error) {
       if (error instanceof sdkTypes.TurnkeyError || error instanceof sdkTypes.TurnkeyNetworkError) {
         callbacks?.onError?.(error);
       } else {
-        callbacks?.onError?.(new sdkTypes.TurnkeyError(`Failed to clear session timeouts`, sdkTypes.TurnkeyErrorCodes.CLEAR_SESSION_TIMEOUTS_ERROR, error));
+        callbacks?.onError?.(new sdkTypes.TurnkeyError("Failed to clear session timeouts", sdkTypes.TurnkeyErrorCodes.CLEAR_SESSION_TIMEOUTS_ERROR, error));
       }
     }
   }
@@ -564,7 +690,12 @@ const ClientProvider = ({
    * @returns A void promise.
    * @throws {TurnkeyError} If the client is not initialized or if there is an error during the process.
    */
-  const handlePostAuth = async () => {
+  const handlePostAuth = async params => {
+    const {
+      method,
+      action,
+      identifier
+    } = params;
     try {
       const sessionKey = await getActiveSessionKey();
       const session = await getSession({
@@ -581,6 +712,12 @@ const ClientProvider = ({
       setAllSessions(allSessions);
       await refreshWallets();
       await refreshUser();
+      callbacks?.onAuthenticationSuccess?.({
+        session,
+        method,
+        action,
+        identifier
+      });
     } catch (error) {
       if (error instanceof sdkTypes.TurnkeyError || error instanceof sdkTypes.TurnkeyNetworkError) {
         callbacks?.onError?.(error);
@@ -594,19 +731,29 @@ const ClientProvider = ({
    * Handles the post-logout flow.
    *
    * - This function is called after a successful logout or session clear.
-   * - It clears all scheduled session expiration and warning timeouts to prevent memory leaks.
-   * - It resets the session state, removes all session and user data from memory, and clears the wallets list.
+   * - It clears all scheduled session expiration and warning timeouts associated to the session key to prevent memory leaks.
+   * - It resets the session state, removes user data from memory, the logged out session from all sessions state, and clears the wallets list.
    * - This ensures that all sensitive information is removed from the provider state after logout.
    * - Called internally after logout or when all sessions are cleared.
    *
    * @returns void
    * @throws {TurnkeyError} If there is an error during the post-logout process.
    */
-  const handlePostLogout = () => {
+  const handlePostLogout = sessionKey => {
     try {
-      clearSessionTimeouts();
+      clearSessionTimeouts(sessionKey ? [sessionKey, `${sessionKey}-warning`] : undefined);
+      setAllSessions(prev => {
+        if (!prev) return prev;
+        if (sessionKey) {
+          const next = {
+            ...prev
+          };
+          delete next[sessionKey];
+          return next;
+        }
+        return {};
+      });
       setSession(undefined);
-      setAllSessions(undefined);
       setUser(undefined);
       setWallets([]);
     } catch (error) {
@@ -625,8 +772,14 @@ const ClientProvider = ({
     if (!client) {
       throw new sdkTypes.TurnkeyError("Client is not initialized.", sdkTypes.TurnkeyErrorCodes.CLIENT_NOT_INITIALIZED);
     }
-    await utils.withTurnkeyErrorHandling(() => client.logout(params), callbacks, "Failed to logout");
-    handlePostLogout();
+    await utils.withTurnkeyErrorHandling(async () => {
+      // If no sessionKey is provided, we try to get the active one.
+      let sessionKey = params?.sessionKey;
+      if (!sessionKey) sessionKey = await getActiveSessionKey();
+      await client.logout(params);
+      // We only handle post logout if the sessionKey is defined since that means we actually logged out of a session.
+      if (sessionKey) handlePostLogout(sessionKey);
+    }, callbacks, "Failed to logout");
     return;
   }, [client, callbacks]);
   const loginWithPasskey = react.useCallback(async params => {
@@ -639,7 +792,11 @@ const ClientProvider = ({
       expirationSeconds
     }), callbacks, "Failed to login with passkey");
     if (res) {
-      await handlePostAuth();
+      await handlePostAuth({
+        method: base.AuthMethod.Passkey,
+        action: sdkTypes.AuthAction.LOGIN,
+        identifier: res.credentialId
+      });
     }
     return res;
   }, [client, callbacks]);
@@ -674,7 +831,11 @@ const ClientProvider = ({
       expirationSeconds
     }), callbacks, "Failed to sign up with passkey");
     if (res) {
-      await handlePostAuth();
+      await handlePostAuth({
+        method: base.AuthMethod.Passkey,
+        action: sdkTypes.AuthAction.SIGNUP,
+        identifier: res.credentialId
+      });
     }
     return res;
   }, [client, callbacks]);
@@ -705,7 +866,7 @@ const ClientProvider = ({
     await client.disconnectWalletAccount(walletProvider);
     // we only refresh the wallets if:
     // 1. there is an active session. This is needed because for WalletConnect
-    //    you can unlink a wallet before actually being logged in
+    //    you can disconnect a wallet before actually being logged in
     //
     // 2. it was a WalletConnect provider that we just disconnected. Since
     //    native providers emit a disconnect event which will already refresh
@@ -734,7 +895,11 @@ const ClientProvider = ({
       expirationSeconds
     }), callbacks, "Failed to login with wallet");
     if (res) {
-      await handlePostAuth();
+      await handlePostAuth({
+        method: base.AuthMethod.Wallet,
+        action: sdkTypes.AuthAction.LOGIN,
+        identifier: res.address
+      });
     }
     return res;
   }, [client, callbacks]);
@@ -759,7 +924,11 @@ const ClientProvider = ({
       expirationSeconds
     }), callbacks, "Failed to sign up with wallet");
     if (res) {
-      await handlePostAuth();
+      await handlePostAuth({
+        method: base.AuthMethod.Wallet,
+        action: sdkTypes.AuthAction.SIGNUP,
+        identifier: res.address
+      });
     }
     return res;
   }, [client, callbacks, masterConfig]);
@@ -784,7 +953,11 @@ const ClientProvider = ({
       expirationSeconds
     }), callbacks, "Failed to login or sign up with wallet");
     if (res) {
-      await handlePostAuth();
+      await handlePostAuth({
+        method: base.AuthMethod.Wallet,
+        action: res.action,
+        identifier: res.address
+      });
     }
     return res;
   }, [client, callbacks, masterConfig]);
@@ -806,7 +979,11 @@ const ClientProvider = ({
     }
     const res = await utils.withTurnkeyErrorHandling(() => client.loginWithOtp(params), callbacks, "Failed to login with OTP");
     if (res) {
-      await handlePostAuth();
+      await handlePostAuth({
+        method: base.AuthMethod.Otp,
+        action: sdkTypes.AuthAction.LOGIN,
+        identifier: params.verificationToken
+      });
     }
     return res;
   }, [client, callbacks]);
@@ -834,7 +1011,11 @@ const ClientProvider = ({
     };
     const res = await utils.withTurnkeyErrorHandling(() => client.signUpWithOtp(params), callbacks, "Failed to sign up with OTP");
     if (res) {
-      await handlePostAuth();
+      await handlePostAuth({
+        method: base.AuthMethod.Otp,
+        action: sdkTypes.AuthAction.SIGNUP,
+        identifier: params.verificationToken
+      });
     }
     return res;
   }, [client, callbacks, masterConfig]);
@@ -862,11 +1043,29 @@ const ClientProvider = ({
     };
     const res = await utils.withTurnkeyErrorHandling(() => client.completeOtp(params), callbacks, "Failed to complete OTP");
     if (res) {
-      await handlePostAuth();
+      await handlePostAuth({
+        method: base.AuthMethod.Otp,
+        action: res.action,
+        identifier: res.verificationToken
+      });
     }
     return res;
   }, [client, callbacks, masterConfig]);
-  const completeOauth = react.useCallback(async params => {
+  const loginWithOauth = react.useCallback(async params => {
+    if (!client) {
+      throw new sdkTypes.TurnkeyError("Client is not initialized.", sdkTypes.TurnkeyErrorCodes.CLIENT_NOT_INITIALIZED);
+    }
+    const res = await utils.withTurnkeyErrorHandling(() => client.loginWithOauth(params), callbacks, "Failed to login with OAuth");
+    if (res) {
+      await handlePostAuth({
+        method: base.AuthMethod.Oauth,
+        action: sdkTypes.AuthAction.LOGIN,
+        identifier: params.oidcToken
+      });
+    }
+    return res;
+  }, [client, callbacks]);
+  const signUpWithOauth = react.useCallback(async params => {
     if (!client) {
       throw new sdkTypes.TurnkeyError("Client is not initialized.", sdkTypes.TurnkeyErrorCodes.CLIENT_NOT_INITIALIZED);
     }
@@ -874,30 +1073,24 @@ const ClientProvider = ({
       throw new sdkTypes.TurnkeyError("Config is not ready yet!", sdkTypes.TurnkeyErrorCodes.INVALID_CONFIGURATION);
     }
     // If createSubOrgParams is not provided, use the default from masterConfig
-    const createSubOrgParams = params.createSubOrgParams ?? masterConfig.auth?.createSuborgParams?.oauth;
+    let createSubOrgParams = params.createSubOrgParams ?? masterConfig.auth?.createSuborgParams?.oauth;
     params = createSubOrgParams !== undefined ? {
       ...params,
       createSubOrgParams
     } : {
       ...params
     };
-    const res = await utils.withTurnkeyErrorHandling(() => client.completeOauth(params), callbacks, "Failed to complete OAuth");
+    const res = await utils.withTurnkeyErrorHandling(() => client.signUpWithOauth(params), callbacks, "Failed to sign up with OAuth");
     if (res) {
-      await handlePostAuth();
+      await handlePostAuth({
+        method: base.AuthMethod.Oauth,
+        action: sdkTypes.AuthAction.SIGNUP,
+        identifier: params.oidcToken
+      });
     }
     return res;
   }, [client, callbacks, masterConfig]);
-  const loginWithOauth = react.useCallback(async params => {
-    if (!client) {
-      throw new sdkTypes.TurnkeyError("Client is not initialized.", sdkTypes.TurnkeyErrorCodes.CLIENT_NOT_INITIALIZED);
-    }
-    const res = await utils.withTurnkeyErrorHandling(() => client.loginWithOauth(params), callbacks, "Failed to login with OAuth");
-    if (res) {
-      await handlePostAuth();
-    }
-    return res;
-  }, [client, callbacks]);
-  const signUpWithOauth = react.useCallback(async params => {
+  const completeOauth = react.useCallback(async params => {
     if (!client) {
       throw new sdkTypes.TurnkeyError("Client is not initialized.", sdkTypes.TurnkeyErrorCodes.CLIENT_NOT_INITIALIZED);
     }
@@ -905,16 +1098,20 @@ const ClientProvider = ({
       throw new sdkTypes.TurnkeyError("Config is not ready yet!", sdkTypes.TurnkeyErrorCodes.INVALID_CONFIGURATION);
     }
     // If createSubOrgParams is not provided, use the default from masterConfig
-    let createSubOrgParams = params.createSubOrgParams ?? masterConfig.auth?.createSuborgParams?.oauth;
+    const createSubOrgParams = params.createSubOrgParams ?? masterConfig.auth?.createSuborgParams?.oauth;
     params = createSubOrgParams !== undefined ? {
       ...params,
       createSubOrgParams
     } : {
       ...params
     };
-    const res = await utils.withTurnkeyErrorHandling(() => client.signUpWithOauth(params), callbacks, "Failed to sign up with OAuth");
+    const res = await utils.withTurnkeyErrorHandling(() => client.completeOauth(params), callbacks, "Failed to complete OAuth");
     if (res) {
-      await handlePostAuth();
+      await handlePostAuth({
+        method: base.AuthMethod.Oauth,
+        action: res.action,
+        identifier: params.oidcToken
+      });
     }
     return res;
   }, [client, callbacks, masterConfig]);
@@ -930,6 +1127,12 @@ const ClientProvider = ({
     }
     return utils.withTurnkeyErrorHandling(() => client.fetchWalletAccounts(params), callbacks, "Failed to fetch wallet accounts");
   }, [client, callbacks]);
+  const fetchPrivateKeys = react.useCallback(async params => {
+    if (!client) {
+      throw new sdkTypes.TurnkeyError("Client is not initialized.", sdkTypes.TurnkeyErrorCodes.CLIENT_NOT_INITIALIZED);
+    }
+    return utils.withTurnkeyErrorHandling(() => client.fetchPrivateKeys(params), callbacks, "Failed to fetch private keys");
+  }, [client, callbacks]);
   const signMessage = react.useCallback(async params => {
     if (!client) throw new sdkTypes.TurnkeyError("Client is not initialized.", sdkTypes.TurnkeyErrorCodes.CLIENT_NOT_INITIALIZED);
     return utils.withTurnkeyErrorHandling(() => client.signMessage(params), callbacks, "Failed to sign message");
@@ -941,7 +1144,7 @@ const ClientProvider = ({
     if (!client) throw new sdkTypes.TurnkeyError("Client is not initialized.", sdkTypes.TurnkeyErrorCodes.CLIENT_NOT_INITIALIZED);
     return new Promise((resolve, reject) => {
       pushPage({
-        key: "Sign Message",
+        key: "Sign message",
         content: jsxRuntime.jsx(Message.SignMessageModal, {
           message: params.message,
           subText: params?.subText,
@@ -1185,6 +1388,10 @@ const ClientProvider = ({
     }
     await utils.withTurnkeyErrorHandling(() => client.setActiveSession(params), callbacks, "Failed to set active session");
     setSession(session);
+    await utils.withTurnkeyErrorHandling(async () => {
+      await refreshWallets();
+      await refreshUser();
+    }, callbacks, "Failed to refresh data after setting active session");
     return;
   }, [client, callbacks]);
   const getActiveSessionKey = react.useCallback(async () => {
@@ -1229,6 +1436,282 @@ const ClientProvider = ({
       setWallets(wallets);
     }
   }, [client, callbacks, getWalletProviders, fetchWallets]);
+  const handleDiscordOauth = react.useCallback(async params => {
+    const {
+      clientId = masterConfig?.auth?.oauthConfig?.discordClientId,
+      openInPage = masterConfig?.auth?.oauthConfig?.openOauthInPage ?? false,
+      additionalState: additionalParameters
+    } = params || {};
+    try {
+      if (!masterConfig) {
+        throw new sdkTypes.TurnkeyError("Config is not ready yet!", sdkTypes.TurnkeyErrorCodes.INVALID_CONFIGURATION);
+      }
+      if (!clientId) {
+        throw new sdkTypes.TurnkeyError("Discord Client ID is not configured.", sdkTypes.TurnkeyErrorCodes.INVALID_CONFIGURATION);
+      }
+      if (!masterConfig.auth?.oauthConfig?.oauthRedirectUri) {
+        throw new sdkTypes.TurnkeyError("OAuth Redirect URI is not configured.", sdkTypes.TurnkeyErrorCodes.INVALID_CONFIGURATION);
+      }
+      const flow = openInPage ? "redirect" : "popup";
+      const redirectURI = masterConfig.auth?.oauthConfig.oauthRedirectUri;
+      // Create key pair and generate nonce
+      const publicKey = await createApiKeyPair();
+      if (!publicKey) {
+        throw new Error("Failed to create public key for OAuth.");
+      }
+      const nonce = utils$1.bytesToHex(sha2.sha256(publicKey));
+      // Generate PKCE challenge pair
+      const {
+        verifier,
+        codeChallenge
+      } = await utils.generateChallengePair();
+      sessionStorage.setItem("discord_verifier", verifier);
+      // Construct Discord Auth URL
+      const discordAuthUrl = new URL(utils.DISCORD_AUTH_URL);
+      discordAuthUrl.searchParams.set("client_id", clientId);
+      discordAuthUrl.searchParams.set("redirect_uri", redirectURI);
+      discordAuthUrl.searchParams.set("response_type", "code");
+      discordAuthUrl.searchParams.set("code_challenge", codeChallenge);
+      discordAuthUrl.searchParams.set("code_challenge_method", "S256");
+      discordAuthUrl.searchParams.set("scope", "identify email");
+      discordAuthUrl.searchParams.set("state", `provider=discord&flow=${flow}&publicKey=${encodeURIComponent(publicKey)}&nonce=${nonce}`);
+      if (additionalParameters) {
+        const extra = Object.entries(additionalParameters).map(([k, v]) => `${encodeURIComponent(k)}=${encodeURIComponent(v)}`).join("&");
+        if (extra) {
+          discordAuthUrl.searchParams.set("state", discordAuthUrl.searchParams.get("state") + `&${extra}`);
+        }
+      }
+      if (openInPage) {
+        window.location.href = discordAuthUrl.toString();
+        return new Promise((_, reject) => {
+          const timeout = setTimeout(() => {
+            reject(new Error("Authentication timed out."));
+          }, 300000);
+          window.addEventListener("beforeunload", () => clearTimeout(timeout));
+        });
+      } else {
+        const width = utils.popupWidth;
+        const height = utils.popupHeight;
+        const left = window.screenX + (window.innerWidth - width) / 2;
+        const top = window.screenY + (window.innerHeight - height) / 2;
+        const authWindow = window.open("about:blank", "_blank", `width=${width},height=${height},top=${top},left=${left},scrollbars=yes,resizable=yes`);
+        if (!authWindow) {
+          throw new Error("Failed to open Discord login window.");
+        }
+        authWindow.location.href = discordAuthUrl.toString();
+        return new Promise((resolve, reject) => {
+          const interval = setInterval(() => {
+            try {
+              if (authWindow.closed) {
+                clearInterval(interval);
+                reject(new Error("Authentication window was closed."));
+                return;
+              }
+              const url = authWindow.location.href || "";
+              if (url.startsWith(window.location.origin)) {
+                const urlParams = new URLSearchParams(new URL(url).search);
+                const authCode = urlParams.get("code");
+                const stateParam = urlParams.get("state");
+                const sessionKey = stateParam?.split("&").find(param => param.startsWith("sessionKey="))?.split("=")[1];
+                if (authCode && stateParam && stateParam.includes("provider=discord")) {
+                  authWindow.close();
+                  clearInterval(interval);
+                  const verifier = sessionStorage.getItem("discord_verifier");
+                  if (!verifier) {
+                    reject(new Error("Missing PKCE verifier"));
+                    return;
+                  }
+                  client?.httpClient.proxyOAuth2Authenticate({
+                    provider: "OAUTH2_PROVIDER_DISCORD",
+                    authCode,
+                    redirectUri: redirectURI,
+                    codeVerifier: verifier,
+                    nonce: nonce
+                  }).then(resp => {
+                    sessionStorage.removeItem("discord_verifier");
+                    const oidcToken = resp.oidcToken;
+                    if (params?.onOauthSuccess) {
+                      params.onOauthSuccess({
+                        oidcToken,
+                        providerName: "discord",
+                        ...(sessionKey && {
+                          sessionKey
+                        })
+                      });
+                    } else if (callbacks?.onOauthRedirect) {
+                      callbacks.onOauthRedirect({
+                        idToken: oidcToken,
+                        publicKey,
+                        ...(sessionKey && {
+                          sessionKey
+                        })
+                      });
+                    } else {
+                      completeOauth({
+                        oidcToken,
+                        publicKey,
+                        providerName: "discord",
+                        ...(sessionKey && {
+                          sessionKey
+                        })
+                      }).then(() => resolve()).catch(reject);
+                      return;
+                    }
+                    resolve();
+                  }).catch(reject);
+                }
+              }
+            } catch {
+              // ignore cross-origin
+            }
+          }, 500);
+          if (authWindow.closed) {
+            clearInterval(interval);
+          }
+        });
+      }
+    } catch (error) {
+      throw error;
+    }
+  }, [client, callbacks]);
+  const handleXOauth = react.useCallback(async params => {
+    const {
+      clientId = masterConfig?.auth?.oauthConfig?.xClientId,
+      openInPage = masterConfig?.auth?.oauthConfig?.openOauthInPage ?? false,
+      additionalState: additionalParameters
+    } = params || {};
+    try {
+      if (!masterConfig) {
+        throw new sdkTypes.TurnkeyError("Config is not ready yet!", sdkTypes.TurnkeyErrorCodes.INVALID_CONFIGURATION);
+      }
+      if (!clientId) {
+        throw new sdkTypes.TurnkeyError("Twitter Client ID is not configured.", sdkTypes.TurnkeyErrorCodes.INVALID_CONFIGURATION);
+      }
+      if (!masterConfig.auth?.oauthConfig?.oauthRedirectUri) {
+        throw new sdkTypes.TurnkeyError("OAuth Redirect URI is not configured.", sdkTypes.TurnkeyErrorCodes.INVALID_CONFIGURATION);
+      }
+      const flow = openInPage ? "redirect" : "popup";
+      const redirectURI = masterConfig.auth?.oauthConfig.oauthRedirectUri;
+      // Create key pair and generate nonce
+      const publicKey = await createApiKeyPair();
+      if (!publicKey) {
+        throw new Error("Failed to create public key for OAuth.");
+      }
+      const nonce = utils$1.bytesToHex(sha2.sha256(publicKey));
+      // Generate PKCE challenge pair
+      const {
+        verifier,
+        codeChallenge
+      } = await utils.generateChallengePair();
+      sessionStorage.setItem("twitter_verifier", verifier);
+      // Construct Twitter Auth URL
+      const twitterAuthUrl = new URL(utils.X_AUTH_URL);
+      twitterAuthUrl.searchParams.set("client_id", clientId);
+      twitterAuthUrl.searchParams.set("redirect_uri", redirectURI);
+      twitterAuthUrl.searchParams.set("response_type", "code");
+      twitterAuthUrl.searchParams.set("code_challenge", codeChallenge);
+      twitterAuthUrl.searchParams.set("code_challenge_method", "S256");
+      twitterAuthUrl.searchParams.set("scope", "tweet.read users.read");
+      twitterAuthUrl.searchParams.set("state", `provider=twitter&flow=${flow}&publicKey=${encodeURIComponent(publicKey)}&nonce=${nonce}`);
+      if (additionalParameters) {
+        const extra = Object.entries(additionalParameters).map(([k, v]) => `${encodeURIComponent(k)}=${encodeURIComponent(v)}`).join("&");
+        if (extra) {
+          twitterAuthUrl.searchParams.set("state", twitterAuthUrl.searchParams.get("state") + `&${extra}`);
+        }
+      }
+      if (openInPage) {
+        window.location.href = twitterAuthUrl.toString();
+        return new Promise((_, reject) => {
+          const timeout = setTimeout(() => {
+            reject(new Error("Authentication timed out."));
+          }, 300000);
+          window.addEventListener("beforeunload", () => clearTimeout(timeout));
+        });
+      } else {
+        const width = utils.popupWidth;
+        const height = utils.popupHeight;
+        const left = window.screenX + (window.innerWidth - width) / 2;
+        const top = window.screenY + (window.innerHeight - height) / 2;
+        const authWindow = window.open("about:blank", "_blank", `width=${width},height=${height},top=${top},left=${left},scrollbars=yes,resizable=yes`);
+        if (!authWindow) {
+          throw new Error("Failed to open Twitter login window.");
+        }
+        authWindow.location.href = twitterAuthUrl.toString();
+        return new Promise((resolve, reject) => {
+          const interval = setInterval(() => {
+            try {
+              if (authWindow.closed) {
+                clearInterval(interval);
+                reject(new Error("Authentication window was closed."));
+                return;
+              }
+              const url = authWindow.location.href || "";
+              if (url.startsWith(window.location.origin)) {
+                const urlParams = new URLSearchParams(new URL(url).search);
+                const authCode = urlParams.get("code");
+                const stateParam = urlParams.get("state");
+                const sessionKey = stateParam?.split("&").find(param => param.startsWith("sessionKey="))?.split("=")[1];
+                if (authCode && stateParam && stateParam.includes("provider=twitter")) {
+                  authWindow.close();
+                  clearInterval(interval);
+                  const verifier = sessionStorage.getItem("twitter_verifier");
+                  if (!verifier) {
+                    reject(new Error("Missing PKCE verifier"));
+                    return;
+                  }
+                  client?.httpClient.proxyOAuth2Authenticate({
+                    provider: "OAUTH2_PROVIDER_X",
+                    authCode,
+                    redirectUri: redirectURI,
+                    codeVerifier: verifier,
+                    nonce
+                  }).then(resp => {
+                    sessionStorage.removeItem("twitter_verifier");
+                    const oidcToken = resp.oidcToken;
+                    if (params?.onOauthSuccess) {
+                      params.onOauthSuccess({
+                        oidcToken,
+                        providerName: "twitter",
+                        ...(sessionKey && {
+                          sessionKey
+                        })
+                      });
+                    } else if (callbacks?.onOauthRedirect) {
+                      callbacks.onOauthRedirect({
+                        idToken: oidcToken,
+                        publicKey,
+                        ...(sessionKey && {
+                          sessionKey
+                        })
+                      });
+                    } else {
+                      completeOauth({
+                        oidcToken,
+                        publicKey,
+                        providerName: "twitter",
+                        ...(sessionKey && {
+                          sessionKey
+                        })
+                      }).then(() => resolve()).catch(reject);
+                      return;
+                    }
+                    resolve();
+                  }).catch(reject);
+                }
+              }
+            } catch {
+              // ignore cross-origin
+            }
+          }, 500);
+          if (authWindow.closed) {
+            clearInterval(interval);
+          }
+        });
+      }
+    } catch (error) {
+      throw error;
+    }
+  }, [client, callbacks]);
   const handleGoogleOauth = react.useCallback(async params => {
     const {
       clientId = masterConfig?.auth?.oauthConfig?.googleClientId,
@@ -1309,24 +1792,35 @@ const ClientProvider = ({
               if (url.startsWith(window.location.origin)) {
                 const hashParams = new URLSearchParams(url.split("#")[1]);
                 const idToken = hashParams.get("id_token");
+                const stateParams = hashParams.get("state");
+                const sessionKey = stateParams?.split("&").find(param => param.startsWith("sessionKey="))?.split("=")[1];
                 if (idToken) {
                   authWindow.close();
                   clearInterval(interval);
                   if (params?.onOauthSuccess) {
                     params.onOauthSuccess({
                       oidcToken: idToken,
-                      providerName: "google"
+                      providerName: "google",
+                      ...(sessionKey && {
+                        sessionKey
+                      })
                     });
                   } else if (callbacks?.onOauthRedirect) {
                     callbacks.onOauthRedirect({
                       idToken,
-                      publicKey
+                      publicKey,
+                      ...(sessionKey && {
+                        sessionKey
+                      })
                     });
                   } else {
                     completeOauth({
                       oidcToken: idToken,
                       publicKey,
-                      providerName: "google"
+                      providerName: "google",
+                      ...(sessionKey && {
+                        sessionKey
+                      })
                     }).then(() => resolve()).catch(reject);
                     return;
                   }
@@ -1422,24 +1916,35 @@ const ClientProvider = ({
               if (url.startsWith(window.location.origin)) {
                 const hashParams = new URLSearchParams(url.split("#")[1]);
                 const idToken = hashParams.get("id_token");
+                const stateParams = hashParams.get("state");
+                const sessionKey = stateParams?.split("&").find(param => param.startsWith("sessionKey="))?.split("=")[1];
                 if (idToken) {
                   authWindow.close();
                   clearInterval(interval);
                   if (params?.onOauthSuccess) {
                     params.onOauthSuccess({
                       oidcToken: idToken,
-                      providerName: "apple"
+                      providerName: "apple",
+                      ...(sessionKey && {
+                        sessionKey
+                      })
                     });
                   } else if (callbacks?.onOauthRedirect) {
                     callbacks.onOauthRedirect({
                       idToken,
-                      publicKey
+                      publicKey,
+                      ...(sessionKey && {
+                        sessionKey
+                      })
                     });
                   } else {
                     completeOauth({
                       oidcToken: idToken,
                       publicKey,
-                      providerName: "apple"
+                      providerName: "apple",
+                      ...(sessionKey && {
+                        sessionKey
+                      })
                     }).then(() => resolve()).catch(reject);
                     return;
                   }
@@ -1546,6 +2051,7 @@ const ClientProvider = ({
                 const authCode = urlParams.get("code");
                 const stateParam = urlParams.get("state");
                 if (authCode && stateParam && stateParam.includes("provider=facebook")) {
+                  const sessionKey = stateParam?.split("&").find(param => param.startsWith("sessionKey="))?.split("=")[1];
                   authWindow.close();
                   clearInterval(interval);
                   // Exchange code for token
@@ -1559,18 +2065,27 @@ const ClientProvider = ({
                     if (params?.onOauthSuccess) {
                       params.onOauthSuccess({
                         oidcToken: tokenData.id_token,
-                        providerName: "apple"
+                        providerName: "facebook",
+                        ...(sessionKey && {
+                          sessionKey
+                        })
                       });
                     } else if (callbacks?.onOauthRedirect) {
                       callbacks.onOauthRedirect({
                         idToken: tokenData.id_token,
-                        publicKey
+                        publicKey,
+                        ...(sessionKey && {
+                          sessionKey
+                        })
                       });
                     } else {
                       completeOauth({
                         oidcToken: tokenData.id_token,
                         publicKey,
-                        providerName: "facebook"
+                        providerName: "facebook",
+                        ...(sessionKey && {
+                          sessionKey
+                        })
                       }).then(() => resolve()).catch(reject);
                       return;
                     }
@@ -1591,10 +2106,12 @@ const ClientProvider = ({
       throw error;
     }
   }, [client, callbacks]);
-  const handleLogin = react.useCallback(async () => {
+  const handleLogin = react.useCallback(async params => {
     pushPage({
       key: "Log in or sign up",
-      content: jsxRuntime.jsx(index.AuthComponent, {})
+      content: jsxRuntime.jsx(index.AuthComponent, {
+        sessionKey: params?.sessionKey
+      })
     });
   }, [pushPage]);
   const handleExportWallet = react.useCallback(async params => {
@@ -1604,7 +2121,7 @@ const ClientProvider = ({
       stampWith
     } = params;
     pushPage({
-      key: "Export Wallet",
+      key: "Export wallet",
       content: jsxRuntime.jsx(Export.ExportComponent, {
         target: walletId,
         exportType: base.ExportType.Wallet,
@@ -1621,13 +2138,17 @@ const ClientProvider = ({
     const {
       privateKeyId,
       targetPublicKey,
+      keyFormat,
       stampWith
     } = params;
     pushPage({
-      key: "Export Private Key",
+      key: "Export private key",
       content: jsxRuntime.jsx(Export.ExportComponent, {
         target: privateKeyId,
         exportType: base.ExportType.PrivateKey,
+        ...(keyFormat !== undefined ? {
+          keyFormat
+        } : {}),
         ...(targetPublicKey !== undefined ? {
           targetPublicKey
         } : {}),
@@ -1641,13 +2162,17 @@ const ClientProvider = ({
     const {
       address,
       targetPublicKey,
+      keyFormat,
       stampWith
     } = params;
     pushPage({
-      key: "Export Wallet Account",
+      key: "Export wallet account",
       content: jsxRuntime.jsx(Export.ExportComponent, {
         target: address,
         exportType: base.ExportType.WalletAccount,
+        ...(keyFormat !== undefined ? {
+          keyFormat
+        } : {}),
         ...(targetPublicKey !== undefined ? {
           targetPublicKey
         } : {}),
@@ -1665,8 +2190,9 @@ const ClientProvider = ({
     } = params || {};
     try {
       return utils.withTurnkeyErrorHandling(() => new Promise((resolve, reject) => pushPage({
-        key: "Import Wallet",
+        key: "Import wallet",
         content: jsxRuntime.jsx(Import.ImportComponent, {
+          importType: base.ImportType.Wallet,
           onError: error => {
             reject(error);
           },
@@ -1689,6 +2215,39 @@ const ClientProvider = ({
       throw new sdkTypes.TurnkeyError("Failed to import wallet.", sdkTypes.TurnkeyErrorCodes.IMPORT_WALLET_ERROR, error);
     }
   }, [pushPage]);
+  const handleImportPrivateKey = react.useCallback(async params => {
+    const {
+      curve,
+      addressFormats,
+      successPageDuration = 2000,
+      stampWith
+    } = params || {};
+    try {
+      return utils.withTurnkeyErrorHandling(() => new Promise((resolve, reject) => pushPage({
+        key: "Import private key",
+        content: jsxRuntime.jsx(Import.ImportComponent, {
+          importType: base.ImportType.PrivateKey,
+          curve: curve,
+          addressFormats: addressFormats,
+          onError: error => {
+            reject(error);
+          },
+          onSuccess: privateKeyId => resolve(privateKeyId),
+          ...(successPageDuration !== undefined && {
+            successPageDuration
+          }),
+          ...(stampWith !== undefined && {
+            stampWith
+          })
+        })
+      })));
+    } catch (error) {
+      if (error instanceof sdkTypes.TurnkeyError) {
+        throw error;
+      }
+      throw new sdkTypes.TurnkeyError("Failed to import private key.", sdkTypes.TurnkeyErrorCodes.IMPORT_WALLET_ERROR, error);
+    }
+  }, [pushPage]);
   const handleUpdateUserName = react.useCallback(async params => {
     const {
       successPageDuration = 2000,
@@ -2320,6 +2879,22 @@ const ClientProvider = ({
       });
     };
     switch (providerName) {
+      case sdkTypes.OAuthProviders.DISCORD:
+        {
+          await handleDiscordOauth({
+            openInPage: false,
+            onOauthSuccess
+          });
+          break;
+        }
+      case sdkTypes.OAuthProviders.X:
+        {
+          await handleXOauth({
+            openInPage: false,
+            onOauthSuccess
+          });
+          break;
+        }
       case sdkTypes.OAuthProviders.GOOGLE:
         {
           await handleGoogleOauth({
@@ -2350,7 +2925,7 @@ const ClientProvider = ({
         }
     }
   }, [pushPage]);
-  const handleLinkExternalWallet = react.useCallback(async params => {
+  const handleConnectExternalWallet = react.useCallback(async params => {
     const {
       successPageDuration = 2000
     } = params || {};
@@ -2363,8 +2938,8 @@ const ClientProvider = ({
     }
     const providers = await getWalletProviders();
     pushPage({
-      key: "Link Wallet",
-      content: jsxRuntime.jsx(LinkWallet.LinkWalletModal, {
+      key: "Connect wallet",
+      content: jsxRuntime.jsx(ConnectWallet.ConnectWalletModal, {
         providers: providers,
         successPageDuration: successPageDuration
       })
@@ -2563,6 +3138,7 @@ const ClientProvider = ({
       completeOauth,
       fetchWallets,
       fetchWalletAccounts,
+      fetchPrivateKeys,
       refreshWallets,
       signMessage,
       signTransaction,
@@ -2599,12 +3175,15 @@ const ClientProvider = ({
       getProxyAuthConfig,
       handleLogin,
       handleGoogleOauth,
+      handleXOauth,
+      handleDiscordOauth,
       handleAppleOauth,
       handleFacebookOauth,
       handleExportWallet,
       handleExportPrivateKey,
       handleExportWalletAccount,
       handleImportWallet,
+      handleImportPrivateKey,
       handleUpdateUserEmail,
       handleUpdateUserPhoneNumber,
       handleUpdateUserName,
@@ -2615,7 +3194,7 @@ const ClientProvider = ({
       handleAddEmail,
       handleAddPhoneNumber,
       handleSignMessage,
-      handleLinkExternalWallet,
+      handleConnectExternalWallet,
       handleRemoveUserEmail,
       handleRemoveUserPhoneNumber
     },