Assist_Design/apps/portal/src/components/atoms/button.tsx

import type { AnchorHTMLAttributes, ButtonHTMLAttributes, ReactNode } from "react";
import { forwardRef } from "react";
import Link from "next/link";
import { cva, type VariantProps } from "class-variance-authority";
import { cn } from "@/lib/utils";
import { Spinner } from "./Spinner";

const buttonVariants = cva(
  "group inline-flex items-center justify-center rounded-lg text-sm font-medium transition-colors duration-[var(--cp-duration-normal)] focus-visible:outline-none focus-visible:ring-2 focus-visible:ring-ring focus-visible:ring-offset-2 focus-visible:ring-offset-background disabled:opacity-50 disabled:pointer-events-none active:scale-[0.98]",
  {
    variants: {
      variant: {
        default:
          "bg-primary text-primary-foreground hover:bg-primary-hover shadow-[var(--cp-shadow-1)] hover:shadow-[var(--cp-shadow-2)]",
        destructive:
          "bg-danger text-danger-foreground hover:bg-danger/90 shadow-[var(--cp-shadow-1)] hover:shadow-[var(--cp-shadow-2)]",
        outline:
          "border border-border bg-background text-foreground hover:bg-muted shadow-[var(--cp-shadow-1)] hover:shadow-[var(--cp-shadow-2)]",
        secondary:
          "bg-secondary text-secondary-foreground hover:bg-secondary/80 shadow-[var(--cp-shadow-1)] hover:shadow-[var(--cp-shadow-2)]",
        ghost: "text-foreground hover:bg-muted",
        link: "underline-offset-4 hover:underline text-primary",
      },
      size: {
        default: "h-11 py-2.5 px-4",
        sm: "h-9 px-3 text-xs",
        lg: "h-12 px-6 text-base",
      },
    },
    defaultVariants: {
      variant: "default",
      size: "default",
    },
  }
);

interface ButtonExtras {
  leftIcon?: ReactNode;
  rightIcon?: ReactNode;
  loading?: boolean;
  isLoading?: boolean;
  loadingText?: ReactNode;
}

type ButtonAsAnchorProps = {
  as: "a";
  href: string;
} & AnchorHTMLAttributes<HTMLAnchorElement> &
  VariantProps<typeof buttonVariants> &
  ButtonExtras;

type ButtonAsButtonProps = {
  as?: "button";
} & ButtonHTMLAttributes<HTMLButtonElement> &
  VariantProps<typeof buttonVariants> &
  ButtonExtras;

export type ButtonProps = ButtonAsAnchorProps | ButtonAsButtonProps;

const Button = forwardRef<HTMLButtonElement | HTMLAnchorElement, ButtonProps>((props, ref) => {
  const {
    leftIcon,
    rightIcon,
    loading: loadingProp,
    isLoading,
    loadingText,
    children,
    ...rest
  } = props;

  const loading = loadingProp ?? isLoading ?? false;

  if (props.as === "a") {
    const { className, variant, size, as: _as, href, ...anchorProps } = rest as ButtonAsAnchorProps;
    void _as;

    const isExternal = href.startsWith("http") || href.startsWith("mailto:");

    if (isExternal) {
      return (
        <a
          className={cn(buttonVariants({ variant, size, className }))}
          href={href}
          ref={ref as React.Ref<HTMLAnchorElement>}
          aria-busy={loading || undefined}
          {...anchorProps}
        >
          <span className="inline-flex items-center justify-center gap-2">
            {loading ? <Spinner size="sm" /> : leftIcon}
            <span className="flex-1">{loading ? (loadingText ?? children) : children}</span>
            {!loading && rightIcon ? (
              <span className="transition-transform duration-200 group-hover:translate-x-0.5">
                {rightIcon}
              </span>
            ) : null}
          </span>
        </a>
      );
    }

    return (
      <Link
        className={cn(buttonVariants({ variant, size, className }))}
        href={href}
        ref={ref as React.Ref<HTMLAnchorElement>}
        aria-busy={loading || undefined}
        {...anchorProps}
      >
        <span className="inline-flex items-center justify-center gap-2">
          {loading ? <Spinner size="sm" /> : leftIcon}
          <span className="flex-1">{loading ? (loadingText ?? children) : children}</span>
          {!loading && rightIcon ? (
            <span className="transition-transform duration-200 group-hover:translate-x-0.5">
              {rightIcon}
            </span>
          ) : null}
        </span>
      </Link>
    );
  }

  const {
    className,
    variant,
    size,
    as: _as,
    disabled,
    ...buttonProps
  } = rest as ButtonAsButtonProps;
  void _as;
  return (
    <button
      className={cn(buttonVariants({ variant, size, className }))}
      ref={ref as React.Ref<HTMLButtonElement>}
      disabled={loading || disabled}
      aria-busy={loading || undefined}
      {...buttonProps}
    >
      <span className="inline-flex items-center justify-center gap-2">
        {loading ? <Spinner size="sm" /> : leftIcon}
        <span className="flex-1">{loading ? (loadingText ?? children) : children}</span>
        {!loading && rightIcon ? (
          <span className="transition-transform duration-200 group-hover:translate-x-0.5">
            {rightIcon}
          </span>
        ) : null}
      </span>
    </button>
  );
});
Button.displayName = "Button";

export { Button, buttonVariants };