import { type PropsWithChildren, useState } from 'react'

import { tooltipVariants } from './variants'

import styles from './tooltip.module.css'
import { Typography } from '../Typography'

type TooltipPosition = 'left' | 'right' | 'top' | 'bottom'
type VerticalArrow = 'top' | 'bottom' | 'center'
type HorizontalArrow = 'left' | 'right' | 'center'

type ValidArrowMap = {
  left: VerticalArrow
  right: VerticalArrow
  top: HorizontalArrow
  bottom: HorizontalArrow
}

type ValidArrow<P extends TooltipPosition> = P extends keyof ValidArrowMap
  ? ValidArrowMap[P]
  : never

interface TooltipProps<P extends TooltipPosition = TooltipPosition> {
  heading?: string
  text?: string
  position: P
  arrow: ValidArrow<P>
  isTouchable?: boolean
  isVisible?: boolean
}

export function Tooltip<P extends TooltipPosition>({
  heading,
  text,
  position,
  arrow,
  children,
  isTouchable = false,
  isVisible = true,
}: PropsWithChildren<TooltipProps<P>>) {
  const className = tooltipVariants({ position, arrow })
  const [isActive, setIsActive] = useState(false)

  function handleToggle() {
    setIsActive((prevState) => !prevState)
  }

  function handleKeyDown(event: React.KeyboardEvent<HTMLDivElement>) {
    if (event.key === 'Enter' || event.key === ' ') {
      event.preventDefault()
      handleToggle()
    }
  }

  if (!isVisible) {
    return <> {children} </>
  }

  return (
    <div
      className={styles.tooltipContainer}
      role="tooltip"
      aria-label={text}
      tabIndex={0}
      onClick={isTouchable ? undefined : handleToggle}
      onTouchStart={isTouchable ? handleToggle : undefined}
      onKeyDown={handleKeyDown}
      data-active={isActive}
    >
      <div className={className}>
        {heading ? (
          <Typography variant="Body/Supporting text (caption)/smBold">
            <p>{heading}</p>
          </Typography>
        ) : null}
        {text ? (
          <Typography variant="Body/Supporting text (caption)/smRegular">
            <p>{text} </p>
          </Typography>
        ) : null}
      </div>
      {children}
    </div>
  )
}