Add method getCurrentWPUser to SubscribersRepository

[MAILPOET-3268]
This commit is contained in:
Jan Lysý
2021-03-09 15:52:58 +01:00
committed by Veljko V
parent 233ca0b5f3
commit 97dd881e8d

View File

@ -7,6 +7,7 @@ use MailPoet\Entities\SegmentEntity;
use MailPoet\Entities\SubscriberCustomFieldEntity;
use MailPoet\Entities\SubscriberEntity;
use MailPoet\Entities\SubscriberSegmentEntity;
use MailPoet\WP\Functions as WPFunctions;
use MailPoetVendor\Doctrine\DBAL\Connection;
use MailPoetVendor\Doctrine\ORM\EntityManager;
use MailPoetVendor\Doctrine\ORM\Query\Expr\Join;
@ -283,4 +284,12 @@ class SubscribersRepository extends Repository {
->setParameter('emails', $emails)
->getQuery()->getResult();
}
public function getCurrentWPUser(): ?SubscriberEntity {
$wpUser = WPFunctions::get()->wpGetCurrentUser();
if (empty($wpUser->ID)) {
return null; // Don't look up a subscriber for guests
}
return $this->findOneBy(['wpUserId' => $wpUser->ID]);
}
}