import React from 'react';

export interface IBadge {
  children: any;
  className?: string;
  variant?: 'primary' | 'secondary' | 'info' | 'success' | 'warning' | 'danger' | 'light';
  [x: string]: any;
}

export const Badge: React.FC<IBadge> = (props) => {
  const { children, className = '', variant = 'primary', ...otherProps } = props;

  const classNames = () => {
    let classList = `b-badge`;
    if (variant) classList += ` ${variant}`;
    if (className) classList += ` ${className}`;
    return classList;
  };

  return (
    <div className={classNames()} {...otherProps}>
      {children}
    </div>
  );
};