[squash] fix build

Author: el-ll

turnkey/src/features/balances/use-balances.ts

@@ -1,3 +1,4 @@
+"use client";
 import { skipToken, useQuery } from "@tanstack/react-query";
 import { useOneBalanceAccountAddress } from "../onebalance-account/use-onebalance-account";
 import { fetchBalances, fetchBTCBalance } from "./fetch-balances";

turnkey/src/features/onebalance-account/create-btc-wallet-ui.tsx

@@ -1,8 +1,8 @@
 "use client";
-import { useState } from "react";
+import { useTurnkey } from "@turnkey/sdk-react";
 import { useTurnkeyAuth } from "../turnkey/use-turnkey-auth";
 import { useBTCAccount } from "./use-btc-account";
-import { useTurnkey } from "@turnkey/sdk-react";
+import { useRecoverBTCWallet } from "../turnkey/use-recover-btc-wallet";
 
 export const BTCWalletUI = () => {
   const { passkeyClient } = useTurnkey();
@@ -65,7 +65,7 @@ export const CreateBTCWalletUI = ({
   onSubmit: () => void;
   rootOrgId: string;
 }) => {
-  const { btcLogin } = useTurnkeyAuth();
+  const { btcLogin } = useRecoverBTCWallet();
 
   return (
     <div className="p-5 text-left w-1/2 border border-surface-level-2 rounded-r-xl flex flex-col gap-2">

turnkey/src/features/onebalance-account/use-btc-account.ts

@@ -1,23 +1,10 @@
+"use client";
 import { skipToken, useMutation, useQuery } from "@tanstack/react-query";
 import { useLocalStorage } from "@uidotdev/usehooks";
 import { createBTCWallet } from "./create-btc-wallet";
 import { fetchBTCWalletAddress } from "./fetch-btc-wallet-address";
 import { useEnvironment } from "../environment/environment";
-
-export const usePersistedBTCWallet = () => {
-  // persisting these values in local storage is not advisable.
-  // they should be persisted in a database against your user.
-  // this integration example uses local storage to simulate
-  // database persistance. Please exercise caution if copying this example
-  const [value, setValue] = useLocalStorage<
-    { walletId: string; organizationId: string } | "null"
-  >("btc-wallet", "null");
-
-  if (value === "null" || !value || Object.keys(value).length === 0)
-    return [null, setValue] as const;
-
-  return [value, setValue] as const;
-};
+import { usePersistedBTCWallet } from "./use-persisted-btc-wallet";
 
 const useCreateBTCAccount = ({
   onSuccess,

turnkey/src/features/onebalance-account/use-persisted-btc-wallet.ts

@@ -0,0 +1,17 @@
+"use client";
+import { useLocalStorage } from "@uidotdev/usehooks";
+
+export const usePersistedBTCWallet = () => {
+  // persisting these values in local storage is not advisable.
+  // they should be persisted in a database against your user.
+  // this integration example uses local storage to simulate
+  // database persistance. Please exercise caution if copying this example
+  const [value, setValue] = useLocalStorage<
+    { walletId: string; organizationId: string } | "null"
+  >("btc-wallet", "null");
+
+  if (value === "null" || !value || Object.keys(value).length === 0)
+    return [null, setValue] as const;
+
+  return [value, setValue] as const;
+};

turnkey/src/features/swap/use-swap.ts

@@ -1,3 +1,4 @@
+"use client";
 import { useMutation, useQuery } from "@tanstack/react-query";
 import { useTurnkey } from "@turnkey/sdk-react";
 import { FormEvent } from "react";

turnkey/src/features/turnkey/use-recover-btc-wallet.ts

@@ -0,0 +1,42 @@
+"use client";
+import { useMutation } from "@tanstack/react-query";
+import { useTurnkey } from "@turnkey/sdk-react";
+import { usePersistedBTCWallet } from "../onebalance-account/use-persisted-btc-wallet";
+
+export const useRecoverBTCWallet = () => {
+  const { turnkey } = useTurnkey();
+  const [, setBTCWallet] = usePersistedBTCWallet();
+
+  const { mutate: btcLogin } = useMutation({
+    mutationFn: async ({ rootOrgId }: { rootOrgId: string }) => {
+      return turnkey
+        ?.passkeyClient()
+        .getWhoami({
+          organizationId: rootOrgId,
+        })
+        .then((result) => {
+          return turnkey
+            ?.passkeyClient()
+            .getWallets({
+              organizationId: result.organizationId,
+            })
+            .then((wallets) => {
+              return [wallets, result] as const;
+            });
+        })
+        .then(([{ wallets }, { organizationId }]) => {
+          if (!wallets[0].walletName.toLowerCase().includes("btc wallet"))
+            return;
+
+          setBTCWallet({
+            walletId: wallets[0].walletId,
+            organizationId,
+          });
+        });
+    },
+  });
+
+  return {
+    btcLogin,
+  };
+};

turnkey/src/features/turnkey/use-turnkey-auth.ts

@@ -1,8 +1,7 @@
+"use client";
 import { useMutation, useQuery } from "@tanstack/react-query";
 import { useTurnkey } from "@turnkey/sdk-react";
 import { queryClient } from "../react-query";
-import { useEnvironment } from "../environment/environment";
-import { usePersistedBTCWallet } from "../onebalance-account/use-btc-account";
 
 type TurnkeyBrowserSDK = NonNullable<ReturnType<typeof useTurnkey>["turnkey"]>;
 export type TurnkeyPasskeyClient = NonNullable<
@@ -20,7 +19,6 @@ export const useTurnkeyAuth = () => {
       refetch();
     },
   });
-  const [, setBTCWallet] = usePersistedBTCWallet();
 
   const {
     data: user,
@@ -45,35 +43,6 @@ export const useTurnkeyAuth = () => {
     },
   });
 
-  const { mutate: btcLogin } = useMutation({
-    mutationFn: async ({ rootOrgId }: { rootOrgId: string }) => {
-      return turnkey
-        ?.passkeyClient()
-        .getWhoami({
-          organizationId: rootOrgId,
-        })
-        .then((result) => {
-          return turnkey
-            ?.passkeyClient()
-            .getWallets({
-              organizationId: result.organizationId,
-            })
-            .then((wallets) => {
-              return [wallets, result] as const;
-            });
-        })
-        .then(([{ wallets }, { organizationId }]) => {
-          if (!wallets[0].walletName.toLowerCase().includes("btc wallet"))
-            return;
-
-          setBTCWallet({
-            walletId: wallets[0].walletId,
-            organizationId,
-          });
-        });
-    },
-  });
-
   const { data: wallets } = useQuery({
     queryKey: ["wallets"],
     queryFn: async () => {
@@ -102,6 +71,5 @@ export const useTurnkeyAuth = () => {
     isLoginPending,
     isUserLoading,
     refreshAuthStatus: refetch,
-    btcLogin,
   };
 };