dispute.php

<?php
 
namespace Bitrix\Location\Infrastructure\Service\DisputedAreaService;
 
use Bitrix\Location\Entity\Location;
use Bitrix\Location\Infrastructure\Service\CurrentRegionFinderService;
use Bitrix\Location\Entity\Location\Type;
use Bitrix\Location\Entity;
use Bitrix\Location\Entity\Address;
 
abstract class Dispute
{
    protected $currentRegion;
 
    public function __construct()
    {
        $this->currentRegion = CurrentRegionFinderService::getInstance()->getRegion();
    }
 
    public function adjustLocation(Location $location): void
    {
        $info = $this->getInfo();
        if (!$info)
        {
            return;
        }
 
        $locationFieldCollection = $location->getFieldCollection();
 
        if (isset($info['country']))
        {
            $this->adjustLocationField($locationFieldCollection, Type::COUNTRY, $info['country']);
        }
        if (isset($info['state']))
        {
            $this->adjustLocationField($locationFieldCollection, Type::ADM_LEVEL_1, $info['state']);
        }
 
        $address = $location->getAddress();
        if ($address)
        {
            $addressFieldCollection = $address->getFieldCollection();
 
            if (isset($info['country']))
            {
                $this->adjustAddressField(
                    $addressFieldCollection,
                    Entity\Address\FieldType::COUNTRY,
                    $info['country']
                );
            }
 
            if (isset($info['state']))
            {
                $this->adjustAddressField(
                    $addressFieldCollection,
                    Entity\Address\FieldType::ADM_LEVEL_1,
                    $info['state']
                );
            }
 
            $item = $addressFieldCollection->getItemByType(Entity\Address\FieldType::POSTAL_CODE);
            if ($item)
            {
                $item->setValue('');
            }
        }
    }
 
    public function adjustAutocompleteItem(array &$item): void
    {
        $info = $this->getInfo();
        if (!$info)
        {
            return;
        }
 
        unset($item['countrycode']);
        unset($item['postcode']);
 
        if (isset($info['country']))
        {
            $item['country'] = $info['country'];
        }
 
        if (isset($info['state']))
        {
            $item['state'] = $info['state'];
        }
    }
 
    abstract protected function getInfo(): array;
 
    private function adjustLocationField(Entity\Location\FieldCollection $fieldCollection, int $type, $value)
    {
        $item = $fieldCollection->getItemByType($type);
        if ($item)
        {
            $item->setValue($value);
        }
        else
        {
            $fieldCollection->addItem(new Entity\Location\Field($type, $value));
        }
    }
 
    private function adjustAddressField(Entity\Address\FieldCollection $fieldCollection, int $type, $value)
    {
        $item = $fieldCollection->getItemByType($type);
        if ($item)
        {
            $item->setValue($value);
        }
        else
        {
            $fieldCollection->addItem(new Entity\Address\Field($type, $value));
        }
    }
}