@effect-ts/system/_src/Experimental/Stream/_internal/api/repeatElements.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 CL from "../../../../Clock/index.js"
import * as E from "../../../../Either/index.js"
import * as O from "../../../../Option/index.js"
import type * as SC from "../../../../Schedule/index.js"
import type * as C from "../core.js"
import * as Collect from "./collect.js"
import * as RepeatElementsEither from "./repeatElementsEither.js"

/**
 * Repeats each element of the stream using the provided schedule. Repetitions are done in
 * addition to the first execution, which means using `Schedule.recurs(1)` actually results in
 * the original effect, plus an additional recurrence, for a total of two repetitions of each
 * value in the stream.
 */
export function repeatElements_<R, R1, E, A, B>(
  self: C.Stream<R, E, A>,
  schedule: SC.Schedule<R1, any, B>
): C.Stream<R & R1 & CL.HasClock, E, A> {
  return Collect.collect_(
    RepeatElementsEither.repeatElementsEither_(self, schedule),
    E.fold(
      () => O.none,
      (a) => O.some(a)
    )
  )
}

/**
 * Repeats each element of the stream using the provided schedule. Repetitions are done in
 * addition to the first execution, which means using `Schedule.recurs(1)` actually results in
 * the original effect, plus an additional recurrence, for a total of two repetitions of each
 * value in the stream.
 *
 * @ets_data_first repeatElements_
 */
export function repeatElements<R1, B>(schedule: SC.Schedule<R1, any, B>) {
  return <R, E, A>(self: C.Stream<R, E, A>) => repeatElements_(self, schedule)
}