org.jetbrains.kotlinx.jupyter.codegen.TextRenderersProcessorImpl.kt Maven / Gradle / Ivy

Go to download

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

Kotlin Jupyter kernel published as artifact

There is a newer version: 0.12.0-335

Show newest version

package org.jetbrains.kotlinx.jupyter.codegen

import org.jetbrains.kotlinx.jupyter.api.TextRenderer
import org.jetbrains.kotlinx.jupyter.api.TextRendererWithPriority
import org.jetbrains.kotlinx.jupyter.exceptions.LibraryProblemPart
import org.jetbrains.kotlinx.jupyter.exceptions.rethrowAsLibraryException
import org.jetbrains.kotlinx.jupyter.repl.execution.AbstractExtensionsProcessor

class TextRenderersProcessorImpl :
    AbstractExtensionsProcessor(
        latterFirst = true,
    ),
    TextRenderersProcessorWithPreventingRecursion {
    private val cache = mutableListOf()

    override fun registeredRenderers(): List {
        return extensions.elementsWithPriority().map { TextRendererWithPriority(it.first, it.second) }
    }

    override fun render(value: Any?): String {
        val cacheIndex = cache.indexOfFirst { it.k === value }
        if (cacheIndex != -1) {
            return cache[cacheIndex].v ?: ""
        }

        cache.add(CacheEntry(value))
        val cacheEntry = cache.last()

        fun cached(rendered: String): String {
            cacheEntry.v = rendered
            return rendered
        }

        for (renderer in extensions) {
            val res =
                rethrowAsLibraryException(LibraryProblemPart.TEXT_RENDERERS) {
                    renderer.render(this, value)
                }
            if (res != null) return cached(res)
        }
        return cached(value.toString())
    }

    override fun renderPreventingRecursion(value: Any?): String {
        val res =
            try {
                cache.clear()
                render(value)
            } finally {
                cache.clear()
            }
        return res
    }

    private data class CacheEntry(
        var k: Any?,
        var v: String? = null,
    )
}