org.jetbrains.kotlinx.jupyter.messaging.MessageRequestProcessorImpl.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.messaging

import ch.qos.logback.classic.Level
import org.jetbrains.kotlinx.jupyter.api.KernelLoggerFactory
import org.jetbrains.kotlinx.jupyter.api.libraries.RawMessage
import org.jetbrains.kotlinx.jupyter.common.looksLikeReplCommand
import org.jetbrains.kotlinx.jupyter.exceptions.ReplCompilerException
import org.jetbrains.kotlinx.jupyter.execution.JupyterExecutor
import org.jetbrains.kotlinx.jupyter.messaging.comms.CommManagerInternal
import org.jetbrains.kotlinx.jupyter.repl.ReplForJupyter
import org.jetbrains.kotlinx.jupyter.repl.impl.ReplForJupyterImpl

open class MessageRequestProcessorImpl(
    rawIncomingMessage: RawMessage,
    messageFactoryProvider: MessageFactoryProvider,
    socketManager: JupyterBaseSockets,
    commManager: CommManagerInternal,
    executor: JupyterExecutor,
    executionCounter: ExecutionCounter,
    loggerFactory: KernelLoggerFactory,
    repl: ReplForJupyter,
) : IdeCompatibleMessageRequestProcessor(
        rawIncomingMessage,
        messageFactoryProvider,
        socketManager,
        commManager,
        executor,
        executionCounter,
        loggerFactory,
        repl,
    ),
    JupyterCommunicationFacility {
    override fun processIsCompleteRequest(content: IsCompleteRequest) {
        // We are in console mode, so switch off all the loggers
        val loggingManager = (repl as? ReplForJupyterImpl)?.loggingManager
        if (loggingManager != null) {
            if (loggingManager.mainLoggerLevel() != Level.OFF) loggingManager.disableLogging()
        }

        val resStr =
            if (looksLikeReplCommand(content.code)) {
                "complete"
            } else {
                val result =
                    try {
                        val check = repl.checkComplete(content.code)
                        when {
                            check.isComplete -> "complete"
                            else -> "incomplete"
                        }
                    } catch (ex: ReplCompilerException) {
                        "invalid"
                    }
                result
            }
        socketManager.shell.sendMessage(
            messageFactory.makeReplyMessage(MessageType.IS_COMPLETE_REPLY, content = IsCompleteReply(resStr)),
        )
    }
}