org.jetbrains.kotlinx.jupyter.libraries.ResolutionUtil.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.KernelLoggerFactory
import org.jetbrains.kotlinx.jupyter.common.HttpClient
import org.jetbrains.kotlinx.jupyter.common.LibraryDescriptorsManager
import org.jetbrains.kotlinx.jupyter.repl.ResolutionInfoProviderFactory
import java.io.File

fun getStandardResolver(
    loggerFactory: KernelLoggerFactory,
    homeDir: String? = null,
    infoProvider: ResolutionInfoProvider,
    httpClient: HttpClient,
    libraryDescriptorsManager: LibraryDescriptorsManager,
): LibraryResolver {
    // Standard resolver doesn't cache results in memory
    var res: LibraryResolver = FallbackLibraryResolver(httpClient, libraryDescriptorsManager)
    val librariesDir: File? = homeDir?.let { libraryDescriptorsManager.homeLibrariesDir(File(it)) }
    res = LocalLibraryResolver(res, loggerFactory, libraryDescriptorsManager, librariesDir)
    res = DefaultInfoLibraryResolver(res, infoProvider, libraryDescriptorsManager, listOf(libraryDescriptorsManager.userLibrariesDir))
    return res
}

object DefaultResolutionInfoProviderFactory : ResolutionInfoProviderFactory {
    override fun create(
        httpUtil: LibraryHttpUtil,
        loggerFactory: KernelLoggerFactory,
    ): ResolutionInfoProvider {
        return StandardResolutionInfoProvider(
            loggerFactory,
            AbstractLibraryResolutionInfo.ByClasspath,
            httpUtil,
        )
    }
}

fun getDefaultResolutionInfoSwitcher(
    provider: ResolutionInfoProvider,
    libraryInfoCache: LibraryInfoCache,
    defaultDir: File,
    defaultRef: String,
): ResolutionInfoSwitcher {
    val initialInfo = provider.fallback

    val dirInfo =
        if (initialInfo is AbstractLibraryResolutionInfo.ByDir) {
            initialInfo
        } else {
            AbstractLibraryResolutionInfo.ByDir(defaultDir)
        }

    val refInfo =
        if (initialInfo is AbstractLibraryResolutionInfo.ByGitRef) {
            initialInfo
        } else {
            libraryInfoCache.getLibraryInfoByRefWithFallback(defaultRef)
        }

    val classpathInfo = AbstractLibraryResolutionInfo.ByClasspath

    return ResolutionInfoSwitcher(provider, DefaultInfoSwitch.DIRECTORY) { switch ->
        when (switch) {
            DefaultInfoSwitch.DIRECTORY -> dirInfo
            DefaultInfoSwitch.GIT_REFERENCE -> refInfo
            DefaultInfoSwitch.CLASSPATH -> classpathInfo
        }
    }
}