Ability to remove file for ReplacingFile

src/FileAbstraction/ReplacingFile.php

@@ -5,16 +5,54 @@
 namespace Vich\UploaderBundle\FileAbstraction;
 
 use Symfony\Component\HttpFoundation\File\File;
-use Symfony\Component\HttpFoundation\File\UploadedFile;
 
 /**
  * This class can be used to signal that the given file should be "uploaded" into the Vich-abstraction
  * in cases where it is not possible to construct an `UploadedFile`.
  */
 class ReplacingFile extends File
 {
+    protected bool $removeReplacedFile;
+    protected bool $removeReplacedFileOnError;
+
+    public function __construct(
+        string $path,
+        bool $checkPath = true,
+        bool $removeReplacedFile = false,
+        bool $removeReplacedFileOnError = false
+    ) {
+        parent::__construct($path, $checkPath);
+
+        $this->removeReplacedFile = $removeReplacedFile;
+        $this->removeReplacedFileOnError = $removeReplacedFileOnError;
+    }
+
     public function getClientOriginalName(): string
     {
         return $this->getFilename();
     }
+
+    public function isRemoveReplacedFile(): bool
+    {
+        return $this->removeReplacedFile;
+    }
+
+    public function setRemoveReplacedFile(bool $removeReplacedFile): self
+    {
+        $this->removeReplacedFile = $removeReplacedFile;
+
+        return $this;
+    }
+
+    public function isRemoveReplacedFileOnError(): bool
+    {
+        return $this->removeReplacedFileOnError;
+    }
+
+    public function setRemoveReplacedFileOnError(bool $removeReplacedFileOnError): self
+    {
+        $this->removeReplacedFileOnError = $removeReplacedFileOnError;
+
+        return $this;
+    }
 }

src/Storage/AbstractStorage.php

@@ -46,7 +46,23 @@ public function upload(object $obj, PropertyMapping $mapping): void
 
         $dir = $mapping->getUploadDir($obj);
 
-        $this->doUpload($mapping, $file, $dir, $name);
+        try {
+            $this->doUpload($mapping, $file, $dir, $name);
+        } catch (\Exception $e) {
+            if ($file instanceof ReplacingFile
+                && $file->isRemoveReplacedFileOnError()
+            ) {
+                unlink($file->getPathname());
+            }
+
+            throw $e;
+        }
+
+        if ($file instanceof ReplacingFile
+            && $file->isRemoveReplacedFile()
+        ) {
+            unlink($file->getPathname());
+        }
     }
 
     abstract protected function doRemove(PropertyMapping $mapping, ?string $dir, string $name): ?bool;