"use client";

import { cn } from "@aur-ui/ui/lib/utils";
import { Avatar as AvatarPrimitive } from "radix-ui";
import type * as React from "react";

/**
 * Avatar — Aur identity mark.
 *
 * Five size stops (xs 20px → xl 48px) driven by `data-size` on the root so
 * the fallback text, status dot, and group count all scale from one prop.
 * Fallback initials take an optional `tone` from the Badge soft palette;
 * the status dot (`AvatarBadge`) takes a semantic `variant`.
 */

type AvatarSize = "xs" | "sm" | "default" | "lg" | "xl";

function Avatar({
  className,
  size = "default",
  ...props
}: React.ComponentProps<typeof AvatarPrimitive.Root> & {
  size?: AvatarSize;
}) {
  return (
    <AvatarPrimitive.Root
      data-slot="avatar"
      data-size={size}
      // No overflow-hidden here: the status dot (AvatarBadge) sits on the
      // circle's edge and would be clipped. Image and fallback round
      // themselves instead.
      className={cn(
        "group/avatar relative flex size-8 shrink-0 rounded-full select-none",
        "data-[size=xs]:size-5 data-[size=sm]:size-6 data-[size=lg]:size-10 data-[size=xl]:size-12",
        className,
      )}
      {...props}
    />
  );
}

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

const avatarFallbackTones = {
  neutral: "bg-ink-wash text-ink-500",
  brand: "bg-brand-soft text-brand",
  positive: "bg-positive-soft text-positive",
  warning: "bg-warning-soft text-warning",
  danger: "bg-danger-soft text-danger",
} as const;

function AvatarFallback({
  className,
  tone = "neutral",
  ...props
}: React.ComponentProps<typeof AvatarPrimitive.Fallback> & {
  /** Soft tint for the initials, from the Badge palette. */
  tone?: keyof typeof avatarFallbackTones;
}) {
  return (
    <AvatarPrimitive.Fallback
      data-slot="avatar-fallback"
      data-tone={tone}
      className={cn(
        "flex size-full items-center justify-center rounded-full text-sm font-medium",
        avatarFallbackTones[tone],
        "group-data-[size=xs]/avatar:text-2xs group-data-[size=sm]/avatar:text-xs group-data-[size=lg]/avatar:text-md group-data-[size=xl]/avatar:text-lg",
        className,
      )}
      {...props}
    />
  );
}

const avatarBadgeVariants = {
  positive: "bg-positive",
  warning: "bg-warning",
  danger: "bg-danger",
  neutral: "bg-ink-400",
} as const;

function AvatarBadge({
  className,
  variant = "positive",
  ...props
}: React.ComponentProps<"span"> & {
  /** Semantic status colour; positive reads as "online". */
  variant?: keyof typeof avatarBadgeVariants;
}) {
  return (
    <span
      data-slot="avatar-badge"
      data-variant={variant}
      className={cn(
        "absolute right-0 bottom-0 z-10 inline-flex items-center justify-center rounded-full text-white ring-2 ring-surface select-none",
        avatarBadgeVariants[variant],
        "group-data-[size=xs]/avatar:size-1.5 group-data-[size=xs]/avatar:[&>svg]:hidden",
        "group-data-[size=sm]/avatar:size-2 group-data-[size=sm]/avatar:[&>svg]:hidden",
        "group-data-[size=default]/avatar:size-2.5 group-data-[size=default]/avatar:[&>svg]:size-2",
        "group-data-[size=lg]/avatar:size-3 group-data-[size=lg]/avatar:[&>svg]:size-2",
        "group-data-[size=xl]/avatar:size-3.5 group-data-[size=xl]/avatar:[&>svg]:size-2.5",
        className,
      )}
      {...props}
    />
  );
}

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-surface",
        className,
      )}
      {...props}
    />
  );
}

function AvatarGroupCount({
  className,
  ...props
}: React.ComponentProps<"div">) {
  return (
    <div
      data-slot="avatar-group-count"
      className={cn(
        "relative flex size-8 shrink-0 items-center justify-center rounded-full bg-ink-wash text-sm font-medium text-ink-500 ring-2 ring-surface [&>svg]:size-4",
        "group-has-data-[size=xs]/avatar-group:size-5 group-has-data-[size=xs]/avatar-group:text-2xs group-has-data-[size=xs]/avatar-group:[&>svg]:size-2.5",
        "group-has-data-[size=sm]/avatar-group:size-6 group-has-data-[size=sm]/avatar-group:text-xs group-has-data-[size=sm]/avatar-group:[&>svg]:size-3",
        "group-has-data-[size=lg]/avatar-group:size-10 group-has-data-[size=lg]/avatar-group:text-md group-has-data-[size=lg]/avatar-group:[&>svg]:size-5",
        "group-has-data-[size=xl]/avatar-group:size-12 group-has-data-[size=xl]/avatar-group:text-lg group-has-data-[size=xl]/avatar-group:[&>svg]:size-6",
        className,
      )}
      {...props}
    />
  );
}

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