import { cn } from "@/lib/utils"
import { useEffect, useState } from "react"

export default function AnimateCountClient({
  count,
  className,
  minDigits,
}: {
  count: number
  className?: string
  minDigits?: number
}) {
  const [previousCount, setPreviousCount] = useState(count)

  useEffect(() => {
    if (count !== previousCount) {
      setTimeout(() => {
        setPreviousCount(count)
      }, 300)
    }
  }, [count])
  return (
    <AnimateCount
      key={count}
      preCount={previousCount}
      className={cn("inline-flex items-center leading-none", className)}
      minDigits={minDigits}
      data-issues-count-animation
    >
      {count}
    </AnimateCount>
  )
}

export function AnimateCount({
  children: count,
  className,
  preCount,
  minDigits = 1,
  ...props
}: {
  children: number
  className?: string
  preCount?: number
  minDigits?: number
}) {
  const currentDigits = count.toString().split("")
  const previousDigits = (
    preCount !== undefined ? preCount.toString() : count - 1 >= 0 ? (count - 1).toString() : "0"
  ).split("")

  // Ensure both numbers meet the minimum length requirement and maintain the same length for animation
  const maxLength = Math.max(previousDigits.length, currentDigits.length, minDigits)
  while (previousDigits.length < maxLength) {
    previousDigits.unshift("0")
  }
  while (currentDigits.length < maxLength) {
    currentDigits.unshift("0")
  }

  return (
    <div {...props} className={cn("flex h-[1em] items-center", className)}>
      {currentDigits.map((digit, index) => {
        const hasChanged = digit !== previousDigits[index]
        return (
          <div
            key={`${index}-${digit}`}
            className={cn("relative flex h-full min-w-[0.6em] items-center text-center", {
              "min-w-[0.2em]": digit === ".",
            })}
          >
            <div
              aria-hidden
              data-issues-count-exit
              className={cn(
                "absolute inset-0 flex items-center justify-center",
                hasChanged ? "animate" : "opacity-0",
              )}
            >
              {previousDigits[index]}
            </div>
            <div
              data-issues-count-enter
              className={cn(
                "absolute inset-0 flex items-center justify-center",
                hasChanged && "animate",
              )}
            >
              {digit}
            </div>
          </div>
        )
      })}
    </div>
  )
}