'use client';

import * as React from 'react';
import * as AvatarPrimitive from 'radix-ui/avatar';

import { cva, type VariantProps } from '@/lib/cva';

import { cn } from '@/lib/utils';
import { colorPalettes } from '@/lib/cva-presets';

const avatarVariants = cva(
  'group/avatar relative flex shrink-0 items-center justify-center overflow-hidden font-medium select-none',
  {
    variants: {
      /*
       * The fill lives on the root rather than the fallback, so a loaded image
       * simply covers it and there is one place that decides how an avatar is
       * coloured. CBAR draws no `ghost` or `surface` avatar, so this is the
       * three treatments it does draw.
       */
      variant: {
        solid: 'bg-(--ctl-solid) text-(--ctl-solid-fg)',
        subtle: 'bg-(--ctl-subtle) text-(--ctl-fg)',
        outline: 'border border-(--ctl-border) text-(--ctl-fg)',
      },
      colorPalette: colorPalettes,
      size: {
        '2xs': 'size-6 text-[10px]',
        xs: 'size-8 text-xs',
        sm: 'size-9 text-sm',
        md: 'size-10 text-base',
        lg: 'size-12 text-xl',
        xl: 'size-16 text-2xl',
      },
      /** CBAR names these `shape` and `square`; the round one is the default. */
      shape: {
        circle: 'rounded-full',
        square: 'rounded-md',
      },
    },
    defaultVariants: {
      variant: 'subtle',
      colorPalette: 'black',
      size: 'md',
      shape: 'circle',
    },
  }
);

export interface AvatarProps
  extends React.ComponentProps<typeof AvatarPrimitive.Root>,
    VariantProps<typeof avatarVariants> {}

export type AvatarImageProps = React.ComponentProps<typeof AvatarPrimitive.Image>;
export type AvatarFallbackProps = React.ComponentProps<typeof AvatarPrimitive.Fallback>;

/**
 * User or entity image with a text fallback.
 *
 * The fallback renders only after the image fails or is still loading, so the
 * initials never flash over a successfully loaded photo.
 *
 * ```tsx
 * <Avatar size="lg" colorPalette="primary" variant="solid">
 *   <AvatarImage src={user.photo} alt="" />
 *   <AvatarFallback>{initials}</AvatarFallback>
 * </Avatar>
 * ```
 */
function Avatar({
  className,
  variant,
  colorPalette,
  size = 'md',
  shape,
  ...props
}: AvatarProps) {
  return (
    <AvatarPrimitive.Root
      data-slot="avatar"
      data-size={size}
      className={cn(avatarVariants({ variant, colorPalette, size, shape }), className)}
      {...props}
    />
  );
}

function AvatarImage({ className, ...props }: AvatarImageProps) {
  return (
    <AvatarPrimitive.Image
      data-slot="avatar-image"
      className={cn('aspect-square size-full', className)}
      {...props}
    />
  );
}

/**
 * Initials shown until the image loads. Transparent on purpose — the root owns
 * the fill and the type size, so the two can never drift apart.
 */
function AvatarFallback({ className, ...props }: AvatarFallbackProps) {
  return (
    <AvatarPrimitive.Fallback
      data-slot="avatar-fallback"
      className={cn('flex size-full items-center justify-center uppercase', className)}
      {...props}
    />
  );
}

/** Presence dot or small icon pinned to the avatar's bottom-right corner. */
function AvatarBadge({ className, ...props }: React.ComponentProps<'span'>) {
  return (
    <span
      data-slot="avatar-badge"
      className={cn(
        'absolute right-0 bottom-0 z-10 inline-flex items-center justify-center rounded-full bg-primary text-primary-foreground ring-2 ring-background select-none',
        /* Scales with the parent avatar. Below `sm` the dot is too small to
           carry a glyph, so any icon inside it is hidden rather than squashed. */
        'group-data-[size=2xs]/avatar:size-1.5 group-data-[size=2xs]/avatar:[&>svg]:hidden',
        'group-data-[size=xs]/avatar:size-2 group-data-[size=xs]/avatar:[&>svg]:hidden',
        'group-data-[size=sm]/avatar:size-2.5 group-data-[size=sm]/avatar:[&>svg]:size-2',
        'group-data-[size=md]/avatar:size-3 group-data-[size=md]/avatar:[&>svg]:size-2',
        'group-data-[size=lg]/avatar:size-3.5 group-data-[size=lg]/avatar:[&>svg]:size-2.5',
        'group-data-[size=xl]/avatar:size-4 group-data-[size=xl]/avatar:[&>svg]:size-3',
        className
      )}
      {...props}
    />
  );
}

/** Overlapping stack of avatars. Pair with `AvatarGroupCount` for an overflow. */
function AvatarGroup({ className, ...props }: React.ComponentProps<'div'>) {
  return (
    <div
      data-slot="avatar-group"
      className={cn(
        'group/avatar-group flex -space-x-2 *:data-[slot=avatar]:ring-2 *:data-[slot=avatar]:ring-background',
        className
      )}
      {...props}
    />
  );
}

function AvatarGroupCount({ className, ...props }: React.ComponentProps<'div'>) {
  return (
    <div
      data-slot="avatar-group-count"
      className={cn(
        'relative flex size-10 shrink-0 items-center justify-center rounded-full bg-muted text-base text-muted-foreground ring-2 ring-background',
        /* Mirrors the avatar ladder so the overflow chip matches the faces it
           sits beside, whatever size the group was given. */
        'group-has-data-[size=2xs]/avatar-group:size-6 group-has-data-[size=2xs]/avatar-group:text-[10px]',
        'group-has-data-[size=xs]/avatar-group:size-8 group-has-data-[size=xs]/avatar-group:text-xs',
        'group-has-data-[size=sm]/avatar-group:size-9 group-has-data-[size=sm]/avatar-group:text-sm',
        'group-has-data-[size=lg]/avatar-group:size-12 group-has-data-[size=lg]/avatar-group:text-xl',
        'group-has-data-[size=xl]/avatar-group:size-16 group-has-data-[size=xl]/avatar-group:text-2xl',
        '[&>svg]:size-4',
        className
      )}
      {...props}
    />
  );
}

export { Avatar, AvatarImage, AvatarFallback, AvatarBadge, AvatarGroup, AvatarGroupCount };