126 lines
4.5 KiB
PHP
126 lines
4.5 KiB
PHP
<?php
|
|
|
|
/*
|
|
* This file is part of the Symfony package.
|
|
*
|
|
* (c) Fabien Potencier <fabien@symfony.com>
|
|
*
|
|
* For the full copyright and license information, please view the LICENSE
|
|
* file that was distributed with this source code.
|
|
*/
|
|
|
|
namespace Symfony\Bridge\Doctrine\Validator\Constraints;
|
|
|
|
use Doctrine\Common\Persistence\ManagerRegistry;
|
|
use Symfony\Component\Validator\Constraint;
|
|
use Symfony\Component\Validator\Exception\UnexpectedTypeException;
|
|
use Symfony\Component\Validator\Exception\ConstraintDefinitionException;
|
|
use Symfony\Component\Validator\ConstraintValidator;
|
|
|
|
/**
|
|
* Unique Entity Validator checks if one or a set of fields contain unique values.
|
|
*
|
|
* @author Benjamin Eberlei <kontakt@beberlei.de>
|
|
*/
|
|
class UniqueEntityValidator extends ConstraintValidator
|
|
{
|
|
/**
|
|
* @var ManagerRegistry
|
|
*/
|
|
private $registry;
|
|
|
|
/**
|
|
* @param ManagerRegistry $registry
|
|
*/
|
|
public function __construct(ManagerRegistry $registry)
|
|
{
|
|
$this->registry = $registry;
|
|
}
|
|
|
|
/**
|
|
* @param object $entity
|
|
* @param Constraint $constraint
|
|
*/
|
|
public function validate($entity, Constraint $constraint)
|
|
{
|
|
if (!is_array($constraint->fields) && !is_string($constraint->fields)) {
|
|
throw new UnexpectedTypeException($constraint->fields, 'array');
|
|
}
|
|
|
|
if (null !== $constraint->errorPath && !is_string($constraint->errorPath)) {
|
|
throw new UnexpectedTypeException($constraint->errorPath, 'string or null');
|
|
}
|
|
|
|
$fields = (array) $constraint->fields;
|
|
|
|
if (0 === count($fields)) {
|
|
throw new ConstraintDefinitionException('At least one field has to be specified.');
|
|
}
|
|
|
|
if ($constraint->em) {
|
|
$em = $this->registry->getManager($constraint->em);
|
|
} else {
|
|
$em = $this->registry->getManagerForClass(get_class($entity));
|
|
}
|
|
|
|
$className = $this->context->getCurrentClass();
|
|
$class = $em->getClassMetadata($className);
|
|
/* @var $class \Doctrine\Common\Persistence\Mapping\ClassMetadata */
|
|
|
|
$criteria = array();
|
|
foreach ($fields as $fieldName) {
|
|
if (!$class->hasField($fieldName) && !$class->hasAssociation($fieldName)) {
|
|
throw new ConstraintDefinitionException("Only field names mapped by Doctrine can be validated for uniqueness.");
|
|
}
|
|
|
|
$criteria[$fieldName] = $class->reflFields[$fieldName]->getValue($entity);
|
|
|
|
if (null === $criteria[$fieldName]) {
|
|
return;
|
|
}
|
|
|
|
if ($class->hasAssociation($fieldName)) {
|
|
/* Ensure the Proxy is initialized before using reflection to
|
|
* read its identifiers. This is necessary because the wrapped
|
|
* getter methods in the Proxy are being bypassed.
|
|
*/
|
|
$em->initializeObject($criteria[$fieldName]);
|
|
|
|
$relatedClass = $em->getClassMetadata($class->getAssociationTargetClass($fieldName));
|
|
$relatedId = $relatedClass->getIdentifierValues($criteria[$fieldName]);
|
|
|
|
if (count($relatedId) > 1) {
|
|
throw new ConstraintDefinitionException(
|
|
"Associated entities are not allowed to have more than one identifier field to be " .
|
|
"part of a unique constraint in: " . $class->getName() . "#" . $fieldName
|
|
);
|
|
}
|
|
$criteria[$fieldName] = array_pop($relatedId);
|
|
}
|
|
}
|
|
|
|
$repository = $em->getRepository($className);
|
|
$result = $repository->{$constraint->repositoryMethod}($criteria);
|
|
|
|
/* If the result is a MongoCursor, it must be advanced to the first
|
|
* element. Rewinding should have no ill effect if $result is another
|
|
* iterator implementation.
|
|
*/
|
|
if ($result instanceof \Iterator) {
|
|
$result->rewind();
|
|
}
|
|
|
|
/* If no entity matched the query criteria or a single entity matched,
|
|
* which is the same as the entity being validated, the criteria is
|
|
* unique.
|
|
*/
|
|
if (0 === count($result) || (1 === count($result) && $entity === ($result instanceof \Iterator ? $result->current() : current($result)))) {
|
|
return;
|
|
}
|
|
|
|
$errorPath = null !== $constraint->errorPath ? $constraint->errorPath : $fields[0];
|
|
|
|
$this->context->addViolationAtSubPath($errorPath, $constraint->message, array(), $criteria[$fields[0]]);
|
|
}
|
|
}
|