Pretier PHP code to make world a better place

Author: Igloczek

app/code/Magento/Catalog/Block/Product/View/Gallery.php

@@ -235,17 +235,18 @@ private function getGalleryImagesConfig()
     }
 
     /**
-     * @param string $image
      * @return string
      */
-    public function getMainProductImage()
+    public function getMainProductImage($size = 'medium_image_url')
     {
-        $image = $this->getGalleryImages()->toArray()['items'][0]['medium_image_url'];
+        foreach ($this->getGalleryImages() as $image) {
+            $image = $image->getData($size);
 
-        if (!$image) {
-            return $this->_imageHelper->getDefaultPlaceholderUrl('image');
-        }
+            if (!$image) {
+                return $this->_imageHelper->getDefaultPlaceholderUrl('image');
+            }
 
-        return $image;
+            return $image;
+        }
     }
 }