import kotlin.random.Random

fun <T> Iterable<T>.splitAt(n: Int): Pair<List<T>, List<T>> = if (this is List<T>) {
    val end = n.coerceAtMost(size)
    subList(0, end) to subList(end, size)
} else {
    take(n) to drop(n)
}

fun <T> Iterable<T>.chunks(sizes: Sequence<Int>): Sequence<List<T>> = sizes
    .runningFold(emptyList<T>() to this) { (_, list), size -> list.splitAt(size) }
    .drop(1)
    .map { it.first }
    .takeWhile { it.isNotEmpty() }

fun main() {
    (0..10).chunks(generateSequence { Random.nextInt(1, 5) }).forEach(::println)
}