Make context typing more accurate

EskiMojo14 · EskiMojo14 · commit 62728256051f · 2023-06-21T23:04:06.000+01:00
diff --git a/docs/api/Provider.md b/docs/api/Provider.md
@@ -39,9 +39,10 @@ interface ProviderProps<A extends Action = AnyAction, S = any> {
    * to create a context to be used.
    * If this is used, you'll need to customize `connect` by supplying the same
    * context provided to the Provider.
-   * Initial value doesn't matter, as it is overwritten with the internal state of Provider.
+   * Set the initial value to null, and the hooks will error
+   * if this is not overwritten by Provider.
    */
-  context?: Context<ReactReduxContextValue<S, A>>
+  context?: Context<ReactReduxContextValue<S, A> | null>
 
   /** Global configuration for the `useSelector` stability check */
   stabilityCheck?: StabilityCheck
diff --git a/docs/using-react-redux/accessing-store.md b/docs/using-react-redux/accessing-store.md
@@ -27,7 +27,7 @@ Redux store accessible to deeply nested connected components. As of React Redux
 by a single default context object instance generated by `React.createContext()`, called `ReactReduxContext`.
 
 React Redux's `<Provider>` component uses `<ReactReduxContext.Provider>` to put the Redux store and the current store
-state into context, and `connect` uses `<ReactReduxContext.Consumer>` to read those values and handle updates.
+state into context, and `connect` uses `useContext(ReactReduxContext)` to read those values and handle updates.
 
 ## Using the `useStore` Hook
 
@@ -87,8 +87,8 @@ This also provides a natural isolation of the stores as they live in separate co
 
 ```js
 // a naive example
-const ContextA = React.createContext();
-const ContextB = React.createContext();
+const ContextA = React.createContext(null);
+const ContextB = React.createContext(null);
 
 // assuming reducerA and reducerB are proper reducer functions
 const storeA = createStore(reducerA);
diff --git a/src/components/Context.ts b/src/components/Context.ts
@@ -16,12 +16,14 @@ export interface ReactReduxContextValue<
 }
 
 const ContextKey = Symbol.for(`react-redux-context-${ReactVersion}`)
-const gT = globalThis as { [ContextKey]?: Context<ReactReduxContextValue> }
+const gT = globalThis as {
+  [ContextKey]?: Context<ReactReduxContextValue | null>
+}
 
 function getContext() {
   let realContext = gT[ContextKey]
   if (!realContext) {
-    realContext = createContext<ReactReduxContextValue>(null as any)
+    realContext = createContext<ReactReduxContextValue | null>(null)
     if (process.env.NODE_ENV !== 'production') {
       realContext.displayName = 'ReactRedux'
     }
@@ -31,8 +33,8 @@ function getContext() {
 }
 
 export const ReactReduxContext = /*#__PURE__*/ new Proxy(
-  {} as Context<ReactReduxContextValue>,
-  /*#__PURE__*/ new Proxy<ProxyHandler<Context<ReactReduxContextValue>>>(
+  {} as Context<ReactReduxContextValue | null>,
+  /*#__PURE__*/ new Proxy<ProxyHandler<Context<ReactReduxContextValue | null>>>(
     {},
     {
       get(_, handler) {
diff --git a/src/components/Provider.tsx b/src/components/Provider.tsx
@@ -21,9 +21,10 @@ export interface ProviderProps<A extends Action = AnyAction, S = unknown> {
   /**
    * Optional context to be used internally in react-redux. Use React.createContext() to create a context to be used.
    * If this is used, you'll need to customize `connect` by supplying the same context provided to the Provider.
-   * Initial value doesn't matter, as it is overwritten with the internal state of Provider.
+   * Set the initial value to null, and the hooks will error
+   * if this is not overwritten by Provider.
    */
-  context?: Context<ReactReduxContextValue<S, A>>
+  context?: Context<ReactReduxContextValue<S, A> | null>
 
   /** Global configuration for the `useSelector` stability check */
   stabilityCheck?: CheckFrequency
diff --git a/src/components/connect.tsx b/src/components/connect.tsx
@@ -584,7 +584,7 @@ function connect<
         : contextValue!.store
 
       const getServerState = didStoreComeFromContext
-        ? contextValue.getServerState
+        ? contextValue!.getServerState
         : store.getState
 
       const childPropsSelector = useMemo(() => {
diff --git a/src/hooks/useDispatch.ts b/src/hooks/useDispatch.ts
@@ -15,7 +15,7 @@ export function createDispatchHook<
   S = unknown,
   A extends Action = AnyAction
   // @ts-ignore
->(context?: Context<ReactReduxContextValue<S, A>> = ReactReduxContext) {
+>(context?: Context<ReactReduxContextValue<S, A> | null> = ReactReduxContext) {
   const useStore =
     // @ts-ignore
     context === ReactReduxContext ? useDefaultStore : createStoreHook(context)
diff --git a/src/hooks/useReduxContext.ts b/src/hooks/useReduxContext.ts
@@ -10,7 +10,7 @@ import type { ReactReduxContextValue } from '../components/Context'
  * @returns {Function} A `useReduxContext` hook bound to the specified context.
  */
 export function createReduxContextHook(context = ReactReduxContext) {
-  return function useReduxContext(): ReactReduxContextValue | null {
+  return function useReduxContext(): ReactReduxContextValue {
     const contextValue = useContext(context)
 
     if (process.env.NODE_ENV !== 'production' && !contextValue) {
@@ -19,7 +19,7 @@ export function createReduxContextHook(context = ReactReduxContext) {
       )
     }
 
-    return contextValue
+    return contextValue!
   }
 }
 
diff --git a/src/hooks/useSelector.ts b/src/hooks/useSelector.ts
@@ -80,7 +80,7 @@ export function createSelectorHook(context = ReactReduxContext): UseSelector {
       getServerState,
       stabilityCheck: globalStabilityCheck,
       noopCheck: globalNoopCheck,
-    } = useReduxContext()!
+    } = useReduxContext()
 
     const firstRun = useRef(true)
 
diff --git a/src/hooks/useStore.ts b/src/hooks/useStore.ts
@@ -17,7 +17,7 @@ export function createStoreHook<
   S = unknown,
   A extends BasicAction = AnyAction
   // @ts-ignore
->(context?: Context<ReactReduxContextValue<S, A>> = ReactReduxContext) {
+>(context?: Context<ReactReduxContextValue<S, A> | null> = ReactReduxContext) {
   const useReduxContext =
     // @ts-ignore
     context === ReactReduxContext
@@ -29,7 +29,7 @@ export function createStoreHook<
     Action extends BasicAction = A
     // @ts-ignore
   >() {
-    const { store } = useReduxContext()!
+    const { store } = useReduxContext()
     // @ts-ignore
     return store as Store<State, Action>
   }
diff --git a/test/components/connect.spec.tsx b/test/components/connect.spec.tsx
@@ -2130,9 +2130,10 @@ describe('React', () => {
           }
         }
 
-        const context = React.createContext<
-          ReactReduxContextValue<any, AnyAction>
-        >(null as any)
+        const context = React.createContext<ReactReduxContextValue<
+          any,
+          AnyAction
+        > | null>(null)
 
         let actualState
 
@@ -2171,9 +2172,10 @@ describe('React', () => {
           }
         }
 
-        const context = React.createContext<
-          ReactReduxContextValue<any, AnyAction>
-        >(null as any)
+        const context = React.createContext<ReactReduxContextValue<
+          any,
+          AnyAction
+        > | null>(null)
 
         let actualState
 
@@ -2425,9 +2427,8 @@ describe('React', () => {
           (state: RootStateType = 0, action: ActionType) =>
             action.type === 'INC' ? state + 1 : state
         )
-        const customContext = React.createContext<ReactReduxContextValue>(
-          null as any
-        )
+        const customContext =
+          React.createContext<ReactReduxContextValue | null>(null)
 
         class A extends Component {
           render() {
diff --git a/test/hooks/useDispatch.spec.tsx b/test/hooks/useDispatch.spec.tsx
@@ -27,9 +27,8 @@ describe('React', () => {
     })
     describe('createDispatchHook', () => {
       it("returns the correct store's dispatch function", () => {
-        const nestedContext = React.createContext<ReactReduxContextValue>(
-          null as any
-        )
+        const nestedContext =
+          React.createContext<ReactReduxContextValue | null>(null)
         const useCustomDispatch = createDispatchHook(nestedContext)
         const { result } = renderHook(() => useDispatch(), {
           // eslint-disable-next-line react/prop-types
diff --git a/test/hooks/useReduxContext.spec.tsx b/test/hooks/useReduxContext.spec.tsx
@@ -1,6 +1,6 @@
 import { renderHook } from '@testing-library/react-hooks'
 import { createContext } from 'react'
-import { ReactReduxContextValue } from '../../src/components/Context'
+import type { ReactReduxContextValue } from '../../src/components/Context'
 import {
   createReduxContextHook,
   useReduxContext,
@@ -23,7 +23,7 @@ describe('React', () => {
     })
     describe('createReduxContextHook', () => {
       it('throws if component is not wrapped in provider', () => {
-        const customContext = createContext<ReactReduxContextValue>(null as any)
+        const customContext = createContext<ReactReduxContextValue | null>(null)
         const useCustomReduxContext = createReduxContextHook(customContext)
         const spy = jest.spyOn(console, 'error').mockImplementation(() => {})
 
diff --git a/test/hooks/useSelector.spec.tsx b/test/hooks/useSelector.spec.tsx
@@ -153,8 +153,8 @@ describe('React', () => {
           }
 
           const Parent = () => {
-            const { subscription } = useContext(ReactReduxContext)
-            appSubscription = subscription
+            const contextVal = useContext(ReactReduxContext)
+            appSubscription = contextVal && contextVal.subscription
             const count = useNormalSelector((s) => s.count)
             return count === 1 ? <Child /> : null
           }
@@ -179,8 +179,8 @@ describe('React', () => {
           let appSubscription: Subscription | null = null
 
           const Parent = () => {
-            const { subscription } = useContext(ReactReduxContext)
-            appSubscription = subscription
+            const contextVal = useContext(ReactReduxContext)
+            appSubscription = contextVal && contextVal.subscription
             const count = useNormalSelector((s) => s.count)
             return count === 0 ? <Child /> : null
           }
@@ -944,9 +944,8 @@ describe('React', () => {
       })
 
       it('subscribes to the correct store', () => {
-        const nestedContext = React.createContext<ReactReduxContextValue>(
-          null as any
-        )
+        const nestedContext =
+          React.createContext<ReactReduxContextValue | null>(null)
         const useCustomSelector = createSelectorHook(nestedContext)
         let defaultCount: number | null = null
         let customCount: number | null = null
diff --git a/test/typetests/connect-mapstate-mapdispatch.tsx b/test/typetests/connect-mapstate-mapdispatch.tsx
@@ -2,27 +2,25 @@
 
 import * as React from 'react'
 import * as ReactDOM from 'react-dom'
+import type { Dispatch, ActionCreator } from 'redux'
 import {
   Store,
-  Dispatch,
   AnyAction,
-  ActionCreator,
   createStore,
   bindActionCreators,
   ActionCreatorsMapObject,
   Reducer,
 } from 'redux'
+import type { ReactReduxContext, MapDispatchToProps } from '../../src/index'
 import {
   connect,
   ConnectedProps,
   Provider,
   DispatchProp,
   MapStateToProps,
-  ReactReduxContext,
   ReactReduxContextValue,
   Selector,
   shallowEqual,
-  MapDispatchToProps,
   useDispatch,
   useSelector,
   useStore,
diff --git a/test/typetests/connect-options-and-issues.tsx b/test/typetests/connect-options-and-issues.tsx
@@ -2,23 +2,17 @@
 import * as PropTypes from 'prop-types'
 import * as React from 'react'
 import * as ReactDOM from 'react-dom'
-import {
-  Store,
-  Dispatch,
-  AnyAction,
-  ActionCreator,
-  createStore,
-  bindActionCreators,
-  ActionCreatorsMapObject,
-  Reducer,
-} from 'redux'
-import {
-  connect,
+import type { Store, Dispatch, AnyAction, ActionCreator, Reducer } from 'redux'
+import { createStore, bindActionCreators, ActionCreatorsMapObject } from 'redux'
+import type {
   Connect,
   ConnectedProps,
-  Provider,
   DispatchProp,
   MapStateToProps,
+} from '../../src/index'
+import {
+  connect,
+  Provider,
   ReactReduxContext,
   ReactReduxContextValue,
   Selector,
diff --git a/test/typetests/hooks.tsx b/test/typetests/hooks.tsx
@@ -2,16 +2,20 @@
 
 import * as React from 'react'
 import * as ReactDOM from 'react-dom'
-import { Store, Dispatch, configureStore, AnyAction } from '@reduxjs/toolkit'
+import type { Store, Dispatch, AnyAction } from '@reduxjs/toolkit'
+import { configureStore } from '@reduxjs/toolkit'
+import type {
+  ReactReduxContextValue,
+  Selector,
+  TypedUseSelectorHook,
+} from '../../src/index'
 import {
   connect,
   ConnectedProps,
   Provider,
   DispatchProp,
   MapStateToProps,
   ReactReduxContext,
-  ReactReduxContextValue,
-  Selector,
   shallowEqual,
   MapDispatchToProps,
   useDispatch,
@@ -20,17 +24,15 @@ import {
   createDispatchHook,
   createSelectorHook,
   createStoreHook,
-  TypedUseSelectorHook,
 } from '../../src/index'
 
+import type { AppDispatch, RootState } from './counterApp'
 import {
   CounterState,
   counterSlice,
   increment,
   incrementAsync,
-  AppDispatch,
   AppThunk,
-  RootState,
   fetchCount,
 } from './counterApp'
 
@@ -224,9 +226,10 @@ function testCreateHookFunctions() {
     type: 'TEST_ACTION'
   }
 
-  const Context = React.createContext<
-    ReactReduxContextValue<RootState, RootAction>
-  >(null as any)
+  const Context = React.createContext<ReactReduxContextValue<
+    RootState,
+    RootAction
+  > | null>(null)
 
   // No context tests
   expectType<() => Dispatch<AnyAction>>(createDispatchHook())

Original file line number	Diff line number	Diff line change
`@@ -16,12 +16,14 @@ export interface ReactReduxContextValue<`
`16`	`16`	`}`
`17`	`17`
`18`	`18`	const ContextKey = Symbol.for(`react-redux-context-${ReactVersion}`)
`19`		`-const gT = globalThis as { [ContextKey]?: Context<ReactReduxContextValue> }`
	`19`	`+const gT = globalThis as {`
	`20`	`+ [ContextKey]?: Context<ReactReduxContextValue \| null>`
	`21`	`+}`
`20`	`22`
`21`	`23`	`function getContext() {`
`22`	`24`	`let realContext = gT[ContextKey]`
`23`	`25`	`if (!realContext) {`
`24`		`- realContext = createContext<ReactReduxContextValue>(null as any)`
	`26`	`+ realContext = createContext<ReactReduxContextValue \| null>(null)`
`25`	`27`	`if (process.env.NODE_ENV !== 'production') {`
`26`	`28`	`realContext.displayName = 'ReactRedux'`
`27`	`29`	`}`
`@@ -31,8 +33,8 @@ function getContext() {`
`31`	`33`	`}`
`32`	`34`
`33`	`35`	`export const ReactReduxContext = /#__PURE__/ new Proxy(`
`34`		`- {} as Context<ReactReduxContextValue>,`
`35`		`- /#__PURE__/ new Proxy<ProxyHandler<Context<ReactReduxContextValue>>>(`
	`36`	`+ {} as Context<ReactReduxContextValue \| null>,`
	`37`	`+ /#__PURE__/ new Proxy<ProxyHandler<Context<ReactReduxContextValue \| null>>>(`
`36`	`38`	`{},`
`37`	`39`	`{`
`38`	`40`	`get(_, handler) {`
Original file line number	Diff line number	Diff line change
`@@ -10,7 +10,7 @@ import type { ReactReduxContextValue } from '../components/Context'`
`10`	`10`	* @returns {Function} A `useReduxContext` hook bound to the specified context.
`11`	`11`	`*/`
`12`	`12`	`export function createReduxContextHook(context = ReactReduxContext) {`
`13`		`- return function useReduxContext(): ReactReduxContextValue \| null {`
	`13`	`+ return function useReduxContext(): ReactReduxContextValue {`
`14`	`14`	`const contextValue = useContext(context)`
`15`	`15`
`16`	`16`	`if (process.env.NODE_ENV !== 'production' && !contextValue) {`
`@@ -19,7 +19,7 @@ export function createReduxContextHook(context = ReactReduxContext) {`
`19`	`19`	`)`
`20`	`20`	`}`
`21`	`21`
`22`		`- return contextValue`
	`22`	`+ return contextValue!`
`23`	`23`	`}`
`24`	`24`	`}`
`25`	`25`