addressrepository.php

<?php
 
namespace Bitrix\Location\Repository;
 
use Bitrix\Location\Entity;
use Bitrix\Location\Model;
use Bitrix\Main\ArgumentNullException;
use Bitrix\Main\Result;
 
class AddressRepository
{
    protected $tableClass = Model\AddressTable::class;
    protected $fieldTableClass = Model\AddressFieldTable::class;
    protected $linkTableClass = Model\AddressLinkTable::class;
 
    public function findById(int $id)
    {
        $res = $this->tableClass::getList([
            'select' => ['*', 'FIELDS', 'LOCATION', 'LOCATION.NAME', 'LINKS'],
            'filter' => ['=ID' => $id]
        ])->fetchObject();
 
        if(!$res)
        {
            $result = null;
        }
        else
        {
            $result = Entity\Address\Converter\OrmConverter::convertFromOrm($res);
        }
 
        return $result;
    }
 
    public function findByLinkedEntity(string $entityId, string $entityType): Entity\Address\AddressCollection
    {
        $res = $this->tableClass::getList([
            'select' => ['*', 'FIELDS', 'LOCATION', 'LOCATION.NAME', 'LOCATION.FIELDS', 'LINKS'],
            'filter' => [
                '=LINKS.ENTITY_ID' => $entityId,
                '=LINKS.ENTITY_TYPE' => $entityType
            ]
        ])->fetchCollection();
 
        return Entity\Address\Converter\OrmConverter::convertCollectionFromOrm($res);
    }
 
    public function save(Entity\Address $address)
    {
        $result = new Result();
 
        if($location = $address->getLocation())
        {
            // Even if location has id > 0. It could contain unsaved name on other language.
            $res = $location->save();
 
            if(!$res->isSuccess())
            {
                $result->addErrors($res->getErrors());
            }
        }
 
        $fields = Entity\Address\Converter\DbFieldConverter::convertToDbField($address);
 
        if((int)$fields['ID'] > 0)
        {
            $result = $this->tableClass::update($fields['ID'], $fields);
        }
        else
        {
            $result = $this->tableClass::add($fields);
 
            if($result->isSuccess())
            {
                $address->setId($result->getId());
            }
        }
 
        if($address->getId() <= 0 || !$result->isSuccess())
        {
            return  $result;
        }
 
        $res = $this->saveFieldCollection($address);
 
        if(!$res->isSuccess())
        {
            $result->addErrors($res->getErrors());
        }
 
        $res = $this->saveLinkCollection($address);
 
        if(!$res->isSuccess())
        {
            $result->addErrors($res->getErrors());
        }
 
        return $result;
    }
 
    protected function saveFieldCollection(Entity\Address $address): \Bitrix\Main\ORM\Data\Result
    {
        if($address->getId() <= 0)
        {
            throw new ArgumentNullException('Address Id');
        }
 
        $fieldsCollection = Entity\Address\Converter\OrmConverter::convertFieldsToOrm($address);
 
        $this->fieldTableClass::deleteByAddressId($address->getId());
        return $fieldsCollection->save();
    }
 
    protected function saveLinkCollection(Entity\Address $address): \Bitrix\Main\ORM\Data\Result
    {
        if($address->getId() <= 0)
        {
            throw new ArgumentNullException('Address Id');
        }
 
        $this->linkTableClass::deleteByAddressId($address->getId());
        $collectionLinks = Entity\Address\Converter\OrmConverter::convertLinksToOrm($address);
        return $collectionLinks->save();
    }
 
    public function delete(int $addressId)
    {
        if($addressId <= 0)
        {
            throw new ArgumentNullException('Address Id');
        }
 
        $result = $this->tableClass::delete($addressId);
        $this->fieldTableClass::deleteByAddressId($addressId);
        $this->linkTableClass::deleteByAddressId($addressId);
        return $result;
    }
}