import { useMessages } from '@/components/hooks';
import { Plus } from '@/components/icons';
import { DialogButton } from '@/components/input/DialogButton';
import { SegmentEditForm } from './SegmentEditForm';

export function SegmentAddButton({ websiteId }: { websiteId: string }) {
  const { t, labels } = useMessages();

  return (
    <DialogButton icon={<Plus />} label={t(labels.segment)} variant="primary" width="800px">
      {({ close }) => {
        return <SegmentEditForm websiteId={websiteId} onClose={close} />;
      }}
    </DialogButton>
  );
}