[WIP] Remove media gallery assets metadata when a directory removed

Author: rogyar

app/code/Magento/MediaGallery/Model/Asset/Command/DeleteByDirectoryPath.php

@@ -0,0 +1,93 @@
+<?php
+/**
+ * Copyright © Magento, Inc. All rights reserved.
+ * See COPYING.txt for license details.
+ */
+declare(strict_types=1);
+
+namespace Magento\MediaGallery\Model\Asset\Command;
+
+use Magento\Framework\App\ResourceConnection;
+use Magento\Framework\DB\Adapter\AdapterInterface;
+use Magento\Framework\Exception\CouldNotDeleteException;
+use Magento\Framework\Exception\LocalizedException;
+use Magento\MediaGalleryApi\Model\Asset\Command\DeleteByDirectoryPathInterface;
+use Psr\Log\LoggerInterface;
+
+/**
+ * Class DeleteByDirectoryPath
+ */
+class DeleteByDirectoryPath implements DeleteByDirectoryPathInterface
+{
+    private const TABLE_MEDIA_GALLERY_ASSET = 'media_gallery_asset';
+
+    private const MEDIA_GALLERY_ASSET_PATH = 'path';
+
+    /**
+     * @var ResourceConnection
+     */
+    private $resourceConnection;
+
+    /**
+     * @var LoggerInterface
+     */
+    private $logger;
+
+    /**
+     * DeleteById constructor.
+     *
+     * @param ResourceConnection $resourceConnection
+     * @param LoggerInterface $logger
+     */
+    public function __construct(
+        ResourceConnection $resourceConnection,
+        LoggerInterface $logger
+    ) {
+        $this->resourceConnection = $resourceConnection;
+        $this->logger = $logger;
+    }
+
+    /**
+     * Delete media asset by path
+     *
+     * @param string $directoryPath
+     *
+     * @return void
+     * @throws CouldNotDeleteException
+     */
+    public function execute(string $directoryPath): void
+    {
+        try {
+            $this->validateDirectoryPath($directoryPath);
+
+            if (substr($directoryPath, -1) !== '/') {
+                $directoryPath .= '/';
+            }
+
+            /** @var AdapterInterface $connection */
+            $connection = $this->resourceConnection->getConnection();
+            $tableName = $this->resourceConnection->getTableName(self::TABLE_MEDIA_GALLERY_ASSET);
+            $connection->delete($tableName, [self::MEDIA_GALLERY_ASSET_PATH . ' LIKE ?' => $directoryPath . '%']);
+        } catch (\Exception $exception) {
+            $this->logger->critical($exception);
+            $message = __(
+                'Could not delete media assets by path %path: %error',
+                ['path' => $directoryPath, 'error' => $exception->getMessage()]
+            );
+            throw new CouldNotDeleteException($message, $exception);
+        }
+    }
+
+    /**
+     * Validate the directory path
+     *
+     * @param string $directoryPath
+     * @throws LocalizedException
+     */
+    private function validateDirectoryPath(string $directoryPath): void
+    {
+        if (trim($directoryPath) === '') {
+            throw new LocalizedException(__('The directory path cannot be empty'));
+        }
+    }
+}

app/code/Magento/MediaGallery/Plugin/Wysiwyg/Images/Storage.php

@@ -8,12 +8,12 @@
 
 namespace Magento\MediaGallery\Plugin\Wysiwyg\Images;
 
+use Magento\MediaGalleryApi\Model\Asset\Command\DeleteByDirectoryPathInterface;
 use Magento\MediaGalleryApi\Model\Asset\Command\GetByPathInterface;
 use Magento\MediaGalleryApi\Model\Asset\Command\DeleteByPathInterface;
 use Magento\Cms\Model\Wysiwyg\Images\Storage as StorageSubject;
 use Magento\Framework\App\Filesystem\DirectoryList;
 use Magento\Framework\Filesystem;
-use Magento\Framework\Exception\ValidatorException;
 use Psr\Log\LoggerInterface;
 
 /**
@@ -31,6 +31,11 @@ class Storage
      */
     private $deleteMediaAssetByPath;
 
+    /**
+     * @var DeleteByDirectoryPathInterface
+     */
+    private $deleteMediAssetByDirectoryPath;
+
     /**
      * @var Filesystem
      */
@@ -46,17 +51,20 @@ class Storage
      *
      * @param GetByPathInterface $getMediaAssetByPath
      * @param DeleteByPathInterface $deleteMediaAssetByPath
+     * @param DeleteByDirectoryPathInterface $deleteByDirectoryPath
      * @param Filesystem $filesystem
      * @param LoggerInterface $logger
      */
     public function __construct(
         GetByPathInterface $getMediaAssetByPath,
         DeleteByPathInterface $deleteMediaAssetByPath,
+        DeleteByDirectoryPathInterface $deleteByDirectoryPath,
         Filesystem $filesystem,
         LoggerInterface $logger
     ) {
         $this->getMediaAssetByPath = $getMediaAssetByPath;
         $this->deleteMediaAssetByPath = $deleteMediaAssetByPath;
+        $this->deleteMediAssetByDirectoryPath = $deleteByDirectoryPath;
         $this->filesystem = $filesystem;
         $this->logger = $logger;
     }
@@ -69,7 +77,6 @@ public function __construct(
      * @param string $target
      *
      * @return StorageSubject
-     * @throws ValidatorException
      *
      * @SuppressWarnings(PHPMD.UnusedFormalParameter)
      */
@@ -92,4 +99,36 @@ public function afterDeleteFile(StorageSubject $subject, StorageSubject $result,
 
         return $result;
     }
+
+    /**
+     * Delete media data after the folder delete action from Wysiwyg
+     *
+     * @param StorageSubject $subject
+     * @param null $result
+     * @param string $path
+     *
+     * @return null
+     *
+     * @SuppressWarnings(PHPMD.UnusedFormalParameter)
+     */
+    public function afterDeleteDirectory(StorageSubject $subject, $result, $path)
+    {
+        if (!is_string($path)) {
+            return $result;
+        }
+
+        $relativePath = $this->filesystem->getDirectoryRead(DirectoryList::MEDIA)->getRelativePath($path);
+
+        if (!$relativePath) {
+            return $result;
+        }
+
+        try {
+            $this->deleteMediAssetByDirectoryPath->execute($relativePath);
+        } catch (\Exception $exception) {
+            $this->logger->critical($exception);
+        }
+
+        return $result;
+    }
 }

app/code/Magento/MediaGallery/etc/di.xml

@@ -13,6 +13,7 @@
     <preference for="Magento\MediaGalleryApi\Model\Asset\Command\SaveInterface" type="Magento\MediaGallery\Model\Asset\Command\Save"/>
     <preference for="Magento\MediaGalleryApi\Model\Asset\Command\GetByPathInterface" type="Magento\MediaGallery\Model\Asset\Command\GetByPath"/>
     <preference for="Magento\MediaGalleryApi\Model\Asset\Command\DeleteByPathInterface" type="Magento\MediaGallery\Model\Asset\Command\DeleteByPath"/>
+    <preference for="Magento\MediaGalleryApi\Model\Asset\Command\DeleteByDirectoryPathInterface" type="Magento\MediaGallery\Model\Asset\Command\DeleteByDirectoryPath"/>
 
     <preference for="Magento\MediaGalleryApi\Model\Keyword\Command\GetAssetKeywordsInterface" type="Magento\MediaGallery\Model\Keyword\Command\GetAssetKeywords"/>
     <preference for="Magento\MediaGalleryApi\Model\Keyword\Command\SaveAssetKeywordsInterface" type="Magento\MediaGallery\Model\Keyword\Command\SaveAssetKeywords"/>

app/code/Magento/MediaGalleryApi/Model/Asset/Command/DeleteByDirectoryPathInterface.php

@@ -0,0 +1,25 @@
+<?php
+/**
+ * Copyright © Magento, Inc. All rights reserved.
+ * See COPYING.txt for license details.
+ */
+
+declare(strict_types=1);
+
+namespace Magento\MediaGalleryApi\Model\Asset\Command;
+
+/**
+ * A command represents the media gallery assets delete action. A media gallery asset is filtered by directory
+ * path value.
+ */
+interface DeleteByDirectoryPathInterface
+{
+    /**
+     * Delete media assets by directory path
+     *
+     * @param string $directoryPath
+     *
+     * @return void
+     */
+    public function execute(string $directoryPath): void;
+}