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

Forbid restore and trash action for WP list in bulk action

Browse Source 
[MAILPOET-3169]
This commit is contained in:
Jan Lysý
2021-02-01 10:56:35 +01:00
committed by Veljko V
parent b03f301a5a
commit 6299d2736e
2 changed files with 31 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
lib/API/JSON/v1/Segments.php
View File

@ -117,6 +117,11 @@ class Segments extends APIEndpoint {
public function restore($data = []) {
$segment = $this->getSegment($data);
if ($segment instanceof SegmentEntity) {
if (!$this->isTrashOrRestoreAllowed($segment)) {
return $this->errorResponse([
APIError::FORBIDDEN => WPFunctions::get()->__('This list cannot be moved to trash.', 'mailpoet'),
]);
}
// When the segment is of type WP_USERS we want to restore all its subscribers
if ($segment->getType() === SegmentEntity::TYPE_WP_USERS) {
$subscribers = $this->subscribersRepository->findBySegment((int)$segment->getId());
@ -126,7 +131,7 @@ class Segments extends APIEndpoint {
$this->subscribersRepository->bulkRestore($subscriberIds);
}
$this->segmentsRepository->bulkRestore([$segment->getId()]);
$this->segmentsRepository->bulkRestore([$segment->getId()], $segment->getType());
$this->segmentsRepository->refresh($segment);
return $this->successResponse(
$this->segmentsResponseBuilder->build($segment),
@ -142,6 +147,11 @@ class Segments extends APIEndpoint {
public function trash($data = []) {
$segment = $this->getSegment($data);
if ($segment instanceof SegmentEntity) {
if (!$this->isTrashOrRestoreAllowed($segment)) {
return $this->errorResponse([
APIError::FORBIDDEN => WPFunctions::get()->__('This list cannot be moved to trash.', 'mailpoet'),
]);
}
// When the segment is of type WP_USERS we want to trash all subscribers who aren't subscribed in another list
if ($segment->getType() === SegmentEntity::TYPE_WP_USERS) {
$subscribers = $this->subscribersRepository->findExclusiveSubscribersBySegment((int)$segment->getId());
@ -151,7 +161,7 @@ class Segments extends APIEndpoint {
$this->subscribersRepository->bulkTrash($subscriberIds);
}
$this->segmentsRepository->bulkTrash([$segment->getId()]);
$this->segmentsRepository->bulkTrash([$segment->getId()], $segment->getType());
$this->segmentsRepository->refresh($segment);
return $this->successResponse(
$this->segmentsResponseBuilder->build($segment),
@ -231,6 +241,18 @@ class Segments extends APIEndpoint {
return $this->successResponse(null, ['count' => $count]);
}
private function isTrashOrRestoreAllowed(SegmentEntity $segment): bool {
$allowedSegmentTypes = [
SegmentEntity::TYPE_DEFAULT,
SegmentEntity::TYPE_WP_USERS,
];
if (in_array($segment->getType(), $allowedSegmentTypes, true)) {
return true;
}
return false;
}
private function getSegment(array $data): ?SegmentEntity {
return isset($data['id'])
? $this->segmentsRepository->findOneById((int)$data['id'])