Forbid manipulations with FirWrappedArgumentExpression type reference

The type of this expression is bound to the type of wrapped expression,
so it's better not to change it directly.

compiler/fir/resolve/src/org/jetbrains/kotlin/fir/resolve/transformers/FirBodyResolveTransformer.kt

@@ -937,7 +937,7 @@ open class FirBodyResolveTransformer(val session: FirSession, val implicitTypeOn
     }
 
     override fun transformExpression(expression: FirExpression, data: Any?): CompositeTransformResult<FirStatement> {
-        if (expression.resultType is FirImplicitTypeRef) {
+        if (expression.resultType is FirImplicitTypeRef && expression !is FirWrappedArgumentExpression) {
             val type = FirErrorTypeRefImpl(session, expression.psi, "Type calculating for ${expression::class} is not supported")
             expression.resultType = type
         }

compiler/fir/tree/src/org/jetbrains/kotlin/fir/expressions/FirWrappedArgumentExpression.kt

@@ -17,6 +17,10 @@ interface FirWrappedArgumentExpression : FirExpression {
     override val typeRef: FirTypeRef
         get() = expression.typeRef
 
+    override fun replaceTypeRef(newTypeRef: FirTypeRef) {
+        throw AssertionError("We should not try to replace type reference in ${this::class}")
+    }
+
     override fun <R, D> accept(visitor: FirVisitor<R, D>, data: D): R =
         visitor.visitWrappedArgumentExpression(this, data)
 

compiler/fir/tree/src/org/jetbrains/kotlin/fir/expressions/impl/FirLambdaArgumentExpressionImpl.kt

@@ -11,20 +11,15 @@ import org.jetbrains.kotlin.fir.FirSession
 import org.jetbrains.kotlin.fir.expressions.FirExpression
 import org.jetbrains.kotlin.fir.expressions.FirLambdaArgumentExpression
 import org.jetbrains.kotlin.fir.transformSingle
-import org.jetbrains.kotlin.fir.types.FirTypeRef
 import org.jetbrains.kotlin.fir.visitors.FirTransformer
 
 class FirLambdaArgumentExpressionImpl(
     session: FirSession,
     psi: PsiElement?,
     override var expression: FirExpression
-) : FirLambdaArgumentExpression, FirAbstractExpression(session, psi) {
+) : FirLambdaArgumentExpression, FirAbstractStatement(session, psi) {
-    override var typeRef: FirTypeRef
-        get() = super<FirLambdaArgumentExpression>.typeRef
-        set(_) {}
-
     override fun <D> transformChildren(transformer: FirTransformer<D>, data: D): FirElement {
         expression = expression.transformSingle(transformer, data)
-        return super<FirAbstractExpression>.transformChildren(transformer, data)
+        return super<FirAbstractStatement>.transformChildren(transformer, data)
     }
 }

compiler/fir/tree/src/org/jetbrains/kotlin/fir/expressions/impl/FirNamedArgumentExpressionImpl.kt

@@ -11,7 +11,6 @@ import org.jetbrains.kotlin.fir.FirSession
 import org.jetbrains.kotlin.fir.expressions.FirExpression
 import org.jetbrains.kotlin.fir.expressions.FirNamedArgumentExpression
 import org.jetbrains.kotlin.fir.transformSingle
-import org.jetbrains.kotlin.fir.types.FirTypeRef
 import org.jetbrains.kotlin.fir.visitors.FirTransformer
 import org.jetbrains.kotlin.name.Name
 
@@ -21,13 +20,9 @@ class FirNamedArgumentExpressionImpl(
     override val name: Name,
     override val isSpread: Boolean,
     override var expression: FirExpression
-) : FirNamedArgumentExpression, FirAbstractExpression(session, psi) {
+) : FirNamedArgumentExpression, FirAbstractStatement(session, psi) {
-    override var typeRef: FirTypeRef
-        get() = super<FirNamedArgumentExpression>.typeRef
-        set(_) {}
-
     override fun <D> transformChildren(transformer: FirTransformer<D>, data: D): FirElement {
         expression = expression.transformSingle(transformer, data)
-        return super<FirAbstractExpression>.transformChildren(transformer, data)
+        return super<FirAbstractStatement>.transformChildren(transformer, data)
     }
 }

compiler/fir/tree/src/org/jetbrains/kotlin/fir/expressions/impl/FirSpreadArgumentExpressionImpl.kt

@@ -11,20 +11,15 @@ import org.jetbrains.kotlin.fir.FirSession
 import org.jetbrains.kotlin.fir.expressions.FirExpression
 import org.jetbrains.kotlin.fir.expressions.FirSpreadArgumentExpression
 import org.jetbrains.kotlin.fir.transformSingle
-import org.jetbrains.kotlin.fir.types.FirTypeRef
 import org.jetbrains.kotlin.fir.visitors.FirTransformer
 
 class FirSpreadArgumentExpressionImpl(
     session: FirSession,
     psi: PsiElement?,
     override var expression: FirExpression
-) : FirSpreadArgumentExpression, FirAbstractExpression(session, psi) {
+) : FirSpreadArgumentExpression, FirAbstractStatement(session, psi) {
-    override var typeRef: FirTypeRef
-        get() = super<FirSpreadArgumentExpression>.typeRef
-        set(_) {}
-
     override fun <D> transformChildren(transformer: FirTransformer<D>, data: D): FirElement {
         expression = expression.transformSingle(transformer, data)
-        return super<FirAbstractExpression>.transformChildren(transformer, data)
+        return super<FirAbstractStatement>.transformChildren(transformer, data)
     }
 }