File size: 5,038 Bytes
aa2e6af | 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 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 | import { InfiniteData } from '@tanstack/react-query';
export const addData = <TCollection, TData>(
data: InfiniteData<TCollection>,
collectionName: string,
newData: TData,
findIndex: (page: TCollection) => number,
) => {
const dataJson = JSON.parse(JSON.stringify(data)) as InfiniteData<TCollection>;
const { pageIndex, index } = findPage<TCollection>(data, findIndex);
if (pageIndex !== -1 && index !== -1) {
return updateData(data, collectionName, newData, findIndex);
}
dataJson.pages[0][collectionName].unshift({
...newData,
updatedAt: new Date().toISOString(),
});
return dataJson;
};
export const getRecordByProperty = <TCollection, TData>(
data: InfiniteData<TCollection>,
collectionName: string,
findProperty: (item: TData) => boolean,
): TData | undefined => {
// Find the page and the index of the record in that page
const { pageIndex, index } = findPage<TCollection>(data, (page) =>
page[collectionName].findIndex(findProperty),
);
// If found, return the record
if (pageIndex !== -1 && index !== -1) {
return data.pages[pageIndex][collectionName][index];
}
// Return undefined if the record is not found
return undefined;
};
export function findPage<TData>(data: InfiniteData<TData>, findIndex: (page: TData) => number) {
for (let pageIndex = 0; pageIndex < data.pages.length; pageIndex++) {
const page = data.pages[pageIndex];
const index = findIndex(page);
if (index !== -1) {
return { pageIndex, index };
}
}
return { pageIndex: -1, index: -1 }; // Not found
}
export const updateData = <TCollection, TData>(
data: InfiniteData<TCollection>,
collectionName: string,
updatedData: TData,
findIndex: (page: TCollection) => number,
) => {
const newData = JSON.parse(JSON.stringify(data)) as InfiniteData<TCollection>;
const { pageIndex, index } = findPage<TCollection>(data, findIndex);
if (pageIndex !== -1 && index !== -1) {
// Remove the data from its current position
newData.pages[pageIndex][collectionName].splice(index, 1);
// Add the updated data to the top of the first page
newData.pages[0][collectionName].unshift({
...updatedData,
updatedAt: new Date().toISOString(),
});
}
return newData;
};
export const deleteData = <TCollection, TData>(
data: TData,
collectionName: string,
findIndex: (page: TCollection) => number,
): TData => {
const newData = JSON.parse(JSON.stringify(data));
const { pageIndex, index } = findPage<TCollection>(newData, findIndex);
if (pageIndex !== -1 && index !== -1) {
// Delete the data from its current page
newData.pages[pageIndex][collectionName].splice(index, 1);
}
return newData;
};
/**
* Normalize the data so that the number of data on each page is within pageSize
*/
export const normalizeData = <TCollection, TData>(
data: InfiniteData<TCollection>,
collectionName: string,
pageSize: number,
uniqueProperty?: keyof TData,
): InfiniteData<TCollection> => {
const infiniteData = JSON.parse(JSON.stringify(data)) as InfiniteData<TCollection>;
const pageCount = infiniteData.pages.length;
if (pageCount === 0) {
return infiniteData;
}
const pageParams = infiniteData.pageParams;
// Combine all conversations of all pages into one array
let collection = infiniteData.pages.flatMap((page) => page[collectionName]);
if (collection.length === 0) {
return infiniteData;
}
if (uniqueProperty) {
const seen = new Set<TData>();
collection = collection.filter((item) => {
const value = item[uniqueProperty];
if (seen.has(value)) {
return false;
}
seen.add(value);
return true;
});
}
// Create the restructured pages
const restructuredPages = Array.from({ length: pageCount }, (_, i) => ({
...infiniteData.pages[i],
[collectionName]: collection.slice(i * pageSize, (i + 1) * pageSize),
})).filter((page) => page[collectionName].length > 0); // Remove empty pages
return {
pageParams: pageParams.slice(0, restructuredPages.length),
pages: restructuredPages,
};
};
export const updateFields = <TCollection, TData>(
data: InfiniteData<TCollection>,
updatedItem: Partial<TData>,
collectionName: string,
identifierField: keyof TData,
callback?: (newItem: TData) => void,
): InfiniteData<TCollection> => {
const newData = JSON.parse(JSON.stringify(data)) as InfiniteData<TCollection>;
const { pageIndex, index } = findPage<TCollection>(newData, (page) =>
page[collectionName].findIndex(
(item: TData) => item[identifierField] === updatedItem[identifierField],
),
);
if (pageIndex !== -1 && index !== -1) {
const deleted = newData.pages[pageIndex][collectionName].splice(index, 1);
const oldItem = deleted[0];
const newItem = {
...oldItem,
...updatedItem,
updatedAt: new Date().toISOString(),
};
if (callback) {
callback(newItem);
}
newData.pages[0][collectionName].unshift(newItem);
}
return newData;
};
|