Add completion values to --purge option

Author: GromNaN

src/Command/LoadDataFixturesDoctrineODMCommand.php

@@ -7,18 +7,21 @@
 use Doctrine\Bundle\MongoDBBundle\Loader\SymfonyFixturesLoaderInterface;
 use Doctrine\Bundle\MongoDBBundle\ManagerRegistry;
 use Doctrine\Common\DataFixtures\Executor\MongoDBExecutor;
+use Doctrine\Common\DataFixtures\Purger\MongoDBPurgeMode;
 use Doctrine\Common\DataFixtures\Purger\MongoDBPurger;
 use Doctrine\ODM\MongoDB\DocumentManager;
+use InvalidArgumentException;
 use Psr\Log\AbstractLogger;
 use Symfony\Component\Console\Input\InputInterface;
 use Symfony\Component\Console\Input\InputOption;
 use Symfony\Component\Console\Output\OutputInterface;
 use Symfony\Component\Console\Question\ConfirmationQuestion;
 use Symfony\Component\Console\Style\SymfonyStyle;
 
+use function array_column;
 use function assert;
+use function class_exists;
 use function implode;
-use function method_exists;
 use function sprintf;
 
 /**
@@ -41,7 +44,7 @@ protected function configure(): void
             ->addOption('group', null, InputOption::VALUE_IS_ARRAY | InputOption::VALUE_REQUIRED, 'Only load fixtures that belong to this group (use with --services)')
             ->addOption('append', null, InputOption::VALUE_NONE, 'Append the data fixtures instead of flushing the database first.')
             ->addOption('dm', null, InputOption::VALUE_REQUIRED, 'The document manager to use for this command.')
-            ->addOption('purge-with-delete', null, InputOption::VALUE_NONE, 'Use deleteMany() to purge the collections instead of dropping them. This is useful to keep the collections, their metadata and their indexes, but it is slower than dropping them.')
+            ->addOption('purge', null, InputOption::VALUE_OPTIONAL, 'Purge the database before loading the fixtures. If set to "delete", collections will be kept and documents deleted instead of dropping the collections.', null, self::getPurgeModes(...))
             ->setHelp(<<<'EOT'
 The <info>doctrine:mongodb:fixtures:load</info> command loads data fixtures from your application:
 
@@ -55,9 +58,9 @@ protected function configure(): void
 
     <info>php %command.full_name%</info> <comment>--group=group1</comment>
 
-If the collection uses search indexes or encryption, you can use the <info>--purge-with-delete</info> option to keep the collections instead of dropping them when purging the database:
+If the collection uses search indexes or encryption, you can use the <info>--purge=delete</info> option to keep the collections instead of dropping them when purging the database:
 
-  <info>php %command.full_name%</info> --purge-with-delete
+  <info>php %command.full_name%</info> --purge=delete
 EOT
         );
     }
@@ -69,9 +72,9 @@ protected function execute(InputInterface $input, OutputInterface $output): int
 
         $ui = new SymfonyStyle($input, $output);
 
-        if ($input->getOption('purge-with-delete')) {
+        if ($input->hasOption('purge')) {
             if ($input->getOption('append')) {
-                $ui->error('The --purge-with-delete option cannot be used with the --append option.');
+                $ui->error('The --purge option cannot be used with the --append option.');
 
                 return self::INVALID;
             }
@@ -99,14 +102,15 @@ protected function execute(InputInterface $input, OutputInterface $output): int
         }
 
         $purger = new MongoDBPurger($dm);
-        if ($input->getOption('purge-with-delete')) {
-            if (! method_exists($purger, 'setPurgeMode')) {
+        $purge  = $input->getOption('purge');
+        if ($purge) {
+            if (! class_exists(MongoDBPurgeMode::class)) {
                 $ui->error('The --purge-with-delete option requires doctrine/data-fixtures >= 2.1.0.');
 
                 return self::INVALID;
             }
 
-            $purger->setPurgeMode(MongoDBPurger::PURGE_MODE_DELETE);
+            $purger->setPurgeMode(MongoDBPurgeMode::tryFrom($purge) ?? throw new InvalidArgumentException('Invalid purge mode: ' . $purge));
         }
 
         $executor = new MongoDBExecutor($dm, $purger);
@@ -126,4 +130,9 @@ public function log($level, $message, array $context = []): void
 
         return 0;
     }
+
+    private static function getPurgeModes(): array
+    {
+        return class_exists(MongoDBPurgeMode::class) ? array_column(MongoDBPurgeMode::cases(), 'value') : [];
+    }
 }