@btst/stack 1.3.1 → 1.4.1

package/src/plugins/ai-chat/client/components/chat-message.tsx

@@ -0,0 +1,561 @@
+"use client";
+
+import { cn } from "@workspace/ui/lib/utils";
+import { MarkdownContent } from "@workspace/ui/components/markdown-content";
+import { Button } from "@workspace/ui/components/button";
+import {
+	Dialog,
+	DialogContent,
+	DialogTitle,
+} from "@workspace/ui/components/dialog";
+import {
+	User,
+	Bot,
+	FileText,
+	Copy,
+	Check,
+	RefreshCw,
+	Pencil,
+	Send,
+	X,
+} from "lucide-react";
+import type { UIMessage } from "ai";
+import { isToolUIPart, getToolName } from "ai";
+import {
+	useMemo,
+	useState,
+	useRef,
+	useEffect,
+	type ComponentType,
+} from "react";
+import { usePluginOverrides } from "@btst/stack/context";
+import type {
+	AiChatPluginOverrides,
+	ToolCallProps,
+	ToolCallState,
+} from "../overrides";
+import { AI_CHAT_LOCALIZATION } from "../localization";
+import { ToolCallDisplay } from "./tool-call-display";
+
+// Import shared markdown + syntax highlighting styles (same pattern as blog plugin)
+import "@workspace/ui/markdown-content.css";
+import "highlight.js/styles/panda-syntax-light.css";
+
+// Import remend for streaming markdown
+import completeMarkdown from "remend";
+
+// Fallback components (so consumers don't have to provide them)
+const DefaultLink = (props: React.ComponentProps<"a">) => <a {...props} />;
+const DefaultImage = (props: React.ImgHTMLAttributes<HTMLImageElement>) => (
+	<img {...props} />
+);
+
+// Clickable image component that opens in a dialog
+interface ClickableImageProps {
+	src: string;
+	alt: string;
+	/** Width in pixels for the thumbnail */
+	width?: number;
+	/** Height in pixels for the thumbnail */
+	height?: number;
+	className?: string;
+	ImageComponent: ComponentType<React.ImgHTMLAttributes<HTMLImageElement>>;
+}
+
+function ClickableImage({
+	src,
+	alt,
+	width = 200,
+	height = 200,
+	className,
+	ImageComponent,
+}: ClickableImageProps) {
+	const [isOpen, setIsOpen] = useState(false);
+
+	return (
+		<>
+			<button
+				type="button"
+				onClick={() => setIsOpen(true)}
+				className="cursor-zoom-in focus:outline-none focus:ring-2 focus:ring-ring focus:ring-offset-2 rounded-md overflow-hidden"
+				style={{ width, height }}
+			>
+				<ImageComponent
+					src={src}
+					alt={alt}
+					width={width}
+					height={height}
+					className={cn(
+						"object-cover hover:opacity-90 transition-opacity",
+						className,
+					)}
+				/>
+			</button>
+			<Dialog open={isOpen} onOpenChange={setIsOpen}>
+				<DialogContent
+					className="max-w-[90vw] max-h-[90vh] p-2 sm:max-w-[90vw]"
+					showCloseButton
+				>
+					<DialogTitle className="sr-only">{alt}</DialogTitle>
+					<div className="flex items-center justify-center">
+						{/* Use native img for full-size preview to avoid Next.js Image constraints */}
+						<img
+							src={src}
+							alt={alt}
+							className="max-w-full max-h-[85vh] object-contain rounded-md"
+						/>
+					</div>
+				</DialogContent>
+			</Dialog>
+		</>
+	);
+}
+
+interface ChatMessageProps {
+	message: UIMessage;
+	isStreaming?: boolean;
+	variant?: "default" | "compact";
+	/** Callback when user wants to retry/regenerate an AI response */
+	onRetry?: () => void;
+	/** Callback when user edits their message - receives the new text */
+	onEdit?: (newText: string) => void;
+	/** Whether retry is currently in progress */
+	isRetrying?: boolean;
+}
+
+export function ChatMessage({
+	message,
+	isStreaming = false,
+	variant = "default",
+	onRetry,
+	onEdit,
+	isRetrying = false,
+}: ChatMessageProps) {
+	const isUser = message.role === "user";
+
+	const {
+		Link,
+		Image,
+		localization: customLocalization,
+		toolRenderers,
+	} = usePluginOverrides<AiChatPluginOverrides, Partial<AiChatPluginOverrides>>(
+		"ai-chat",
+		{},
+	);
+
+	const localization = { ...AI_CHAT_LOCALIZATION, ...customLocalization };
+
+	// Use provided Image component or fallback to default
+	const ImageComponent = Image ?? DefaultImage;
+
+	// State for copy button
+	const [copied, setCopied] = useState(false);
+
+	// State for editing user messages
+	const [isEditing, setIsEditing] = useState(false);
+	const [editText, setEditText] = useState("");
+	const editTextareaRef = useRef<HTMLTextAreaElement>(null);
+
+	// Auto-resize textarea when editing
+	useEffect(() => {
+		if (isEditing && editTextareaRef.current) {
+			editTextareaRef.current.style.height = "auto";
+			editTextareaRef.current.style.height = `${editTextareaRef.current.scrollHeight}px`;
+			editTextareaRef.current.focus();
+		}
+	}, [isEditing, editText]);
+
+	// Copy message text to clipboard
+	const handleCopy = async () => {
+		try {
+			await navigator.clipboard.writeText(textContent);
+			setCopied(true);
+			setTimeout(() => setCopied(false), 2000);
+		} catch (err) {
+			console.error("Failed to copy text:", err);
+		}
+	};
+
+	// Start editing
+	const handleStartEdit = () => {
+		setEditText(textContent);
+		setIsEditing(true);
+	};
+
+	// Cancel editing
+	const handleCancelEdit = () => {
+		setIsEditing(false);
+		setEditText("");
+	};
+
+	// Save edited message
+	const handleSaveEdit = () => {
+		if (editText.trim() && onEdit) {
+			onEdit(editText.trim());
+			setIsEditing(false);
+			setEditText("");
+		}
+	};
+
+	// Handle keyboard shortcuts in edit mode
+	const handleEditKeyDown = (e: React.KeyboardEvent<HTMLTextAreaElement>) => {
+		if (e.key === "Enter" && !e.shiftKey) {
+			e.preventDefault();
+			handleSaveEdit();
+		} else if (e.key === "Escape") {
+			handleCancelEdit();
+		}
+	};
+
+	// Extract text content from message parts
+	const textContent = useMemo(() => {
+		if (message.parts && Array.isArray(message.parts)) {
+			return message.parts
+				.filter((part: any) => part.type === "text")
+				.map((part: any) => part.text)
+				.join("");
+		}
+		return "";
+	}, [message.parts]);
+
+	// Extract image parts from message
+	const imageParts = useMemo(() => {
+		if (message.parts && Array.isArray(message.parts)) {
+			return message.parts.filter(
+				(part: any) =>
+					part.type === "file" && part.mediaType?.startsWith("image/"),
+			);
+		}
+		return [];
+	}, [message.parts]);
+
+	// Extract non-image file parts from message (PDFs, text files, etc.)
+	const fileParts = useMemo(() => {
+		if (message.parts && Array.isArray(message.parts)) {
+			return message.parts.filter(
+				(part: any) =>
+					part.type === "file" && !part.mediaType?.startsWith("image/"),
+			);
+		}
+		return [];
+	}, [message.parts]);
+
+	// Extract tool call parts from message (AI SDK v5 format: type is "tool-{toolName}")
+	const toolParts = useMemo(() => {
+		if (message.parts && Array.isArray(message.parts)) {
+			return message.parts.filter((part: any) => isToolUIPart(part));
+		}
+		return [];
+	}, [message.parts]);
+
+	// Use remend to complete partial markdown when streaming
+	const displayContent = useMemo(() => {
+		if (!textContent) return "";
+		if (isStreaming && !isUser) {
+			// Complete any unclosed markdown elements during streaming
+			return completeMarkdown(textContent);
+		}
+		return textContent;
+	}, [textContent, isStreaming, isUser]);
+
+	const isCompact = variant === "compact";
+
+	return (
+		<div
+			className={cn(
+				"flex gap-3 w-full",
+				isCompact ? "mb-3" : "mb-4",
+				isUser ? "justify-end" : "justify-start",
+			)}
+			aria-label={
+				isUser
+					? localization.A11Y_USER_MESSAGE
+					: localization.A11Y_ASSISTANT_MESSAGE
+			}
+		>
+			{/* Assistant Avatar */}
+			{!isUser && (
+				<div
+					className={cn(
+						"rounded-full bg-muted flex items-center justify-center shrink-0",
+						isCompact ? "w-7 h-7" : "w-8 h-8",
+					)}
+				>
+					<Bot className={cn(isCompact ? "w-4 h-4" : "w-5 h-5")} />
+				</div>
+			)}
+
+			{/* Message Content with Action Buttons */}
+			<div className="flex flex-col gap-1 max-w-[80%] group">
+				<div
+					className={cn(
+						"rounded-lg",
+						isCompact ? "px-3 py-2" : "px-4 py-3",
+						isUser ? "bg-primary text-primary-foreground" : "bg-muted",
+					)}
+				>
+					{isUser ? (
+						// User messages: files + images + text (or edit mode)
+						<div className="space-y-2">
+							{/* Attached files (non-images) */}
+							{fileParts.length > 0 && (
+								<div className="flex flex-wrap gap-2">
+									{fileParts.map((part: any, index: number) => (
+										<a
+											key={index}
+											href={part.url}
+											target="_blank"
+											rel="noopener noreferrer"
+											className="flex items-center gap-2 px-3 py-2 rounded-md bg-primary-foreground/10 hover:bg-primary-foreground/20 transition-colors"
+										>
+											<FileText className="h-4 w-4 shrink-0" />
+											<span className="text-xs truncate max-w-[150px]">
+												{part.filename || "File"}
+											</span>
+										</a>
+									))}
+								</div>
+							)}
+							{/* Attached images */}
+							{imageParts.length > 0 && (
+								<div className="flex flex-wrap gap-2">
+									{imageParts.map((part: any, index: number) => (
+										<ClickableImage
+											key={index}
+											src={part.url}
+											alt={part.filename || `Attached image ${index + 1}`}
+											width={150}
+											height={150}
+											className="rounded-md"
+											ImageComponent={ImageComponent}
+										/>
+									))}
+								</div>
+							)}
+							{/* Text content or edit mode */}
+							{isEditing ? (
+								<div className="space-y-2">
+									<textarea
+										ref={editTextareaRef}
+										value={editText}
+										onChange={(e) => setEditText(e.target.value)}
+										onKeyDown={handleEditKeyDown}
+										className={cn(
+											"w-full min-w-[200px] resize-none rounded-md border-0 bg-primary-foreground/10 p-2 text-sm",
+											"focus:outline-none focus:ring-1 focus:ring-primary-foreground/30",
+											"text-primary-foreground placeholder:text-primary-foreground/50",
+										)}
+										rows={1}
+									/>
+									<div className="flex items-center justify-end gap-2">
+										<Button
+											type="button"
+											size="icon-sm"
+											variant="ghost"
+											onClick={handleCancelEdit}
+											className="h-7 w-7 text-primary-foreground hover:bg-primary-foreground/20"
+											title={localization.MESSAGE_CANCEL}
+										>
+											<X className="h-4 w-4" />
+										</Button>
+										<Button
+											type="button"
+											size="icon-sm"
+											variant="ghost"
+											onClick={handleSaveEdit}
+											disabled={!editText.trim()}
+											className="h-7 w-7 text-primary-foreground hover:bg-primary-foreground/20 disabled:opacity-50"
+											title={localization.MESSAGE_SAVE}
+										>
+											<Send className="h-4 w-4" />
+										</Button>
+									</div>
+								</div>
+							) : (
+								textContent && (
+									<p
+										className={cn(
+											"whitespace-pre-wrap wrap-break-word",
+											isCompact ? "text-sm" : "text-sm",
+										)}
+									>
+										{textContent}
+									</p>
+								)
+							)}
+						</div>
+					) : (
+						// Assistant messages: rendered markdown + files + images + tool calls
+						<div className="wrap-break-word space-y-2">
+							{/* Any attached files (non-images) */}
+							{fileParts.length > 0 && (
+								<div className="flex flex-wrap gap-2">
+									{fileParts.map((part: any, index: number) => (
+										<a
+											key={index}
+											href={part.url}
+											target="_blank"
+											rel="noopener noreferrer"
+											className="flex items-center gap-2 px-3 py-2 rounded-md bg-muted-foreground/10 hover:bg-muted-foreground/20 transition-colors border"
+										>
+											<FileText className="h-4 w-4 shrink-0 text-muted-foreground" />
+											<span className="text-xs truncate max-w-[150px]">
+												{part.filename || "File"}
+											</span>
+										</a>
+									))}
+								</div>
+							)}
+							{/* Any attached/generated images */}
+							{imageParts.length > 0 && (
+								<div className="flex flex-wrap gap-2">
+									{imageParts.map((part: any, index: number) => (
+										<ClickableImage
+											key={index}
+											src={part.url}
+											alt={part.filename || `Image ${index + 1}`}
+											width={200}
+											height={200}
+											className="rounded-md"
+											ImageComponent={ImageComponent}
+										/>
+									))}
+								</div>
+							)}
+							{/* Tool calls */}
+							{toolParts.length > 0 && (
+								<div className="space-y-2">
+									{toolParts.map((part: any) => {
+										const toolName = getToolName(part);
+										const toolCallId = part.toolCallId;
+										const state = part.state as ToolCallState;
+										const input = part.input;
+										const output = part.output;
+										const errorText = part.errorText;
+										const isLoading =
+											state === "input-streaming" ||
+											state === "input-available";
+
+										const toolCallProps: ToolCallProps = {
+											toolCallId,
+											toolName,
+											state,
+											input,
+											output,
+											errorText,
+											isLoading,
+										};
+
+										// Check if there's a custom renderer for this tool
+										const CustomRenderer = toolRenderers?.[toolName];
+										if (CustomRenderer) {
+											return (
+												<CustomRenderer key={toolCallId} {...toolCallProps} />
+											);
+										}
+
+										// Use default tool call display
+										return (
+											<ToolCallDisplay key={toolCallId} {...toolCallProps} />
+										);
+									})}
+								</div>
+							)}
+							{/* Text content */}
+							{displayContent ? (
+								<MarkdownContent
+									markdown={displayContent}
+									variant="chat"
+									LinkComponent={Link ?? DefaultLink}
+									ImageComponent={ImageComponent}
+								/>
+							) : imageParts.length === 0 &&
+								fileParts.length === 0 &&
+								toolParts.length === 0 ? (
+								<span className="text-muted-foreground">...</span>
+							) : null}
+						</div>
+					)}
+				</div>
+
+				{/* Action Buttons */}
+				{!isStreaming && !isEditing && (
+					<div
+						className={cn(
+							"flex items-center gap-1 opacity-0 group-hover:opacity-100 transition-opacity",
+							isUser ? "justify-end" : "justify-start",
+						)}
+					>
+						{isUser ? (
+							// User message actions: Edit
+							onEdit && (
+								<Button
+									type="button"
+									size="icon-sm"
+									variant="ghost"
+									onClick={handleStartEdit}
+									className="h-7 w-7 bg-transparent text-muted-foreground hover:text-foreground hover:bg-muted/50"
+									title={localization.MESSAGE_EDIT}
+								>
+									<Pencil className="h-3.5 w-3.5" />
+								</Button>
+							)
+						) : (
+							// Assistant message actions: Copy and Retry
+							<>
+								<Button
+									type="button"
+									size="icon-sm"
+									variant="ghost"
+									onClick={handleCopy}
+									className="h-7 w-7 bg-transparent text-muted-foreground hover:text-foreground hover:bg-muted/50"
+									title={
+										copied
+											? localization.MESSAGE_COPIED
+											: localization.MESSAGE_COPY
+									}
+								>
+									{copied ? (
+										<Check className="h-3.5 w-3.5" />
+									) : (
+										<Copy className="h-3.5 w-3.5" />
+									)}
+								</Button>
+								{onRetry && (
+									<Button
+										type="button"
+										size="icon-sm"
+										variant="ghost"
+										onClick={onRetry}
+										disabled={isRetrying}
+										className="h-7 w-7 bg-transparent text-muted-foreground hover:text-foreground hover:bg-muted/50 disabled:opacity-50"
+										title={localization.MESSAGE_RETRY}
+									>
+										<RefreshCw
+											className={cn(
+												"h-3.5 w-3.5",
+												isRetrying && "animate-spin",
+											)}
+										/>
+									</Button>
+								)}
+							</>
+						)}
+					</div>
+				)}
+			</div>
+
+			{/* User Avatar */}
+			{isUser && (
+				<div
+					className={cn(
+						"rounded-full bg-primary flex items-center justify-center shrink-0 text-primary-foreground",
+						isCompact ? "w-7 h-7" : "w-8 h-8",
+					)}
+				>
+					<User className={cn(isCompact ? "w-4 h-4" : "w-5 h-5")} />
+				</div>
+			)}
+		</div>
+	);
+}