fix remote wallet terminate API proxy impl

solana-mobile · Nov 12, 2024 · 0ebe3e7 · 0ebe3e7
1 parent 656983e
commit 0ebe3e7
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/js/packages/mobile-wallet-adapter-protocol/src/transact.ts b/js/packages/mobile-wallet-adapter-protocol/src/transact.ts
@@ -455,11 +455,14 @@ export async function transactRemote<TReturn>(
                     try {
                         resolve(await callback(new Proxy(wallet as RemoteMobileWallet, {
                             get<TMethodName extends keyof RemoteMobileWallet>(target: RemoteMobileWallet, p: TMethodName) {
-                                if (p === 'terminateSession') {
-                                    disposeSocket();
-                                    socket.close();
-                                    return;
-                                } else return target[p]
+                                if (p === 'terminateSession' && target[p] == null) {
+                                    target['terminateSession'] = async function () {
+                                        disposeSocket();
+                                        socket.close();
+                                        return;
+                                    };
+                                }
+                                return target[p];
                             },
                         })))
                     } catch (e) {