import type { Task } from "../Task";
import * as T from "../Task/_core";
import { failure } from "./constructors";
import type { Exit } from "./model";

/**
 * Applies the function `f` to the successful result of the `Exit` and
 * returns the result in a new `Exit`.
 */
export const foreachTask_ = <E2, A2, R, E, A>(
   exit: Exit<E2, A2>,
   f: (a: A2) => Task<R, E, A>
): Task<R, never, Exit<E | E2, A>> => {
   switch (exit._tag) {
      case "Failure": {
         return T.pure(failure(exit.cause));
      }
      case "Success": {
         return T.result(f(exit.value));
      }
   }
};

/**
 * Applies the function `f` to the successful result of the `Exit` and
 * returns the result in a new `Exit`.
 */
export const foreachTask = <A2, R, E, A>(f: (a: A2) => Task<R, E, A>) => <E2>(exit: Exit<E2, A2>) =>
   foreachTask_(exit, f);

export const mapTask_ = <R, E, E1, A, A1>(
   exit: Exit<E, A>,
   f: (a: A) => Task<R, E1, A1>
): Task<R, never, Exit<E | E1, A1>> => {
   switch (exit._tag) {
      case "Failure":
         return T.pure(failure(exit.cause));
      case "Success":
         return T.result(f(exit.value));
   }
};

export const mapTask = <R, E1, A, A1>(f: (a: A) => Task<R, E1, A1>) => <E>(
   exit: Exit<E, A>
): Task<R, never, Exit<E | E1, A1>> => mapTask_(exit, f);