Aurora ChatMessage

1"use client"
2 
3import * as React from "react"
4import { Copy, Link2, RotateCcw, Shield } from "lucide-react"
5 
6import { Button } from "./button"
7import { cn } from "@/lib/utils"
8import { safeHttpUrl } from "@/registry/aurora/lib/safe-url"
9 
10// Styles: registry/aurora/styles/aurora-components.css (@layer aurora-components).
11 
12/* -------------------------------------------------------------------------- */
13/*  Types                                                                     */
14/* -------------------------------------------------------------------------- */
15 
16export type ChatMessageRole = "user" | "assistant"
17 
18export interface ChatMessageCitation {
19  label: string
20  href: string
21}
22 
23export interface ChatMessageProps extends React.HTMLAttributes<HTMLDivElement> {
24  /** Who is speaking. User turns align right; assistant turns align left with an avatar. */
25  role: ChatMessageRole
26  /** Display name shown in the assistant header. */
27  author?: React.ReactNode
28  /** Timestamp shown next to the author. */
29  time?: React.ReactNode
30  /** Source citations rendered as monospace chips below the bubble. */
31  citations?: ChatMessageCitation[]
32  /** Optional copy handler (reserved for action affordances). */
33  onCopy?: () => void
34  /** Optional retry handler (reserved for action affordances). */
35  onRetry?: () => void
36  children?: React.ReactNode
37}
38 
39/* -------------------------------------------------------------------------- */
40/*  Component                                                                 */
41/* -------------------------------------------------------------------------- */
42 
43function ChatMessage(
44  {
45    role,
46    author,
47    time,
48    citations,
49    onCopy,
50    onRetry,
51    className,
52    children,
53    ref,
54    ...props
55  }: ChatMessageProps & { ref?: React.Ref<HTMLDivElement> },
56) {
57  const isAssistant = role === "assistant"
58  const showHead = isAssistant && (author != null || time != null)
59  const showActions = isAssistant && (onCopy != null || onRetry != null)
60 
61  return (
62    <div
63      ref={ref}
64      data-role={role}
65      className={cn("aurora-chat-message", className)}
66      {...props}
67    >
68      {showHead ? (
69        <div className="aurora-chat-message__head">
70          <span className="aurora-chat-message__avatar" aria-hidden="true">
71            <Shield size={18} strokeWidth={1.7} />
72          </span>
73          <span className="aurora-chat-message__meta">
74            {author != null ? (
75// … truncated