Function reconcileEditedMessageInCache

site/src/api/queries/chatMessageEdits.ts:117–148

({
	currentData,
	optimisticMessageId,
	responseMessage,
}: {
	currentData: InfiniteData<TypesGen.ChatMessagesResponse> | undefined;
	optimisticMessageId: number;
	responseMessage: TypesGen.ChatMessage;
})

};

export const reconcileEditedMessageInCache = ({
	currentData,
	optimisticMessageId,
	responseMessage,
}: {
	currentData: InfiniteData<TypesGen.ChatMessagesResponse> | undefined;
	optimisticMessageId: number;
	responseMessage: TypesGen.ChatMessage;
}): InfiniteData<TypesGen.ChatMessagesResponse> | undefined => {
	if (!currentData?.pages?.length) {
		return currentData;
	}

	const replacedPages = currentData.pages.map((page, pageIndex) => {
		const preservedMessages = page.messages.filter(
			(message) =>
				message.id !== optimisticMessageId && message.id !== responseMessage.id,
		);
		if (pageIndex !== 0) {
			return { ...page, messages: preservedMessages };
		}
		return {
			...page,
			messages: upsertFirstPageMessage(preservedMessages, responseMessage),
		};
	});

	return {
		...currentData,
		pages: replacedPages,
	};
};