import {
  ButtonHTMLAttributes,
  ComponentPropsWithoutRef,
  ElementType,
  ForwardedRef,
  forwardRef,
} from 'react';
import classnames from 'classnames';
import { Icon } from './Icon';

export interface ButtonProps<T extends ElementType>
  extends ButtonHTMLAttributes<HTMLButtonElement> {
  color?: 'primary' | 'secondary';
  size?: 'xs' | 'sm' | 'md';
  justify?: 'start' | 'center';
  forDropdown?: boolean;
  as?: T;
}

export const Button = forwardRef(function Button<T extends ElementType>(
  {
    className,
    as,
    justify = 'center',
    children,
    size = 'md',
    forDropdown,
    color,
    ...props
  }: ButtonProps<T> & Omit<ComponentPropsWithoutRef<T>, keyof ButtonProps<T>>,
  ref: ForwardedRef<HTMLButtonElement>,
) {
  const Component = as || 'button';
  return (
    <Component
      ref={ref}
      className={classnames(
        className,
        'rounded-md flex items-center',
        // 'active:translate-y-[0.5px] active:scale-[0.99]',
        justify === 'start' && 'justify-start',
        justify === 'center' && 'justify-center',
        size === 'md' && 'h-10 px-4',
        size === 'sm' && 'h-8 px-3 text-sm',
        size === 'xs' && 'h-6 px-2 text-sm',
        color === undefined && 'hover:bg-gray-500/[0.1] text-gray-800 hover:text-gray-900',
        color === 'primary' && 'bg-blue-500 hover:bg-blue-500/90 text-white',
        color === 'secondary' && 'bg-violet-500 hover:bg-violet-500/90 text-white',
      )}
      {...props}
    >
      {children}
      {forDropdown && <Icon icon="triangle-down" className="ml-1 -mr-1" />}
    </Component>
  );
});
