CLOUDINARY-525 - Additional graphql media enhancements

Author: arielien

Model/Api/ProductGalleryManagement.php

@@ -850,7 +850,7 @@ private function getProductCldUrlsBySku($sku)
                     $urls['thumbnail'] = $gallItem->getUrl();
                 }
             }
-            $urls['gallery_widget_parameters'] = $this->jsonHelper->jsonEncode($this->productGalleryHelper->getCloudinaryPGOptions());
+            $urls['gallery_widget_parameters'] = $this->jsonHelper->jsonEncode($this->productGalleryHelper->getCloudinaryPGOptions(true, true));
         } catch (\Exception $e) {
             $urls = [
                 'error' => 1,

Model/GraphQLResolver/ProductAttributeCldResolver.php

@@ -2,12 +2,16 @@
     declare(strict_types=1);
     namespace Cloudinary\Cloudinary\Model\GraphQLResolver;
 
+    use Cloudinary\Cloudinary\Model\Configuration;
+    use Magento\Framework\Exception\LocalizedException;
     use Magento\Framework\GraphQl\Config\Element\Field;
     use Magento\Framework\GraphQl\Query\ResolverInterface;
     use Magento\Framework\GraphQl\Schema\Type\ResolveInfo;
     use Cloudinary\Cloudinary\Model\Api\ProductGalleryManagement;
     use Magento\Framework\GraphQl\Type\Definition\ObjectType;
-
+    use Magento\Catalog\Api\ProductRepositoryInterface;
+    use Magento\Store\Model\StoreManagerInterface;
+    use Magento\Framework\UrlInterface;
     /**
      * Class ProductAttributeCldResolver
      **/
@@ -18,25 +22,78 @@ class ProductAttributeCldResolver implements ResolverInterface
          */
         private $productGalleryManagement;
 
+        protected $productRepository;
+
+        private $urlBuilder;
+        private $storeManager;
+
+        protected $_sku;
+
+        protected $_configuration;
+
         /**
          * ProductAttributeCldResolver constructor.
          * @param ProductGalleryManagement $productGalleryManagement
          */
         public function __construct(
-            ProductGalleryManagement $productGalleryManagement
+            ProductGalleryManagement $productGalleryManagement,
+            ProductRepositoryInterface $productRepository,
+            Configuration $configuration,
+            UrlInterface $urlBuilder,
+            StoreManagerInterface $storeManager,
         ) {
             $this->productGalleryManagement = $productGalleryManagement;
+            $this->productRepository = $productRepository;
+            $this->urlBuilder = $urlBuilder;
+            $this->storeManager = $storeManager;
+            $this->_configuration = $configuration;
         }
 
         /**
          * @inheritdoc
          */
         public function resolve(Field $field, $context, ResolveInfo $info, array $value = null, array $args = null)
         {
-            $productId = $value['sku'];
-            $productMediaStr = $this->productGalleryManagement->getProductMedia($productId);
+            $sku = $value['sku'] ?? null;
+            if ($sku) {
+                $this->_sku = $sku;
+            }
+
+            $attributeCodes = $args['attribute_codes'] ?? null;
+            if ($attributeCodes && is_array($attributeCodes)) {
+                $this->checkEnabled();
+                $product = $this->productRepository->get($this->_sku);
+                $mediaAttributes = [];
+                  foreach ($attributeCodes as $attributeCode) {
+                      $attrValue = $product->getData($attributeCode);
+                      if ($attrValue) {
+                          foreach ($product->getMediaGalleryImages() as $gallItem) {
+                              if ($attrValue == $gallItem->getFile()) {
+                                  $mediaAttributes[] = [
+                                      'attribute_code' => $attributeCode,
+                                      'url' => $gallItem->getUrl()
+                                  ];
+                              }
+                          }
+
+                      }
+                  }
+                  return $mediaAttributes;
+            }
+            $productMediaStr = $this->productGalleryManagement->getProductMedia($this->_sku);
             $jsonDecoder = new \Magento\Framework\Serialize\Serializer\Json();
             $productMedia = $jsonDecoder->unserialize($productMediaStr);
+
             return $productMedia['data'];
         }
-    }
+
+
+        private function checkEnabled() {
+            if (!$this->_configuration->isEnabled()) {
+                throw new LocalizedException(
+                    __("Cloudinary module is disabled. Please enable it first in order to use this API.")
+                );
+            }
+            return $this;
+        }
+    }

etc/module.xml

@@ -1,6 +1,6 @@
 <?xml version="1.0"?>
 <config xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:noNamespaceSchemaLocation="urn:magento:framework:Module/etc/module.xsd">
-  <module name="Cloudinary_Cloudinary" setup_version="1.20.4">
+  <module name="Cloudinary_Cloudinary" setup_version="1.20.5">
       <sequence>
           <module name="Magento_ProductVideo"/>
           <module name="Magento_PageBuilder"/>

etc/schema.graphqls

@@ -1,9 +1,17 @@
+type CustomMediaAttribute {
+    attribute_code: String
+    url: String
+}
+
 type CloudinaryData {
     image: String
     small_image: String
     thumbnail: String
     media_gallery: [String]
     gallery_widget_parameters: String @doc(description: "Cloudinary gallery widget parameters (json)")
+    custom_media_attributes(attribute_codes: [String]!): [CustomMediaAttribute]
+    @resolver(class: "\\Cloudinary\\Cloudinary\\Model\\GraphQLResolver\\ProductAttributeCldResolver")
+    @doc(description: "Fetch a custom media attributes for the product based on the provided attribute code.")
 }
 
 interface ProductInterface {

marketplace.composer.json

@@ -5,7 +5,7 @@
   "version": "1.18.0",
   "license": "MIT",
   "require": {
-    "cloudinary/cloudinary_php": "^1.20.4"
+    "cloudinary/cloudinary_php": "^1.20.5"
   },
   "autoload": {
     "files": [