loophp\collection\Operation\Times

Complexity

Total Complexity

5

Size/Duplication

Total Lines	45
Duplicated Lines	0 %

Importance

Changes	1
Bugs	0	Features	0

2 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	4	1
A	on()	0	18	4

<?php

declare(strict_types=1);

namespace loophp\collection\Operation;

use Closure;
use Generator;
use loophp\collection\Contract\Operation;

/**
 * Class Times.
 */
final class Times implements Operation
{
    /**
     * @var callable|null
     */
    private $callback;

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

    /**
     * Times constructor.
     *
     * @param int $number
     * @param callable|null $callback
     */
    public function __construct(int $number, ?callable $callback = null)
    {
        $this->number = $number;
        $this->callback = $callback;
    }

    /**
     * {@inheritdoc}
     */
    public function on(iterable $collection): Closure
    {
        $number = $this->number;
        $callback = $this->callback;

        return static function () use ($number, $callback): Generator {
            if (1 > $number) {
                return;
            }

            if (null === $callback) {
                $callback = static function ($value) {
                    return $value;
                };
            }

            for ($current = 1; $current <= $number; ++$current) {
                yield $callback($current);
            }
        };
    }
}