// ets_tracing: off

import type { NonEmptyArray } from "../Collections/Immutable/NonEmptyArray/index.js"
import type * as Tp from "../Collections/Immutable/Tuple/index.js"
import type { _A, _E, _R, ForcedArray } from "../Utils/index.js"
import type { Effect } from "./effect.js"
import { map_ } from "./map.js"
import { tuple, tuplePar, tupleParN } from "./tuple.js"

/**
 * Sequentially zips the specified effects using the specified combiner
 * function.
 *
 * @ets_data_first mapN_
 */
export function mapN<T extends NonEmptyArray<Effect<any, any, any>>, B>(
  f: (..._: ForcedArray<{ [k in keyof T]: _A<T[k]> }>) => B,
  __trace?: string
): (t: Tp.Tuple<T>) => Effect<_R<T[number]>, _E<T[number]>, B> {
  return (t) => mapN_(t, f, __trace)
}

/**
 * Sequentially zips the specified effects using the specified combiner
 * function.
 */
export function mapN_<T extends NonEmptyArray<Effect<any, any, any>>, B>(
  t: Tp.Tuple<T>,
  f: (..._: ForcedArray<{ [k in keyof T]: _A<T[k]> }>) => B,
  __trace?: string
): Effect<_R<T[number]>, _E<T[number]>, B> {
  // @ts-expect-error
  return map_(tuple<T>(...t.tuple), (x) => f(...x.tuple), __trace)
}

/**
 * Zips the specified effects in parallel using the specified combiner
 * function.
 *
 * @ets_data_first mapNPar_
 */
export function mapNPar<T extends NonEmptyArray<Effect<any, any, any>>, B>(
  f: (..._: ForcedArray<{ [k in keyof T]: _A<T[k]> }>) => B,
  __trace?: string
): (t: Tp.Tuple<T>) => Effect<_R<T[number]>, _E<T[number]>, B> {
  return (t) => mapNPar_(t, f, __trace)
}

/**
 * Zips the specified effects in parallel using the specified combiner
 * function.
 */
export function mapNPar_<T extends NonEmptyArray<Effect<any, any, any>>, B>(
  t: Tp.Tuple<T>,
  f: (..._: ForcedArray<{ [k in keyof T]: _A<T[k]> }>) => B,
  __trace?: string
): Effect<_R<T[number]>, _E<T[number]>, B> {
  // @ts-expect-error
  return map_(tuplePar<T>(...t.tuple), (x) => f(...x.tuple), __trace)
}

/**
 * Zips the specified effects in parallel using the specified combiner
 * function.
 *
 * This variant uses up to N fibers.
 */
export function mapNParN<T extends NonEmptyArray<Effect<any, any, any>>, B>(
  n: number,
  f: (..._: ForcedArray<{ [k in keyof T]: _A<T[k]> }>) => B,
  __trace?: string
): (t: Tp.Tuple<T>) => Effect<_R<T[number]>, _E<T[number]>, B> {
  return (t) => mapNParN_(t, n, f, __trace)
}

/**
 * Zips the specified effects in parallel using the specified combiner
 * function.
 *
 * This variant uses up to N fibers.
 */
export function mapNParN_<T extends NonEmptyArray<Effect<any, any, any>>, B>(
  t: Tp.Tuple<T>,
  n: number,
  f: (..._: ForcedArray<{ [k in keyof T]: _A<T[k]> }>) => B,
  __trace?: string
): Effect<_R<T[number]>, _E<T[number]>, B> {
  // @ts-expect-error
  return map_(tupleParN(n)(...t.tuple), (x) => f(...x.tuple), __trace)
}