Zenstruck\Foundry\FactoryCollection

Complexity

Total Complexity

5

Size/Duplication

Total Lines	36
Duplicated Lines	0 %

Importance

Changes	1
Bugs	0	Features	1

3 Methods

Rating	Name	Duplication	Size	Complexity
A	createFor()	0	8	2
A	create()	0	3	1
A	__construct()	0	8	2

<?php

namespace Zenstruck\Foundry;

/**
 * @author Kevin Bond <kevinbond@gmail.com>
 */
final class FactoryCollection
{
    /** @var Factory */
    private $factory;

    /** @var int */
    private $count;

    public function __construct(Factory $factory, int $count)
    {
        if ($count < 1) {
            throw new \InvalidArgumentException('Count must be greater than 0.');
        }

        $this->factory = $factory;
        $this->count = $count;
    }

    /**
     * @param array|callable $attributes
     *
     * @return Proxy[]|object[]
     */
    public function create($attributes = []): array
    {
        return $this->factory->createMany($this->count, $attributes);
    }

    public function createFor(Proxy $proxy): void
    {
        $attribute = \lcfirst((new \ReflectionClass($proxy->object()))->getShortName());

        $this->create([$attribute => $proxy]);

        if ($proxy->isPersisted()) {
            $proxy->refresh();
        }
    }
}