diff --git a/lib/Subscribers/ImportExport/ImportExportFactory.php b/lib/Subscribers/ImportExport/ImportExportFactory.php index 5ec499d14a..35badc3192 100644 --- a/lib/Subscribers/ImportExport/ImportExportFactory.php +++ b/lib/Subscribers/ImportExport/ImportExportFactory.php @@ -4,8 +4,10 @@ namespace MailPoet\Subscribers\ImportExport; use MailPoet\DI\ContainerWrapper; use MailPoet\DynamicSegments\FreePluginConnectors\AddToNewslettersSegments; +use MailPoet\Entities\SegmentEntity; use MailPoet\Models\CustomField; use MailPoet\Models\Segment; +use MailPoet\Segments\SegmentSubscribersRepository; use MailPoet\Util\Helpers; class ImportExportFactory { @@ -18,14 +20,21 @@ class ImportExportFactory { /** @var AddToNewslettersSegments */ private $addToNewslettersSegments; + /** @var SegmentSubscribersRepository */ + private $segmentSubscribersRepository; + public function __construct($action = null) { $this->action = $action; $this->addToNewslettersSegments = ContainerWrapper::getInstance()->get(AddToNewslettersSegments::class); + $this->segmentSubscribersRepository = ContainerWrapper::getInstance()->get(SegmentSubscribersRepository::class); } public function getSegments() { + $segments = $this->segmentSubscribersRepository->getSimpleSegmentListWithSubscribersCounts(); if ($this->action === self::IMPORT_ACTION) { - $segments = Segment::getSegmentsForImport(); + $segments = array_values(array_filter($segments, function($segment) { + return in_array($segment['type'], [SegmentEntity::TYPE_DEFAULT, SegmentEntity::TYPE_WP_USERS]); + })); } else { $segments = Segment::getSegmentsForExport(); $segments = $this->addToNewslettersSegments->add($segments);