Phraseanet/lib/Alchemy/Phrasea/Databox/Caption/CaptionService.php

<?php
/*
 * This file is part of Phraseanet
 *
 * (c) 2005-2016 Alchemy
 *
 * For the full copyright and license information, please view the LICENSE
 * file that was distributed with this source code.
 */

namespace Alchemy\Phrasea\Databox\Caption;

use Alchemy\Phrasea\Databox\DataboxBoundRepositoryProvider;
use Alchemy\Phrasea\Model\RecordReferenceInterface;
use Alchemy\Phrasea\Record\RecordReferenceCollection;

class CaptionService
{
    /**
     * @var DataboxBoundRepositoryProvider
     */
    private $repositoryProvider;

    public function __construct(DataboxBoundRepositoryProvider $repositoryProvider)
    {
        $this->repositoryProvider = $repositoryProvider;
    }

    public function findByReferenceCollection($references)
    {
        $references = $this->normalizeReferenceCollection($references);

        $groups = [];

        foreach ($references->getDataboxIds() as $databoxId) {
            $recordIds = $references->getDataboxRecordIds($databoxId);

            $groups[$databoxId] = $this->getRepositoryForDatabox($databoxId)->findByRecordIds($recordIds);
        }

        return $this->reorderInstances($references, $groups);
    }

    /**
     * @param RecordReferenceInterface[]|RecordReferenceCollection $references
     * @return RecordReferenceCollection
     */
    private function normalizeReferenceCollection($references)
    {
        if ($references instanceof RecordReferenceCollection) {
            return $references;
        }

        return new RecordReferenceCollection($references);
    }

    /**
     * @param int $databoxId
     * @return CaptionRepository
     */
    private function getRepositoryForDatabox($databoxId)
    {
        return $this->repositoryProvider->getRepositoryForDatabox($databoxId);
    }

    /**
     * @param RecordReferenceCollection $references
     * @param \caption_record[][] $groups
     * @return \caption_record[]
     */
    private function reorderInstances(RecordReferenceCollection $references, array $groups)
    {
        $captions = [];

        foreach ($groups as $databoxId => $group) {
            $captions[$databoxId] = array_reduce($group, function (array &$carry, \caption_record $caption) {
                $carry[$caption->getRecordReference()->getRecordId()] = $caption;

                return $carry;
            }, []);
        }

        $instances = [];

        foreach ($references as $index => $reference) {
            $databoxId = $reference->getDataboxId();
            $recordId = $reference->getRecordId();

            if (isset($captions[$databoxId][$recordId])) {
                $instances[$index] = $captions[$databoxId][$recordId];
            }
        }

        return $instances;
    }
}