feat: Hide disconnect button in account modal (#99)

* hide disconnect modal prop * format * changeset * fix * actual fix
TreasureProject · Aug 22, 2024 · d4d5e77 · d4d5e77
1 parent 62fdaa6
commit d4d5e77
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 3 deletions.
diff --git a/.changeset/spicy-rockets-build.md b/.changeset/spicy-rockets-build.md
@@ -0,0 +1,5 @@
+---
+"@treasure-dev/tdk-react": patch
+---
+
+Added the hideDisconnect prop
diff --git a/examples/connect-electron/src/renderer/src/App.tsx b/examples/connect-electron/src/renderer/src/App.tsx
@@ -16,7 +16,10 @@ function App(): JSX.Element {
   return (
     <>
       <p>
-        <ConnectButton redirectUrl={`${window.location.origin}/auth/`} />
+        <ConnectButton
+          redirectUrl={`${window.location.origin}/auth/`}
+          hideDisconnect={true}
+        />
       </p>
       <Versions />
     </>

diff --git a/packages/react/src/hooks/useConnect.tsx b/packages/react/src/hooks/useConnect.tsx
@@ -17,6 +17,7 @@ import { getLocaleId } from "../i18n";
 export type Options = ConnectModalOptions & {
   supportedChainIds?: number[];
   connectModalSize?: ConnectModalProps["size"];
+  hideDisconnect?: boolean;
 };
 
 type Props = Options;
@@ -61,8 +62,12 @@ const SUPPORTED_TOKENS = [
 export const useConnect = (props?: Props) => {
   const { chain, client, logOut, setRootElement } = useTreasure();
   const { open: openWalletDetailsModal } = useWalletDetailsModal();
-  const { supportedChainIds, connectModalSize, ...connectModalProps } =
-    props ?? {};
+  const {
+    supportedChainIds,
+    connectModalSize,
+    hideDisconnect,
+    ...connectModalProps
+  } = props ?? {};
 
   const chains =
     supportedChainIds && supportedChainIds.length > 0
@@ -93,6 +98,7 @@ export const useConnect = (props?: Props) => {
         })).filter(({ address }) => address !== ZERO_ADDRESS);
         return acc;
       }, {} as SupportedTokens),
+      hideDisconnect,
       onDisconnect: () => {
         logOut();
       },