import React from 'react'

import { Box, BoxOwnProps } from './Box'
import { SVG, SVGProps } from './SVG'
import { Assign, ForwardRef } from './types'
import { __internalProps } from './util'

const CheckboxChecked = (props: SVGProps) => (
  <SVG {...props}>
    <path d="M19 3H5c-1.11 0-2 .9-2 2v14c0 1.1.89 2 2 2h14c1.11 0 2-.9 2-2V5c0-1.1-.89-2-2-2zm-9 14l-5-5 1.41-1.41L10 14.17l7.59-7.59L19 8l-9 9z" />
  </SVG>
)

const CheckboxUnchecked = (props: SVGProps) => (
  <SVG {...props}>
    <path d="M19 5v14H5V5h14m0-2H5c-1.1 0-2 .9-2 2v14c0 1.1.9 2 2 2h14c1.1 0 2-.9 2-2V5c0-1.1-.9-2-2-2z" />
  </SVG>
)

const CheckboxIcon = (props: SVGProps) => (
  <React.Fragment>
    <CheckboxChecked
      {...props}
      {...__internalProps({
        __css: {
          display: 'none',
          'input:checked ~ &': {
            display: 'block',
          },
        },
      })}
    />
    <CheckboxUnchecked
      {...props}
      {...__internalProps({
        __css: {
          display: 'block',
          'input:checked ~ &': {
            display: 'none',
          },
        },
      })}
    />
  </React.Fragment>
)

export interface CheckboxProps
  extends Assign<React.ComponentPropsWithRef<'input'>, BoxOwnProps> {}

/**
 * Form checkbox input component
 *
 * Checkbox variants can be defined in `theme.forms` and the
 * component uses the `theme.forms.checkbox` variant by default.
 * @see https://theme-ui.com/components/checkbox/
 */
export const Checkbox: ForwardRef<HTMLInputElement, CheckboxProps> =
  React.forwardRef(function Checkbox(
    { className, sx, variant = 'checkbox', children, ...props },
    ref
  ) {
    return (
      <Box sx={{ minWidth: 'min-content' }}>
        <Box
          ref={ref}
          as="input"
          type="checkbox"
          {...props}
          sx={{
            position: 'absolute',
            opacity: 0,
            zIndex: -1,
            width: 1,
            height: 1,
            overflow: 'hidden',
          }}
        />
        <Box
          as={CheckboxIcon}
          aria-hidden="true"
          variant={variant}
          className={className}
          sx={sx}
          {...__internalProps({
            __themeKey: 'forms',
            __css: {
              mr: 2,
              borderRadius: 4,
              color: 'gray',
              flexShrink: 0,
              'input:checked ~ &': {
                color: 'primary',
              },
              'input:focus ~ &': {
                color: 'primary',
                bg: 'highlight',
              },
            },
          })}
        />
        {children}
      </Box>
    )
  })