Move dynamic segments from Premium plugin

[MAILPOET-2382]

lib/DynamicSegments/Persistence/Loading/Loader.php

@@ -0,0 +1,35 @@
+<?php
+
+namespace MailPoet\Premium\DynamicSegments\Persistence\Loading;
+
+use MailPoet\Premium\DynamicSegments\Mappers\DBMapper;
+use MailPoet\Premium\Models\DynamicSegment;
+use MailPoet\Premium\Models\DynamicSegmentFilter;
+
+class Loader {
+
+  /** @var DBMapper */
+  private $mapper;
+
+  public function __construct(DBMapper $mapper) {
+    $this->mapper = $mapper;
+  }
+
+  /**
+   * @return DynamicSegment[]
+   */
+  function load() {
+    $segments = DynamicSegment::findAll();
+    return $this->loadFilters($segments);
+  }
+
+  private function loadFilters(array $segments) {
+    $ids = array_map(function($segment) {
+      return $segment->id;
+    }, $segments);
+    $filters = DynamicSegmentFilter::getAllBySegmentIds($ids);
+
+    return $this->mapper->mapSegments($segments, $filters);
+  }
+
+}

lib/DynamicSegments/Persistence/Loading/SingleSegmentLoader.php

@@ -0,0 +1,35 @@
+<?php
+
+namespace MailPoet\Premium\DynamicSegments\Persistence\Loading;
+
+use MailPoet\Premium\DynamicSegments\Mappers\DBMapper;
+use MailPoet\Premium\Models\DynamicSegment;
+
+class SingleSegmentLoader {
+
+  /** @var DBMapper */
+  private $mapper;
+
+  public function __construct(DBMapper $mapper) {
+    $this->mapper = $mapper;
+  }
+
+  /**
+   * @param string|int $segment_id
+   * @return DynamicSegment
+   */
+  function load($segment_id) {
+
+    $segment = DynamicSegment::findOne($segment_id);
+    if (!$segment instanceof DynamicSegment) {
+      throw new \InvalidArgumentException('Segment not found');
+    }
+
+    $filters = $segment->dynamicSegmentFilters()->findMany();
+
+    return $this->mapper->mapSegment($segment, $filters);
+  }
+
+
+
+}

lib/DynamicSegments/Persistence/Loading/SubscribersCount.php

@@ -0,0 +1,23 @@
+<?php
+
+namespace MailPoet\Premium\DynamicSegments\Persistence\Loading;
+
+use MailPoet\Models\Subscriber;
+use MailPoet\Premium\Models\DynamicSegment;
+
+class SubscribersCount {
+
+  /**
+   * @param DynamicSegment $dynamic_segment
+   *
+   * @return int
+   */
+  function getSubscribersCount(DynamicSegment $dynamic_segment) {
+    $orm = Subscriber::selectExpr('count(distinct ' . Subscriber::$_table . '.id) as cnt');
+    foreach ($dynamic_segment->getFilters() as $filter) {
+      $orm = $filter->toSql($orm);
+    }
+    return $orm->findOne()->cnt;
+  }
+
+}

lib/DynamicSegments/Persistence/Loading/SubscribersIds.php

@@ -0,0 +1,29 @@
+<?php
+
+namespace MailPoet\Premium\DynamicSegments\Persistence\Loading;
+
+use MailPoet\Models\Subscriber;
+use MailPoet\Premium\Models\DynamicSegment;
+
+class SubscribersIds {
+
+  /**
+   * Finds subscribers in a dynamic segment and returns their ids.
+   *
+   * @param DynamicSegment $dynamic_segment
+   * @param array $limit_to_subscribers_ids If passed the result will be limited only to ids within this array
+   *
+   * @return Subscriber[]
+   */
+  function load(DynamicSegment $dynamic_segment, $limit_to_subscribers_ids = null) {
+    $orm = Subscriber::selectExpr(Subscriber::$_table . '.id');
+    foreach ($dynamic_segment->getFilters() as $filter) {
+      $orm = $filter->toSql($orm);
+    }
+    if ($limit_to_subscribers_ids) {
+      $orm->whereIn(Subscriber::$_table . '.id', $limit_to_subscribers_ids);
+    }
+    return $orm->findMany();
+  }
+
+}

lib/DynamicSegments/Persistence/Saver.php

@@ -0,0 +1,77 @@
+<?php
+
+namespace MailPoet\Premium\DynamicSegments\Persistence;
+
+use MailPoet\Models\Model;
+use MailPoet\Premium\DynamicSegments\Exceptions\ErrorSavingException;
+use MailPoet\Premium\DynamicSegments\Filters\Filter;
+use MailPoet\Premium\Models\DynamicSegment;
+use MailPoet\Premium\Models\DynamicSegmentFilter;
+
+class Saver {
+
+  /**
+   * @param DynamicSegment $segment
+   *
+   * @return integer
+   * @throws ErrorSavingException
+   */
+  function save(DynamicSegment $segment) {
+    $db = \ORM::get_db();
+    $db->beginTransaction();
+
+    $data_segment = $this->saveSegment($db, $segment);
+    $this->saveFilters($db, $segment, $data_segment->id());
+    $db->commit();
+    return $data_segment->id();
+  }
+
+  /**
+   * @throws ErrorSavingException
+   */
+  private function saveSegment(\PDO $db, DynamicSegment $segment) {
+    $segment->save();
+    $this->checkErrors($db, $segment);
+    return $segment;
+  }
+
+  /**
+   * @throws ErrorSavingException
+   */
+  private function checkErrors(\PDO $db, Model $model) {
+    $errors = $model->getErrors();
+    if ($errors) {
+      $code = null;
+      if (array_key_exists(Model::DUPLICATE_RECORD, $errors)) {
+        $code = Model::DUPLICATE_RECORD;
+      }
+      $db->rollBack();
+      throw new ErrorSavingException(join(", ", $model->getErrors()), $code);
+    }
+  }
+
+  /**
+   * @throws ErrorSavingException
+   */
+  private function saveFilters(\PDO $db, DynamicSegment $segment, $saved_data_id) {
+    $this->deleteFilters($segment);
+    foreach ($segment->getFilters() as $filter) {
+      $data_filter = $this->saveFilter($filter, $saved_data_id);
+      $this->checkErrors($db, $data_filter);
+    }
+  }
+
+  private function deleteFilters(DynamicSegment $segment) {
+    DynamicSegmentFilter::deleteAllBySegmentIds([$segment->id]);
+  }
+
+  private function saveFilter(Filter $filter, $data_segment_id) {
+    $data_filter = DynamicSegmentFilter::create();
+    if ($data_filter instanceof DynamicSegmentFilter) {
+      $data_filter->segment_id = $data_segment_id;
+      $data_filter->filter_data = $filter->toArray();
+      $data_filter->save();
+    }
+    return $data_filter;
+  }
+}