@effect-ts/system/_src/Experimental/Stream/_internal/api/flattenExitOption.ts

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

// ets_tracing: off

import * as CS from "../../../../Cause/index.js"
import * as CK from "../../../../Collections/Immutable/Chunk/index.js"
import * as Ex from "../../../../Exit/index.js"
import * as O from "../../../../Option/index.js"
import * as CH from "../../Channel/index.js"
import * as C from "../core.js"

/**
 * Unwraps `Exit` values that also signify end-of-stream by failing with `None`.
 *
 * For `Exit<E, A>` values that do not signal end-of-stream, prefer:
 */
export function flattenExitOption<R, E, E1, A>(
  self: C.Stream<R, E, Ex.Exit<O.Option<E1>, A>>
): C.Stream<R, E | E1, A> {
  const processChunk = (
    chunk: CK.Chunk<Ex.Exit<O.Option<E1>, A>>,
    cont: CH.Channel<
      R,
      E,
      CK.Chunk<Ex.Exit<O.Option<E1>, A>>,
      unknown,
      E | E1,
      CK.Chunk<A>,
      any
    >
  ): CH.Channel<
    R,
    E,
    CK.Chunk<Ex.Exit<O.Option<E1>, A>>,
    unknown,
    E | E1,
    CK.Chunk<A>,
    any
  > => {
    const {
      tuple: [toEmit, rest]
    } = CK.splitWhere_(chunk, (_) => !Ex.succeeded(_))
    const next = O.fold_(
      CK.head(rest),
      () => cont,
      Ex.fold(
        (cause) =>
          O.fold_(
            CS.flipCauseOption(cause),
            () => CH.end<void>(undefined),
            (cause) => CH.failCause(cause)
          ),
        () => CH.end<void>(undefined)
      )
    )

    return CH.zipRight_(
      CH.write(
        CK.collect_(
          toEmit,
          Ex.fold(
            () => O.none,
            (a) => O.some(a)
          )
        )
      ),
      next
    )
  }

  const process: CH.Channel<
    R,
    E,
    CK.Chunk<Ex.Exit<O.Option<E1>, A>>,
    unknown,
    E | E1,
    CK.Chunk<A>,
    any
  > = CH.readWithCause(
    (chunk) => processChunk(chunk, process),
    (cause) => CH.failCause(cause),
    (_) => CH.end(undefined)
  )

  return new C.Stream(self.channel[">>>"](process))
}