DataInstaller.php 2.3 KB
<?php
/**
 * Copyright © Magento, Inc. All rights reserved.
 * See COPYING.txt for license details.
 */

namespace Magento\Directory\Setup;

use Magento\Framework\App\ResourceConnection;
use Magento\Framework\DB\Adapter\AdapterInterface;

/**
 * Class DataInstaller
 * @package Magento\Directory\Setup
 */
class DataInstaller
{
    /**
     * @var \Magento\Directory\Helper\Data
     */
    private $data;

    /**
     * @var ResourceConnection
     */
    private $resourceConnection;

    /**
     * DatInstaller constructor.
     * @param \Magento\Directory\Helper\Data $data
     * @param ResourceConnection $resourceConnection
     */
    public function __construct(
        \Magento\Directory\Helper\Data $data,
        ResourceConnection $resourceConnection
    ) {
        $this->data = $data;
        $this->resourceConnection = $resourceConnection;
    }

    /**
     * Add country-region data.
     *
     * @param AdapterInterface $adapter
     * @param array $data
     */
    public function addCountryRegions(AdapterInterface $adapter, array $data)
    {
        /**
         * Fill table directory/country_region
         * Fill table directory/country_region_name for en_US locale
         */
        foreach ($data as $row) {
            $bind = ['country_id' => $row[0], 'code' => $row[1], 'default_name' => $row[2]];
            $adapter->insert($this->resourceConnection->getTableName('directory_country_region'), $bind);
            $regionId = $adapter->lastInsertId($this->resourceConnection->getTableName('directory_country_region'));
            $bind = ['locale' => 'en_US', 'region_id' => $regionId, 'name' => $row[2]];
            $adapter->insert($this->resourceConnection->getTableName('directory_country_region_name'), $bind);
        }
        /**
         * Upgrade core_config_data general/region/state_required field.
         */
        $countries = $this->data->getCountryCollection()->getCountriesWithRequiredStates();
        $adapter->update(
            $this->resourceConnection->getTableName('core_config_data'),
            [
                'value' => implode(',', array_keys($countries))
            ],
            [
                'scope="default"',
                'scope_id=0',
                'path=?' => \Magento\Directory\Helper\Data::XML_PATH_STATES_REQUIRED
            ]
        );
    }
}