Function updateFeedAndAdsCache

packages/shared/src/lib/query.ts:771–817

(
  postId: string,
  feedQueryKey: QueryKey,
  queryClient: QueryClient,
  update: Partial<Post>,
)

};

export const updateFeedAndAdsCache = (
  postId: string,
  feedQueryKey: QueryKey,
  queryClient: QueryClient,
  update: Partial<Post>,
): void => {
  // Update the main feed cache
  const updateFeedPost = updateCachedPagePost(feedQueryKey, queryClient);
  const feedData =
    queryClient.getQueryData<InfiniteData<AnyFeedData>>(feedQueryKey);

  if (feedData) {
    const { pageIndex, index } = findIndexOfPostInData(feedData, postId, true);
    if (index > -1) {
      const currentPost = getFeedApiItemPost(
        feedData.pages[pageIndex].page.edges[index].node,
      );

      if (!currentPost) {
        throw new Error(
          `Missing post-backed feed item at page ${pageIndex} index ${index}`,
        );
      }

      updateFeedPost(pageIndex, index, {
        ...currentPost,
        ...update,
      });
    }
  }

  // Update the ads cache if the post exists there
  const adsQueryKey = [RequestKey.Ads, ...feedQueryKey];
  const adsData = queryClient.getQueryData<InfiniteData<Ad>>(adsQueryKey);

  if (adsData) {
    const existingAdPost = adsData.pages.find(
      (page) => page.data?.post?.id === postId,
    )?.data?.post;

    if (existingAdPost) {
      queryClient.setQueryData(adsQueryKey, (currentData: InfiniteData<Ad>) => {
        return updateAdPostInCache(postId, currentData, update);
      });
    }
  }
};