org.jetbrains.kotlin.backend.common.ScopeStack.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-2024 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.backend.common

internal class ScopeStack {
    private class Scope(val isGlobal: Boolean) {
        val values = mutableSetOf()
    }

    // The outer list is never empty
    private val scopes = mutableListOf>>(mutableListOf())

    private fun  withNestedScope(isGlobalScope: Boolean, populateScope: MutableSet.() -> Unit = {}, block: () -> R): R =
        scopes.last().temporarilyPushing(Scope(isGlobalScope)) {
            it.values.populateScope()
            block()
        }

    /**
     * Creates a new scope and runs [block] with it.
     *
     * @param isGlobalScope Whether values of this new scope are always visible in nested scopes,
     *   even for those nested scopes where [outerScopesAreInvisible] is `true`.
     * @param outerScopesAreInvisible Whether values of outer scopes are invisible in this new scope, except when an outer scope is global.
     */
    fun  withNewScope(
        isGlobalScope: Boolean = false,
        outerScopesAreInvisible: Boolean = false,
        populateScope: MutableSet.() -> Unit = {},
        block: () -> R,
    ): R =
        if (outerScopesAreInvisible) {
            scopes.temporarilyPushing(scopes.last().filterTo(mutableListOf(), Scope::isGlobal)) {
                withNestedScope(isGlobalScope = false, populateScope, block)
            }
        } else {
            withNestedScope(isGlobalScope, populateScope, block)
        }

    fun addToCurrentScope(element: E) {
        scopes.last().lastOrNull()?.values?.add(element)
    }

    fun isVisibleInCurrentScope(element: E): Boolean = scopes.last().any { it.values.contains(element) }
}