Added new option with a new sql query considering the relation between media and product; added new option to delete only the cache images

Author: fabrizio

Console/Command/RemoveUnusedMediaCommand.php

@@ -17,6 +17,11 @@ class RemoveUnusedMediaCommand extends Command
 {
     private const OPTION_DRY_RUN = 'dry-run';
     private const OPTION_INCLUDING_CACHE = 'including-cache';
+
+    private const OPTION_ONLY_CACHE = 'only-cache';
+
+    private const OPTION_INCLUDING_RELATION_ENTITY = 'including-relation';
+
     private const OPTION_FORCE = 'force';
     private const COMMAND_NAME_EAV_MEDIA_REMOVE_UNUSED = 'eav:media:remove-unused';
 
@@ -44,6 +49,11 @@ public function execute(InputInterface $input, OutputInterface $output): int
         $isForce = $input->getOption(self::OPTION_FORCE);
         $isDryRun = $input->getOption(self::OPTION_DRY_RUN);
         $deleteCacheAsWell = $input->getOption(self::OPTION_INCLUDING_CACHE);
+        $deleteOnlyCache = $input->getOption(self::OPTION_ONLY_CACHE);
+        if ($deleteOnlyCache) {
+            $deleteCacheAsWell=true;
+        }
+        $deleteNotInRelation = $input->getOption(self::OPTION_INCLUDING_RELATION_ENTITY);
 
         if (!$isDryRun && !$isForce) {
             if (!$input->isInteractive()) {
@@ -69,13 +79,22 @@ public function execute(InputInterface $input, OutputInterface $output): int
         }
 
         $imageDir = $this->getImageDir();
+
         $connection = $this->resourceConnection->getConnection('core_read');
         $mediaGalleryTable = $this->resourceConnection->getTableName('catalog_product_entity_media_gallery');
 
         $directoryIterator = new RecursiveDirectoryIterator($imageDir);
 
         $imagesToKeep = $connection->fetchCol('SELECT value FROM ' . $mediaGalleryTable);
 
+        // begin new code
+        if ($deleteNotInRelation) {
+            $mediaGalleryToEntityTable = $this->resourceConnection->getTableName('catalog_product_entity_media_gallery_value_to_entity');
+            $sql='SELECT value FROM ' . $mediaGalleryTable. ' where value_id IN (SELECT value_id from '.$mediaGalleryToEntityTable.')';
+            $imagesToKeep = $connection->fetchCol($sql);
+        }
+        // end
+
         foreach (new RecursiveIteratorIterator($directoryIterator) as $file) {
             // Directory guard
             if (is_dir($file)) {
@@ -87,6 +106,11 @@ public function execute(InputInterface $input, OutputInterface $output): int
                 continue;
             }
 
+            // Original image guard if option --only-cache
+            if (!$this->isInCachePath($file) && $deleteOnlyCache) {
+                continue;
+            }
+
             $filePath = str_replace($imageDir, "", $file);
             // Filepath guard
             if (empty($filePath)) {
@@ -164,6 +188,21 @@ protected function configure(): void
             null,
             'Also clear the ./cache/* entries for the corresponding images'
         );
+
+        $this->addOption(
+            self::OPTION_ONLY_CACHE,
+            'k',
+            null,
+            'Clear only the ./cache/* entries for the corresponding images nad not the corresponding images'
+        );
+
+        $this->addOption(
+            self::OPTION_INCLUDING_RELATION_ENTITY,
+            'r',
+            null,
+            'Also clear the media not in relation table "catalog_product_entity_media_gallery_value_to_entity"'
+        );
+
         $this->addOption(
             self::OPTION_DRY_RUN,
             'd',