org.jetbrains.kotlinx.jupyter.libraries.LibrariesProcessorImpl.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-333

Show newest version

package org.jetbrains.kotlinx.jupyter.libraries

import org.jetbrains.kotlinx.jupyter.api.KotlinKernelVersion
import org.jetbrains.kotlinx.jupyter.api.libraries.LibraryDefinition
import org.jetbrains.kotlinx.jupyter.api.libraries.LibraryDefinitionProducer
import org.jetbrains.kotlinx.jupyter.api.libraries.LibraryResolutionRequest
import org.jetbrains.kotlinx.jupyter.exceptions.ReplException
import org.jetbrains.kotlinx.jupyter.magics.splitLibraryCalls

class LibrariesProcessorImpl(
    private val libraryResolver: LibraryResolver?,
    private val libraryReferenceParser: LibraryReferenceParser,
    private val kernelVersion: KotlinKernelVersion?,
) : LibrariesProcessor {
    private val _requests = mutableListOf()
    override val requests: Collection
        get() = _requests

    private fun checkKernelVersionRequirements(
        name: String,
        library: LibraryDefinition,
    ) {
        library.minKernelVersion?.let { minVersion ->
            kernelVersion?.let { currentVersion ->
                if (currentVersion < minVersion) {
                    throw ReplException(
                        """
                        Library '$name' requires at least $minVersion version of kernel.
                        Current kernel version is $currentVersion.
                        Please update kernel, see https://github.com/Kotlin/kotlin-jupyter#updating for more info.
                        """.trimIndent(),
                    )
                }
            }
        }
    }

    override fun processNewLibraries(arg: String): List =
        splitLibraryCalls(arg).map {
            val (libRef, vars) = libraryReferenceParser.parseReferenceWithArgs(it)
            val library =
                libraryResolver?.resolve(libRef, vars)
                    ?: throw ReplException("Unknown library '$libRef'")

            _requests.add(LibraryResolutionRequest(libRef, vars, library))

            checkKernelVersionRequirements(libRef.toString(), library)

            TrivialLibraryDefinitionProducer(library)
        }
}