From eecf33a31db1f89b1ea655f835c5d17934c7f23c Mon Sep 17 00:00:00 2001
From: Mikhail Glukhikh <mikhail.glukhikh@jetbrains.com>
Date: Tue, 28 Apr 2020 07:43:34 +0300
Subject: [PATCH] FIR resolve completion: add delegated completion for variable
 access

---
 .../FirCallCompletionResultsWriterTransformer.kt            | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/compiler/fir/resolve/src/org/jetbrains/kotlin/fir/resolve/transformers/FirCallCompletionResultsWriterTransformer.kt b/compiler/fir/resolve/src/org/jetbrains/kotlin/fir/resolve/transformers/FirCallCompletionResultsWriterTransformer.kt
index ece45249e76..54aab3891e2 100644
--- a/compiler/fir/resolve/src/org/jetbrains/kotlin/fir/resolve/transformers/FirCallCompletionResultsWriterTransformer.kt
+++ b/compiler/fir/resolve/src/org/jetbrains/kotlin/fir/resolve/transformers/FirCallCompletionResultsWriterTransformer.kt
@@ -92,6 +92,12 @@ class FirCallCompletionResultsWriterTransformer(
         result.replaceTypeRef(resultType)
         result.replaceTypeArguments(typeArguments)
 
+        if (mode == Mode.DelegatedPropertyCompletion) {
+            candidateFir.transformSingle(declarationWriter, null)
+            val typeUpdater = TypeUpdaterForDelegateArguments()
+            result.transformExplicitReceiver(typeUpdater, null)
+        }
+
         return result.compose()
     }