org.jetbrains.kotlinx.jupyter.libraries.libraryParsingUtil.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-356

Show newest version

package org.jetbrains.kotlinx.jupyter.libraries

import org.jetbrains.kotlinx.jupyter.api.libraries.LibraryReference
import org.jetbrains.kotlinx.jupyter.api.libraries.LibraryResolutionInfo
import org.jetbrains.kotlinx.jupyter.api.libraries.Variable
import java.io.File
import java.net.URL

fun parseReferenceWithArgs(str: String): Pair> {
    val (fullName, vars) = parseLibraryName(str)
    val reference = parseLibraryReference(fullName)
    return reference to vars
}

private fun parseResolutionInfo(string: String): LibraryResolutionInfo {
    // In case of empty string after `@`: %use lib@
    if (string.isBlank()) return AbstractLibraryResolutionInfo.Default()

    val (type, vars) = parseCall(string, Brackets.SQUARE)
    return defaultParsers[type]?.getInfo(vars) ?: AbstractLibraryResolutionInfo.Default(type)
}

fun parseLibraryReference(string: String): LibraryReference {
    val sepIndex = string.indexOf('@')
    if (sepIndex == -1) return LibraryReference(AbstractLibraryResolutionInfo.Default(), string)

    val nameString = string.substring(0, sepIndex)
    val infoString = string.substring(sepIndex + 1)
    val info = parseResolutionInfo(infoString)
    return LibraryReference(info, nameString)
}

private val defaultParsers = listOf(
    LibraryResolutionInfoParser.make("ref", listOf(Parameter.Required("ref"))) { args ->
        AbstractLibraryResolutionInfo.getInfoByRef(args["ref"] ?: error("Argument 'ref' should be specified"))
    },
    LibraryResolutionInfoParser.make("file", listOf(Parameter.Required("path"))) { args ->
        AbstractLibraryResolutionInfo.ByFile(File(args["path"] ?: error("Argument 'path' should be specified")))
    },
    LibraryResolutionInfoParser.make("dir", listOf(Parameter.Required("dir"))) { args ->
        AbstractLibraryResolutionInfo.ByDir(File(args["dir"] ?: error("Argument 'dir' should be specified")))
    },
    LibraryResolutionInfoParser.make("url", listOf(Parameter.Required("url"))) { args ->
        AbstractLibraryResolutionInfo.ByURL(URL(args["url"] ?: error("Argument 'url' should be specified")))
    },
    LibraryResolutionInfoParser.make("classpath", listOf()) {
        AbstractLibraryResolutionInfo.ByClasspath
    },
).associateBy { it.name }