index.tsx•3.64 kB
import { cva, type VariantProps } from 'class-variance-authority';
import * as React from 'react';
import { Avatar, AvatarImage, AvatarFallback } from '@/components/ui/avatar';
import { Button, ButtonProps } from '@/components/ui/button';
import { cn } from '@/lib/utils';
import MessageLoading from './message-loading';
// ChatBubble
const chatBubbleVariant = cva('flex gap-2 w-full items-start relative group', {
variants: {
variant: {
received: 'self-start',
sent: 'self-end flex-row-reverse',
},
layout: {
default: '',
ai: 'max-w-full w-full items-center',
},
},
defaultVariants: {
variant: 'received',
layout: 'default',
},
});
interface ChatBubbleProps
extends React.HTMLAttributes<HTMLDivElement>,
VariantProps<typeof chatBubbleVariant> {}
const ChatBubble = React.forwardRef<HTMLDivElement, ChatBubbleProps>(
({ className, variant, layout, children, ...props }, ref) => (
<div
className={cn(
chatBubbleVariant({ variant, layout, className }),
'relative group',
)}
ref={ref}
{...props}
>
{React.Children.map(children, (child) =>
React.isValidElement(child) && typeof child.type !== 'string'
? React.cloneElement(child, {
variant,
layout,
} as React.ComponentProps<typeof child.type>)
: child,
)}
</div>
),
);
ChatBubble.displayName = 'ChatBubble';
// ChatBubbleAvatar
interface ChatBubbleAvatarProps {
src?: string;
fallback?: React.ReactNode;
className?: string;
}
const ChatBubbleAvatar: React.FC<ChatBubbleAvatarProps> = ({
src,
fallback,
className,
}) => (
<Avatar>
<AvatarImage
src={src}
alt="Avatar"
className={cn('aspect-square p-2', className)}
/>
<AvatarFallback className="bg-background border">{fallback}</AvatarFallback>
</Avatar>
);
// ChatBubbleMessage
const chatBubbleMessageVariants = cva('px-1', {
variants: {
variant: {
received: 'bg-background text-foreground rounded-3xl py-2',
sent: 'bg-accent text-accent-foreground rounded-3xl py-3 px-5',
},
layout: {
default: '',
ai: 'border-t w-full rounded-none bg-transparent',
},
},
defaultVariants: {
variant: 'received',
layout: 'default',
},
});
interface ChatBubbleMessageProps
extends React.HTMLAttributes<HTMLDivElement>,
VariantProps<typeof chatBubbleMessageVariants> {
isLoading?: boolean;
}
const ChatBubbleMessage = React.forwardRef<
HTMLDivElement,
ChatBubbleMessageProps
>(
(
{ className, variant, layout, isLoading = false, children, ...props },
ref,
) => (
<div
className={cn(
chatBubbleMessageVariants({ variant, layout, className }),
'break-words max-w-full whitespace-pre-wrap overflow-x-auto',
)}
ref={ref}
{...props}
>
{isLoading ? (
<div className="flex items-center space-x-2">
<MessageLoading />
</div>
) : (
children
)}
</div>
),
);
ChatBubbleMessage.displayName = 'ChatBubbleMessage';
// ChatBubbleAction
type ChatBubbleActionProps = ButtonProps & {
icon: React.ReactNode;
};
const ChatBubbleAction: React.FC<ChatBubbleActionProps> = ({
icon,
onClick,
className,
variant = 'ghost',
size = 'icon',
...props
}) => (
<Button
variant={variant}
size={size}
className={className}
onClick={onClick}
{...props}
>
{icon}
</Button>
);
export {
ChatBubble,
ChatBubbleAvatar,
ChatBubbleMessage,
chatBubbleVariant,
chatBubbleMessageVariants,
ChatBubbleAction,
};