org.jetbrains.kotlin.diagnostics.rendering.RenderingContext.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of kotlin-compiler-embeddable Show documentation

the Kotlin compiler embeddable

There is a newer version: 2.1.0-RC

Show newest version

/*
 * Copyright 2010-2021 JetBrains s.r.o. and Kotlin Programming Language contributors.
 * Use of this source code is governed by the Apache 2.0 license that can be found in the license/LICENSE.txt file.
 */

package org.jetbrains.kotlin.diagnostics.rendering

// holds data about the parameters of the diagnostic we're about to render
sealed class RenderingContext {
    abstract operator fun  get(key: Key): T

    abstract class Key(val name: String) {
        abstract fun compute(objectsToRender: Collection): T
    }

    class Impl(private val objectsToRender: Collection) : RenderingContext() {
        private val data = linkedMapOf, Any?>()

        @Suppress("UNCHECKED_CAST")
        override fun  get(key: Key): T {
            return data[key] as? T ?: key.compute(objectsToRender).also { data[key] = it }
        }
    }

    object Empty : RenderingContext() {
        override fun  get(key: Key): T {
            return key.compute(emptyList())
        }
    }

    companion object {
        @JvmStatic
        fun of(vararg objectsToRender: Any?): RenderingContext {
            return Impl(objectsToRender.toList())
        }
    }
}