/
masa-client-provider.tsx
86 lines (77 loc) · 2.15 KB
/
masa-client-provider.tsx
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
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
import React, { ReactNode, createContext, useContext, useMemo } from 'react';
import { useAsync } from 'react-use';
import { useMasaClient } from './masa-client/use-masa-client';
import { useIdentity } from './masa/use-identity';
import { useWallet } from './wallet-client/wallet/use-wallet';
import { useSession } from './masa/use-session';
export interface MasaClientProviderValue {
masa?: ReturnType<typeof useMasaClient>['sdk'];
session?: ReturnType<typeof useSession>['session'];
identity?: ReturnType<typeof useIdentity>['identity'];
}
export const MasaClientContext = createContext({} as MasaClientProviderValue);
export const MasaClientProvider = ({ children }: { children: ReactNode }) => {
const { masa, masaAddress } = useMasaClient();
const {
session,
isLoadingSession,
hasSession,
loginSession,
logoutSession,
sessionAddress,
} = useSession();
const { isDisconnected } = useWallet();
const { identity } = useIdentity();
// * useEffect to handle account switches and disconnect
useAsync(async () => {
if (isLoadingSession) return;
if (
session &&
masaAddress &&
masaAddress === session?.user.address &&
hasSession
) {
return;
}
if (isDisconnected) {
await logoutSession();
return;
}
if (
hasSession &&
sessionAddress &&
masaAddress &&
sessionAddress !== masaAddress
) {
await logoutSession();
}
}, [
isLoadingSession,
sessionAddress,
masaAddress,
isDisconnected,
logoutSession,
hasSession,
session,
]);
const masaClientProviderValue: MasaClientProviderValue = useMemo(
() =>
({
masa,
session,
sessionAddress,
loginSession,
logoutSession,
identity,
} as MasaClientProviderValue),
[masa, session, identity, sessionAddress, loginSession, logoutSession]
);
return (
<MasaClientContext.Provider value={masaClientProviderValue}>
{children}
</MasaClientContext.Provider>
);
};
export const useMasaClientProvider = (): MasaClientProviderValue =>
useContext(MasaClientContext);
export default MasaClientProvider;