<?php

namespace MailPoet\Doctrine\EventListeners;

if (!defined('ABSPATH')) exit;


use MailPoet\Doctrine\Validator\ValidationException;
use MailPoetVendor\Doctrine\ORM\Event\OnFlushEventArgs;
use MailPoetVendor\Symfony\Component\Validator\Validator\ValidatorInterface;

class ValidationListener {
  /** @var ValidatorInterface */
  private $validator;

  public function __construct(ValidatorInterface $validator) {
    $this->validator = $validator;
  }

  public function onFlush(OnFlushEventArgs $eventArgs) {
    $unitOfWork = $eventArgs->getEntityManager()->getUnitOfWork();

    foreach ($unitOfWork->getScheduledEntityInsertions() as $entity) {
      $this->validate($entity);
    }

    foreach ($unitOfWork->getScheduledEntityUpdates() as $entity) {
      $this->validate($entity);
    }
  }

  private function validate($entity) {
    $violations = $this->validator->validate($entity);
    if ($violations->count() > 0) {
      throw new ValidationException(get_class($entity), $violations);
    }
  }
}
