Add a session_delete handler

safe-global · Sep 21, 2023 · 3303802 · 3303802
1 parent f05fc0b
commit 3303802
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 7 deletions.
diff --git a/src/components/walletconnect/SessionManager/index.tsx b/src/components/walletconnect/SessionManager/index.tsx
@@ -14,14 +14,19 @@ const SessionManager = () => {
   const { safe, safeAddress } = useSafeInfo()
   const { chainId } = safe
   const { walletConnect, error: walletConnectError } = useContext(WalletConnectContext)
-  const [sessions, setSessions] = useState<Record<string, SessionTypes.Struct>>(walletConnect.getActiveSessions() || {})
+  const [sessions, setSessions] = useState<Record<string, SessionTypes.Struct>>(walletConnect.getActiveSessions())
   const [proposal, setProposal] = useState<Web3WalletTypes.SessionProposal>()
   const [error, setError] = useState<Error>()
 
   // Subscribe to session proposals
   useEffect(() => {
-    return walletConnect.addOnSessionPropose(async (event) => {
-      setProposal(event)
+    return walletConnect.onSessionPropose(setProposal)
+  }, [walletConnect])
+
+  // Subscribe to session deletes
+  useEffect(() => {
+    return walletConnect.onSessionDelete(() => {
+      setSessions(walletConnect.getActiveSessions())
     })
   }, [walletConnect])
 
@@ -37,8 +42,7 @@ const SessionManager = () => {
     }
 
     setProposal(undefined)
-    // Update sessions
-    setSessions(walletConnect.getActiveSessions() || {})
+    setSessions(walletConnect.getActiveSessions())
   }, [proposal, walletConnect])
 
   // On session reject
@@ -59,7 +63,6 @@ const SessionManager = () => {
   const onDisconnect = async (session: SessionTypes.Struct) => {
     try {
       await walletConnect.disconnectSession(session)
-      setSessions(walletConnect.getActiveSessions() || {})
     } catch (error) {
       setError(asError(error))
     }

diff --git a/src/services/walletconnect/WalletConnectWallet.ts b/src/services/walletconnect/WalletConnectWallet.ts
@@ -144,7 +144,7 @@ class WalletConnectWallet {
   /**
    * Subscribe to session proposals
    */
-  public addOnSessionPropose(onSessionPropose: (e: Web3WalletTypes.SessionProposal) => void) {
+  public onSessionPropose(onSessionPropose: (e: Web3WalletTypes.SessionProposal) => void) {
     // Subscribe to the session proposal event
     this.web3Wallet?.on('session_proposal', onSessionPropose)
 
@@ -154,6 +154,17 @@ class WalletConnectWallet {
     }
   }
 
+  /**
+   * Subscribe to session delete
+   */
+  public onSessionDelete(handler: () => void) {
+    this.web3Wallet?.on('session_delete', handler)
+
+    return () => {
+      this.web3Wallet?.off('session_delete', handler)
+    }
+  }
+
   /**
    * Disconnect a session
    */