Your IP : 216.73.217.95
<?php
/**
* Copyright © Magento, Inc. All rights reserved.
* See COPYING.txt for license details.
*/
declare(strict_types=1);
namespace Magento\ConfigurableProduct\Model\Product;
use Magento\Catalog\Model\Product\Type as ProductType;
use Magento\Framework\Exception\LocalizedException;
/**
* Variation Handler
* @SuppressWarnings(PHPMD.CouplingBetweenObjects)
* @api
* @since 100.0.2
*/
class VariationHandler
{
/**
* @var \Magento\Catalog\Model\Product\Gallery\Processor
* @since 100.1.0
*/
protected $mediaGalleryProcessor;
/**
* @var \Magento\ConfigurableProduct\Model\Product\Type\Configurable
*/
protected $configurableProduct;
/**
* @var \Magento\Eav\Model\Entity\Attribute\SetFactory
*/
protected $attributeSetFactory;
/**
* @var \Magento\Eav\Model\EntityFactory
*/
protected $entityFactory;
/**
* @var \Magento\Catalog\Model\ProductFactory
*/
protected $productFactory;
/**
* @var \Magento\Eav\Model\Entity\Attribute\AbstractAttribute[]
*/
private $attributes;
/**
* @var \Magento\CatalogInventory\Api\StockConfigurationInterface
* @deprecated 100.1.0
*/
protected $stockConfiguration;
/**
* @param Type\Configurable $configurableProduct
* @param \Magento\Eav\Model\Entity\Attribute\SetFactory $attributeSetFactory
* @param \Magento\Eav\Model\EntityFactory $entityFactory
* @param \Magento\Catalog\Model\ProductFactory $productFactory
* @param \Magento\CatalogInventory\Api\StockConfigurationInterface $stockConfiguration
* @param \Magento\Catalog\Model\Product\Gallery\Processor $mediaGalleryProcessor
*/
public function __construct(
Type\Configurable $configurableProduct,
\Magento\Eav\Model\Entity\Attribute\SetFactory $attributeSetFactory,
\Magento\Eav\Model\EntityFactory $entityFactory,
\Magento\Catalog\Model\ProductFactory $productFactory,
\Magento\CatalogInventory\Api\StockConfigurationInterface $stockConfiguration,
\Magento\Catalog\Model\Product\Gallery\Processor $mediaGalleryProcessor
) {
$this->configurableProduct = $configurableProduct;
$this->attributeSetFactory = $attributeSetFactory;
$this->entityFactory = $entityFactory;
$this->productFactory = $productFactory;
$this->stockConfiguration = $stockConfiguration;
$this->mediaGalleryProcessor = $mediaGalleryProcessor;
}
/**
* Generate simple products to link with configurable
*
* @param \Magento\Catalog\Model\Product $parentProduct
* @param array $productsData
* @return array
* @throws \Magento\Framework\Exception\LocalizedException
*/
public function generateSimpleProducts($parentProduct, $productsData)
{
$generatedProductIds = [];
$this->attributes = null;
$productsData = $this->duplicateImagesForVariations($productsData);
foreach ($productsData as $simpleProductData) {
$newSimpleProduct = $this->productFactory->create();
if (isset($simpleProductData['configurable_attribute'])) {
$configurableAttribute = json_decode($simpleProductData['configurable_attribute'], true);
unset($simpleProductData['configurable_attribute']);
} else {
throw new LocalizedException(
__('Contribution must have attributes specified. Enter attributes and try again.')
);
}
$this->fillSimpleProductData(
$newSimpleProduct,
$parentProduct,
// phpcs:ignore Magento2.Performance.ForeachArrayMerge
array_merge($simpleProductData, $configurableAttribute)
);
$newSimpleProduct->save();
$generatedProductIds[] = $newSimpleProduct->getId();
}
return $generatedProductIds;
}
/**
* Prepare attribute set comprising all selected configurable attributes
*
* @deprecated 100.1.0
* @param \Magento\Catalog\Model\Product $product
* @return void
*/
protected function prepareAttributeSetToBeBaseForNewVariations(\Magento\Catalog\Model\Product $product)
{
$this->prepareAttributeSet($product);
}
/**
* Prepare attribute set comprising all selected configurable attributes
*
* @param \Magento\Catalog\Model\Product $product
* @return void
* @since 100.1.0
*/
public function prepareAttributeSet(\Magento\Catalog\Model\Product $product)
{
$attributes = $this->configurableProduct->getUsedProductAttributes($product);
$attributeSetId = $product->getNewVariationsAttributeSetId();
/** @var $attributeSet \Magento\Eav\Model\Entity\Attribute\Set */
$attributeSet = $this->attributeSetFactory->create()->load($attributeSetId);
$attributeSet->addSetInfo(
$this->entityFactory->create()->setType(\Magento\Catalog\Model\Product::ENTITY)->getTypeId(),
$attributes
);
foreach ($attributes as $attribute) {
/* @var $attribute \Magento\Catalog\Model\Entity\Attribute */
if (!$attribute->isInSet($attributeSetId)) {
$attribute->setAttributeSetId(
$attributeSetId
)->setAttributeGroupId(
$attributeSet->getDefaultGroupId($attributeSetId)
)->save();
}
}
}
/**
* Fill simple product data during generation
*
* @param \Magento\Catalog\Model\Product $product
* @param \Magento\Catalog\Model\Product $parentProduct
* @param array $postData
* @SuppressWarnings(PHPMD.CyclomaticComplexity)
* @SuppressWarnings(PHPMD.NPathComplexity)
*
* @return void
*/
protected function fillSimpleProductData(
\Magento\Catalog\Model\Product $product,
\Magento\Catalog\Model\Product $parentProduct,
$postData
) {
$typeId = isset($postData['weight']) && !empty($postData['weight'])
? ProductType::TYPE_SIMPLE
: ProductType::TYPE_VIRTUAL;
$product->setStoreId(
\Magento\Store\Model\Store::DEFAULT_STORE_ID
)->setTypeId(
$typeId
)->setAttributeSetId(
$parentProduct->getNewVariationsAttributeSetId()
);
if ($this->attributes === null) {
$this->attributes = $product->getTypeInstance()->getSetAttributes($product);
}
foreach ($this->attributes as $attribute) {
if ($attribute->getIsUnique() ||
$attribute->getAttributeCode() == 'url_key' ||
$attribute->getFrontend()->getInputType() == 'gallery' ||
$attribute->getFrontend()->getInputType() == 'media_image' ||
!$attribute->getIsVisible()
) {
continue;
}
$product->setData($attribute->getAttributeCode(), $parentProduct->getData($attribute->getAttributeCode()));
}
$keysFilter = ['item_id', 'product_id', 'stock_id', 'type_id', 'website_id'];
$postData['stock_data'] = array_diff_key((array)$parentProduct->getStockData(), array_flip($keysFilter));
$stockStatus = $parentProduct->getQuantityAndStockStatus();
if (isset($stockStatus['is_in_stock'])) {
$postData['stock_data']['is_in_stock'] = $stockStatus['is_in_stock'];
}
$postData = $this->processMediaGallery($product, $postData);
$postData['status'] = isset($postData['status'])
? $postData['status']
: \Magento\Catalog\Model\Product\Attribute\Source\Status::STATUS_ENABLED;
$product->addData(
$postData
)->setWebsiteIds(
$parentProduct->getWebsiteIds()
)->setVisibility(
\Magento\Catalog\Model\Product\Visibility::VISIBILITY_NOT_VISIBLE
);
}
/**
* Duplicate images for variations
*
* @param array $productsData
*
* @return array
* @SuppressWarnings(PHPMD.CyclomaticComplexity)
*/
public function duplicateImagesForVariations($productsData)
{
$imagesForCopy = [];
foreach ($productsData as $variationId => $simpleProductData) {
if (!isset($simpleProductData['media_gallery']['images'])) {
continue;
}
foreach ($simpleProductData['media_gallery']['images'] as $imageId => $image) {
$image['variation_id'] = $variationId;
$imagesForCopy[$imageId][] = $image;
}
}
foreach ($imagesForCopy as $imageId => $variationImages) {
foreach ($variationImages as $image) {
$file = $image['file'];
$variationId = $image['variation_id'];
$newFile = $this->mediaGalleryProcessor->duplicateImageFromTmp($file);
$productsData[$variationId]['media_gallery']['images'][$imageId]['file'] = $newFile;
foreach ($this->mediaGalleryProcessor->getMediaAttributeCodes() as $attribute) {
if (isset($productsData[$variationId][$attribute])
&& $productsData[$variationId][$attribute] == $file
) {
$productsData[$variationId][$attribute] = $newFile;
}
}
}
}
return $productsData;
}
/**
* Process media gallery for product
*
* @param \Magento\Catalog\Model\Product $product
* @param array $productData
*
* @return array
*/
public function processMediaGallery($product, $productData)
{
if (!empty($productData['image'])) {
$image = $productData['image'];
if (!isset($productData['media_gallery']['images'])) {
$productData['media_gallery']['images'] = [];
}
if (false === array_search($image, array_column($productData['media_gallery']['images'], 'file'))) {
$productData['small_image'] = $productData['thumbnail'] = $image;
$productData['media_gallery']['images'][] = [
'position' => 1,
'file' => $image,
'disabled' => 0,
'label' => '',
];
}
}
if ($product->getMediaGallery('images') && !empty($productData['media_gallery']['images'])) {
$gallery = array_map(
function ($image) {
$image['removed'] = 1;
return $image;
},
$product->getMediaGallery('images')
);
$gallery = array_merge($productData['media_gallery']['images'], $gallery);
$productData['media_gallery']['images'] = $gallery;
}
return $productData;
}
}