Function updateMessageInPages

packages/kilo-chat-hooks/src/use-messages.ts:293–315

(
  old: MessageInfiniteData<TPageParam>,
  messageId: string,
  updater: (message: Message) => Message
)

  return undefined;
}

export function updateMessageInPages<TPageParam>(
  old: MessageInfiniteData<TPageParam>,
  messageId: string,
  updater: (message: Message) => Message
): MessageInfiniteData<TPageParam> {
  for (let pageIndex = 0; pageIndex < old.pages.length; pageIndex++) {
    const page = old.pages[pageIndex];
    if (!page) continue;
    const messageIndex = page.messages.findIndex(msg => msg.id === messageId);
    if (messageIndex === -1) continue;

    const pages = old.pages.slice();
    const updatedMessages = page.messages.slice();
    const message = updatedMessages[messageIndex];
    if (!message) return old;
    const updatedMessage = updater(message);
    if (updatedMessage === message) return old;
    updatedMessages[messageIndex] = updatedMessage;
    pages[pageIndex] = withPageMessages(page, updatedMessages);
    return { ...old, pages };
  }
  return old;
}

function restoreOptimisticMessage(