miquido/data-structure

<?php

declare(strict_types=1);

namespace Miquido\DataStructure\TypedCollection;

use Miquido\DataStructure\Map\Map;
use Miquido\DataStructure\Map\MapInterface;
use Webmozart\Assert\Assert;

final class ObjectCollection implements ObjectCollectionInterface
{
    private $objects;

    public function __construct(object ...$objects)
    {
        $this->objects = $objects;
    }

    public static function fromIterator(iterable $iterable): ObjectCollectionInterface
    {
        $objects = [];
        foreach ($iterable as $object) {
            Assert::object($object, \sprintf('Accepts only objects, got %s', \gettype($object)));
            $objects[] = $object;
        }

        return new ObjectCollection(...$objects);
    }

    public function toMap(callable $keyProvider = null): MapInterface
    {
        if (!\is_callable($keyProvider)) {
            return new Map($this->objects);
        }

        return \array_reduce(
            $this->objects,
            function (MapInterface $map, object $object) use ($keyProvider): MapInterface {
                $key = $keyProvider($object);
                if (!\is_string($key)) {
                    throw new \RuntimeException(\sprintf('Key provider should return a string, got %s', \gettype($key)));
                }
                if ($map->has($key)) {
                    throw new \LogicException(\sprintf('Key "%s" already in use', $key));
                }

                return $map->set($key, $object);
            },
            new Map()
        );
    }

    public function getAll(): array
    {
        return $this->objects;
    }

    public function count(): int
    {
        return \count($this->objects);
    }
}