// ets_tracing: off

import type { StateInURI, StateOutURI } from "../../Modules/index.js"
import * as HKT from "../../Prelude/HKT/index.js"
import type { Monad } from "../../Prelude/index.js"
import * as StateT from "../Classic/index.js"

/**
 * Take over ownership of "S" making it fixed to provided "S"
 */
export type V<C, S> = HKT.CleanParam<C, "S"> & HKT.Fix<"S", S>

/**
 * State Input URI with local override of "S", this makes it safe to be
 * stacked multiple times
 */
export interface PSIn<S> extends HKT.URI<StateInURI, HKT.Fix<"S", S>> {}

/**
 * State Output URI with local override of "S", this makes it safe to be
 * stacked multiple times
 */
export interface PSOut<S> extends HKT.URI<StateOutURI, HKT.Fix<"S", S>> {}

/**
 * Construct the transformed URI as [StateIn, F, StateOut]
 */
export type ParametricStateT<F extends HKT.URIS, S> = [PSIn<S>, ...F, PSOut<S>]

export function monad<S>() {
  return <F extends HKT.URIS, C>(M: Monad<F, C>) => getMonad_<F, C, S>(M)
}

function getMonad_<F extends HKT.URIS, C, S>(
  M: Monad<F, C>
): Monad<ParametricStateT<F, S>, V<C, S>>
function getMonad_<F, S>(
  M: Monad<HKT.UHKT<F>>
): Monad<ParametricStateT<HKT.UHKT<F>, S>, V<HKT.Auto, S>> {
  return HKT.instance(StateT.monad(M))
}