import { Badge, BadgeVariant, MantineSize } from '@mantine/core';
import { useTranslation } from 'next-i18next';
import Dockerode from 'dockerode';

export interface ContainerStateProps {
  state: Dockerode.ContainerInfo['State'];
}

export default function ContainerState(props: ContainerStateProps) {
  const { state } = props;

  const { t } = useTranslation('modules/docker-module');

  const options: {
    size: MantineSize;
    radius: MantineSize;
    variant: BadgeVariant;
  } = {
    size: 'md',
    radius: 'md',
    variant: 'outline',
  };
  switch (state) {
    case 'running': {
      return (
        <Badge color="green" {...options}>
          {t('table.states.running')}
        </Badge>
      );
    }
    case 'created': {
      return (
        <Badge color="cyan" {...options}>
          {t('table.states.created')}
        </Badge>
      );
    }
    case 'exited': {
      return (
        <Badge color="red" {...options}>
          {t('table.states.stopped')}
        </Badge>
      );
    }
    default: {
      return (
        <Badge color="purple" {...options}>
          {t('table.states.unknown')}
        </Badge>
      );
    }
  }
}
