<?php declare(strict_types = 1);

namespace MailPoet\Test\Subscription;

use Codeception\Stub;
use MailPoet\Config\Renderer;
use MailPoet\Cron\Workers\SendingQueue\Tasks\Links;
use MailPoet\Cron\Workers\StatsNotifications\NewsletterLinkRepository;
use MailPoet\DI\ContainerWrapper;
use MailPoet\Entities\NewsletterEntity;
use MailPoet\Entities\SegmentEntity;
use MailPoet\Entities\SendingQueueEntity;
use MailPoet\Entities\StatisticsUnsubscribeEntity;
use MailPoet\Entities\SubscriberCustomFieldEntity;
use MailPoet\Entities\SubscriberEntity;
use MailPoet\Entities\SubscriberSegmentEntity;
use MailPoet\Form\AssetsController;
use MailPoet\Newsletter\NewslettersRepository;
use MailPoet\Newsletter\Scheduler\WelcomeScheduler;
use MailPoet\Newsletter\Sending\ScheduledTasksRepository;
use MailPoet\Newsletter\Sending\SendingQueuesRepository;
use MailPoet\Settings\SettingsController;
use MailPoet\Settings\TrackingConfig;
use MailPoet\Statistics\StatisticsClicksRepository;
use MailPoet\Statistics\StatisticsUnsubscribesRepository;
use MailPoet\Statistics\Track\SubscriberHandler;
use MailPoet\Statistics\Track\Unsubscribes;
use MailPoet\Subscribers\LinkTokens;
use MailPoet\Subscribers\NewSubscriberNotificationMailer;
use MailPoet\Subscribers\SubscriberSaveController;
use MailPoet\Subscribers\SubscriberSegmentRepository;
use MailPoet\Subscribers\SubscribersRepository;
use MailPoet\Subscription\CaptchaFormRenderer;
use MailPoet\Subscription\ManageSubscriptionFormRenderer;
use MailPoet\Subscription\Pages;
use MailPoet\Subscription\SubscriptionUrlFactory;
use MailPoet\Test\DataFactories\CustomField;
use MailPoet\Test\DataFactories\Newsletter;
use MailPoet\Test\DataFactories\NewsletterLink;
use MailPoet\Test\DataFactories\NewsletterOption as NewsletterOptionFactory;
use MailPoet\Test\DataFactories\Segment as SegmentFactory;
use MailPoet\Test\DataFactories\Subscriber;
use MailPoet\WP\Functions as WPFunctions;
use MailPoetVendor\Carbon\Carbon;
use MailPoetVendor\Doctrine\ORM\EntityManager;

class PagesTest extends \MailPoetTest {
  private $testData = [];

  /** @var SubscriberEntity */
  private $subscriber;

  /** @var SubscribersRepository */
  private $subscribersRepository;

  /*** @var StatisticsUnsubscribesRepository */
  private $statisticsUnsubscribesRepository;

  /*** @var StatisticsClicksRepository */
  private $statisticsClicksRepository;

  /*** @var LinkTokens */
  private $linkTokens;

  /*** @var Links */
  private $links;

  public function _before() {
    parent::_before();
    $this->subscribersRepository = $this->diContainer->get(SubscribersRepository::class);
    $this->statisticsUnsubscribesRepository = $this->diContainer->get(StatisticsUnsubscribesRepository::class);
    $this->statisticsClicksRepository = $this->diContainer->get(StatisticsClicksRepository::class);
    $this->subscriber = new SubscriberEntity();
    $this->subscriber->setEmail('jane.doe@example.com');
    $this->subscriber->setStatus(SubscriberEntity::STATUS_UNCONFIRMED);
    $this->subscribersRepository->persist($this->subscriber);
    $this->subscribersRepository->flush();
    $this->linkTokens = $this->diContainer->get(LinkTokens::class);
    $this->links = $this->diContainer->get(Links::class);

    $this->testData = [
        'email' => $this->subscriber->getEmail(),

    ];
    $this->testData['email'] = $this->subscriber->getEmail();
    $this->testData['token'] = $this->linkTokens->getToken($this->subscriber);
  }

  public function testItConfirmsSubscription() {
    $newSubscriberNotificationSender = $this->makeEmpty(NewSubscriberNotificationMailer::class, ['sendWithSubscriberAndSegmentEntities' => Stub\Expected::once()]);
    $pages = $this->getPages($newSubscriberNotificationSender);
    $subscription = $pages->init(false, $this->testData, false, false);
    $subscription->confirm();

    $confirmedSubscriber = $this->subscribersRepository->findOneById($this->subscriber->getId());
    $this->assertInstanceOf(SubscriberEntity::class, $confirmedSubscriber);
    verify($confirmedSubscriber->getStatus())->equals(SubscriberEntity::STATUS_SUBSCRIBED);
    $this->assertTrue(Carbon::parse($confirmedSubscriber->getLastSubscribedAt())->isToday());
  }

  public function testItUpdatesUnconfirmedDataWhenConfirmingSubscription() {
    $customField = (new CustomField())
      ->withName('nickname')
      ->create();

    $firstName = 'Jane';
    $lastName = 'Doe';
    $this->subscriber->setUnconfirmedData((string)json_encode([
      'first_name' => $firstName,
      'last_name' => $lastName,
      'email' => 'jane.doe@example.com',
      'cf_' . $customField->getId() => 'Nickname',
    ]));
    $this->entityManager->persist($this->subscriber);
    $this->entityManager->flush();

    $newSubscriberNotificationSender = $this->makeEmpty(NewSubscriberNotificationMailer::class, ['sendWithSubscriberAndSegmentEntities' => Stub\Expected::once()]);
    $pages = $this->getPages($newSubscriberNotificationSender);
    $subscription = $pages->init(false, $this->testData, false, false);

    $subscription->confirm();

    $confirmedSubscriber = $this->subscribersRepository->findOneById($this->subscriber->getId());
    $this->assertInstanceOf(SubscriberEntity::class, $confirmedSubscriber);
    $this->assertSame(SubscriberEntity::STATUS_SUBSCRIBED, $confirmedSubscriber->getStatus());
    $this->assertSame($firstName, $confirmedSubscriber->getFirstName());
    $this->assertSame($lastName, $confirmedSubscriber->getLastName());
    $this->assertNull($confirmedSubscriber->getUnconfirmedData());
    $subscriberCustomField = $confirmedSubscriber->getSubscriberCustomField($customField);
    $this->assertInstanceOf(SubscriberCustomFieldEntity::class, $subscriberCustomField);
    $this->assertSame('Nickname', $subscriberCustomField->getValue());
  }

  public function testItUpdatesSubscriptionOnDuplicateAttemptButDoesntSendNotification() {
    $newSubscriberNotificationSender = $this->makeEmpty(NewSubscriberNotificationMailer::class, ['send' => Stub\Expected::never()]);
    $pages = $this->getPages($newSubscriberNotificationSender);
    $subscriber = $this->subscriber;
    $subscriber->setStatus(SubscriberEntity::STATUS_SUBSCRIBED);
    $subscriber->setFirstName('First name');
    $subscriber->setUnconfirmedData(null);
    $subscriber->setLastSubscribedAt(Carbon::now()->millisecond(0)->subDays(10));
    $subscriber->setConfirmedIp('111.111.111.111');
    $this->entityManager->flush();
    $subscription = $pages->init(false, $this->testData, false, false);
    $subscription->confirm();
    $this->entityManager->clear();
    $confirmedSubscriber = $this->subscribersRepository->findOneById($subscriber->getId());
    $this->assertInstanceOf(SubscriberEntity::class, $confirmedSubscriber);
    verify($confirmedSubscriber->getStatus())->equals(SubscriberEntity::STATUS_SUBSCRIBED);
    verify($confirmedSubscriber->getConfirmedAt())->greaterThanOrEqual(Carbon::now()->millisecond(0)->subSecond());
    verify($confirmedSubscriber->getConfirmedAt())->lessThanOrEqual(Carbon::now()->millisecond(0)->addSecond());
    verify($confirmedSubscriber->getLastSubscribedAt())->greaterThanOrEqual(Carbon::now()->millisecond(0)->subSecond());
    verify($confirmedSubscriber->getLastSubscribedAt())->lessThanOrEqual(Carbon::now()->millisecond(0)->addSecond());
    verify($confirmedSubscriber->getFirstName())->equals('First name');
  }

  public function testItSendsWelcomeNotificationUponConfirmingSubscription() {
    $scheduledTasksRepository = $this->diContainer->get(ScheduledTasksRepository::class);
    $newslettersRepository = $this->diContainer->get(NewslettersRepository::class);
    $newSubscriberNotificationSender = $this->makeEmpty(NewSubscriberNotificationMailer::class, ['sendWithSubscriberAndSegmentEntities' => Stub\Expected::once()]);
    $pages = $this->getPages($newSubscriberNotificationSender);
    $subscription = $pages->init($action = false, $this->testData, false, false);

    $segment = $this->createSegment();
    $this->createSubscriberSegment($segment);

    // create welcome notification newsletter and relevant scheduling options
    $newsletter = new NewsletterEntity();
    $newsletter->setSubject('Some subject');
    $newsletter->setType(NewsletterEntity::TYPE_WELCOME);
    $newsletter->setStatus(NewsletterEntity::STATUS_ACTIVE);
    $this->entityManager->persist($newsletter);

    $newsletterOptions = [
      'event' => 'segment',
      'segment' => $segment->getId(),
      'afterTimeType' => 'days',
      'afterTimeNumber' => 1,
    ];
    (new NewsletterOptionFactory())->createMultipleOptions($newsletter, $newsletterOptions);

    // confirm subscription and ensure that welcome email is scheduled
    $subscription->confirm();
    $newsletterEntity = $newslettersRepository->findOneById($newsletter->getId());
    $this->assertInstanceOf(NewsletterEntity::class, $newsletterEntity);
    $scheduledNotifications = $scheduledTasksRepository->findByNewsletterAndStatus($newsletterEntity, SendingQueueEntity::STATUS_SCHEDULED);
    verify(count($scheduledNotifications))->equals(1);

    // Does not schedule another on repeated confirmation
    $subscription->confirm();
    $scheduledNotifications = $scheduledTasksRepository->findByNewsletterAndStatus($newsletterEntity, SendingQueueEntity::STATUS_SCHEDULED);
    verify(count($scheduledNotifications))->equals(1);
  }

  public function testItUnsubscribes() {
    $segment = $this->createSegment();
    $this->createSubscriberSegment($segment);

    $pages = $this->getPages()->init($action = 'unsubscribe', $this->testData);
    $pages->unsubscribe(StatisticsUnsubscribeEntity::METHOD_LINK);

    $updatedSubscriber = $this->subscribersRepository->findOneById($this->subscriber->getId());
    $this->assertInstanceOf(SubscriberEntity::class, $updatedSubscriber);
    verify($updatedSubscriber->getStatus())->equals(SubscriberEntity::STATUS_UNSUBSCRIBED);

    $subscriberSegments = $updatedSubscriber->getSubscriberSegments();
    foreach ($subscriberSegments as $subscriberSegment) {
      $this->assertSame(SubscriberEntity::STATUS_UNSUBSCRIBED, $subscriberSegment->getStatus());
    }
  }

  public function testItTrackUnsubscribeWhenTrackingIsEnabled() {
    $unsubscribesMock = $this->make(Unsubscribes::class, ['track' => Stub\Expected::once()]);
    $this->testData['queueId'] = 1;
    SettingsController::getInstance()->set('tracking.level', TrackingConfig::LEVEL_PARTIAL);
    $pages = $this->getPages(null, $unsubscribesMock)->init($action = 'unsubscribe', $this->testData);
    $pages->unsubscribe(StatisticsUnsubscribeEntity::METHOD_LINK);
  }

  public function testItDontTrackUnsubscribeWhenTrackingIsDisabled() {
    $unsubscribesMock = $this->make(Unsubscribes::class, ['track' => Stub\Expected::never()]);
    $this->testData['queueId'] = 1;
    SettingsController::getInstance()->set('tracking.level', TrackingConfig::LEVEL_BASIC);
    $pages = $this->getPages(null, $unsubscribesMock)->init($action = 'unsubscribe', $this->testData);
    $pages->unsubscribe(StatisticsUnsubscribeEntity::METHOD_LINK);
  }

  public function testItDoesntUnsubscribeWhenPreviewing() {
    $this->testData['preview'] = 1;
    $pages = $this->getPages()->init($action = 'unsubscribe', $this->testData);
    $pages->unsubscribe(StatisticsUnsubscribeEntity::METHOD_LINK);

    $updatedSubscriber = $this->subscribersRepository->findOneById($this->subscriber->getId());
    $this->assertInstanceOf(SubscriberEntity::class, $updatedSubscriber);
    verify($updatedSubscriber->getStatus())->notEquals(SubscriberEntity::STATUS_UNSUBSCRIBED);
  }

  public function testItUnsubscribesAndTracksCorrectMethod() {
    SettingsController::getInstance()->set('tracking.level', TrackingConfig::LEVEL_PARTIAL);
    $subscriber = (new Subscriber())->withStatus(SubscriberEntity::STATUS_SUBSCRIBED)->create();
    $testData = [
      'email' => $subscriber->getEmail(),
      'queueId' => 1, // just a random queueId,
      'token' => $this->linkTokens->getToken($subscriber),
    ];
    $pages = $this->getPages()->init('unsubscribe', $testData);

    // with link method
    $pages->unsubscribe(StatisticsUnsubscribeEntity::METHOD_LINK);
    $updatedSubscriber = $this->subscribersRepository->findOneById($subscriber->getId());
    $this->assertInstanceOf(SubscriberEntity::class, $updatedSubscriber);

    verify($updatedSubscriber->getStatus())->equals(SubscriberEntity::STATUS_UNSUBSCRIBED);
    $unsubscriptionStat = $this->statisticsUnsubscribesRepository->findOneBy(['subscriber' => $updatedSubscriber->getId()]);
    verify($unsubscriptionStat->getMethod())->equals(StatisticsUnsubscribeEntity::METHOD_LINK);
    verify($unsubscriptionStat->getSource())->equals(StatisticsUnsubscribeEntity::SOURCE_NEWSLETTER);

    $this->statisticsUnsubscribesRepository->remove($unsubscriptionStat);
    $this->statisticsUnsubscribesRepository->flush();

    // with one-click method
    $subscriber = (new Subscriber())->withStatus(SubscriberEntity::STATUS_SUBSCRIBED)->create();
    $testData = [
      'email' => $subscriber->getEmail(),
      'queueId' => 1, // just a random queueId,
      'token' => $this->linkTokens->getToken($subscriber),
    ];
    $pages = $this->getPages()->init('unsubscribe', $testData);
    $pages->unsubscribe(StatisticsUnsubscribeEntity::METHOD_ONE_CLICK);
    $updatedSubscriber = $this->subscribersRepository->findOneById($subscriber->getId());
    $this->assertInstanceOf(SubscriberEntity::class, $updatedSubscriber);

    verify($updatedSubscriber->getStatus())->equals(SubscriberEntity::STATUS_UNSUBSCRIBED);
    $unsubscriptionStat = $this->statisticsUnsubscribesRepository->findOneBy(['subscriber' => $updatedSubscriber->getId()]);
    verify($unsubscriptionStat->getMethod())->equals(StatisticsUnsubscribeEntity::METHOD_ONE_CLICK);
    verify($unsubscriptionStat->getSource())->equals(StatisticsUnsubscribeEntity::SOURCE_NEWSLETTER);

  }

  public function testItUnsubscribesAndRecordsClickForOneClickMethod() {
    SettingsController::getInstance()->set('tracking.level', TrackingConfig::LEVEL_PARTIAL);
    $newsletter = (new Newsletter())->withSendingQueue()->create();
    $this->testData['queueId'] = $newsletter->getLatestQueue()->getId();
    (new NewsletterLink($newsletter))->withHash($newsletter->getHash())->create();
    $renderedNewsletter = ['html' => '', 'text' => ''];
    $this->links->process($renderedNewsletter, $newsletter, $newsletter->getLatestQueue());
    $pages = $this->getPages()->init('unsubscribe', $this->testData);

    // with one-click method
    $pages->unsubscribe(StatisticsUnsubscribeEntity::METHOD_ONE_CLICK);
    $updatedSubscriber = $this->subscribersRepository->findOneById($this->subscriber->getId());
    $this->assertInstanceOf(SubscriberEntity::class, $updatedSubscriber);
    $clickStat = $this->statisticsClicksRepository->getAllForSubscriber($this->subscriber)->getQuery()->getResult();
    verify($updatedSubscriber->getStatus())->equals(SubscriberEntity::STATUS_UNSUBSCRIBED);
    verify($clickStat)->arrayCount(1);
  }

  private function getPages(
    NewSubscriberNotificationMailer $newSubscriberNotificationsMock = null,
    Unsubscribes $unsubscribesMock = null
  ): Pages {
    $container = ContainerWrapper::getInstance();
    return new Pages(
      $newSubscriberNotificationsMock ?? $container->get(NewSubscriberNotificationMailer::class),
      $container->get(WPFunctions::class),
      $container->get(CaptchaFormRenderer::class),
      $container->get(WelcomeScheduler::class),
      $container->get(LinkTokens::class),
      $container->get(SubscriptionUrlFactory::class),
      $container->get(AssetsController::class),
      $container->get(Renderer::class),
      $unsubscribesMock ?? $container->get(Unsubscribes::class),
      $container->get(ManageSubscriptionFormRenderer::class),
      $container->get(SubscriberHandler::class),
      $this->subscribersRepository,
      $container->get(TrackingConfig::class),
      $container->get(EntityManager::class),
      $container->get(SubscriberSaveController::class),
      $container->get(SubscriberSegmentRepository::class),
      $container->get(NewsletterLinkRepository::class),
      $container->get(StatisticsClicksRepository::class),
      $container->get(SendingQueuesRepository::class)
    );
  }

  private function createSegment(): SegmentEntity {
    $segmentFactory = new SegmentFactory();
    $segment = $segmentFactory->withName('List #1')->create();

    return $segment;
  }

  private function createSubscriberSegment(SegmentEntity $segment) {
    $subscriberSegment = new SubscriberSegmentEntity($segment, $this->subscriber, 'subscribed');
    $this->entityManager->persist($subscriberSegment);
    $this->subscriber->getSubscriberSegments()->add($subscriberSegment);
    $this->entityManager->persist($this->subscriber);
    $this->entityManager->flush();
  }
}