@effect-ts/system/_src/Experimental/Stream/Sink/collectAllWhile.ts

Effect-TS is a zero dependency set of libraries to write highly productive, purely functional TypeScript at scale.

// ets_tracing: off

import * as CK from "../../../Collections/Immutable/Chunk/index.js"
import * as L from "../../../Collections/Immutable/List/index.js"
import * as Tp from "../../../Collections/Immutable/Tuple/index.js"
import type { Predicate } from "../../../Function/index.js"
import { pipe } from "../../../Function/index.js"
import type * as C from "./core.js"
import * as Fold from "./fold.js"
import * as Map from "./map.js"

/**
 * Accumulates incoming elements into a chunk as long as they verify predicate `p`.
 */
export function collectAllWhile<Err, In>(
  p: Predicate<In>
): C.Sink<unknown, Err, In, Err, In, CK.Chunk<In>> {
  return pipe(
    Fold.fold<Err, In, Tp.Tuple<[L.List<In>, boolean]>>(
      Tp.tuple(L.empty(), true),
      Tp.get(1),
      ({ tuple: [as, _] }, a) => {
        if (p(a)) {
          return Tp.tuple(L.prepend_(as, a), true)
        } else {
          return Tp.tuple(as, false)
        }
      }
    ),
    Map.map(({ tuple: [is, _] }) => CK.from(L.reverse(is)))
  )
}