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

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

// ets_tracing: off

import type * as CK from "../../../Collections/Immutable/Chunk/index.js"
import type * as T from "../../../Effect/index.js"
import type { Predicate } from "../../../Function/index.js"
import * as CH from "../Channel/index.js"
import * as C from "./core.js"

/**
 * A sink that effectfully folds its input chunks with the provided function, termination predicate and initial state.
 * `contFn` condition is checked only for the initial value and at the end of processing of each chunk.
 * `f` and `contFn` must preserve chunking-invariance.
 */
export function foldChunksEffect<Env, Err, In, S>(
  z: S,
  contFn: Predicate<S>,
  f: (s: S, chunk: CK.Chunk<In>) => T.Effect<Env, Err, S>
): C.Sink<Env, Err, In, Err, unknown, S> {
  const reader = (s: S): CH.Channel<Env, Err, CK.Chunk<In>, unknown, Err, never, S> =>
    CH.readWith(
      (in_) =>
        CH.chain_(CH.fromEffect(f(s, in_)), (nextS) => {
          if (contFn(nextS)) {
            return reader(nextS)
          } else {
            return CH.end(nextS)
          }
        }),
      (err) => CH.fail(err),
      (_) => CH.end(s)
    )

  return new C.Sink(contFn(z) ? reader(z) : CH.end(z))
}