InitializeStoresAndWebsites.php

Go to the documentation of this file.

<?php
namespace Magento\Store\Setup\Patch\Schema;

use Magento\Catalog\Helper\DefaultCategory;
use Magento\Framework\Setup\SchemaSetupInterface;
use Magento\Framework\Setup\Patch\PatchVersionInterface;
use Magento\Framework\Setup\Patch\SchemaPatchInterface;
use Magento\Store\Api\Data\WebsiteInterface;

class InitializeStoresAndWebsites implements SchemaPatchInterface, PatchVersionInterface
{
    private $schemaSetup;

    private $defaultCategory;

    private $defaultCategoryFactory;

    public function __construct(
        SchemaSetupInterface $schemaSetup,
        \Magento\Catalog\Helper\DefaultCategoryFactory $defaultCategoryFactory
    ) {
        $this->schemaSetup = $schemaSetup;
        $this->defaultCategoryFactory = $defaultCategoryFactory;
    }

    public function apply()
    {
        $this->schemaSetup->startSetup();
        $connection = $this->schemaSetup->getConnection();
        $select = $connection->select()
            ->from($this->schemaSetup->getTable('store_website'))
            ->where('website_id = ?', 0);

        if ($connection->fetchOne($select) === false) {
            $connection->insertForce(
                $this->schemaSetup->getTable('store_website'),
                [
                    'website_id' => 0,
                    'code' => WebsiteInterface::ADMIN_CODE,
                    'name' => 'Admin',
                    'sort_order' => 0,
                    'default_group_id' => 0,
                    'is_default' => 0
                ]
            );
            $connection->insertForce(
                $this->schemaSetup->getTable('store_website'),
                [
                    'website_id' => 1,
                    'code' => 'base',
                    'name' => 'Main Website',
                    'sort_order' => 0,
                    'default_group_id' => 1,
                    'is_default' => 1
                ]
            );

            $connection->insertForce(
                $this->schemaSetup->getTable('store_group'),
                [
                    'group_id' => 0,
                    'website_id' => 0,
                    'name' => 'Default',
                    'root_category_id' => 0,
                    'default_store_id' => 0
                ]
            );
            $connection->insertForce(
                $this->schemaSetup->getTable('store_group'),
                [
                    'group_id' => 1,
                    'website_id' => 1,
                    'name' => 'Main Website Store',
                    'root_category_id' => $this->getDefaultCategory()->getId(),
                    'default_store_id' => 1
                ]
            );

            $connection->insertForce(
                $this->schemaSetup->getTable('store'),
                [
                    'store_id' => 0,
                    'code' => 'admin',
                    'website_id' => 0,
                    'group_id' => 0,
                    'name' => 'Admin',
                    'sort_order' => 0,
                    'is_active' => 1
                ]
            );
            $connection->insertForce(
                $this->schemaSetup->getTable('store'),
                [
                    'store_id' => 1,
                    'code' => 'default',
                    'website_id' => 1,
                    'group_id' => 1,
                    'name' => 'Default Store View',
                    'sort_order' => 0,
                    'is_active' => 1
                ]
            );
            $this->schemaSetup->endSetup();
        }
    }

    private function getDefaultCategory()
    {
        if ($this->defaultCategory === null) {
            $this->defaultCategory = $this->defaultCategoryFactory->create();
        }
        return $this->defaultCategory;
    }

    public static function getDependencies()
    {
        return [];
    }

    public static function getVersion()
    {
        return '2.0.0';
    }

    public function getAliases()
    {
        return [];
    }
}