vega-frontend-monorepo/libs/ui-toolkit/src/components/button/button.tsx

import { AnchorHTMLAttributes, ButtonHTMLAttributes, forwardRef } from 'react';
import classNames from 'classnames';
import { Icon, IconName } from '../icon';
import {
  includesLeftPadding,
  includesRightPadding,
  includesBorderWidth,
} from '../../utils/class-names';

interface CommonProps {
  children?: React.ReactNode;
  variant?: 'primary' | 'secondary' | 'accent' | 'inline' | 'inline-link';
  className?: string;
  prependIconName?: IconName;
  appendIconName?: IconName;
}
export interface ButtonProps
  extends ButtonHTMLAttributes<HTMLButtonElement>,
    CommonProps {}

export interface AnchorButtonProps
  extends AnchorHTMLAttributes<HTMLAnchorElement>,
    CommonProps {}

const getClassName = (
  className: CommonProps['className'],
  variant: CommonProps['variant']
) => {
  const paddingLeftProvided = includesLeftPadding(className);
  const paddingRightProvided = includesRightPadding(className);
  const borderWidthProvided = includesBorderWidth(className);
  return classNames(
    [
      'inline-flex',
      'items-center',
      'justify-center',
      'box-border',
      'h-28',
      'hover:underline',
      'disabled:no-underline',
      'transition-all',
    ],
    {
      'text-ui': variant !== 'inline-link',
      'no-underline': variant !== 'inline-link',
      'pl-28': !(
        paddingLeftProvided ||
        variant === 'inline' ||
        variant === 'inline-link'
      ),
      'pr-28': !(
        paddingRightProvided ||
        variant === 'inline' ||
        variant === 'inline-link'
      ),

      border: !borderWidthProvided,

      'hover:border-black dark:hover:border-white':
        variant !== 'inline' && variant !== 'inline-link',
      'active:border-black dark:active:border-white': true,

      'bg-black dark:bg-white': variant === 'primary',
      'border-black-60 dark:border-white-60':
        variant === 'primary' || variant === 'secondary',
      'text-white dark:text-black': variant === 'primary',
      'hover:bg-black-80 dark:hover:bg-white-80': variant === 'primary',
      'active:bg-white dark:active:bg-black':
        variant === 'primary' || variant === 'accent',
      'active:text-black dark:active:text-white':
        variant === 'primary' || variant === 'accent',

      'bg-white dark:bg-black': variant === 'secondary',
      'text-black dark:text-white': variant === 'secondary',
      'hover:bg-black-25 dark:hover:bg-white-25': variant === 'secondary',
      'hover:text-black dark:hover:text-white':
        variant === 'secondary' || variant === 'accent',
      'active:bg-black dark:active:bg-white': variant === 'secondary',
      'active:text-white dark:active:text-black': variant === 'secondary',

      uppercase: variant === 'accent',
      'bg-vega-yellow dark:bg-vega-yellow': variant === 'accent',
      'border-transparent dark:border-transparent':
        variant === 'accent' ||
        variant === 'inline' ||
        variant === 'inline-link',
      'hover:bg-vega-yellow-dark dark:hover:bg-vega-yellow/30':
        variant === 'accent',
      'text-black dark:text-black': variant === 'accent',
      'hover:text-white dark:hover:text-white': variant === 'accent',

      'pl-4': variant === 'inline' && !paddingLeftProvided,
      'pr-4': variant === 'inline' && !paddingRightProvided,
      'border-0':
        (variant === 'inline' || variant === 'inline-link') &&
        !borderWidthProvided,
      underline: variant === 'inline' || variant === 'inline-link',
      'hover:no-underline': variant === 'inline',
      'hover:border-transparent dark:hover:border-transparent':
        variant === 'inline' || variant === 'inline-link',
      'active:border-transparent dark:active:border-transparent':
        variant === 'inline' || variant === 'inline-link',
      'active:text-black dark:active:text-vega-yellow':
        variant === 'inline' || variant === 'inline-link',
      'text-black-95 dark:text-white-95':
        variant === 'inline' || variant === 'inline-link',
      'hover:text-black hover:dark:text-white':
        variant === 'inline' || variant === 'inline-link',

      'disabled:bg-black-10 dark:disabled:bg-white-10':
        variant !== 'inline' && variant !== 'inline-link',
      'disabled:text-black-60 dark:disabled:text-white-60':
        variant !== 'inline' && variant !== 'inline-link',
      'disabled:border-black-25 dark:disabled:border-white-25':
        variant !== 'inline' && variant !== 'inline-link',
    },
    className
  );
};

const getContent = (
  children: React.ReactNode,
  prependIconName?: IconName,
  appendIconName?: IconName
) => {
  const iconName = prependIconName || appendIconName;
  if (iconName === undefined) {
    return children;
  }
  const iconClassName = classNames(['fill-current'], {
    'mr-8': prependIconName,
    'ml-8': appendIconName,
  });
  const icon = <Icon name={iconName} className={iconClassName} size={16} />;
  return (
    <>
      {prependIconName && icon}
      {children}
      {appendIconName && icon}
    </>
  );
};

export const Button = forwardRef<HTMLButtonElement, ButtonProps>(
  (
    {
      variant = 'primary',
      type = 'button',
      children,
      className,
      prependIconName,
      appendIconName,
      ...props
    },
    ref
  ) => {
    return (
      <button
        ref={ref}
        className={getClassName(className, variant)}
        type={type}
        {...props}
      >
        {getContent(children, prependIconName, appendIconName)}
      </button>
    );
  }
);

export const AnchorButton = forwardRef<HTMLAnchorElement, AnchorButtonProps>(
  (
    {
      variant = 'primary',
      children,
      className,
      prependIconName,
      appendIconName,
      ...prosp
    },
    ref
  ) => {
    return (
      <a ref={ref} className={getClassName(className, variant)} {...prosp}>
        {getContent(children, prependIconName, appendIconName)}
      </a>
    );
  }
);