add methods getError(), getSize() to UploadControlEntity

contributte · Jul 20, 2021 · 7520a6b · 7520a6b
1 parent 7880912
commit 7520a6b
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/src/Bridge/Nette/Form/Entity/UploadControlEntity.php b/src/Bridge/Nette/Form/Entity/UploadControlEntity.php
@@ -2,6 +2,7 @@
 
 namespace Contributte\Imagist\Bridge\Nette\Form\Entity;
 
+use Contributte\Imagist\Bridge\Nette\Uploader\FileUploadUploader;
 use Contributte\Imagist\Entity\EmptyImageInterface;
 use Contributte\Imagist\Entity\PersistentImageInterface;
 use Contributte\Imagist\Entity\StorableImageInterface;
@@ -22,6 +23,26 @@ public function __construct(?StorableImageInterface $value = null, ?PersistentIm
 		$this->default = $default;
 	}
 
+	public function getSize(): ?int
+	{
+		$uploader = $this->value?->getUploader();
+		if ($uploader instanceof FileUploadUploader) {
+			return $uploader->getFileUpload()->getSize();
+		}
+
+		return null;
+	}
+
+	public function getError(): ?int
+	{
+		$uploader = $this->value?->getUploader();
+		if ($uploader instanceof FileUploadUploader) {
+			return $uploader->getFileUpload()->getError();
+		}
+
+		return null;
+	}
+
 	public function getDefault(): ?PersistentImageInterface
 	{
 		return $this->default;