withCachedChildNavigation.js
1.84 KB
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
/* @flow */
import * as React from 'react';
import addNavigationHelpers from './addNavigationHelpers';
import type {
NavigationScreenProp,
NavigationState,
NavigationRoute,
} from './TypeDefinition';
type InputProps<S: NavigationState> = {
navigation: NavigationScreenProp<S>,
};
type OutputProps = {
childNavigationProps: {
+[key: string]: NavigationScreenProp<NavigationRoute>,
},
};
/**
* HOC which caches the child navigation items.
*/
export default function withCachedChildNavigation<
S: NavigationState,
T: InputProps<S>
>(Comp: React.ComponentType<OutputProps & T>): React.ComponentType<T> {
const displayName: string = Comp.displayName || Comp.name;
return class extends React.PureComponent<T> {
static displayName = `withCachedChildNavigation(${displayName})`;
componentWillMount() {
this._updateNavigationProps(this.props.navigation);
}
componentWillReceiveProps(nextProps: T) {
this._updateNavigationProps(nextProps.navigation);
}
_childNavigationProps: {
[key: string]: NavigationScreenProp<NavigationRoute>,
};
_updateNavigationProps = (
navigation: NavigationScreenProp<NavigationState>
) => {
// Update props for each child route
if (!this._childNavigationProps) {
this._childNavigationProps = {};
}
navigation.state.routes.forEach((route: *) => {
const childNavigation = this._childNavigationProps[route.key];
if (childNavigation && childNavigation.state === route) {
return;
}
this._childNavigationProps[route.key] = addNavigationHelpers({
dispatch: navigation.dispatch,
state: route,
});
});
};
render() {
return (
<Comp
{...this.props}
childNavigationProps={this._childNavigationProps}
/>
);
}
};
}