feat: add method to find images by type

monsieurbiz · Nov 28, 2024 · b511dc6 · b511dc6
1 parent 11f2bde
commit b511dc6
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/src/Model/Product/ProductDTO.php b/src/Model/Product/ProductDTO.php
@@ -17,4 +17,15 @@
 
 class ProductDTO extends Eater
 {
+    public function getImagesByType(string $type): array
+    {
+        $images = $this->getData('images') ?? [];
+        if (!\is_array($images)) {
+            return [];
+        }
+
+        return array_filter($images, function ($image) use ($type) {
+            return \is_object($image) && method_exists($image, 'getType') && $image->getType() === $type;
+        });
+    }
 }