/**
 * Copyright 2025 LY Corporation
 *
 * LY Corporation licenses this file to you under the Apache License,
 * version 2.0 (the "License"); you may not use this file except in compliance
 * with the License. You may obtain a copy of the License at:
 *
 *   https://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
 * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
 * License for the specific language governing permissions and limitations
 * under the License.
 */
import * as React from 'react';
import { Slot } from '@radix-ui/react-slot';
import type { VariantProps } from 'class-variance-authority';
import { cva } from 'class-variance-authority';

import type { Color, Radius } from '../lib/types';
import { cn } from '../lib/utils';
import useTheme from './use-theme';

type BadgeVariant = 'bold' | 'subtle' | 'outline';

const badgeVariants = cva('badge', {
  variants: {
    radius: {
      large: 'badge-radius-large',
      medium: 'badge-radius-medium',
      small: 'badge-radius-small',
    },
    variant: {
      bold: '',
      subtle: '',
      outline: '',
    },
    color: {
      default: '',
      blue: '',
      orange: '',
      red: '',
      green: '',
    },
  },
  compoundVariants: [
    {
      variant: 'bold',
      color: 'default',
      className: 'badge-bold-default',
    },
    {
      variant: 'bold',
      color: 'blue',
      className: 'badge-bold-blue',
    },
    {
      variant: 'bold',
      color: 'orange',
      className: 'badge-bold-orange',
    },
    {
      variant: 'bold',
      color: 'red',
      className: 'badge-bold-red',
    },
    {
      variant: 'bold',
      color: 'green',
      className: 'badge-bold-green',
    },
    {
      variant: 'subtle',
      color: 'default',
      className: 'badge-subtle-default',
    },
    {
      variant: 'subtle',
      color: 'blue',
      className: 'badge-subtle-blue',
    },
    {
      variant: 'subtle',
      color: 'orange',
      className: 'badge-subtle-orange',
    },
    {
      variant: 'subtle',
      color: 'red',
      className: 'badge-subtle-red',
    },
    {
      variant: 'subtle',
      color: 'green',
      className: 'badge-subtle-green',
    },
    {
      variant: 'outline',
      color: 'default',
      className: 'badge-outline-default',
    },
    {
      variant: 'outline',
      color: 'blue',
      className: 'badge-outline-blue',
    },
    {
      variant: 'outline',
      color: 'orange',
      className: 'badge-outline-orange',
    },
    {
      variant: 'outline',
      color: 'red',
      className: 'badge-outline-red',
    },
    {
      variant: 'outline',
      color: 'green',
      className: 'badge-outline-green',
    },
  ],
  defaultVariants: {
    radius: undefined,
    variant: 'bold',
    color: 'default',
  },
});

interface BadgeProps
  extends React.HTMLAttributes<HTMLDivElement>,
    VariantProps<typeof badgeVariants> {
  radius?: Radius;
  variant?: BadgeVariant;
  color?: Color;
  asChild?: boolean;
}

const Badge = React.forwardRef<HTMLDivElement, BadgeProps>(
  (
    {
      radius,
      variant = 'bold',
      color = 'default',
      className,
      asChild,
      ...props
    },
    ref,
  ) => {
    const { themeRadius } = useTheme();
    const Comp = asChild ? Slot : 'div';

    return (
      <Comp
        ref={ref}
        className={cn(
          badgeVariants({ radius: radius ?? themeRadius, variant, color }),
          className,
        )}
        {...props}
      />
    );
  },
);

Badge.displayName = 'Badge';

export { Badge, type BadgeProps };