From ab1e135aa5f701e89576b53af4daf175b6255b5f Mon Sep 17 00:00:00 2001
From: fzaninotto <fzaninotto@gmail.com>
Date: Fri, 5 Feb 2021 22:45:32 +0100
Subject: [PATCH] Fix ReferenceManyField throws error after insert

---
 .../src/dataProvider/useGetManyReference.ts   | 54 +++++++++++++------
 1 file changed, 37 insertions(+), 17 deletions(-)

diff --git a/packages/ra-core/src/dataProvider/useGetManyReference.ts b/packages/ra-core/src/dataProvider/useGetManyReference.ts
index 3731fc1812b..2eb8c6bfa03 100644
--- a/packages/ra-core/src/dataProvider/useGetManyReference.ts
+++ b/packages/ra-core/src/dataProvider/useGetManyReference.ts
@@ -1,4 +1,6 @@
-import { useSelector, shallowEqual } from 'react-redux';
+import get from 'lodash/get';
+import { useMemo } from 'react';
+
 import { CRUD_GET_MANY_REFERENCE } from '../actions/dataActions/crudGetManyReference';
 import {
     PaginationPayload,
@@ -8,12 +10,13 @@ import {
 } from '../types';
 import useQueryWithStore from './useQueryWithStore';
 import {
-    getReferences,
     getIds,
     getTotal,
     nameRelatedTo,
 } from '../reducer/admin/references/oneToMany';
-import { useMemo } from 'react';
+
+const defaultIds = [];
+const defaultData = {};
 
 /**
  * Call the dataProvider.getManyReference() method and return the resolved result
@@ -75,29 +78,46 @@ const useGetManyReference = (
         [filter, resource, id, referencingResource, target]
     );
 
-    const { data: ids, total, error, loading, loaded } = useQueryWithStore(
+    const {
+        data: { ids, allRecords },
+        total,
+        error,
+        loading,
+        loaded,
+    } = useQueryWithStore(
         {
             type: 'getManyReference',
             resource: resource,
             payload: { target, id, pagination, sort, filter },
         },
         { ...options, relatedTo, action: CRUD_GET_MANY_REFERENCE },
-        selectIds(relatedTo),
-        selectTotal(relatedTo)
+        // ids and data selector
+        (state: ReduxState) => ({
+            ids: getIds(state, relatedTo) || defaultIds,
+            allRecords: get(
+                state.admin.resources,
+                [resource, 'data'],
+                defaultData
+            ),
+        }),
+        (state: ReduxState) => getTotal(state, relatedTo)
+    );
+
+    const data = useMemo(
+        () =>
+            ids === null
+                ? defaultData
+                : ids
+                      .map(id => allRecords[id])
+                      .reduce((acc, record) => {
+                          if (!record) return acc;
+                          acc[record.id] = record;
+                          return acc;
+                      }, {}),
+        [ids, allRecords]
     );
-    const data = useSelector(selectData(resource, relatedTo), shallowEqual);
 
     return { data, ids, total, error, loading, loaded };
 };
 
 export default useGetManyReference;
-
-const selectData = (reference: string, relatedTo: string) => (
-    state: ReduxState
-) => getReferences(state, reference, relatedTo);
-
-const selectIds = (relatedTo: string) => (state: ReduxState) =>
-    getIds(state, relatedTo);
-
-const selectTotal = (relatedTo: string) => (state: ReduxState) =>
-    getTotal(state, relatedTo);