allow to reassign timesheets from delete-user form (#2159)

src/Repository/Query/UserFormTypeQuery.php

@@ -22,6 +22,10 @@ final class UserFormTypeQuery extends BaseFormTypeQuery
      * @var User[]
      */
     private $includeUsers = [];
+    /**
+     * @var User[]
+     */
+    private $ignoredUsers = [];
 
     /**
      * Sets a list of users which must be included in the result always.
@@ -45,4 +49,27 @@ final class UserFormTypeQuery extends BaseFormTypeQuery
     {
         return $this->includeUsers;
     }
+
+    /**
+     * Given user will be excluded from the result set.
+     *
+     * @param User $user
+     * @return $this
+     */
+    public function addUserToIgnore(User $user): UserFormTypeQuery
+    {
+        $this->ignoredUsers[] = $user;
+
+        return $this;
+    }
+
+    /**
+     * Returns the list of users that should not be loaded.
+     *
+     * @return User[]
+     */
+    public function getUsersToIgnore(): array
+    {
+        return $this->ignoredUsers;
+    }
 }

src/Repository/UserRepository.php

@@ -9,7 +9,9 @@
 
 namespace App\Repository;
 
+use App\Entity\Invoice;
 use App\Entity\Role;
+use App\Entity\Timesheet;
 use App\Entity\User;
 use App\Repository\Loader\UserIdLoader;
 use App\Repository\Loader\UserLoader;
@@ -165,6 +167,14 @@ class UserRepository extends EntityRepository implements UserLoaderInterface
             $qb->andWhere($or);
         }
 
+        if (\count($query->getUsersToIgnore()) > 0) {
+            $ids = array_map(function (User $user) {
+                return $user->getId();
+            }, $query->getUsersToIgnore());
+
+            $qb->andWhere($qb->expr()->notIn('u.id', $ids));
+        }
+
         $qb->orderBy('u.username', 'ASC');
 
         $this->addPermissionCriteria($qb, $query->getUser(), $query->getTeams());
@@ -353,4 +363,41 @@ class UserRepository extends EntityRepository implements UserLoaderInterface
 
         return $results;
     }
+
+    public function deleteUser(User $delete, ?User $replace = null)
+    {
+        $em = $this->getEntityManager();
+        $em->beginTransaction();
+
+        try {
+            if (null !== $replace) {
+                $qb = $em->createQueryBuilder();
+                $qb
+                    ->update(Timesheet::class, 't')
+                    ->set('t.user', ':replace')
+                    ->where('t.user = :delete')
+                    ->setParameter('delete', $delete)
+                    ->setParameter('replace', $replace)
+                    ->getQuery()
+                    ->execute();
+
+                $qb = $em->createQueryBuilder();
+                $qb
+                    ->update(Invoice::class, 'i')
+                    ->set('i.user', ':replace')
+                    ->where('i.user = :delete')
+                    ->setParameter('delete', $delete)
+                    ->setParameter('replace', $replace)
+                    ->getQuery()
+                    ->execute();
+            }
+
+            $em->remove($delete);
+            $em->flush();
+            $em->commit();
+        } catch (ORMException $ex) {
+            $em->rollback();
+            throw $ex;
+        }
+    }
 }