org.jetbrains.kotlinx.jupyter.magics.CompoundCodePreprocessor.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.magics

import org.jetbrains.kotlinx.jupyter.api.CodePreprocessor
import org.jetbrains.kotlinx.jupyter.api.ExtensionsProcessor
import org.jetbrains.kotlinx.jupyter.api.KotlinKernelHost
import org.jetbrains.kotlinx.jupyter.exceptions.KernelInternalObject
import org.jetbrains.kotlinx.jupyter.exceptions.LibraryProblemPart
import org.jetbrains.kotlinx.jupyter.exceptions.rethrowAsLibraryException
import org.jetbrains.kotlinx.jupyter.util.PriorityList

/**
 * Containing [preprocessors]' [process] are run in reversed order: last added processors
 * are run first
 */
class CompoundCodePreprocessor(vararg preprocessors: CodePreprocessor) : CodePreprocessor, ExtensionsProcessor {
    private val preprocessors = PriorityList()

    init {
        for (preprocessor in preprocessors) {
            register(preprocessor)
        }
    }

    override fun process(
        code: String,
        host: KotlinKernelHost,
    ): CodePreprocessor.Result {
        var result = CodePreprocessor.Result(code, emptyList())

        fun iteration(preprocessor: CodePreprocessor) {
            if (preprocessor.accepts(result.code)) {
                val newResult = preprocessor.process(result.code, host)
                result = CodePreprocessor.Result(newResult.code, result.libraries + newResult.libraries)
            }
        }

        for (preprocessor in preprocessors) {
            if (preprocessor is KernelInternalObject) {
                iteration(preprocessor)
            } else {
                rethrowAsLibraryException(LibraryProblemPart.CODE_PREPROCESSORS) {
                    iteration(preprocessor)
                }
            }
        }
        return result
    }

    override fun register(
        extension: CodePreprocessor,
        priority: Int,
    ) {
        preprocessors.add(extension, priority)
    }

    override fun unregister(extension: CodePreprocessor) {
        preprocessors.remove(extension)
    }

    override fun unregisterAll() {
        preprocessors.clear()
    }

    override fun registeredExtensions(): Collection {
        return preprocessors.elements()
    }

    override fun registeredExtensionsWithPriority(): List> {
        return preprocessors.elementsWithPriority()
    }
}