"use client"

import Image from "next/image"
import * as React from "react"
import { Avatar as AvatarPrimitive } from "radix-ui"

import { cn } from "@/shared/lib/utils"

type AvatarImageStatus = "idle" | "loading" | "loaded" | "error"

type AvatarImageStatusContextValue = {
  status: AvatarImageStatus
  setStatus: React.Dispatch<React.SetStateAction<AvatarImageStatus>>
}

const AvatarImageStatusContext =
  React.createContext<AvatarImageStatusContextValue | null>(null)

function Avatar({
  className,
  size = "default",
  ...props
}: React.ComponentProps<typeof AvatarPrimitive.Root> & {
  size?: "default" | "sm" | "lg"
}) {
  const [status, setStatus] = React.useState<AvatarImageStatus>("idle")

  return (
    <AvatarImageStatusContext.Provider value={{ status, setStatus }}>
      <AvatarPrimitive.Root
        data-slot="avatar"
        data-size={size}
        className={cn(
          "group/avatar relative flex size-8 shrink-0 rounded-full select-none after:absolute after:inset-0 after:rounded-full after:border after:border-border after:mix-blend-darken data-[size=lg]:size-10 data-[size=sm]:size-6 dark:after:mix-blend-lighten",
          className
        )}
        {...props}
      />
    </AvatarImageStatusContext.Provider>
  )
}

function AvatarImage({
  className,
  sizes = "100%",
  onLoadingStatusChange,
  onLoad,
  onError,
  ...props
}: Omit<React.ComponentPropsWithoutRef<typeof Image>, "fill"> &
  Pick<React.ComponentProps<typeof AvatarPrimitive.Image>, "onLoadingStatusChange">) {
  const imageStatusContext = React.useContext(AvatarImageStatusContext)
  const { setStatus } = imageStatusContext ?? {}

  React.useEffect(() => {
    setStatus?.("loading")
    onLoadingStatusChange?.("loading")
  }, [onLoadingStatusChange, setStatus])

  return (
    <Image
      data-slot="avatar-image"
      fill
      sizes={sizes}
      className={cn(
        "aspect-square size-full rounded-full object-cover",
        className
      )}
      onLoad={(event) => {
        setStatus?.("loaded")
        onLoadingStatusChange?.("loaded")
        onLoad?.(event)
      }}
      onError={(event) => {
        setStatus?.("error")
        onLoadingStatusChange?.("error")
        onError?.(event)
      }}
      {...props}
    />
  )
}

function AvatarFallback({
  className,
  ...props
}: React.ComponentProps<typeof AvatarPrimitive.Fallback>) {
  const imageStatusContext = React.useContext(AvatarImageStatusContext)

  if (imageStatusContext?.status === "loaded") {
    return null
  }

  return (
    <AvatarPrimitive.Fallback
      data-slot="avatar-fallback"
      className={cn(
        "flex size-full items-center justify-center rounded-full bg-muted text-sm text-muted-foreground group-data-[size=sm]/avatar:text-xs",
        className
      )}
      {...props}
    />
  )
}

function AvatarBadge({ className, ...props }: React.ComponentProps<"span">) {
  return (
    <span
      data-slot="avatar-badge"
      className={cn(
        "absolute end-0 bottom-0 z-10 inline-flex items-center justify-center rounded-full bg-primary text-primary-foreground bg-blend-color ring-2 ring-background select-none",
        "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",
        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-background",
        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-muted text-sm text-muted-foreground ring-2 ring-background group-has-data-[size=lg]/avatar-group:size-10 group-has-data-[size=sm]/avatar-group:size-6 [&>svg]:size-4 group-has-data-[size=lg]/avatar-group:[&>svg]:size-5 group-has-data-[size=sm]/avatar-group:[&>svg]:size-3",
        className
      )}
      {...props}
    />
  )
}

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