Allow to extend useEditController

`useCreateController` accepts a onSuccess and a onFailure props, but `useEditController` don't So, I made this PR to have the same behavior for both of them
marmelab · Nov 28, 2019 · 803223e · 803223e
1 parent 7996e7d
commit 803223e
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/packages/ra-core/src/controller/useEditController.ts b/packages/ra-core/src/controller/useEditController.ts
@@ -92,12 +92,12 @@ const useEditController = (props: EditProps): EditControllerProps => {
     );
 
     const save = useCallback(
-        (data: Partial<Record>, redirectTo = 'list') =>
+        (data: Partial<Record>, redirectTo = 'list', { onSuccess, onFailure } = {}) =>
             update(
                 { payload: { data } },
                 {
                     action: CRUD_UPDATE,
-                    onSuccess: () => {
+                    onSuccess: onSuccess ? onSuccess : () => {
                         notify(
                             successMessage || 'ra.notification.updated',
                             'info',
@@ -108,7 +108,7 @@ const useEditController = (props: EditProps): EditControllerProps => {
                         );
                         redirect(redirectTo, basePath, data.id, data);
                     },
-                    onFailure: error =>
+                    onFailure: onFailure ? onFailure : error =>
                         notify(
                             typeof error === 'string'
                                 ? error