curry createAsyncThunk

Author: phryneas

src/createAsyncThunk.ts

@@ -62,7 +62,7 @@ export const miniSerializeError = (value: any): SerializedError => {
   return { message: String(value) }
 }
 
-type AsyncThunkConfig = {
+export type AsyncThunkConfig = {
   state?: unknown
   dispatch?: Dispatch
   extra?: unknown
@@ -191,7 +191,7 @@ type AsyncThunkActionCreator<
     : (arg: ThunkArg) => AsyncThunkAction<Returned, ThunkArg, ThunkApiConfig>
 >
 
-interface AsyncThunkOptions<
+export interface AsyncThunkOptions<
   ThunkArg = void,
   ThunkApiConfig extends AsyncThunkConfig = {}
 > {

src/curriedTypes/createAsyncThunk.ts

@@ -0,0 +1,29 @@
+export interface CurryableTypes {
+  createAsyncThunk: typeof import('../createAsyncThunk').createAsyncThunk
+}
+
+export interface CurriedType<RootState, Dispatch> {
+  createAsyncThunk: CuriedCreateAsyncThunk<RootState, Dispatch>
+}
+
+/* eslint-disable import/first */
+import {
+  AsyncThunk,
+  AsyncThunkPayloadCreator,
+  AsyncThunkConfig,
+  AsyncThunkOptions
+} from '../createAsyncThunk'
+
+type CuriedCreateAsyncThunk<RootState, Dispatch> = <
+  Returned,
+  ThunkArg = void,
+  ThunkApiConfig extends Omit<AsyncThunkConfig, 'dispatch' | 'state'> = {}
+>(
+  typePrefix: string,
+  payloadCreator: AsyncThunkPayloadCreator<
+    Returned,
+    ThunkArg,
+    ThunkApiConfig & { dispatch: Dispatch; state: RootState }
+  >,
+  options?: AsyncThunkOptions<ThunkArg, ThunkApiConfig>
+) => AsyncThunk<Returned, ThunkArg, ThunkApiConfig>

src/curriedTypes/index.ts

@@ -2,6 +2,11 @@ import {
   CurriedType as RRCurriedType,
   CurryableTypes as RRCurryableType
 } from './react-redux'
+import {
+  CurriedType as CatCurriedType,
+  CurryableTypes as CatCurryableType
+} from './createAsyncThunk'
+
 import { UnionToIntersection } from '../tsHelpers'
 
 type CurrySingleType<RootState, Dispatch> = UnionToIntersection<
@@ -25,7 +30,8 @@ export type CurryType<RootState, Dispatch> = CurrySingleType<
 > &
   CurryMultipleTypes<RootState, Dispatch>
 
-export interface CurryableTypes extends RRCurryableType {}
+export interface CurryableTypes extends RRCurryableType, CatCurryableType {}
 
 export interface CurriedType<RootState, Dispatch>
-  extends RRCurriedType<RootState, Dispatch> {}
+  extends RRCurriedType<RootState, Dispatch>,
+    CatCurriedType<RootState, Dispatch> {}

type-tests/files/curriedTypes.typetest.ts

@@ -1,5 +1,5 @@
 import { useDispatch, useSelector } from 'react-redux'
-import { configureStore } from 'src'
+import { configureStore, createAsyncThunk } from 'src'
 
 type DispatchType = typeof store.dispatch
 type StateType = ReturnType<typeof store.getState>
@@ -60,3 +60,11 @@ function expectType<T>(t: T) {
     connectAdvanced
   } = store.withCurriedTypes(require('react-redux'))
 }
+
+{
+  const curriedCAT = store.withCurriedTypes(createAsyncThunk)
+  curriedCAT('foo', (arg: any, { getState, dispatch }) => {
+    expectType<DispatchType>(dispatch)
+    expectType<StateType>(getState())
+  })
+}