DEVX-813: adding ProductVariantUtil

Author: ShipilA

commercetools/commercetools-importapi-utils/src/main/java/com/commercetools/sdk/ProductUtil.java

@@ -129,23 +129,26 @@ private static ProductVariantDraftImport extractProductVariantDraftImport(Produc
         return ProductVariantDraftImport.builder()
                 .key(variant.getKey())
                 .sku(variant.getSku())
-                .images(variant.getImages()
-                        .stream()
-                        .map(i -> com.commercetools.importapi.models.common.Image.builder()
-                                .dimensions(d -> com.commercetools.importapi.models.common.AssetDimensions.builder()
-                                        .w(i.getDimensions().getW())
-                                        .h(i.getDimensions().getH()))
-                                .url(i.getUrl())
-                                .label(i.getLabel())
-                                .build())
-                        .collect(Collectors.toList()))
+                .images(toImportImages(variant.getImages()))
                 .prices(mapPricesToImportApi(variant))
                 .attributes(
                     variant.getAttributes().stream().map(ProductUtil::mapAttribute).collect(Collectors.toList()))
                 .assets(importAssets(variant.getAssets()))
                 .build();
     }
 
+    public static @NotNull List<Image> toImportImages(List<com.commercetools.api.models.common.Image> images) {
+        return images
+                .stream()
+                .map(i -> Image.builder()
+                        .dimensions(
+                                d -> AssetDimensions.builder().w(i.getDimensions().getW()).h(i.getDimensions().getH()))
+                        .url(i.getUrl())
+                        .label(i.getLabel())
+                        .build())
+                .collect(Collectors.toList());
+    }
+
     private static List<PriceDraftImport> mapPricesToImportApi(ProductVariant variant) {
         return variant.getPrices()
                 .stream()
@@ -163,7 +166,7 @@ private List<CategoryKeyReference> extractCategoryKeyReference(ProductProjection
                 .collect(Collectors.toList());
     }
 
-    private static Attribute mapAttribute(com.commercetools.api.models.product.Attribute attribute) {
+    public static Attribute mapAttribute(com.commercetools.api.models.product.Attribute attribute) {
         Object value = attribute.getValue();
 
         if (value instanceof String) {

commercetools/commercetools-importapi-utils/src/main/java/com/commercetools/sdk/ProductVariantUtil.java

@@ -0,0 +1,26 @@
+package com.commercetools.sdk;
+
+import com.commercetools.api.models.product.ProductProjection;
+import com.commercetools.api.models.product.ProductVariant;
+import com.commercetools.importapi.models.productvariants.ProductVariantImport;
+
+import java.util.stream.Collectors;
+
+import static com.commercetools.sdk.CommonImportUtil.importAssets;
+import static com.commercetools.sdk.ProductUtil.toImportImages;
+
+public class ProductVariantUtil {
+    public ProductVariantImport toProductVariantImport(ProductProjection product, ProductVariant variant, String productKey) {
+        return ProductVariantImport.builder()
+                .key(variant.getKey()) // required field
+                .sku(variant.getSku())
+                .isMasterVariant(variant.getId().equals(product.getMasterVariant().getId())) // required field
+                .attributes(variant.getAttributes().stream()
+                        .map(ProductUtil::mapAttribute)
+                        .collect(Collectors.toList()))
+                .images(toImportImages(variant.getImages()))
+                .assets(importAssets(variant.getAssets()))
+                .product(p -> p.key(productKey)) // required field
+                .build();
+    }
+}