/
session.ts
54 lines (48 loc) · 1.37 KB
/
session.ts
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
import { useCallback, useEffect } from 'react';
import { useQuery } from 'react-query';
import { queryClient } from '../../masa-query-client';
import { Masa } from '@masa-finance/masa-sdk';
export const useSession = (
masa: Masa | null,
walletAddress: string | undefined
): {
loggedIn?: boolean;
login: () => void;
logout: () => void;
status: string;
isLoading: boolean;
error: unknown;
} => {
const {
data: loggedIn,
status,
isLoading,
error,
} = useQuery(`session-${walletAddress}`, () => masa?.session.checkLogin(), {
enabled: !!masa && !!walletAddress,
});
useEffect(() => {
if (loggedIn && walletAddress) {
void queryClient.invalidateQueries(`session-${walletAddress}`);
}
}, [walletAddress, loggedIn]);
const login = useCallback(async () => {
const logged = await masa?.session.login();
if (logged) {
await queryClient.invalidateQueries(`session-${walletAddress}`);
await queryClient.refetchQueries();
}
}, [masa, walletAddress]);
const logout = useCallback(
async (callback?: () => void) => {
await masa?.session.logout();
await queryClient.invalidateQueries(`session-${walletAddress}`);
await queryClient.refetchQueries();
if (callback) {
callback();
}
},
[masa, walletAddress]
);
return { loggedIn, login, logout, status, isLoading, error };
};