FilesExpand file tree

main

/

useUpdateCache.ts

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

66 lines (62 loc) · 1.94 KB

main

/

useUpdateCache.ts

Copy path

Top

File metadata and controls

• Code
• Blame

66 lines (62 loc) · 1.94 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

import { useCallback } from 'react';

import type { GetListResult, Identifier } from 'react-admin';

import { useQueryClient } from '@tanstack/react-query';

import type { ApiPlatformAdminRecord } from '../types.js';

const useUpdateCache = () => {

const queryClient = useQueryClient();

// From https://github.com/marmelab/react-admin/blob/next/packages/ra-core/src/dataProvider/useUpdate.ts

return useCallback(

({

resource,

id,

data,

}: {

resource: string;

id: Identifier;

data: ApiPlatformAdminRecord;

}) => {

const updateColl = (old: ApiPlatformAdminRecord[]) => {

const index = old.findIndex(

// eslint-disable-next-line eqeqeq

(record) => record.id == id,

);

if (index === -1) {

return old;

}

return [

...old.slice(0, index),

{ ...old[index], ...data },

...old.slice(index + 1),

];

};

queryClient.setQueryData(

[resource, 'getOne', { id: String(id) }],

(record: ApiPlatformAdminRecord | undefined) => ({

...record,

...data,

}),

);

queryClient.setQueriesData(

{ queryKey: [resource, 'getList'] },

(res: GetListResult<ApiPlatformAdminRecord> | undefined) =>

res?.data

? { data: updateColl(res.data), total: res.total }

: { data: [data] },

);

queryClient.setQueriesData(

{ queryKey: [resource, 'getMany'] },

(coll: ApiPlatformAdminRecord[] | undefined) =>

coll && coll.length > 0 ? updateColl(coll) : [data],

);

queryClient.setQueriesData(

{ queryKey: [resource, 'getManyReference'] },

(res: GetListResult<ApiPlatformAdminRecord> | undefined) =>

res?.data

? { data: updateColl(res.data), total: res.total }

: { data: [data] },

);

},

[queryClient],

);

};

export default useUpdateCache;