org.jetbrains.kotlinx.jupyter.util.StdIO.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of kotlin-jupyter-shared-compiler Show documentation

Implementation of REPL compiler and preprocessor for Jupyter dialect of Kotlin (IDE-compatible)

There is a newer version: 0.12.0-290

Show newest version

package org.jetbrains.kotlinx.jupyter.util

import java.io.Closeable
import kotlin.reflect.KMutableProperty0

var systemErrStream
    get() = System.err
    set(value) {
        System.setErr(value)
    }

var systemOutStream
    get() = System.out
    set(value) {
        System.setOut(value)
    }

var systemInStream
    get() = System.`in`
    set(value) {
        System.setIn(value)
    }

fun  withSubstitutedStream(
    standardStreamProp: KMutableProperty0,
    newStreamFactory: (StreamT) -> StreamT,
    body: (StreamT) -> R,
): R {
    val originalStream = standardStreamProp.get()
    val newStream = newStreamFactory(originalStream)
    return try {
        standardStreamProp.set(newStream)
        body(newStream)
    } finally {
        newStream.close()
        standardStreamProp.set(originalStream)
    }
}