1 | import * as React from 'react';
|
2 | import useNavigation from './useNavigation';
|
3 |
|
4 |
|
5 |
|
6 |
|
7 |
|
8 |
|
9 | export default function useNavigationState(selector) {
|
10 | const navigation = useNavigation();
|
11 |
|
12 |
|
13 | const [, setResult] = React.useState(() => selector(navigation.getState()));
|
14 |
|
15 | const selectorRef = React.useRef(selector);
|
16 | React.useEffect(() => {
|
17 | selectorRef.current = selector;
|
18 | });
|
19 | React.useEffect(() => {
|
20 | const unsubscribe = navigation.addListener('state', e => {
|
21 | setResult(selectorRef.current(e.data.state));
|
22 | });
|
23 | return unsubscribe;
|
24 | }, [navigation]);
|
25 | return selector(navigation.getState());
|
26 | }
|
27 |
|
\ | No newline at end of file |