feat(command): Allow specifying group

Author: pan93412

src/Command/CreateUsersCommand.php

@@ -4,6 +4,7 @@
 
 namespace App\Command;
 
+use App\Entity\Group;
 use App\Entity\User;
 use Doctrine\ORM\EntityManagerInterface;
 use Random\RandomException;
@@ -61,7 +62,7 @@ protected function execute(InputInterface $input, OutputInterface $output): int
         $io->title("Creating users from $filename");
 
         /**
-         * @var array{email: string, name: string, roles: list<string>}[] $users
+         * @var array{email: string, name: string, roles: list<string>, group: string|null}[] $users
          */
         $users = $io->progressIterate(self::parseUsers($filename));
 
@@ -75,7 +76,25 @@ protected function execute(InputInterface $input, OutputInterface $output): int
         foreach ($users as $user) {
             $password = self::generateRandomPassword();
 
-            $user = (new User())->setName($user['name'])->setEmail($user['email'])->setRoles($user['roles']);
+            /**
+             * @var Group|null $group
+             */
+            $group = null;
+
+            if (null !== $user['group']) {
+                $group = $this->entityManager->getRepository(Group::class)->findOneBy(['name' => $user['group']]);
+                if (null === $group) {
+                    $io->warning("Group {$user['group']} not found for user {$user['email']}. Skipping.");
+                    continue;
+                }
+            }
+
+            $user = (new User())
+                ->setName($user['name'])
+                ->setEmail($user['email'])
+                ->setRoles($user['roles'])
+                ->setGroup($group);
+
             $hashedPassword = $this->passwordHasher->hashPassword($user, $password);
             $user->setPassword($hashedPassword);
 
@@ -102,7 +121,7 @@ protected function execute(InputInterface $input, OutputInterface $output): int
     }
 
     /**
-     * @return array{email: string, name: string, roles: list<string>}[]
+     * @return array{email: string, name: string, roles: list<string>, group: string|null}[]
      */
     private static function parseUsers(string $filename): array
     {
@@ -119,6 +138,7 @@ private static function parseUsers(string $filename): array
         $emailIndex = array_search('email', $header, true);
         $nameIndex = array_search('name', $header, true);
         $rolesIndex = array_search('roles', $header, true);
+        $groupIndex = array_search('group', $header, true);
 
         if (!\is_int($emailIndex) || !\is_int($nameIndex) || !\is_int($rolesIndex)) {
             throw new \RuntimeException("Could not find email, name, or roles in the header of $filename.");
@@ -129,15 +149,17 @@ private static function parseUsers(string $filename): array
             $email = $row[$emailIndex];
             $name = $row[$nameIndex];
             $roles = $row[$rolesIndex];
+            $group = false !== $groupIndex ? $row[$groupIndex] : null;
 
-            if (!\is_string($email) || !\is_string($name) || !\is_string($roles)) {
+            if (!\is_string($email) || !\is_string($name) || !\is_string($roles) || (!\is_string($group) && null !== $group)) {
                 throw new \RuntimeException("Invalid row in $filename.");
             }
 
             $users[] = [
                 'email' => $email,
                 'name' => $name,
                 'roles' => explode(',', $roles),
+                'group' => $group,
             ];
         }