import type { AcpMessage } from "@/lib/api/acp";

export interface CoalescedMessage {
  role: "user" | "agent";
  text: string;
  time: string;
  messageId?: string;
}

export function groupMessages(messages: AcpMessage[]): CoalescedMessage[] {
  const result: CoalescedMessage[] = [];

  for (const message of messages) {
    const last = result[result.length - 1];

    if (last && last.role === message.role) {
      last.text += message.text;
    } else {
      result.push({
        role: message.role,
        text: message.text,
        time: message.time,
        messageId: (message as AcpMessage & { messageId?: string }).messageId,
      });
    }
  }

  return result;
}