From 07be2ae5c702167e013bb16ded005a3ffe4efab9 Mon Sep 17 00:00:00 2001
From: Denis Zharkov <denis.zharkov@jetbrains.com>
Date: Thu, 7 May 2020 16:03:18 +0300
Subject: [PATCH] FIR: Use containers structure in
 FirContractResolveTransformer

transformDeclarationContent call remembers the container
---
 .../transformers/contracts/FirContractResolveTransformer.kt   | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/compiler/fir/resolve/src/org/jetbrains/kotlin/fir/resolve/transformers/contracts/FirContractResolveTransformer.kt b/compiler/fir/resolve/src/org/jetbrains/kotlin/fir/resolve/transformers/contracts/FirContractResolveTransformer.kt
index c0e156a70d2..26c2f176444 100644
--- a/compiler/fir/resolve/src/org/jetbrains/kotlin/fir/resolve/transformers/contracts/FirContractResolveTransformer.kt
+++ b/compiler/fir/resolve/src/org/jetbrains/kotlin/fir/resolve/transformers/contracts/FirContractResolveTransformer.kt
@@ -166,7 +166,9 @@ class FirContractResolveTransformer(
             context.storeClass(regularClass)
             regularClass.transformCompanionObject(this, data)
             withTypeParametersOf(regularClass) {
-                regularClass.transformDeclarations(this, data)
+                context.withContainer(regularClass) {
+                    regularClass.transformDeclarations(this, data)
+                }
             }
             return regularClass.compose()
         }