|
| 1 | +// REACT IMPORTS |
| 2 | +import { useEffect, useCallback } from 'react'; |
| 3 | + |
| 4 | +// REDUX IMPORTS |
| 5 | +import { useDispatch, useSelector } from 'react-redux'; |
| 6 | +import { setAuthLoading, setSession, setBalance } from '@redux/auth'; |
| 7 | + |
| 8 | +// LENS IMPORTS |
| 9 | +import { useSession, useLogout } from '@lens-protocol/react-web'; |
| 10 | + |
| 11 | +// NOTIFICATIONS IMPORTS |
| 12 | +import { ERRORS } from '@notifications/errors'; |
| 13 | +import { notifyError } from '@notifications/internal-notifications'; |
| 14 | + |
| 15 | +// WEB3AUTH IMPORTS |
| 16 | +import { useWeb3Auth } from '@src/hooks/use-web3-auth'; |
| 17 | +import { useWeb3Session } from '@src/hooks/use-web3-session'; |
| 18 | + |
| 19 | +// ---------------------------------------------------------------------- |
| 20 | + |
| 21 | +interface UseAccountSessionProps { |
| 22 | + /** |
| 23 | + * Whether to automatically run the expiration checks |
| 24 | + * on mount + every interval (default: true). |
| 25 | + */ |
| 26 | + autoCheck?: boolean; |
| 27 | +} |
| 28 | + |
| 29 | +interface UseAccountSessionHook { |
| 30 | + /** |
| 31 | + * Combined logout for Lens + Web3Auth |
| 32 | + */ |
| 33 | + logout: () => Promise<void>; |
| 34 | + |
| 35 | + /** |
| 36 | + * Check session validity on demand: |
| 37 | + * - Verifies Web3Auth is fully connected (or still connecting) |
| 38 | + * - Verifies localStorage expiration |
| 39 | + * - Logs out + notifies error if invalid |
| 40 | + */ |
| 41 | + checkSessionValidity: () => void; |
| 42 | +} |
| 43 | + |
| 44 | +// ---------------------------------------------------------------------- |
| 45 | + |
| 46 | +/** |
| 47 | + * This hook consolidates: |
| 48 | + * 1. Lens session fetching & Redux updates. |
| 49 | + * 2. Web3Auth session validation (connected, bundler, smartAccount). |
| 50 | + * 3. LocalStorage expiration checks + auto-logout (optionally). |
| 51 | + * |
| 52 | + * If `autoCheck` is false, the hook won't run the checks automatically, |
| 53 | + * but you can still call `checkSessionValidity()` manually. |
| 54 | + */ |
| 55 | +export const useAccountSession = ( |
| 56 | + { autoCheck = true }: UseAccountSessionProps = {} |
| 57 | +): UseAccountSessionHook => { |
| 58 | + const dispatch = useDispatch(); |
| 59 | + const { execute: lensLogout } = useLogout(); |
| 60 | + const { web3Auth } = useWeb3Auth(); |
| 61 | + const { data, loading } = useSession(); |
| 62 | + const isUpdatingMetadata: boolean = useSelector( |
| 63 | + (state: any) => state.auth.isUpdatingMetadata |
| 64 | + ); |
| 65 | + const { bundlerClient, smartAccount } = useWeb3Session(); |
| 66 | + |
| 67 | + const parsedSessionConnected = JSON.stringify(web3Auth.connected); |
| 68 | + const parsedSessionData = JSON.stringify(data); |
| 69 | + |
| 70 | + // Keep Redux in sync with Lens loading state |
| 71 | + useEffect(() => { |
| 72 | + // If autoCheck is disabled, skip |
| 73 | + if (!autoCheck) return; |
| 74 | + dispatch(setAuthLoading({ isSessionLoading: loading })); |
| 75 | + }, [loading, autoCheck]); |
| 76 | + |
| 77 | + // Keep Redux in sync with actual Lens session data |
| 78 | + useEffect(() => { |
| 79 | + // If autoCheck is disabled, skip |
| 80 | + if (!autoCheck) return; |
| 81 | + if (!isUpdatingMetadata) { |
| 82 | + dispatch(setSession({ session: data })); |
| 83 | + } |
| 84 | + }, [parsedSessionData, isUpdatingMetadata, autoCheck]); |
| 85 | + |
| 86 | + // LOGOUT (Lens + Web3Auth) |
| 87 | + const logout = useCallback(async () => { |
| 88 | + try { |
| 89 | + // 1) Logout from Lens |
| 90 | + await lensLogout(); |
| 91 | + // 2) Logout from Web3Auth |
| 92 | + await web3Auth?.logout(); |
| 93 | + // 3) Clear Redux state & localStorage |
| 94 | + dispatch(setBalance({ balance: 0 })); |
| 95 | + localStorage.removeItem('sessionExpiration'); |
| 96 | + } catch (err) { |
| 97 | + console.error('Error during logout:', err); |
| 98 | + localStorage.removeItem('sessionExpiration'); |
| 99 | + } |
| 100 | + }, [lensLogout, web3Auth, dispatch]); |
| 101 | + |
| 102 | + // Decide if Web3Auth is in a valid/connecting state |
| 103 | + const isValidWeb3AuthSession = useCallback((): boolean => { |
| 104 | + const isConnecting = |
| 105 | + web3Auth.status === 'connecting' || |
| 106 | + web3Auth.status === 'not_ready'; |
| 107 | + |
| 108 | + const isFullyValid = |
| 109 | + web3Auth.connected && |
| 110 | + web3Auth.status === 'connected' && |
| 111 | + !!bundlerClient && |
| 112 | + !!smartAccount; |
| 113 | + |
| 114 | + console.log('isValidWeb3AuthSession') |
| 115 | + console.log(isConnecting) |
| 116 | + console.log(isFullyValid) |
| 117 | + console.log(isConnecting || isFullyValid) |
| 118 | + |
| 119 | + // Return true if either "still connecting" or "fully valid" |
| 120 | + return isConnecting || isFullyValid; |
| 121 | + }, [web3Auth.connected, web3Auth.status, bundlerClient, smartAccount]); |
| 122 | + |
| 123 | + // If session is invalid or expired, do logout + show error |
| 124 | + const handleSessionExpired = useCallback(() => { |
| 125 | + logout(); |
| 126 | + notifyError(ERRORS.BUNDLER_UNAVAILABLE); |
| 127 | + }, [logout]); |
| 128 | + |
| 129 | + const checkSessionValidity = useCallback(() => { |
| 130 | + console.log('checkSessionValidity') |
| 131 | + // 1) If Web3Auth isn't valid (and not just connecting), expire |
| 132 | + if (!isValidWeb3AuthSession()) { |
| 133 | + handleSessionExpired(); |
| 134 | + return; |
| 135 | + } |
| 136 | + |
| 137 | + // 2) Otherwise, check localStorage for expiration |
| 138 | + const expirationStr = localStorage.getItem('sessionExpiration'); |
| 139 | + if (!expirationStr) return; |
| 140 | + |
| 141 | + const expirationTime = parseInt(expirationStr, 10); |
| 142 | + if (Date.now() >= expirationTime) { |
| 143 | + handleSessionExpired(); |
| 144 | + } |
| 145 | + }, [isValidWeb3AuthSession, handleSessionExpired]); |
| 146 | + |
| 147 | + // Automatic checks on mount + interval |
| 148 | + useEffect(() => { |
| 149 | + // If autoCheck is disabled, skip |
| 150 | + if (!autoCheck) return; |
| 151 | + |
| 152 | + // If Lens or Web3Auth is still loading, skip checks until loaded |
| 153 | + if (loading || web3Auth.status === 'connecting' || web3Auth.status === 'not_ready') return; |
| 154 | + |
| 155 | + // If user is not authenticated in Lens, skip |
| 156 | + if (!data?.authenticated) { |
| 157 | + return; |
| 158 | + } |
| 159 | + |
| 160 | + // If Web3Auth is in "ready" state but not connected, log out |
| 161 | + // (meaning: it's done with any connecting state, but the user is not actually connected) |
| 162 | + if (!web3Auth.connected && web3Auth.status === 'ready') { |
| 163 | + logout(); |
| 164 | + return; |
| 165 | + } |
| 166 | + |
| 167 | + // Check once immediately |
| 168 | + checkSessionValidity(); |
| 169 | + |
| 170 | + // Then check every 60 seconds |
| 171 | + const intervalId = setInterval(() => { |
| 172 | + checkSessionValidity(); |
| 173 | + }, 60 * 1000); |
| 174 | + |
| 175 | + // Cleanup |
| 176 | + return () => clearInterval(intervalId); |
| 177 | + }, [ |
| 178 | + autoCheck, |
| 179 | + loading, |
| 180 | + parsedSessionConnected, |
| 181 | + parsedSessionData |
| 182 | + ]); |
| 183 | + |
| 184 | + return { |
| 185 | + logout, |
| 186 | + checkSessionValidity, |
| 187 | + }; |
| 188 | +}; |
0 commit comments