From 59f87d01fa8636b4986be20b11af239546ccd763 Mon Sep 17 00:00:00 2001
From: Gildas Garcia <1122076+djhi@users.noreply.github.com>
Date: Mon, 1 Feb 2021 17:15:48 +0100
Subject: [PATCH] Fix SaveButton Breaking Change Regarding FormContext

---
 .../src/button/SaveButton.spec.tsx            | 33 +++++--------------
 .../src/button/SaveButton.tsx                 | 26 ++++++++++++---
 2 files changed, 30 insertions(+), 29 deletions(-)

diff --git a/packages/ra-ui-materialui/src/button/SaveButton.spec.tsx b/packages/ra-ui-materialui/src/button/SaveButton.spec.tsx
index e2f61127212..17b57bc6469 100644
--- a/packages/ra-ui-materialui/src/button/SaveButton.spec.tsx
+++ b/packages/ra-ui-materialui/src/button/SaveButton.spec.tsx
@@ -57,9 +57,7 @@ describe('<SaveButton />', () => {
             <TestContext>
                 <ThemeProvider theme={theme}>
                     <SaveContextProvider value={saveContextValue}>
-                        <FormContextProvider value={formContextValue}>
-                            <SaveButton {...invalidButtonDomProps} />
-                        </FormContextProvider>
+                        <SaveButton {...invalidButtonDomProps} />
                     </SaveContextProvider>
                 </ThemeProvider>
             </TestContext>
@@ -78,9 +76,7 @@ describe('<SaveButton />', () => {
             <TestContext>
                 <ThemeProvider theme={theme}>
                     <SaveContextProvider value={saveContextValue}>
-                        <FormContextProvider value={formContextValue}>
-                            <SaveButton disabled={true} />
-                        </FormContextProvider>
+                        <SaveButton disabled={true} />
                     </SaveContextProvider>
                 </ThemeProvider>
             </TestContext>
@@ -92,9 +88,7 @@ describe('<SaveButton />', () => {
         const { getByLabelText } = render(
             <TestContext>
                 <SaveContextProvider value={saveContextValue}>
-                    <FormContextProvider value={formContextValue}>
-                        <SaveButton submitOnEnter />
-                    </FormContextProvider>
+                    <SaveButton submitOnEnter />
                 </SaveContextProvider>
             </TestContext>
         );
@@ -107,9 +101,7 @@ describe('<SaveButton />', () => {
         const { getByLabelText } = render(
             <TestContext>
                 <SaveContextProvider value={saveContextValue}>
-                    <FormContextProvider value={formContextValue}>
-                        <SaveButton submitOnEnter={false} />
-                    </FormContextProvider>
+                    <SaveButton submitOnEnter={false} />
                 </SaveContextProvider>
             </TestContext>
         );
@@ -144,12 +136,7 @@ describe('<SaveButton />', () => {
         const { getByLabelText } = render(
             <TestContext>
                 <SaveContextProvider value={saveContextValue}>
-                    <FormContextProvider value={formContextValue}>
-                        <SaveButton
-                            handleSubmitWithRedirect={onSubmit}
-                            saving
-                        />
-                    </FormContextProvider>
+                    <SaveButton handleSubmitWithRedirect={onSubmit} saving />
                 </SaveContextProvider>
             </TestContext>
         );
@@ -168,12 +155,10 @@ describe('<SaveButton />', () => {
                     dispatchSpy = jest.spyOn(store, 'dispatch');
                     return (
                         <SaveContextProvider value={saveContextValue}>
-                            <FormContextProvider value={formContextValue}>
-                                <SaveButton
-                                    handleSubmitWithRedirect={onSubmit}
-                                    invalid
-                                />
-                            </FormContextProvider>
+                            <SaveButton
+                                handleSubmitWithRedirect={onSubmit}
+                                invalid
+                            />
                         </SaveContextProvider>
                     );
                 }}
diff --git a/packages/ra-ui-materialui/src/button/SaveButton.tsx b/packages/ra-ui-materialui/src/button/SaveButton.tsx
index 67227880609..f524d4c1ce4 100644
--- a/packages/ra-ui-materialui/src/button/SaveButton.tsx
+++ b/packages/ra-ui-materialui/src/button/SaveButton.tsx
@@ -81,21 +81,37 @@ const SaveButton: FC<SaveButtonProps> = props => {
     const classes = useStyles(props);
     const notify = useNotify();
     const translate = useTranslate();
-    const { setOnSave } = useFormContext();
+    const saveContext = useFormContext();
     const { setOnSuccess, setOnFailure, setTransform } = useSaveContext(props);
 
     const handleClick = event => {
         // deprecated: use onSuccess and transform instead of onSave
         if (typeof onSave === 'function') {
             if (process.env.NODE_ENV !== 'production') {
-                console.log(
+                console.warn(
                     '<SaveButton onSave> prop is deprecated, use the onSuccess prop instead.'
                 );
+                if (!saveContext || !saveContext.setOnSave) {
+                    console.warn(
+                        'Using <SaveButton> outside a FormContext is deprecated.'
+                    );
+                }
+            }
+            if (saveContext && saveContext.setOnSave) {
+                saveContext.setOnSave(onSave);
             }
-            setOnSave(onSave);
         } else {
-            // we reset to the Form default save function
-            setOnSave();
+            if (
+                (process.env.NODE_ENV !== 'production' && !saveContext) ||
+                !saveContext.setOnSave
+            ) {
+                console.warn(
+                    'Using <SaveButton> outside a FormContext is deprecated.'
+                );
+            } else {
+                // we reset to the Form default save function
+                saveContext.setOnSave();
+            }
         }
         if (onSuccess) {
             setOnSuccess(onSuccess);