Use SegmentEntity in endpoint Segment::save()

[MAILPOET-3167]

lib/Segments/SegmentSaveController.php

@@ -0,0 +1,62 @@
+<?php
+
+namespace MailPoet\Segments;
+
+use InvalidArgumentException;
+use MailPoet\Entities\SegmentEntity;
+use MailPoet\NotFoundException;
+
+class SegmentSaveController {
+  /** @var SegmentsRepository */
+  private $segmentsRepository;
+
+  public function __construct(
+    SegmentsRepository $segmentsRepository
+  ) {
+    $this->segmentsRepository = $segmentsRepository;
+  }
+
+  public function save(array $data = []): SegmentEntity {
+    $this->checkSegmenUniqueName($data['name'] ?? '');
+
+    $segment = $this->createOrUpdateSegment($data);
+    return $segment;
+  }
+
+  private function checkSegmenUniqueName(string $name): void {
+    $segment = $this->segmentsRepository->findOneBy(['name' => $name]);
+    if ($segment) {
+      throw new InvalidArgumentException("Segment with name: '{$name}' already exists.");
+    }
+  }
+
+  private function createOrUpdateSegment(array $data): SegmentEntity {
+    $name = $data['name'] ?? '';
+    $description = $data['description'] ?? '';
+
+    if (isset($data['id'])) {
+      $segment = $this->getSegment((int)$data['id']);
+      $segment->setName($name);
+      $segment->setDescription($description);
+    } else {
+      $segment = new SegmentEntity(
+        $name,
+        SegmentEntity::TYPE_DEFAULT,
+        $description
+      );
+      $this->segmentsRepository->persist($segment);
+    }
+
+    $this->segmentsRepository->flush();
+    return $segment;
+  }
+
+  private function getSegment(int $segmentId): SegmentEntity {
+    $segment = $this->segmentsRepository->findOneById($segmentId);
+    if (!$segment) {
+      throw new NotFoundException();
+    }
+
+    return $segment;
+  }
+}