diff --git a/lib/Config/Shortcodes.php b/lib/Config/Shortcodes.php index 846399c942..30dd85b330 100644 --- a/lib/Config/Shortcodes.php +++ b/lib/Config/Shortcodes.php @@ -2,10 +2,13 @@ namespace MailPoet\Config; +use MailPoet\Entities\NewsletterEntity; +use MailPoet\Entities\ScheduledTaskEntity; +use MailPoet\Entities\SendingQueueEntity; use MailPoet\Entities\SubscriberEntity; use MailPoet\Form\Widget; -use MailPoet\Models\Newsletter; use MailPoet\Models\Subscriber; +use MailPoet\Newsletter\NewslettersRepository; use MailPoet\Newsletter\Url as NewsletterUrl; use MailPoet\Segments\SegmentSubscribersRepository; use MailPoet\Subscribers\SubscribersRepository; @@ -28,18 +31,23 @@ class Shortcodes { /** @var NewsletterUrl */ private $newsletterUrl; + /** @var NewslettersRepository */ + private $newslettersRepository; + public function __construct( Pages $subscriptionPages, WPFunctions $wp, SegmentSubscribersRepository $segmentSubscribersRepository, SubscribersRepository $subscribersRepository, - NewsletterUrl $newsletterUrl + NewsletterUrl $newsletterUrl, + NewslettersRepository $newslettersRepository ) { $this->subscriptionPages = $subscriptionPages; $this->wp = $wp; $this->segmentSubscribersRepository = $segmentSubscribersRepository; $this->subscribersRepository = $subscribersRepository; $this->newsletterUrl = $newsletterUrl; + $this->newslettersRepository = $newslettersRepository; } public function init() { @@ -110,7 +118,7 @@ class Shortcodes { $html = ''; - $newsletters = Newsletter::getArchives($segmentIds); + $newsletters = $this->newslettersRepository->getArchives($segmentIds); $subscriber = $this->subscribersRepository->getCurrentWPUser(); $subscriber = $subscriber ? Subscriber::findOne($subscriber->getId()) : null; @@ -127,10 +135,17 @@ class Shortcodes { } $html .= '