org.jetbrains.kotlinx.jupyter.repl.ResultsRendering.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)

The newest version!

package org.jetbrains.kotlinx.jupyter.repl

import kotlinx.serialization.json.Json
import kotlinx.serialization.json.JsonPrimitive
import kotlinx.serialization.json.encodeToJsonElement
import org.jetbrains.kotlinx.jupyter.api.DisplayResult
import org.jetbrains.kotlinx.jupyter.api.InMemoryMimeTypedResult
import org.jetbrains.kotlinx.jupyter.api.MimeTypedResultEx
import org.jetbrains.kotlinx.jupyter.api.Renderable
import org.jetbrains.kotlinx.jupyter.api.libraries.ExecutionHost
import org.jetbrains.kotlinx.jupyter.api.outputs.standardMetadataModifiers
import org.jetbrains.kotlinx.jupyter.api.textResult
import org.jetbrains.kotlinx.jupyter.repl.notebook.MutableNotebook

fun renderValue(
    notebook: MutableNotebook,
    executor: ExecutionHost,
    value: Any?,
    id: String?,
): DisplayResult? {
    val inMemoryReplResultsHolder = notebook.sharedReplContext!!.inMemoryReplResultsHolder
    val renderedValue = notebook.renderersProcessor.renderValue(executor, value)
    val rendered =
        if (renderedValue is InMemoryMimeTypedResult) {
            // Strip out in-memory values and put them into the InMemoryReplResultsHolder
            val inMemoryValue = renderedValue.inMemoryOutput.result
            val displayId =
                if (id != null) {
                    inMemoryReplResultsHolder.setReplResult(id, inMemoryValue)
                    id
                } else {
                    inMemoryReplResultsHolder.addReplResult(inMemoryValue)
                }
            val mimeData = renderedValue.fallbackResult + Pair(renderedValue.inMemoryOutput.mimeType, JsonPrimitive(displayId))
            MimeTypedResultEx(mimeData = Json.encodeToJsonElement(mimeData), id, standardMetadataModifiers())
        } else {
            renderedValue
        }
    return notebook.postRender(rendered)
}

fun MutableNotebook.postRender(value: Any?): DisplayResult? {
    fun renderAsText(obj: Any?): String = textRenderersProcessor.renderPreventingRecursion(obj)
    return when (value) {
        null -> textResult(renderAsText(null))
        is DisplayResult -> value
        is Renderable -> value.render(this)
        is Unit -> null
        else -> textResult(renderAsText(value))
    }
}