@effect-ts/system/_src/Experimental/Stream/Channel/api/mapOutEffect.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 T from "../../../../Effect/index.js"
import * as C from "../core.js"
import * as ReadWith from "./readWith.js"
import * as ZipRight from "./zipRight.js"

const mapOutMReader = <Env, Env1, OutErr, OutErr1, OutElem, OutElem1, OutDone>(
  f: (o: OutElem) => T.Effect<Env1, OutErr1, OutElem1>
): C.Channel<
  Env & Env1,
  OutErr,
  OutElem,
  OutDone,
  OutErr | OutErr1,
  OutElem1,
  OutDone
> =>
  ReadWith.readWith(
    (out) =>
      ZipRight.zipRight_(
        C.chain_(C.fromEffect(f(out)), (_) => C.write(_)),
        mapOutMReader(f)
      ),
    (e) => C.fail(e),
    (z) => C.end(z)
  )

export function mapOutEffect_<
  Env,
  Env1,
  InErr,
  InElem,
  InDone,
  OutErr,
  OutErr1,
  OutElem,
  OutElem1,
  OutDone
>(
  self: C.Channel<Env, InErr, InElem, InDone, OutErr, OutElem, OutDone>,
  f: (o: OutElem) => T.Effect<Env1, OutErr1, OutElem1>
): C.Channel<Env & Env1, InErr, InElem, InDone, OutErr | OutErr1, OutElem1, OutDone> {
  return C.pipeTo_(self, mapOutMReader(f))
}

/**
 * @ets_data_first mapOutEffect_
 */
export function mapOutEffect<Env1, OutErr1, OutElem, OutElem1>(
  f: (o: OutElem) => T.Effect<Env1, OutErr1, OutElem1>
) {
  return <Env, InErr, InElem, InDone, OutErr, OutDone>(
    self: C.Channel<Env, InErr, InElem, InDone, OutErr, OutElem, OutDone>
  ) => mapOutEffect_(self, f)
}