@effect-ts/system/_src/Effect/repeatN.ts

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

// ets_tracing: off

import { chain_, succeed, yieldNow } from "./core.js"
import type { Effect } from "./effect.js"
import { zipRight_ } from "./zips.js"

/**
 * Repeats this effect the specified number of times.
 *
 * @ets_data_first repeatN_
 */
export function repeatN(n: number, __trace?: string) {
  return <R, E, A>(self: Effect<R, E, A>): Effect<R, E, A> => repeatN_(self, n, __trace)
}

/**
 * Repeats this effect the specified number of times.
 */
export function repeatN_<R, E, A>(
  self: Effect<R, E, A>,
  n: number,
  __trace?: string
): Effect<R, E, A> {
  return chain_(
    self,
    (a) => (n <= 0 ? succeed(a) : zipRight_(yieldNow, repeatN_(self, n - 1))),
    __trace
  )
}