import { Button } from "@/shared/ui/button"; import { ButtonGroup, ButtonGroupText } from "@/shared/ui/button-group"; import { Tooltip, TooltipContent, TooltipProvider, TooltipTrigger, } from "@/shared/ui/tooltip"; import { isExternalHref } from "@/shared/lib/isExternalHref"; import { LinkSafetyModal } from "@/shared/ui/ai-elements/link-safety-modal"; import { cn } from "@/shared/lib/cn"; import { cjk } from "@streamdown/cjk"; import { code } from "@streamdown/code"; import { math } from "@streamdown/math"; import { mermaid } from "@streamdown/mermaid"; import type { UIMessage } from "ai"; import { ChevronLeftIcon, ChevronRightIcon } from "lucide-react"; import type { ComponentProps, HTMLAttributes, ReactElement } from "react"; import { createContext, memo, useCallback, useContext, useEffect, useMemo, useState, } from "react"; import { Streamdown } from "streamdown"; import { useTranslation } from "react-i18next"; export type MessageProps = HTMLAttributes & { from: UIMessage["role"]; }; export const Message = ({ className, from, ...props }: MessageProps) => (

); export type MessageContentProps = HTMLAttributes; export const MessageContent = ({ children, className, ...props }: MessageContentProps) => (

{children}

); export type MessageActionsProps = ComponentProps<"div">; export const MessageActions = ({ className, children, ...props }: MessageActionsProps) => (

{children}

); export type MessageActionProps = ComponentProps & { tooltip?: string; label?: string; }; export const MessageAction = ({ tooltip, children, label, variant = "ghost", size = "icon-sm", ...props }: MessageActionProps) => { const button = ( ); if (tooltip) { return ( {button}

{tooltip}

); } return button; }; interface MessageBranchContextType { currentBranch: number; totalBranches: number; goToPrevious: () => void; goToNext: () => void; branches: ReactElement[]; setBranches: (branches: ReactElement[]) => void; } const MessageBranchContext = createContext( null, ); const useMessageBranch = () => { const context = useContext(MessageBranchContext); if (!context) { throw new Error( "MessageBranch components must be used within MessageBranch", ); } return context; }; export type MessageBranchProps = HTMLAttributes & { defaultBranch?: number; onBranchChange?: (branchIndex: number) => void; }; export const MessageBranch = ({ defaultBranch = 0, onBranchChange, className, ...props }: MessageBranchProps) => { const [currentBranch, setCurrentBranch] = useState(defaultBranch); const [branches, setBranches] = useState([]); const handleBranchChange = useCallback( (newBranch: number) => { setCurrentBranch(newBranch); onBranchChange?.(newBranch); }, [onBranchChange], ); const goToPrevious = useCallback(() => { const newBranch = currentBranch > 0 ? currentBranch - 1 : branches.length - 1; handleBranchChange(newBranch); }, [currentBranch, branches.length, handleBranchChange]); const goToNext = useCallback(() => { const newBranch = currentBranch < branches.length - 1 ? currentBranch + 1 : 0; handleBranchChange(newBranch); }, [currentBranch, branches.length, handleBranchChange]); const contextValue = useMemo( () => ({ branches, currentBranch, goToNext, goToPrevious, setBranches, totalBranches: branches.length, }), [branches, currentBranch, goToNext, goToPrevious], ); return (

div]:pb-0", className)} {...props} /> ); }; export type MessageBranchContentProps = HTMLAttributes; export const MessageBranchContent = ({ children, ...props }: MessageBranchContentProps) => { const { currentBranch, setBranches, branches } = useMessageBranch(); const childrenArray = useMemo( () => (Array.isArray(children) ? children : [children]), [children], ); // Use useEffect to update branches when they change useEffect(() => { if (branches.length !== childrenArray.length) { setBranches(childrenArray); } }, [childrenArray, branches, setBranches]); return childrenArray.map((branch, index) => (

div]:pb-0", index === currentBranch ? "block" : "hidden", )} key={branch.key} {...props} > {branch}

)); }; export type MessageBranchSelectorProps = ComponentProps; export const MessageBranchSelector = ({ className, ...props }: MessageBranchSelectorProps) => { const { totalBranches } = useMessageBranch(); // Don't render if there's only one branch if (totalBranches <= 1) { return null; } return ( *:not(:first-child)]:rounded-l-md [&>*:not(:last-child)]:rounded-r-md", className, )} orientation="horizontal" {...props} /> ); }; export type MessageBranchPreviousProps = ComponentProps; export const MessageBranchPrevious = ({ children, ...props }: MessageBranchPreviousProps) => { const { t } = useTranslation("common"); const { goToPrevious, totalBranches } = useMessageBranch(); return ( ); }; export type MessageBranchNextProps = ComponentProps; export const MessageBranchNext = ({ children, ...props }: MessageBranchNextProps) => { const { t } = useTranslation("common"); const { goToNext, totalBranches } = useMessageBranch(); return ( ); }; export type MessageBranchPageProps = HTMLAttributes; export const MessageBranchPage = ({ className, ...props }: MessageBranchPageProps) => { const { t } = useTranslation("common"); const { currentBranch, totalBranches } = useMessageBranch(); return ( {t("components.messageBranch.page", { current: currentBranch + 1, total: totalBranches, })} ); }; export type MessageResponseProps = ComponentProps; const streamdownPlugins = { cjk, code, math, mermaid }; type OpenLinkSafetyModal = (url: string) => void; const LinkSafetyContext = createContext(null); /** * Custom link component that splits behavior by link type: * - External links → with preventDefault that opens a LinkSafetyModal via context * - Internal links → plain so useArtifactLinkHandler can intercept via closest("a") * * Both render as elements. useArtifactLinkHandler has an early return for external * hrefs, so there is no conflict with its delegated click handler. * * This replaces Streamdown's built-in linkSafety which renders