blob: d4b20daeb12003f896a968597ca4ff84faef7655 (
plain)
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
|
import React, { useContext } from 'react';
import App from './app';
export interface IAppContext {
app: App;
}
export const AppContext = React.createContext<IAppContext | undefined>(undefined);
if (window.runningInDevelopment) {
AppContext.displayName = 'AppContext';
}
const missingContextError = new Error(
'The context value is empty. Make sure to wrap the component in AppContext.Provider.',
);
export default function withAppContext<Props>(BaseComponent: React.ComponentType<Props>) {
// Exclude the IAppContext from props since those are injected props
const wrappedComponent = (props: Omit<Props, keyof IAppContext>) => {
return (
<AppContext.Consumer>
{(context) => {
if (context) {
// Enforce type because Typescript does not recognize that
// (Props ~ IAppContext & IAppContext) is identical to Props.
const mergedProps = ({ ...props, ...context } as unknown) as Props;
return <BaseComponent {...mergedProps} />;
} else {
throw missingContextError;
}
}}
</AppContext.Consumer>
);
};
if (window.runningInDevelopment) {
wrappedComponent.displayName =
'withAppContext(' + (BaseComponent.displayName || BaseComponent.name) + ')';
}
return wrappedComponent;
}
export function useAppContext(): App {
const appContext = useContext(AppContext);
if (appContext) {
return appContext.app;
} else {
throw missingContextError;
}
}
|
