Add context to render empty, error and loading components

Author: fzaninotto

packages/ra-core/src/controller/field/ReferenceArrayFieldBase.tsx

@@ -109,10 +109,20 @@ export const ReferenceArrayFieldBase = <
     }
 
     if (controllerProps.isPending && loading) {
-        return loading;
+        return (
+            <ResourceContextProvider value={reference}>
+                {loading}
+            </ResourceContextProvider>
+        );
     }
     if (controllerProps.error && error) {
-        return error;
+        return (
+            <ResourceContextProvider value={reference}>
+                <ListContextProvider value={controllerProps}>
+                    {error}
+                </ListContextProvider>
+            </ResourceContextProvider>
+        );
     }
     if (
         // there is an empty page component
@@ -133,7 +143,11 @@ export const ReferenceArrayFieldBase = <
         // the user didn't set any filters
         !Object.keys(controllerProps.filterValues).length
     ) {
-        return empty;
+        return (
+            <ResourceContextProvider value={reference}>
+                {empty}
+            </ResourceContextProvider>
+        );
     }
 
     return (

packages/ra-core/src/controller/field/ReferenceFieldBase.tsx

@@ -60,10 +60,20 @@ export const ReferenceFieldBase = <
     }
 
     if (controllerProps.isPending && loading) {
-        return loading;
+        return (
+            <ResourceContextProvider value={props.reference}>
+                {loading}
+            </ResourceContextProvider>
+        );
     }
     if (controllerProps.error && error) {
-        return error;
+        return (
+            <ResourceContextProvider value={props.reference}>
+                <ReferenceFieldContextProvider value={controllerProps}>
+                    {error}
+                </ReferenceFieldContextProvider>
+            </ResourceContextProvider>
+        );
     }
     if (
         (empty &&
@@ -74,7 +84,11 @@ export const ReferenceFieldBase = <
             !controllerProps.isPending &&
             !controllerProps.referenceRecord)
     ) {
-        return empty;
+        return (
+            <ResourceContextProvider value={props.reference}>
+                {empty}
+            </ResourceContextProvider>
+        );
     }
 
     return (

packages/ra-core/src/controller/field/ReferenceManyFieldBase.tsx

@@ -105,10 +105,20 @@ export const ReferenceManyFieldBase = <
     }
 
     if (controllerProps.isPending && loading) {
-        return loading;
+        return (
+            <ResourceContextProvider value={reference}>
+                {loading}
+            </ResourceContextProvider>
+        );
     }
     if (controllerProps.error && error) {
-        return error;
+        return (
+            <ResourceContextProvider value={reference}>
+                <ListContextProvider value={controllerProps}>
+                    {error}
+                </ListContextProvider>
+            </ResourceContextProvider>
+        );
     }
     if (
         // there is an empty page component
@@ -129,7 +139,11 @@ export const ReferenceManyFieldBase = <
         // the user didn't set any filters
         !Object.keys(controllerProps.filterValues).length
     ) {
-        return empty;
+        return (
+            <ResourceContextProvider value={reference}>
+                {empty}
+            </ResourceContextProvider>
+        );
     }
 
     return (

packages/ra-core/src/controller/field/ReferenceOneFieldBase.tsx

@@ -79,16 +79,30 @@ export const ReferenceOneFieldBase = <
 
     const recordFromContext = useRecordContext<RecordType>(props);
     if (controllerProps.isPending && loading) {
-        return loading;
+        return (
+            <ResourceContextProvider value={reference}>
+                {loading}
+            </ResourceContextProvider>
+        );
     }
     if (controllerProps.error && error) {
-        return error;
+        return (
+            <ResourceContextProvider value={reference}>
+                <ReferenceFieldContextProvider value={context}>
+                    {error}
+                </ReferenceFieldContextProvider>
+            </ResourceContextProvider>
+        );
     }
     if (
         !recordFromContext ||
         (!controllerProps.isPending && controllerProps.referenceRecord == null)
     ) {
-        return empty;
+        return (
+            <ResourceContextProvider value={reference}>
+                {empty}
+            </ResourceContextProvider>
+        );
     }
 
     return (