OptimCloud-gw-ui/src/components/Buttons/StepButton/index.tsx

import React, { useCallback, useMemo } from 'react';
import { Button, IconButton, Tooltip, useBreakpoint } from '@chakra-ui/react';
import { ArrowRight, FloppyDisk } from 'phosphor-react';
import { useTranslation } from 'react-i18next';

export interface StepButtonProps {
  onNext: () => void;
  onSave?: () => void;
  currentStep: number;
  lastStep: number;
  isDisabled?: boolean;
  isLoading?: boolean;
  isCompact?: boolean;
  ml?: string | number;
}

const _StepButton: React.FC<StepButtonProps> = ({
  onNext,
  onSave,
  isDisabled,
  isLoading,
  isCompact,
  currentStep,
  lastStep,
  ml,
  ...props
}) => {
  const { t } = useTranslation();
  const breakpoint = useBreakpoint();

  const onClick = useCallback(
    () => (currentStep === lastStep && onSave ? onSave() : onNext()),
    [currentStep, lastStep, onNext, onSave],
  );
  const icon = useMemo(
    () => (currentStep === lastStep ? <FloppyDisk size={20} /> : <ArrowRight size={20} />),
    [currentStep, lastStep],
  );
  const label = useMemo(
    () => (currentStep === lastStep ? t('common.save') : t('common.next')),
    [currentStep, lastStep],
  );

  if (!isCompact && breakpoint !== 'base' && breakpoint !== 'sm') {
    return (
      <Button
        type="button"
        colorScheme="blue"
        onClick={onClick}
        rightIcon={icon}
        isLoading={isLoading}
        isDisabled={isDisabled}
        ml={ml}
        {...props}
      >
        {label}
      </Button>
    );
  }

  return (
    <Tooltip label={label}>
      <IconButton
        colorScheme="blue"
        aria-label="next"
        type="button"
        onClick={onClick}
        icon={icon}
        isLoading={isLoading}
        isDisabled={isDisabled}
        ml={ml}
        {...props}
      />
    </Tooltip>
  );
};

export const StepButton = React.memo(_StepButton);