import {
  Select as AriaSelect,
  SelectValue,
  Popover,
  ListBox,
  Button,
} from 'react-aria-components'
import { cx } from 'class-variance-authority'

import { MaterialIcon } from '../Icons/MaterialIcon'
import { Typography } from '../Typography'
import { SelectItem } from './SelectItem'
import { SelectFilter } from './SelectFilter'

import type { SelectProps, SelectFilterProps } from './types'

import styles from './select.module.css'
import { useState } from 'react'

export function Select({
  name,
  label,
  items,
  isDisabled,
  icon,
  itemIcon,
  ...props
}: SelectProps | SelectFilterProps) {
  const [isOpen, setIsOpen] = useState(false)

  if (props.enableFiltering) {
    return (
      <SelectFilter
        name={name}
        label={label}
        items={items}
        icon={icon}
        itemIcon={itemIcon}
        {...props}
      />
    )
  }
  const iconColor = isDisabled ? 'Icon/Interactive/Disabled' : 'Icon/Default'

  return (
    <AriaSelect
      className={styles.select}
      name={name}
      aria-label={label}
      isDisabled={isDisabled}
      onOpenChange={setIsOpen}
      {...props}
    >
      <Button className={cx(styles.inner, styles.button)}>
        {icon ? (
          <MaterialIcon
            icon={icon}
            size={24}
            color={iconColor}
            aria-hidden="true"
          />
        ) : null}
        <SelectValue className={cx(styles.displayText, styles.selectValue)}>
          {({ selectedText }) => {
            return (
              <>
                <Typography
                  variant={
                    selectedText || isOpen
                      ? 'Label/xsRegular'
                      : 'Body/Paragraph/mdRegular'
                  }
                >
                  <span className={styles.label}>{label}</span>
                </Typography>
                <Typography variant="Body/Paragraph/mdRegular">
                  <span className={styles.selectedText}>{selectedText}</span>
                </Typography>
              </>
            )
          }}
        </SelectValue>

        <MaterialIcon
          icon="chevron_right"
          size={24}
          color={iconColor}
          aria-hidden="true"
          className={styles.chevron}
        />
      </Button>

      <Popover className={styles.popover} shouldFlip={false}>
        <ListBox className={styles.listBox}>
          {items.map((item, idx) => (
            <SelectItem
              key={`${item.value}-${idx}`}
              id={item.value}
              icon={item.icon || itemIcon}
              isDisabled={item.isDisabled}
            >
              {item.label}
            </SelectItem>
          ))}
        </ListBox>
      </Popover>
    </AriaSelect>
  )
}
