PriceCollection.php

<?php
 
namespace Bitrix\Catalog\v2\Price;
 
use Bitrix\Catalog\v2\BaseCollection;
 
class PriceCollection extends BaseCollection
{
    protected $factory;
 
    public function __construct(PriceFactory $factory)
    {
        $this->factory = $factory;
    }
 
    public function findBasePrice(): ?BasePrice
    {
        foreach ($this->getIterator() as $price)
        {
            if ($price->isPriceBase())
            {
                return $price;
            }
        }
 
        return null;
    }
 
    public function findByGroupId(int $groupId): ?BasePrice
    {
        foreach ($this->getIterator() as $price)
        {
            if ($price->getGroupId() === $groupId)
            {
                return $price;
            }
        }
 
        return null;
    }
 
    public function create(): BasePrice
    {
        $price = $this->factory->createEntity();
 
        $this->add($price);
 
        return $price;
    }
 
    public function setValues(array $values): self
    {
        $preparedValues = $this->prepareValues($values);
 
        foreach ($preparedValues as $id => $fields)
        {
            $price = $this->findByGroupId($id);
 
            if ($price === null)
            {
                // ToDo make all collections with factory methods?
                $price = $this->create()->setGroupId($id);
            }
 
            if ($price)
            {
                $price->setPrice($fields['PRICE'] ?? null);
 
                if (isset($fields['CURRENCY']))
                {
                    $price->setCurrency($fields['CURRENCY']);
                }
            }
        }
 
        return $this;
    }
 
    public function getValues(): array
    {
        $values = [];
 
        foreach ($this->getIterator() as $item)
        {
            if ($item->hasField('PRICE') && $item->hasField('CURRENCY'))
            {
                $values[$item->getGroupId()] = [
                    'PRICE' => $item->getPrice(),
                    'CURRENCY' => $item->getCurrency(),
                ];
            }
        }
 
        return $values;
    }
 
    public function hasBasePrice(): bool
    {
        $basePrice = $this->findBasePrice();
 
        return $basePrice && $basePrice->hasPrice();
    }
 
    public function hasAnyPrice(): bool
    {
        return !empty($this->getValues());
    }
 
    private function prepareValues(array $values): array
    {
        // ToDo check required properties + QUANTITY_FROM etc
        $prepared = [];
 
        foreach ($values as $id => $fields)
        {
            if (!is_array($fields))
            {
                if (is_numeric($fields) && is_finite($fields))
                {
                    $fields = ['PRICE' => $fields];
                }
                else
                {
                    continue;
                }
            }
 
            $fields = array_intersect_key($fields, ['PRICE' => true, 'CURRENCY' => true]);
            if (!empty($fields))
            {
                if (isset($fields['PRICE']))
                {
                    if (is_numeric($fields['PRICE']) && is_finite($fields['PRICE']))
                    {
                        $fields['PRICE'] = (float)$fields['PRICE'];
                    }
                    else
                    {
                        $fields['PRICE'] = null;
                    }
                }
 
                if (isset($fields['CURRENCY']) && !is_string($fields['CURRENCY']))
                {
                    unset($fields['CURRENCY']);
                }
            }
 
            if (!empty($fields))
            {
                if (is_numeric($id))
                {
                    $id = (int)$id;
                    $prepared[$id] = $fields;
                }
                elseif ($id === 'BASE')
                {
                    $basePrice = \CCatalogGroup::GetBaseGroup();
 
                    if (!empty($basePrice['ID']))
                    {
                        $id = (int)$basePrice['ID'];
                        $prepared[$id] = $fields;
                    }
                }
            }
        }
 
        return $prepared;
    }
}