|
1 | | -import React, { PropsWithChildren, useState } from 'react'; |
| 1 | +import React, { PropsWithChildren, useEffect, useState } from 'react'; |
2 | 2 |
|
3 | | -import { Group } from '../../api'; |
4 | | -import { SelectedGroupContext } from '../../constants'; |
| 3 | +import { Group, useGetInitialGroup } from '../../api'; |
| 4 | +import { SELECTED_GROUP_ID_LOCAL_STORAGE_KEY, SelectedGroupContext } from '../../constants'; |
5 | 5 |
|
6 | 6 | export const SelectedGroupProvider = ({ children }: PropsWithChildren) => { |
| 7 | + const { data: initialGroup } = useGetInitialGroup({ |
| 8 | + id: Number(localStorage.getItem(SELECTED_GROUP_ID_LOCAL_STORAGE_KEY)), |
| 9 | + }); |
| 10 | + |
7 | 11 | const [selectedGroup, setSelectedGroup] = useState<Group | null>(null); |
8 | 12 |
|
| 13 | + useEffect(() => { |
| 14 | + if (initialGroup) { |
| 15 | + setSelectedGroup(initialGroup); |
| 16 | + } |
| 17 | + }, [initialGroup]); |
| 18 | + |
| 19 | + const handleSelectGroup = (group: Group) => { |
| 20 | + setSelectedGroup(group); |
| 21 | + localStorage.setItem(SELECTED_GROUP_ID_LOCAL_STORAGE_KEY, group.data.id.toString()); |
| 22 | + }; |
| 23 | + |
| 24 | + const handleCleanGroup = () => { |
| 25 | + setSelectedGroup(null); |
| 26 | + localStorage.removeItem(SELECTED_GROUP_ID_LOCAL_STORAGE_KEY); |
| 27 | + }; |
| 28 | + |
9 | 29 | const contextValue = { |
10 | 30 | group: selectedGroup, |
11 | | - selectGroup: setSelectedGroup, |
12 | | - cleanGroup: () => setSelectedGroup(null), |
| 31 | + selectGroup: handleSelectGroup, |
| 32 | + cleanGroup: handleCleanGroup, |
13 | 33 | }; |
14 | 34 |
|
15 | 35 | return <SelectedGroupContext.Provider value={contextValue}>{children}</SelectedGroupContext.Provider>; |
|
0 commit comments