[AA] introduce FirReceiverParameter

^KT-54417

analysis/analysis-api-fir/src/org/jetbrains/kotlin/analysis/api/fir/annotations/KtFirAnnotationListForReceiverParameter.kt

@@ -0,0 +1,66 @@
+/*
+ * 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.fir.annotations
+
+import org.jetbrains.kotlin.analysis.api.annotations.KtAnnotationApplication
+import org.jetbrains.kotlin.analysis.api.annotations.KtAnnotationsList
+import org.jetbrains.kotlin.analysis.api.fir.toKtAnnotationApplication
+import org.jetbrains.kotlin.analysis.api.impl.base.annotations.KtEmptyAnnotationsList
+import org.jetbrains.kotlin.analysis.api.lifetime.KtLifetimeToken
+import org.jetbrains.kotlin.analysis.api.lifetime.withValidityAssertion
+import org.jetbrains.kotlin.fir.FirAnnotationContainer
+import org.jetbrains.kotlin.fir.FirSession
+import org.jetbrains.kotlin.fir.declarations.fullyExpandedClassId
+import org.jetbrains.kotlin.fir.symbols.impl.FirCallableSymbol
+import org.jetbrains.kotlin.fir.symbols.resolvedAnnotationClassIds
+import org.jetbrains.kotlin.fir.symbols.resolvedAnnotationsWithArguments
+import org.jetbrains.kotlin.name.ClassId
+
+internal class KtFirAnnotationListForReceiverParameter private constructor(
+    private val firCallableSymbol: FirCallableSymbol<*>,
+    private val receiverParameter: FirAnnotationContainer,
+    private val useSiteSession: FirSession,
+    override val token: KtLifetimeToken,
+) : KtAnnotationsList() {
+
+    override val annotations: List<KtAnnotationApplication>
+        get() = withValidityAssertion {
+            receiverParameter.resolvedAnnotationsWithArguments(firCallableSymbol).map { annotation ->
+                annotation.toKtAnnotationApplication(useSiteSession)
+            }
+        }
+
+    override fun hasAnnotation(classId: ClassId): Boolean = withValidityAssertion {
+        classId in receiverParameter.resolvedAnnotationClassIds(firCallableSymbol)
+    }
+
+    override fun annotationsByClassId(classId: ClassId): List<KtAnnotationApplication> = withValidityAssertion {
+        receiverParameter.resolvedAnnotationsWithArguments(firCallableSymbol).mapNotNull { annotation ->
+            if (annotation.fullyExpandedClassId(useSiteSession) != classId) return@mapNotNull null
+            annotation.toKtAnnotationApplication(useSiteSession)
+        }
+    }
+
+    override val annotationClassIds: Collection<ClassId>
+        get() = withValidityAssertion {
+            receiverParameter.resolvedAnnotationClassIds(firCallableSymbol)
+        }
+
+    companion object {
+        fun create(
+            firCallableSymbol: FirCallableSymbol<*>,
+            useSiteSession: FirSession,
+            token: KtLifetimeToken,
+        ): KtAnnotationsList {
+            val receiverParameter = firCallableSymbol.receiverParameter
+            return if (receiverParameter?.annotations?.isEmpty() != false) {
+                KtEmptyAnnotationsList(token)
+            } else {
+                KtFirAnnotationListForReceiverParameter(firCallableSymbol, receiverParameter, useSiteSession, token)
+            }
+        }
+    }
+}

analysis/analysis-api-fir/src/org/jetbrains/kotlin/analysis/api/fir/symbols/KtFirReceiverParameterSymbol.kt

@@ -7,7 +7,9 @@ package org.jetbrains.kotlin.analysis.api.fir.symbols
 
 import com.intellij.psi.PsiElement
 import org.jetbrains.kotlin.analysis.api.analyze
+import org.jetbrains.kotlin.analysis.api.annotations.KtAnnotationsList
 import org.jetbrains.kotlin.analysis.api.fir.KtSymbolByFirBuilder
+import org.jetbrains.kotlin.analysis.api.fir.annotations.KtFirAnnotationListForReceiverParameter
 import org.jetbrains.kotlin.analysis.api.fir.findPsi
 import org.jetbrains.kotlin.analysis.api.fir.symbols.pointers.KtFirReceiverParameterSymbolPointer
 import org.jetbrains.kotlin.analysis.api.fir.symbols.pointers.requireOwnerPointer
@@ -46,4 +48,8 @@ internal class KtFirReceiverParameterSymbol(
     override fun createPointer(): KtSymbolPointer<KtReceiverParameterSymbol> = withValidityAssertion {
         KtFirReceiverParameterSymbolPointer(owningCallableSymbol.createPointer())
     }
+
+    override val annotationsList: KtAnnotationsList by cached {
+        KtFirAnnotationListForReceiverParameter.create(firSymbol, firResolveSession.useSiteFirSession, token)
+    }
 }

analysis/analysis-api-fir/src/org/jetbrains/kotlin/analysis/api/fir/symbols/KtFirTypeAndAnnotations.kt

@@ -1,5 +1,5 @@
 /*
- * Copyright 2010-2020 JetBrains s.r.o. and Kotlin Programming Language contributors.
+ * 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.
  */
 
@@ -12,6 +12,7 @@ import org.jetbrains.kotlin.fir.resolve.transformers.resolveSupertypesInTheAir
 import org.jetbrains.kotlin.fir.symbols.impl.FirCallableSymbol
 import org.jetbrains.kotlin.fir.symbols.impl.FirClassSymbol
 import org.jetbrains.kotlin.fir.symbols.impl.FirRegularClassSymbol
+import org.jetbrains.kotlin.fir.types.FirResolvedTypeRef
 import org.jetbrains.kotlin.fir.types.FirTypeRef
 
 

analysis/analysis-api/src/org/jetbrains/kotlin/analysis/api/symbols/KtCallableSymbol.kt

@@ -1,11 +1,12 @@
 /*
- * Copyright 2010-2020 JetBrains s.r.o. and Kotlin Programming Language contributors.
+ * 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.symbols
 
 import org.jetbrains.kotlin.analysis.api.base.KtContextReceiversOwner
+import org.jetbrains.kotlin.analysis.api.symbols.markers.KtAnnotatedSymbol
 import org.jetbrains.kotlin.analysis.api.symbols.markers.KtPossibleMemberSymbol
 import org.jetbrains.kotlin.analysis.api.symbols.markers.KtSymbolWithKind
 import org.jetbrains.kotlin.analysis.api.symbols.pointers.KtSymbolPointer
@@ -26,7 +27,7 @@ public sealed class KtCallableSymbol : KtSymbolWithKind, KtPossibleMemberSymbol,
  * Symbol for a receiver parameter of a function or property. For example, consider code `fun String.foo() {...}`, the declaration of
  * `String` receiver parameter is such a symbol.
  */
-public abstract class KtReceiverParameterSymbol : KtSymbol {
+public abstract class KtReceiverParameterSymbol : KtAnnotatedSymbol {
     public abstract val type: KtType
 
     /**

analysis/analysis-api/testData/components/callResolver/resolveCall/builderInference.descriptors.txt

@@ -4,6 +4,7 @@ KtSuccessCallInfo:
     partiallyAppliedSymbol = KtPartiallyAppliedSymbol:
       dispatchReceiver = KtImplicitReceiverValue:
         symbol = KtReceiverParameterSymbol:
+          annotationsList: []
           origin: SOURCE
           owningCallableSymbol: KtAnonymousFunctionSymbol(<local>/<no name provided>)
           type: test/Target<kotlin/String>

analysis/analysis-api/testData/components/callResolver/resolveCall/builderInference.txt

@@ -4,6 +4,7 @@ KtSuccessCallInfo:
     partiallyAppliedSymbol = KtPartiallyAppliedSymbol:
       dispatchReceiver = KtImplicitReceiverValue:
         symbol = KtReceiverParameterSymbol:
+          annotationsList: []
           origin: SOURCE
           owningCallableSymbol: KtAnonymousFunctionSymbol(<local>/<no name provided>)
           type: test/Target<kotlin/String>
@@ -30,4 +31,4 @@ KtSuccessCallInfo:
         returnType = T
         symbol = t: T
         callableIdIfNonLocal = null)
-    }
+    }

analysis/analysis-api/testData/components/callResolver/resolveCall/javaPropertyGetter_unqualified.txt

@@ -3,6 +3,7 @@ KtSuccessCallInfo:
     partiallyAppliedSymbol = KtPartiallyAppliedSymbol:
       dispatchReceiver = KtImplicitReceiverValue:
         symbol = KtReceiverParameterSymbol:
+          annotationsList: []
           origin: SOURCE
           owningCallableSymbol: KtFunctionSymbol(/call)
           type: JavaClass
@@ -16,4 +17,4 @@ KtSuccessCallInfo:
         callableIdIfNonLocal = /JavaClass.foo
     simpleAccess = Read:
 
-    typeArgumentsMapping = {}
+    typeArgumentsMapping = {}

analysis/analysis-api/testData/components/callResolver/resolveCall/javaPropertySetter_unqualified.txt

@@ -3,6 +3,7 @@ KtSuccessCallInfo:
     partiallyAppliedSymbol = KtPartiallyAppliedSymbol:
       dispatchReceiver = KtImplicitReceiverValue:
         symbol = KtReceiverParameterSymbol:
+          annotationsList: []
           origin: SOURCE
           owningCallableSymbol: KtFunctionSymbol(/call)
           type: JavaClass
@@ -16,4 +17,4 @@ KtSuccessCallInfo:
         callableIdIfNonLocal = /JavaClass.foo
     simpleAccess = Write:
       value = 42
-    typeArgumentsMapping = {}
+    typeArgumentsMapping = {}

analysis/analysis-api/testData/components/callResolver/resolveCall/kotlinPropertyGetter_unqualified.txt

@@ -3,6 +3,7 @@ KtSuccessCallInfo:
     partiallyAppliedSymbol = KtPartiallyAppliedSymbol:
       dispatchReceiver = KtImplicitReceiverValue:
         symbol = KtReceiverParameterSymbol:
+          annotationsList: []
           origin: SOURCE
           owningCallableSymbol: KtFunctionSymbol(/foo)
           type: A
@@ -16,4 +17,4 @@ KtSuccessCallInfo:
         callableIdIfNonLocal = /A.i
     simpleAccess = Read:
 
-    typeArgumentsMapping = {}
+    typeArgumentsMapping = {}

analysis/analysis-api/testData/components/callResolver/resolveCall/kotlinPropertySetter_unqualified.txt

@@ -3,6 +3,7 @@ KtSuccessCallInfo:
     partiallyAppliedSymbol = KtPartiallyAppliedSymbol:
       dispatchReceiver = KtImplicitReceiverValue:
         symbol = KtReceiverParameterSymbol:
+          annotationsList: []
           origin: SOURCE
           owningCallableSymbol: KtFunctionSymbol(/foo)
           type: A
@@ -16,4 +17,4 @@ KtSuccessCallInfo:
         callableIdIfNonLocal = /A.i
     simpleAccess = Write:
       value = 1
-    typeArgumentsMapping = {}
+    typeArgumentsMapping = {}

analysis/analysis-api/testData/components/callResolver/resolveCall/smartCastImplicitDispatchReceiver.txt

@@ -4,6 +4,7 @@ KtSuccessCallInfo:
       dispatchReceiver = KtSmartCastedReceiverValue:
         original = KtImplicitReceiverValue:
           symbol = KtReceiverParameterSymbol:
+            annotationsList: []
             origin: SOURCE
             owningCallableSymbol: KtFunctionSymbol(/test)
             type: kotlin/Any
@@ -18,4 +19,4 @@ KtSuccessCallInfo:
         callableIdIfNonLocal = kotlin/String.length
     simpleAccess = Read:
 
-    typeArgumentsMapping = {}
+    typeArgumentsMapping = {}

analysis/analysis-api/testData/components/callResolver/resolveCall/smartCastImplicitExtensionReceiver.txt

@@ -6,6 +6,7 @@ KtSuccessCallInfo:
       extensionReceiver = KtSmartCastedReceiverValue:
         original = KtImplicitReceiverValue:
           symbol = KtReceiverParameterSymbol:
+            annotationsList: []
             origin: SOURCE
             owningCallableSymbol: KtFunctionSymbol(/test)
             type: kotlin/Any
@@ -18,4 +19,4 @@ KtSuccessCallInfo:
         valueParameters = []
         callableIdIfNonLocal = /foo
     typeArgumentsMapping = {}
-    argumentMapping = {}
+    argumentMapping = {}

analysis/analysis-api/testData/components/callResolver/resolveCandidates/singleCandidate/builderInference.descriptors.txt

@@ -4,6 +4,7 @@ KtApplicableCallCandidateInfo:
     partiallyAppliedSymbol = KtPartiallyAppliedSymbol:
       dispatchReceiver = KtImplicitReceiverValue:
         symbol = KtReceiverParameterSymbol:
+          annotationsList: []
           origin: SOURCE
           owningCallableSymbol: KtAnonymousFunctionSymbol(<local>/<no name provided>)
           type: test/Target<kotlin/String>

analysis/analysis-api/testData/components/callResolver/resolveCandidates/singleCandidate/builderInference.txt

@@ -4,6 +4,7 @@ KtApplicableCallCandidateInfo:
     partiallyAppliedSymbol = KtPartiallyAppliedSymbol:
       dispatchReceiver = KtImplicitReceiverValue:
         symbol = KtReceiverParameterSymbol:
+          annotationsList: []
           origin: SOURCE
           owningCallableSymbol: KtAnonymousFunctionSymbol(<local>/<no name provided>)
           type: test/Target<kotlin/String>

analysis/analysis-api/testData/symbols/symbolByReference/functionReceiverParameter.pretty.txt

@@ -1,4 +1,8 @@
 KtReceiverParameterSymbol:
+  annotationsList: [
+    ReceiverAnnotation()
+      psi: KtAnnotationEntry
+  ]
   origin: SOURCE
   owningCallableSymbol: KtFunctionSymbol(/foo)
   type: [

analysis/analysis-api/testData/symbols/symbolByReference/functionReceiverParameter.txt

@@ -1,4 +1,8 @@
 KtReceiverParameterSymbol:
+  annotationsList: [
+    ReceiverAnnotation()
+      psi: KtAnnotationEntry
+  ]
   origin: SOURCE
   owningCallableSymbol: KtFunctionSymbol(/foo)
   type: [

analysis/analysis-api/testData/symbols/symbolByReference/propertyReceiverParameter.pretty.txt

@@ -1,4 +1,8 @@
 KtReceiverParameterSymbol:
+  annotationsList: [
+    ReceiverAnnotation()
+      psi: KtAnnotationEntry
+  ]
   origin: SOURCE
   owningCallableSymbol: KtKotlinPropertySymbol(/prop)
   type: [

analysis/analysis-api/testData/symbols/symbolByReference/propertyReceiverParameter.txt

@@ -1,4 +1,8 @@
 KtReceiverParameterSymbol:
+  annotationsList: [
+    ReceiverAnnotation()
+      psi: KtAnnotationEntry
+  ]
   origin: SOURCE
   owningCallableSymbol: KtKotlinPropertySymbol(/prop)
   type: [

compiler/fir/checkers/checkers.native/src/org/jetbrains/kotlin/fir/analysis/native/checkers/FirNativeObjCNameChecker.kt

@@ -125,6 +125,7 @@ object FirNativeObjCNameChecker : FirBasicDeclarationChecker() {
         is FirFunctionSymbol<*> -> buildList {
             add((this@getObjCNames as FirBasedSymbol<*>).getObjCName())
             add(resolvedReceiverTypeRef?.getObjCName())
+            add(receiverParameter?.getObjCName())
             valueParameterSymbols.forEach { add(it.getObjCName()) }
         }
 

compiler/fir/tree/src/org/jetbrains/kotlin/fir/symbols/FirBasedSymbol.kt

@@ -1,11 +1,12 @@
 /*
- * Copyright 2010-2018 JetBrains s.r.o. and Kotlin Programming Language contributors.
+ * 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.fir.symbols
 
 import org.jetbrains.kotlin.KtSourceElement
+import org.jetbrains.kotlin.fir.FirAnnotationContainer
 import org.jetbrains.kotlin.fir.FirModuleData
 import org.jetbrains.kotlin.fir.declarations.FirDeclaration
 import org.jetbrains.kotlin.fir.declarations.FirDeclarationOrigin
@@ -42,22 +43,28 @@ abstract class FirBasedSymbol<E : FirDeclaration> {
         get() = fir.annotations
 
     val resolvedAnnotationsWithArguments: List<FirAnnotation>
-        get() {
+        get() = fir.resolvedAnnotationsWithArguments(this)
-            lazyResolveToPhase(FirResolvePhase.ANNOTATIONS_ARGUMENTS_MAPPING)
-            return fir.annotations
-        }
 
     val resolvedAnnotationsWithClassIds: List<FirAnnotation>
-        get() {
+        get() = fir.resolvedAnnotationsWithClassIds(this)
-            lazyResolveToPhase(FirResolvePhase.TYPES)
-            return fir.annotations
-        }
 
     val resolvedAnnotationClassIds: List<ClassId>
-        get() {
+        get() = fir.resolvedAnnotationClassIds(this)
-            lazyResolveToPhase(FirResolvePhase.TYPES)
+}
-            return fir.annotations.mapNotNull { (it.annotationTypeRef.coneType as? ConeClassLikeType)?.lookupTag?.classId }
+
-        }
+fun FirAnnotationContainer.resolvedAnnotationsWithArguments(anchorElement: FirBasedSymbol<*>): List<FirAnnotation> {
+    anchorElement.lazyResolveToPhase(FirResolvePhase.ANNOTATIONS_ARGUMENTS_MAPPING)
+    return annotations
+}
+
+fun FirAnnotationContainer.resolvedAnnotationsWithClassIds(anchorElement: FirBasedSymbol<*>): List<FirAnnotation> {
+    anchorElement.lazyResolveToPhase(FirResolvePhase.TYPES)
+    return annotations
+}
+
+fun FirAnnotationContainer.resolvedAnnotationClassIds(anchorElement: FirBasedSymbol<*>): List<ClassId> {
+    anchorElement.lazyResolveToPhase(FirResolvePhase.TYPES)
+    return annotations.mapNotNull { (it.annotationTypeRef.coneType as? ConeClassLikeType)?.lookupTag?.classId }
 }
 
 @RequiresOptIn

compiler/fir/tree/src/org/jetbrains/kotlin/fir/symbols/impl/FirCallableSymbol.kt

@@ -1,11 +1,12 @@
 /*
- * Copyright 2010-2019 JetBrains s.r.o. and Kotlin Programming Language contributors.
+ * 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.fir.symbols.impl
 
 import org.jetbrains.kotlin.config.ApiVersion
+import org.jetbrains.kotlin.fir.FirAnnotationContainer
 import org.jetbrains.kotlin.fir.declarations.*
 import org.jetbrains.kotlin.fir.symbols.FirBasedSymbol
 import org.jetbrains.kotlin.fir.symbols.lazyResolveToPhase
@@ -32,6 +33,12 @@ abstract class FirCallableSymbol<D : FirCallableDeclaration> : FirBasedSymbol<D>
             return fir.receiverParameter?.type as FirResolvedTypeRef?
         }
 
+    val receiverParameter: FirAnnotationContainer?
+        get() {
+            ensureType(fir.receiverParameter?.type)
+            return fir.receiverParameter
+        }
+
     val resolvedContextReceivers: List<FirContextReceiver>
         get() {
             lazyResolveToPhase(FirResolvePhase.TYPES)