loophp\collection\Operation\Chunk

Complexity

Total Complexity

5

Size/Duplication

Total Lines	42
Duplicated Lines	0 %

Test Coverage

Coverage

100%

Importance

Changes

0

1 Method

Rating	Name	Duplication	Size	Complexity
A	__invoke()	0	37	5

<?php

declare(strict_types=1);

namespace loophp\collection\Operation;

use Closure;
use Generator;

use function count;

/**
 * @immutable
 *
 * @template TKey
 * @template T
 */
final class Chunk extends AbstractOperation
{
    /**
     * @return Closure(int...): Closure(iterable<TKey, T>): Generator<int, non-empty-list<T>>
     */
    public function __invoke(): Closure
    {
        return
            /**
             * @return Closure(iterable<TKey, T>): Generator<int, non-empty-list<T>>
             */
            static fn (int ...$sizes): Closure =>
                /**
                 * @param iterable<TKey, T> $iterable
                 *
                 * @return Generator<int, non-empty-list<T>>
                 */
                static function (iterable $iterable) use ($sizes): Generator {
                    $sizesCount = count($sizes);
                    $chunkIndex = 0;
                    $chunk = [];

                    foreach ($iterable as $value) {
                        $size = $sizes[$chunkIndex % $sizesCount];

                        if (0 >= $size) {
                            return;
                        }

                        $chunk[] = $value;

                        if (count($chunk) >= $size) {
                            ++$chunkIndex;

                            yield $chunk;

                            $chunk = [];
                        }
                    }

                    if ([] !== $chunk) {
                        yield $chunk;
                    }
                };
    }
}