fix: update disconnect method (#76)

babylonlabs-io · Dec 1, 2024 · 1717074 · 1717074
1 parent c241d94
commit 1717074
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 3 deletions.
diff --git a/.changeset/lovely-hotels-lay.md b/.changeset/lovely-hotels-lay.md
@@ -0,0 +1,5 @@
+---
+"@babylonlabs-io/bbn-wallet-connect": patch
+---
+
+update disconnect method
diff --git a/src/hooks/useWalletConnect.ts b/src/hooks/useWalletConnect.ts
@@ -1,17 +1,26 @@
 import { useCallback, useMemo } from "react";
+
+import { useChainProviders } from "@/context/Chain.context";
 import { useWidgetState } from "./useWidgetState";
 
 export function useWalletConnect() {
-  const { confirmed, chains: chainMap, selectedWallets, open: openModal, close, reset } = useWidgetState();
+  const { confirmed, chains: chainMap, selectedWallets, open: openModal, reset } = useWidgetState();
+  const connectors = useChainProviders();
 
   const open = useCallback(() => {
     reset?.();
     openModal?.();
   }, [openModal, reset]);
 
-  const disconnect = useCallback(() => {
+  const disconnect = useCallback(async () => {
+    for (const connector of Object.values(connectors)) {
+      if (!connector) continue;
+
+      await connector.disconnect();
+    }
+
     reset?.();
-  }, [close]);
+  }, []);
 
   const selected = useMemo(() => {
     const chains = Object.values(chainMap).filter(Boolean);