blob: f91495d24cb34adf158a69cce9801d766948fc19 (
plain) (
blame)
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
|
import React from "react"
const EMPTY: unique symbol = Symbol()
export interface ContainerProviderProps<State = void> {
initialState?: State
children: React.ReactNode
}
export interface Container<Value, State = void> {
Provider: React.ComponentType<ContainerProviderProps<State>>
useContainer: () => Value
}
export function createContainer<Value, State = void>(
useHook: (initialState?: State) => Value,
): Container<Value, State> {
let Context = React.createContext<Value | typeof EMPTY>(EMPTY)
function Provider(props: ContainerProviderProps<State>) {
let value = useHook(props.initialState)
return <Context.Provider value={value}>{props.children}</Context.Provider>
}
function useContainer(): Value {
let value = React.useContext(Context)
if (value === EMPTY) {
throw new Error("Component must be wrapped with <Container.Provider>")
}
// @ts-ignore
return value
}
return { Provider, useContainer }
}
export function useContainer<Value, State = void>(
container: Container<Value, State>,
): Value {
return container.useContainer()
}
|
