import { ChatEvent } from "../../../widgets";
import { usePart } from "../context";
import { EventPartType } from "../types";

export interface EventPartProps {
  className?: string;
  renderData?: (data: ChatEvent["data"]) => React.ReactNode;
}

/**
 * Render an event inside a ChatMessage, return null if current part is not event type
 * This component is useful to show an event from the assistant.
 * Normally, it will start with "Loading" status and then change to "Success" with a result
 * @param className - custom styles for the event
 */
export function EventPartUI({ className, renderData }: EventPartProps) {
  const part = usePart(EventPartType);
  if (!part) return null;
  return (
    <ChatEvent
      event={part.data}
      className={className}
      renderData={renderData}
    />
  );
}