loophp\collection\Operation\Chunk::__invoke()

Complexity

Conditions	5
Paths	1

Size

Total Lines	37
Code Lines	17

Duplication

Lines	0
Ratio	0 %

Code Coverage

Tests	18
CRAP Score	5

Importance

Changes

0

<?php
    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;
                    }