src/components/Chat/MessageBubble.tsx

import UserMessage from './UserMessage';
import AssistantMessage from './AssistantMessage';
import type { Message } from '@/types/agent';

interface MessageBubbleProps {
  message: Message;
  /** True if this is the user message that starts the last turn. */
  isLastTurn?: boolean;
  /** Callback to undo (remove) the last turn. */
  onUndoTurn?: () => void;
  /** Whether the agent is currently processing. */
  isProcessing?: boolean;
  /** True when this message is actively receiving streaming chunks. */
  isStreaming?: boolean;
}

/**
 * Thin dispatcher — routes each message to the correct
 * specialised component based on its role / content.
 */
export default function MessageBubble({
  message,
  isLastTurn = false,
  onUndoTurn,
  isProcessing = false,
  isStreaming = false,
}: MessageBubbleProps) {
  // Legacy approval-only messages (from old localStorage data) — skip them.
  // Approvals are now rendered inline within ToolCallGroup.
  if (message.approval && !message.content && !message.segments?.length) {
    return null;
  }

  if (message.role === 'user') {
    return (
      <UserMessage
        message={message}
        isLastTurn={isLastTurn}
        onUndoTurn={onUndoTurn}
        isProcessing={isProcessing}
      />
    );
  }

  if (message.role === 'assistant') {
    return <AssistantMessage message={message} isStreaming={isStreaming} />;
  }

  // Fallback (tool messages, etc.)
  return null;
}