org.jetbrains.kotlin.sir.providers.impl.SirOneToOneModuleProvider.kt Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of sir-providers Show documentation

SIR Providers - family of classes, that transforms KaSymbol into corresponding SIR nodes

The 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.sir.providers.impl

import org.jetbrains.kotlin.analysis.api.KaExperimentalApi
import org.jetbrains.kotlin.analysis.api.projectStructure.KaLibraryModule
import org.jetbrains.kotlin.analysis.api.projectStructure.KaModule
import org.jetbrains.kotlin.analysis.api.projectStructure.KaSourceModule
import org.jetbrains.kotlin.sir.SirModule
import org.jetbrains.kotlin.sir.builder.buildModule
import org.jetbrains.kotlin.sir.providers.SirModuleProvider

/**
 * A module provider implementation that generates a [SirModule] for each given [KaModule]
 */
public class SirOneToOneModuleProvider : SirModuleProvider {

    private val moduleCache = mutableMapOf()

    public val modules: Map
        get() = moduleCache.toMap()

    override fun KaModule.sirModule(): SirModule = moduleCache.getOrPut(this) {
        buildModule {
            name = getModuleName(this@sirModule)
        }
    }

    private fun getModuleName(ktModule: KaModule): String {
        return ktModule.moduleName
    }
}

@OptIn(KaExperimentalApi::class)
private val KaModule.moduleName: String
    get() = when(this) {
        is KaLibraryModule -> libraryName
        is KaSourceModule -> name
        else -> error("Tried to calculate KaModule.moduleName for unknown module type: $this, described as: ${this.moduleDescription}")
    }