/
session.ts
58 lines (51 loc) · 1.42 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
55
56
57
58
import { useCallback, useEffect, useMemo } 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 queryKey: any[] = useMemo(() => {
return ['session', walletAddress];
}, [walletAddress]);
const {
data: loggedIn,
status,
isLoading,
error,
} = useQuery(queryKey, () => masa?.session.checkLogin(), {
enabled: !!masa && !!walletAddress
});
useEffect(() => {
if (loggedIn && !walletAddress) {
void queryClient.invalidateQueries('session');
}
}, [walletAddress, loggedIn, queryKey]);
const login = useCallback(async () => {
const logged = await masa?.session.login();
if (logged) {
await queryClient.invalidateQueries('session');
await queryClient.refetchQueries();
}
}, [masa, queryKey]);
const logout = useCallback(
async (callback?: () => void) => {
await masa?.session.logout();
await queryClient.invalidateQueries('session');
await queryClient.refetchQueries();
if (callback) {
callback();
}
},
[masa, queryKey]
);
return { loggedIn, login, logout, status, isLoading, error };
};