ConfigurableProductGenerator.php

Go to the documentation of this file.

<?php
namespace Magento\Setup\Model\FixtureGenerator;

class ConfigurableProductGenerator
{
    private $productGeneratorFactory;

    private $autoIncrement;

    public function __construct(
        ProductGeneratorFactory $productGeneratorFactory,
        AutoIncrement $autoIncrement
    ) {
        $this->productGeneratorFactory = $productGeneratorFactory;
        $this->autoIncrement = $autoIncrement;
    }

    public function generate($products, $fixtureMap)
    {
        $this->productGeneratorFactory->create([
            'customTableMap' => [
                'catalog_product_super_attribute_label' => [
                    'entity_id_field' => EntityGenerator::SKIP_ENTITY_ID_BINDING,
                    'handler' => function ($productId, $entityNumber, $fixture, $binds) {
                        foreach ($binds as &$bind) {
                            $bind['product_super_attribute_id'] = $this->generateSuperAttributeId(
                                $bind['product_super_attribute_id'],
                                $entityNumber,
                                $fixture
                            );
                        }
                        return $binds;
                    },
                ],
                'catalog_product_super_link' => [
                    'entity_id_field' => EntityGenerator::SKIP_ENTITY_ID_BINDING,
                    'handler' => function ($productId, $entityNumber, $fixture, $binds) {
                        foreach ($binds as &$bind) {
                            $bind['parent_id'] = $productId;
                            $bind['product_id'] = $this->generateSimpleProductId(
                                $bind['product_id'],
                                $entityNumber,
                                $fixture
                            );
                        }
                        return $binds;
                    },
                ],
                'catalog_product_relation' => [
                    'entity_id_field' => EntityGenerator::SKIP_ENTITY_ID_BINDING,
                    'handler' => function ($productId, $entityNumber, $fixture, $binds) {
                        foreach ($binds as &$bind) {
                            $bind['parent_id'] = $productId;
                            $bind['child_id'] = $this->generateSimpleProductId(
                                $bind['child_id'],
                                $entityNumber,
                                $fixture
                            );
                        }
                        return $binds;
                    },
                ],
            ]
        ])->generate($products, $fixtureMap);
    }

    private function generateSuperAttributeId($superAttributeId, $entityNumber, array $fixture)
    {
        return $superAttributeId + ($entityNumber + 1) * $fixture['_attributes_count']
            * $this->autoIncrement->getIncrement();
    }

    private function generateSimpleProductId($previousProductId, $entityNumber, array $fixture)
    {
        return $previousProductId + $entityNumber * $fixture['_variation_count'];
    }
}