All Downloads are FREE. Search and download functionalities are using the official Maven repository.

org.jetbrains.kotlinx.jupyter.util.JupyterClientDetector.kt Maven / Gradle / Ivy

Go to download

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.util

import org.jetbrains.kotlinx.jupyter.api.JupyterClientType
import org.jetbrains.kotlinx.jupyter.api.KernelLoggerFactory
import org.jetbrains.kotlinx.jupyter.api.getLogger

class JupyterClientDetector(
    loggerFactory: KernelLoggerFactory,
) {
    private val logger = loggerFactory.getLogger(this::class)

    fun detect(): JupyterClientType {
        return try {
            doDetect()
        } catch (e: LinkageError) {
            logger.error("Unable to detect Jupyter client type because of incompatible JVM version", e)
            JupyterClientType.UNKNOWN
        }
    }

    private fun doDetect(): JupyterClientType {
        logger.info("Detecting Jupyter client type")
        val currentHandle = ProcessHandle.current()
        val ancestors = generateSequence(currentHandle) { it.parent().orElse(null) }.toList()

        for (handle in ancestors) {
            val info = handle.info()
            val command = info.command().orElse("")
            val arguments = info.arguments().orElse(emptyArray()).toList()

            logger.info("Inspecting process: $command ${arguments.joinToString(" ")}")
            val correctDetector = detectors.firstOrNull { it.isThisClient(command, arguments) } ?: continue

            logger.info("Detected type is ${correctDetector.type}")
            return correctDetector.type
        }

        logger.info("Client type has not been detected")
        return JupyterClientType.UNKNOWN
    }

    private interface Detector {
        val type: JupyterClientType

        fun isThisClient(
            command: String,
            arguments: List,
        ): Boolean
    }

    private fun detector(
        type: JupyterClientType,
        predicate: (String, List) -> Boolean,
    ) = object : Detector {
        override val type: JupyterClientType get() = type

        override fun isThisClient(
            command: String,
            arguments: List,
        ): Boolean {
            return predicate(command, arguments)
        }
    }

    private val detectors =
        listOf(
            detector(JupyterClientType.JUPYTER_NOTEBOOK) { command, args ->
                "jupyter-notebook" in command || args.any { "jupyter-notebook" in it }
            },
            detector(JupyterClientType.JUPYTER_LAB) { command, args ->
                "jupyter-lab" in command || args.any { "jupyter-lab" in it }
            },
            detector(JupyterClientType.KERNEL_TESTS) { command, _ ->
                command.endsWith("idea64.exe")
            },
        )
}




© 2015 - 2024 Weber Informatics LLC | Privacy Policy