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

The newest version!

package org.jetbrains.kotlinx.jupyter.messaging

import kotlinx.serialization.json.JsonElement
import kotlinx.serialization.json.jsonPrimitive
import org.jetbrains.kotlinx.jupyter.api.libraries.RawMessage

class MessageFactoryImpl : MessageFactory {
    private var _messageId: List = listOf(byteArrayOf(1))
    override val messageId: List
        get() = _messageId

    private var _sessionId = ""
    override val sessionId: String
        get() = _sessionId

    private var _username = ""
    override val username: String
        get() = _username

    private var _contextMessage: RawMessage? = null
    override val contextMessage: RawMessage?
        get() = _contextMessage

    override fun updateSessionInfo(message: RawMessage) {
        val header = message.header
        header["session"]?.jsonPrimitive?.content?.let { _sessionId = it }
        header["username"]?.jsonPrimitive?.content?.let { _username = it }
        _messageId = message.id
    }

    override fun updateContextMessage(contextMessage: RawMessage?) {
        _contextMessage = contextMessage
    }

    override fun makeReplyMessageOrNull(
        msgType: MessageType?,
        sessionId: String?,
        header: MessageHeader?,
        parentHeader: MessageHeader?,
        metadata: JsonElement?,
        content: MessageContent?,
    ): Message? {
        val myContextMessage = contextMessage ?: return null

        return makeReplyMessage(
            myContextMessage,
            msgType,
            sessionId,
            header,
            parentHeader,
            metadata,
            content,
        )
    }
}