CollectionProvider.php

Go to the documentation of this file.

<?php
namespace Magento\Catalog\Model\ProductLink;

use Magento\Catalog\Model\ProductLink\Converter\ConverterPool;
use Magento\Framework\Exception\NoSuchEntityException;

class CollectionProvider
{
    protected $providers;

    protected $converterPool;

    public function __construct(ConverterPool $converterPool, array $providers = [])
    {
        $this->converterPool = $converterPool;
        $this->providers = $providers;
    }

    public function getCollection(\Magento\Catalog\Model\Product $product, $type)
    {
        if (!isset($this->providers[$type])) {
            throw new NoSuchEntityException(__("The collection provider isn't registered."));
        }

        $products = $this->providers[$type]->getLinkedProducts($product);
        $converter = $this->converterPool->getConverter($type);
        $output = [];
        $sorterItems = [];
        foreach ($products as $item) {
            $output[$item->getId()] = $converter->convert($item);
        }

        foreach ($output as $item) {
            $itemPosition = $item['position'];
            if (!isset($sorterItems[$itemPosition])) {
                $sorterItems[$itemPosition] = $item;
            } else {
                $newPosition = $itemPosition + 1;
                $sorterItems[$newPosition] = $item;
            }
        }
        ksort($sorterItems);
        return $sorterItems;
    }
}