ProductTemplateGeneratorFactory.php

Go to the documentation of this file.

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

use Magento\Bundle\Model\Product\Type as BundleType;
use Magento\Catalog\Model\Product\Type;
use Magento\ConfigurableProduct\Model\Product\Type\Configurable;
use Magento\Framework\ObjectManagerInterface;

class ProductTemplateGeneratorFactory
{
    private $objectManager;

    private $templateEntityMap = [
        Type::TYPE_SIMPLE => SimpleProductTemplateGenerator::class,
        BundleType::TYPE_CODE => BundleProductTemplateGenerator::class,
        Configurable::TYPE_CODE => ConfigurableProductTemplateGenerator::class,
    ];

    public function __construct(ObjectManagerInterface $objectManager)
    {
        $this->objectManager = $objectManager;
    }

    public function create(array $fixture)
    {
        $type = isset($fixture['type_id']) ? $fixture['type_id'] : Type::TYPE_SIMPLE;
        if (!isset($this->templateEntityMap[$type])) {
            throw new \InvalidArgumentException(sprintf(
                'Cannot instantiate product template generator. Wrong type_id "%s" passed',
                $type
            ));
        }

        return $this->objectManager->create($this->templateEntityMap[$type], ['fixture' => $fixture]);
    }
}