* * This program is free software: you can redistribute it and/or modify * it under the terms of the GNU Affero General Public License as published by * the Free Software Foundation, either version 3 of the License, or * (at your option) any later version. * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU Affero General Public License for more details. * * You should have received a copy of the GNU Affero General Public License * along with this program. If not, see . */ namespace Chill\PersonBundle\Actions\Remove; use Doctrine\ORM\EntityManagerInterface; use Chill\PersonBundle\Entity\Person; use Chill\PersonBundle\Entity\AccompanyingPeriod; use Doctrine\ORM\Mapping\ClassMetadata; /** * Move all person to a new one, and delete the old record. * */ class PersonMove { /** * * @var EntityManagerInterface */ protected $em; public function __construct(EntityManagerInterface $em) { $this->em = $em; } public function getSQL(Person $from, Person $to) { $sqls = []; foreach ($this->em->getMetadataFactory()->getAllMetadata() as $metadata) { if ($metadata->isMappedSuperclass) { continue; } foreach ($metadata->getAssociationMappings() as $field => $mapping) { if ($mapping['targetEntity'] === Person::class) { if (\in_array($metadata->getName(), $this->deleteEntities())) { $sqls[] = $this->createDeleteSQL($metadata, $from, $field); } else { $sqls[] = $this->createMoveSQL($metadata, $from, $to, $field); } } } } $personMetadata = $this->em->getClassMetadata(Person::class); $sqls[] = sprintf("DELETE FROM %s WHERE id = %d", $this->getTableName($personMetadata), $from->getId()); return $sqls ?? []; } protected function createMoveSQL(ClassMetadata $metadata, Person $from, Person $to, $field): string { $mapping = $metadata->getAssociationMapping($field); // Set part of the query, aka in "UPDATE table SET " $sets = []; foreach ($mapping["joinColumns"] as $columns) { $sets[] = sprintf("%s = %d", $columns["name"], $to->getId()); } $conditions = []; foreach ($mapping["joinColumns"] as $columns) { $conditions[] = sprintf("%s = %d", $columns["name"], $from->getId()); } return \sprintf("UPDATE %s SET %s WHERE %s", $this->getTableName($metadata), \implode(" ", $sets), \implode(" AND ", $conditions) ); } protected function createDeleteSQL(ClassMetadata $metadata, Person $from, $field): string { $mapping = $metadata->getAssociationMapping($field); $conditions = []; foreach ($mapping["joinColumns"] as $columns) { $conditions[] = sprintf("%s = %d", $columns["name"], $from->getId()); } return \sprintf("DELETE FROM %s WHERE %s", $this->getTableName($metadata), \implode(" AND ", $conditions) ); } /** * return an array of classes where entities should be deleted * instead of moved * * @return array */ protected function deleteEntities(): array { return [ AccompanyingPeriod::class ]; } /** * get the full table name with schema if it does exists */ private function getTableName(ClassMetadata $metadata): string { return empty($metadata->getSchemaName()) ? $metadata->getTableName() : $metadata->getSchemaName().".".$metadata->getTableName(); } }