[Analysis API] introduce API to work with signatures substitution

2022-06-07 18:06:04 +02:00
parent 2db73af75e
commit 78c63c082b
9 changed files with 246 additions and 64 deletions
@@ -0,0 +1,95 @@
+/*
+ * Copyright 2010-2022 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.analysis.api.impl.base.components
+
+import org.jetbrains.kotlin.analysis.api.components.KtSignatureSubsitutor
+import org.jetbrains.kotlin.analysis.api.symbols.*
+import org.jetbrains.kotlin.analysis.api.types.KtSubstitutor
+import org.jetbrains.kotlin.analysis.utils.errors.unexpectedElementError
+
+abstract class AbstractKtSignatureSubsitutorImpl : KtSignatureSubsitutor() {
+    @Suppress("UNCHECKED_CAST")
+    override fun <S : KtCallableSymbol> substitute(signature: KtSignature<S>, substitutor: KtSubstitutor): KtSignature<S> {
+        return when (signature) {
+            is KtFunctionLikeSignature -> {
+                substitute(signature as KtFunctionLikeSignature<KtFunctionLikeSymbol>, substitutor) as KtSignature<S>
+            }
+            is KtVariableLikeSignature -> {
+                substitute(signature as KtVariableLikeSignature<KtVariableLikeSymbol>, substitutor) as KtSignature<S>
+            }
+        }
+    }
+
+    override fun <S : KtVariableLikeSymbol> substitute(
+        signature: KtVariableLikeSignature<S>,
+        substitutor: KtSubstitutor
+    ): KtVariableLikeSignature<S> {
+        if (substitutor is KtSubstitutor.Empty) return signature
+        return KtVariableLikeSignature(
+            signature.symbol,
+            substitutor.substituteOrSelf(signature.returnType),
+            signature.receiverType?.let { substitutor.substituteOrSelf(it) },
+        )
+    }
+
+    override fun <S : KtFunctionLikeSymbol> substitute(
+        signature: KtFunctionLikeSignature<S>,
+        substitutor: KtSubstitutor
+    ): KtFunctionLikeSignature<S> {
+        if (substitutor is KtSubstitutor.Empty) return signature
+        return KtFunctionLikeSignature(
+            signature.symbol,
+            substitutor.substituteOrSelf(signature.returnType),
+            signature.receiverType?.let { substitutor.substituteOrSelf(it) },
+            signature.valueParameters.map { substitute(it, substitutor) }
+        )
+    }
+
+
+    @Suppress("UNCHECKED_CAST")
+    override fun <S : KtCallableSymbol> substitute(symbol: S, substitutor: KtSubstitutor): KtSignature<S> {
+        return when (symbol) {
+            is KtFunctionLikeSymbol -> substitute(symbol, substitutor)
+            is KtVariableLikeSymbol -> substitute(symbol, substitutor)
+            else -> unexpectedElementError("symbol", symbol)
+        }
+    }
+
+    override fun <S : KtFunctionLikeSymbol> substitute(symbol: S, substitutor: KtSubstitutor): KtFunctionLikeSignature<S> {
+        if (substitutor is KtSubstitutor.Empty) return asSignature(symbol)
+        return KtFunctionLikeSignature(
+            symbol,
+            substitutor.substituteOrSelf(symbol.returnType),
+            symbol.receiverType?.let { substitutor.substituteOrSelf(it) },
+            symbol.valueParameters.map { substitute(it, substitutor) }
+        )
+    }
+
+    override fun <S : KtVariableLikeSymbol> substitute(symbol: S, substitutor: KtSubstitutor): KtVariableLikeSignature<S> {
+        if (substitutor is KtSubstitutor.Empty) return asSignature(symbol)
+        return KtVariableLikeSignature(
+            symbol,
+            substitutor.substituteOrSelf(symbol.returnType),
+            symbol.receiverType?.let { substitutor.substituteOrSelf(it) },
+        )
+    }
+
+    override fun <S : KtCallableSymbol> asSignature(symbol: S): KtSignature<S> {
+        return when (symbol) {
+            is KtFunctionLikeSymbol -> asSignature(symbol)
+            is KtVariableLikeSymbol -> asSignature(symbol)
+            else -> unexpectedElementError("symbol", symbol)
+        }
+    }
+
+    override fun <S : KtFunctionLikeSymbol> asSignature(symbol: S): KtFunctionLikeSignature<S> {
+        return KtFunctionLikeSignature(symbol, symbol.returnType, symbol.receiverType, symbol.valueParameters.map { asSignature(it) })
+    }
+
+    override fun <S : KtVariableLikeSymbol> asSignature(symbol: S): KtVariableLikeSignature<S> {
+        return KtVariableLikeSignature(symbol, symbol.returnType, symbol.receiverType)
+    }
+}