Add onResponseChange prop for convenience

techniq · Dec 19, 2019 · 60f08bf · 60f08bf
1 parent d6a8eb1
commit 60f08bf
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/index.d.ts b/index.d.ts
@@ -37,6 +37,7 @@ export interface FetchProps<TData = any, TError = Error> {
     | { [type: string]: (res: TData) => Promise<any> };
   fetchFunction?: (url: string, options: RequestInit) => Promise<any>;
   onDataChange?: (newData: TData, data: TData) => any;
+  onResponseChange?: (response: Response) => any;
   onChange?: (result: FetchResult<TData, TError>) => void;
   deps?: [any];
 }

diff --git a/src/Fetch.js b/src/Fetch.js
@@ -145,7 +145,7 @@ function useFetch(props) {
       promises.current.splice(0, index + 1);
     }
 
-    const { onChange, onDataChange } = props;
+    const { onChange, onDataChange, onResponseChange } = props;
 
     let data = undefined;
     if (
@@ -159,6 +159,14 @@ function useFetch(props) {
       );
     }
 
+    if (
+      nextState.response &&
+      nextState.response !== state.response &&
+      isFunction(onResponseChange)
+    ) {
+      data = onResponseChange(nextState.response);
+    }
+
     if (isFunction(onChange)) {
       // Always call onChange even if unmounted.  Useful for `POST` requests with a redirect
       onChange({