import type { EffectURI } from "@effect/core/io/Effect/definition/base"

type NonEmptyArrayEffect = Array<Effect<any, any, any>> & { readonly 0: Effect<any, any, any> }

export type TupleA<T extends NonEmptyArrayEffect> = {
  [K in keyof T]: [T[K]] extends [Effect<any, any, infer A>] ? A : never
}

/**
 * Like `forEach` + `identity` with a tuple type.
 *
 * @tsplus static effect/core/io/Effect.Ops tuple
 */
export function tuple<T extends NonEmptyArrayEffect>(
  ...t: T & {
    0: Effect<any, any, any>
  }
): Effect<
  [T[number]] extends [{ [EffectURI]: { _R: (_: never) => infer R } }] ? R : never,
  [T[number]] extends [{ [EffectURI]: { _E: (_: never) => infer E } }] ? E : never,
  TupleA<T>
> {
  return Effect.collectAll(t).map((c) => c.toArray) as any
}

/**
 * Like tuple but parallel, same as `forEachPar` + `identity` with a tuple type.
 *
 * @tsplus static effect/core/io/Effect.Ops tuplePar
 */
export function tuplePar<T extends NonEmptyArrayEffect>(
  ...t: T & {
    0: Effect<any, any, any>
  }
): Effect<
  [T[number]] extends [{ [EffectURI]: { _R: (_: never) => infer R } }] ? R : never,
  [T[number]] extends [{ [EffectURI]: { _E: (_: never) => infer E } }] ? E : never,
  TupleA<T>
> {
  return Effect.collectAllPar(t).map((c) => c.toArray) as any
}