kio-algebra.kt

sealed class KIO<out A> {
    fun <B> flatMap(f: (A) -> KIO<B>): KIO<B> {
        return FlatMap(this, f)
    }

    fun <B> map(f: (A) -> B): KIO<B> {
        return flatMap { a ->
            Effect {
                f(a)
            }
        }
    }
}

data class Effect<out A>(val f: () -> A) : KIO<A>()

data class FlatMap<A, out B>(
    val input: KIO<A>,
    val f: (A) -> KIO<B>
) : KIO<B>()