import * as React from "react";

import { cn } from "../utils";
import type { TButtonSizes, TButtonVariant } from "./helper";
import { getButtonStyling, getIconStyling } from "./helper";

export interface ButtonProps extends React.ButtonHTMLAttributes<HTMLButtonElement> {
  variant?: TButtonVariant;
  size?: TButtonSizes;
  className?: string;
  loading?: boolean;
  disabled?: boolean;
  appendIcon?: any;
  prependIcon?: any;
  children: React.ReactNode;
}

const Button = React.forwardRef(function Button(
  props: ButtonProps,
  ref: React.ForwardedRef<HTMLButtonElement>,
) {
  const {
    variant = "primary",
    size = "md",
    className = "",
    type = "button",
    loading = false,
    disabled = false,
    prependIcon = null,
    appendIcon = null,
    children,
    ...rest
  } = props;

  const buttonStyle = getButtonStyling(variant, size, disabled || loading);
  const buttonIconStyle = getIconStyling(size);

  return (
    <button
      ref={ref}
      type={type}
      className={cn(buttonStyle, className)}
      disabled={disabled || loading}
      {...rest}
    >
      {prependIcon && (
        <div className={buttonIconStyle}>
          {React.cloneElement(prependIcon, { strokeWidth: 2 })}
        </div>
      )}
      {children}
      {appendIcon && (
        <div className={buttonIconStyle}>
          {React.cloneElement(appendIcon, { strokeWidth: 2 })}
        </div>
      )}
    </button>
  );
});

Button.displayName = "buzzwareTech-ui-button";

export { Button };