<?php
/**
*
* ..::..
* ..::::::::::::..
* ::'''''':''::'''''::
* ::.. ..: : ....::
* :::: ::: : : ::
* :::: ::: : ''' ::
* ::::..:::..::.....::
* ''::::::::::::''
* ''::''
*
*
* NOTICE OF LICENSE
*
* This source file is subject to the Creative Commons License.
* It is available through the world-wide-web at this URL:
* http://creativecommons.org/licenses/by-nc-nd/3.0/nl/deed.en_US
* If you are unable to obtain it through the world-wide-web, please send an email
* to servicedesk@tig.nl so we can send you a copy immediately.
*
* DISCLAIMER
*
* Do not edit or add to this file if you wish to upgrade this module to newer
* versions in the future. If you wish to customize this module for your
* needs please contact servicedesk@tig.nl for more information.
*
* @copyright Copyright (c) Total Internet Group B.V. https://tig.nl/copyright
* @license http://creativecommons.org/licenses/by-nc-nd/3.0/nl/deed.en_US
*/

namespace TIG\PostNL\Service\Validation;

use Magento\Directory\Api\Data\CountryInformationInterface;
use Magento\Directory\Helper\Data as DirectoryHelper;
use Magento\Directory\Model\AllowedCountries;
use Magento\Directory\Model\Data\CountryInformation;
use Magento\Directory\Model\Data\CountryInformationFactory;
use Magento\Directory\Model\Data\RegionInformation;
use Magento\Directory\Model\Data\RegionInformationFactory;
use Magento\Directory\Model\ResourceModel\Country as CountryResource;
use Magento\Directory\Model\ResourceModel\Country\Collection as CountryCollection;
use Magento\Framework\App\Config\ScopeConfigInterface;
use Magento\Store\Model\ScopeInterface;
use Magento\Store\Model\StoreManagerInterface;

class Country implements ContractInterface
{
/** @var array */
private $countryList = [];

/** @var null|int */
private $websiteId = null;

/**@var ScopeConfigInterface */
private $scopeConfig;

/** @var StoreManagerInterface */
private $storeManager;

/** @var AllowedCountries */
private $allowedCountries;

/** @var CountryCollection */
private $countryCollection;

/** @var DirectoryHelper */
private $directoryHelper;

/** @var CountryInformationFactory */
private $countryInformationFactory;

/** @var RegionInformationFactory */
private $regionInformationFactory;

/**
* @param ScopeConfigInterface $scopeConfig
* @param StoreManagerInterface $storeManager
* @param AllowedCountries $allowedCountries
* @param CountryCollection $countryCollection
* @param DirectoryHelper $directoryHelper
* @param CountryInformationFactory $countryInformationFactory
* @param RegionInformationFactory $regionInformationFactory
*/
public function __construct(
ScopeConfigInterface $scopeConfig,
StoreManagerInterface $storeManager,
AllowedCountries $allowedCountries,
CountryCollection $countryCollection,
DirectoryHelper $directoryHelper,
CountryInformationFactory $countryInformationFactory,
RegionInformationFactory $regionInformationFactory
) {
$this->scopeConfig = $scopeConfig;
$this->storeManager = $storeManager;
$this->allowedCountries = $allowedCountries;
$this->countryCollection = $countryCollection;
$this->directoryHelper = $directoryHelper;
$this->countryInformationFactory = $countryInformationFactory;
$this->regionInformationFactory = $regionInformationFactory;
}

/**
* Validate the data. Returns false when the
*
* @param $line
*
* @return bool|mixed
*/
public function validate($line)
{
if (strpos($line, ',') !== false) {
return $this->validateArray($line);
}

return $this->validateSingle($line);
}

/**
* @param $line
*
* @return bool|string
*/
private function validateSingle($line)
{
if ($line == '*') {
return 0;
}

$countries = $this->getCountryList();

if (!array_key_exists($line, $countries)) {
return false;
}

return $countries[$line]['id'];
}

/**
* @param $line
*
* @return bool|string
*/
private function validateArray($line)
{
$parts = explode(',', $line);
$pieces = array_map(function ($part) {
return $this->validateSingle($part);
}, $parts);

if (in_array(false, $pieces, true)) {
return false;
}

return implode(',', $pieces);
}

/**
* @param int $websiteId
*/
public function setWebsiteId($websiteId)
{
$this->websiteId = $websiteId;
}

/**
* @return array
*/
public function getCountryList()
{
if (!empty($this->countryList)) {
return $this->countryList;
}

$countriesInfo = $this->getCountriesInfo();

foreach ($countriesInfo as $country) {
$regions = $this->getRegions($country);

$data = [
'id' => $country->getId(),
'regions' => $regions,
];

$this->countryList[$country->getTwoLetterAbbreviation()] = $data;
$this->countryList[$country->getThreeLetterAbbreviation()] = $data;
}

return $this->countryList;
}

/**
* @return array
*/
private function getCountriesInfo()
{
$countriesInfo = [];
$website = $this->storeManager->getWebsite($this->websiteId);
$storeLocale = $this->scopeConfig->getValue('general/locale/code', ScopeInterface::SCOPE_WEBSITE, $website);
$allowedCountries = $this->allowedCountries->getAllowedCountries(ScopeInterface::SCOPE_WEBSITE, $website);
$countryCollection = $this->countryCollection->addFieldToFilter("country_id", ['in' => $allowedCountries]);
$regions = $this->directoryHelper->getRegionData();

foreach ($countryCollection as $data) {
$countryInfo = $this->setCountryInfo($data, $regions, $storeLocale);
$countriesInfo[] = $countryInfo;
}

return $countriesInfo;
}

/**
* @param CountryResource $country
* @param array $regions
* @param string $storeLocale
*
* @return CountryInformation
*/
private function setCountryInfo($country, $regions, $storeLocale)
{
$countryId = $country->getCountryId();

/** @var CountryInformation $countryInfo */
$countryInfo = $this->countryInformationFactory->create();
$countryInfo->setId($countryId);
$countryInfo->setTwoLetterAbbreviation($country->getData('iso2_code'));
$countryInfo->setThreeLetterAbbreviation($country->getData('iso3_code'));
$countryInfo->setFullNameLocale($country->getName($storeLocale));
$countryInfo->setFullNameEnglish($country->getName('en_US'));

if (array_key_exists($countryId, $regions)) {
$regionsInfo = [];
foreach ($regions[$countryId] as $id => $regionData) {
/** @var RegionInformation $regionInfo */
$regionInfo = $this->regionInformationFactory->create();
$regionInfo->setId($id);
$regionInfo->setCode($regionData['code']);
$regionInfo->setName($regionData['name']);
$regionsInfo[] = $regionInfo;
}
$countryInfo->setAvailableRegions($regionsInfo);
}

return $countryInfo;
}

/**
* @param CountryInformationInterface $country
*
* @return array
*/
private function getRegions(CountryInformationInterface $country)
{
$regions = $country->getAvailableRegions();

if ($regions === null) {
return [];
}

$output = [];
foreach ($regions as $region) {
$output[$region->getCode()] = $region->getId();
$output[$region->getName()] = $region->getId();
}

return $output;
}
}