Cavemanon/piratepoet
7
Fork 0
Code Pull Requests Actions Releases 16 Activity

Refactor getting segments for import into doctrine

Browse Source 
[MAILPOET-3164]
This commit is contained in:
Rostislav Wolny
2021-01-19 15:50:23 +01:00
committed by Veljko V
parent 6c7000430d
commit c7152e9448
1 changed files with 10 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
lib/Subscribers/ImportExport/ImportExportFactory.php
View File

@ -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);