IBX-5385: Added option content-type to reindex command

src/bundle/Core/Command/ReindexCommand.php

@@ -11,6 +11,10 @@
 use const DIRECTORY_SEPARATOR;
 use Generator;
 use Ibexa\Contracts\Core\Persistence\Content\Location\Handler;
+use Ibexa\Contracts\Core\Repository\ContentService;
+use Ibexa\Contracts\Core\Repository\Values\Content\Content;
+use Ibexa\Contracts\Core\Repository\Values\Content\Query;
+use Ibexa\Contracts\Core\Repository\Values\Filter\Filter;
 use Ibexa\Contracts\Core\Search\Content\IndexerGateway;
 use Ibexa\Core\Base\Exceptions\InvalidArgumentException;
 use Ibexa\Core\Search\Common\IncrementalIndexer;
@@ -25,6 +29,7 @@
 use Symfony\Component\Console\Style\SymfonyStyle;
 use Symfony\Component\Process\PhpExecutableFinder;
 use Symfony\Component\Process\Process;
+use Traversable;
 
 class ReindexCommand extends Command implements BackwardCompatibleCommand
 {
@@ -55,6 +60,9 @@ class ReindexCommand extends Command implements BackwardCompatibleCommand
     /** @var \Ibexa\Contracts\Core\Persistence\Content\Location\Handler */
     private $locationHandler;
 
+    /** @var \Ibexa\Contracts\Core\Repository\ContentService */
+    private $contentService;
+
     public function __construct(
         $searchIndexer,
         Handler $locationHandler,
@@ -64,6 +72,7 @@ public function __construct(
         string $env,
         bool $isDebug,
         string $projectDir,
+        ContentService $contentService,
         string $phpPath = null
     ) {
         $this->gateway = $gateway;
@@ -75,6 +84,7 @@ public function __construct(
         $this->env = $env;
         $this->isDebug = $isDebug;
         $this->projectDir = $projectDir;
+        $this->contentService = $contentService;
         $this->phpPath = $phpPath;
 
         parent::__construct();
@@ -128,23 +138,28 @@ protected function configure()
                 'since',
                 null,
                 InputOption::VALUE_OPTIONAL,
-                'Refresh changes since a time provided in any format understood by DateTime. Implies "no-purge", cannot be combined with "content-ids" or "subtree"'
+                'Refresh changes since a time provided in any format understood by DateTime. Implies "no-purge", cannot be combined with "content-ids", "subtree", or "content-type"'
             )->addOption(
                 'content-ids',
                 null,
                 InputOption::VALUE_OPTIONAL,
-                'Comma-separated list of content ID\'s to refresh (deleted/updated/added). Implies "no-purge", cannot be combined with "since" or "subtree"'
+                'Comma-separated list of content ID\'s to refresh (deleted/updated/added). Implies "no-purge", cannot be combined with "since", "subtree", or "content-type"'
             )->addOption(
                 'subtree',
                 null,
                 InputOption::VALUE_OPTIONAL,
-                'Location ID whose subtree will be indexed (including the Location itself). Implies "no-purge", cannot be combined with "since" or "content-ids"'
+                'Location ID whose subtree will be indexed (including the Location itself). Implies "no-purge", cannot be combined with "since", "content-ids", or "content-type"'
             )->addOption(
                 'processes',
                 null,
                 InputOption::VALUE_OPTIONAL,
                 'Number of child processes to run in parallel for iterations, if set to "auto" it will set to number of CPU cores -1, set to "1" or "0" to disable',
                 'auto'
+            )->addOption(
+                'content-type',
+                null,
+                InputOption::VALUE_REQUIRED,
+                'Content type identifier to refresh (deleted/updated/added). Implies "no-purge", cannot be combined with "since", "subtree", or "content-ids"'
             )->setHelp(
                 <<<EOT
                     The command <info>%command.name%</info> indexes the current configured database in the configured search engine index.
@@ -256,6 +271,18 @@ protected function indexIncrementally(
             $count = $this->gateway->countContentInSubtree($location->pathString);
             $generator = $this->gateway->getContentInSubtree($location->pathString, $iterationCount);
             $purge = false;
+        } elseif ($contentType = $input->getOption('content-type')) {
+            $filter = new Filter();
+            $filter
+                ->withCriterion(
+                    new Query\Criterion\ContentTypeIdentifier($contentType)
+                );
+            /** @var \Ibexa\Contracts\Core\Repository\Values\Content\ContentList $contentList */
+            $contentList = $this->contentService->find($filter);
+            $count = $contentList->getTotalCount();
+            $contentListIterator = $contentList->getIterator();
+            $generator = $this->fetchIterationFromContentListIterator($contentListIterator, $iterationCount);
+            $purge = false;
         } else {
             $count = $this->gateway->countAllContent();
             $generator = $this->gateway->getAllContent($iterationCount);
@@ -454,6 +481,26 @@ public function getDeprecatedAliases(): array
     {
         return ['ezplatform:reindex'];
     }
+
+    /**
+     * @param Traversable<Content> $contentListIterator
+     * @param int $iterationCount
+     *
+     * @return \Generator
+     */
+    private function fetchIterationFromContentListIterator(Traversable $contentListIterator, int $iterationCount): Generator
+    {
+        $i = 0;
+        $contentIds = [];
+        foreach ($contentListIterator as $content) {
+            $contentIds[] = $content->id;
+            if ($iterationCount <= $i) {
+                break;
+            }
+            ++$i;
+        }
+        yield $contentIds;
+    }
 }
 
 class_alias(ReindexCommand::class, 'eZ\Bundle\EzPublishCoreBundle\Command\ReindexCommand');

src/bundle/Core/Resources/config/commands.yml

@@ -42,6 +42,7 @@ services:
             $env: '%kernel.environment%'
             $projectDir: '%kernel.project_dir%'
             $isDebug: '%kernel.debug%'
+            $contentService: '@Ibexa\Contracts\Core\Repository\ContentService'
         tags:
             - { name: console.command }