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

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

// ets_tracing: off

import { pipe } from "../../../../Function/index.js"
import type * as TK from "../../Take/index.js"
import type * as C from "../core.js"
import * as FlattenChunks from "./flattenChunks.js"
import * as FlattenExitOption from "./flattenExitOption.js"
import * as Map from "./map.js"

/**
 * Unwraps `Exit` values and flatten chunks that also signify end-of-stream by failing with `None`.
 */
export function flattenTake<R, E, E1, A>(
  self: C.Stream<R, E, TK.Take<E1, A>>
): C.Stream<R, E | E1, A> {
  return pipe(
    self,
    Map.map((_) => _.exit),
    FlattenExitOption.flattenExitOption,
    FlattenChunks.flattenChunks
  )
}