fix(ui-react): set current modal state on first call of use effect in…

… useTonConnectModal
ton-connect · Apr 21, 2024 · a881c7f · a881c7f
1 parent 3f2954a
commit a881c7f
Showing 1 changed file with 4 additions and 11 deletions.
diff --git a/packages/ui-react/src/hooks/useTonConnectModal.ts b/packages/ui-react/src/hooks/useTonConnectModal.ts
@@ -11,23 +11,16 @@ export function useTonConnectModal(): Omit<WalletsModal, 'onStateChange'> {
 
     useEffect(() => {
         if (tonConnectUI) {
+            setState(tonConnectUI.modal.state);
             return tonConnectUI.onModalStateChange((value: WalletsModalState) => {
                 setState(value);
             });
         }
     }, [tonConnectUI]);
 
     return {
-        state,
-        open: () => {
-            if (tonConnectUI) {
-                return tonConnectUI.modal.open();
-            }
-        },
-        close: () => {
-            if (tonConnectUI) {
-                tonConnectUI.modal.close();
-            }
-        }
+        state: state,
+        open: () => tonConnectUI?.modal.open(),
+        close: () => tonConnectUI?.modal.close()
     };
 }