From c5fd248e69a364f02b7137c37b512457abf87984 Mon Sep 17 00:00:00 2001
From: Dmitriy Novozhilov <dmitriy.novozhilov@jetbrains.com>
Date: Tue, 20 Sep 2022 11:48:20 +0300
Subject: [PATCH] [FIR] Add pretty toString to ResolutionMode

---
 .../kotlin/fir/resolve/ResolutionMode.kt      | 63 +++++++++++++++----
 1 file changed, 51 insertions(+), 12 deletions(-)

diff --git a/compiler/fir/resolve/src/org/jetbrains/kotlin/fir/resolve/ResolutionMode.kt b/compiler/fir/resolve/src/org/jetbrains/kotlin/fir/resolve/ResolutionMode.kt
index ee879a212ec..6a5457893a1 100644
--- a/compiler/fir/resolve/src/org/jetbrains/kotlin/fir/resolve/ResolutionMode.kt
+++ b/compiler/fir/resolve/src/org/jetbrains/kotlin/fir/resolve/ResolutionMode.kt
@@ -6,31 +6,58 @@
 package org.jetbrains.kotlin.fir.resolve
 
 import org.jetbrains.kotlin.fir.declarations.FirDeclarationStatus
-import org.jetbrains.kotlin.fir.types.ConeKotlinType
-import org.jetbrains.kotlin.fir.types.FirResolvedTypeRef
-import org.jetbrains.kotlin.fir.types.FirTypeRef
+import org.jetbrains.kotlin.fir.render
+import org.jetbrains.kotlin.fir.resolve.ResolutionMode.Companion.prettyString
+import org.jetbrains.kotlin.fir.types.*
 import org.jetbrains.kotlin.fir.types.builder.buildResolvedTypeRef
 
 sealed class ResolutionMode {
-    object ContextDependent : ResolutionMode()
-    object ContextDependentDelegate : ResolutionMode()
-    object ContextIndependent : ResolutionMode()
-    object ReceiverResolution : ResolutionMode()
+    object ContextDependent : ResolutionMode() {
+        override fun toString(): String = "ContextDependent"
+    }
+
+    object ContextDependentDelegate : ResolutionMode() {
+        override fun toString(): String = "ContextDependentDelegate"
+    }
+
+    object ContextIndependent : ResolutionMode() {
+        override fun toString(): String = "ContextIndependent"
+    }
+
+    object ReceiverResolution : ResolutionMode() {
+        override fun toString(): String = "ReceiverResolution"
+    }
 
     // TODO: it's better not to use WithExpectedType(FirImplicitTypeRef)
     class WithExpectedType(
         val expectedTypeRef: FirTypeRef,
         val mayBeCoercionToUnitApplied: Boolean = false,
         val expectedTypeMismatchIsReportedInChecker: Boolean = false,
-    ) : ResolutionMode()
+    ) : ResolutionMode() {
+        override fun toString(): String {
+            return "WithExpectedType: ${expectedTypeRef.prettyString()}"
+        }
+    }
 
-    class WithStatus(val status: FirDeclarationStatus) : ResolutionMode()
+    class WithStatus(val status: FirDeclarationStatus) : ResolutionMode() {
+        override fun toString(): String {
+            return "WithStatus: ${status.render()}"
+        }
+    }
 
-    class LambdaResolution(val expectedReturnTypeRef: FirResolvedTypeRef?) : ResolutionMode()
+    class LambdaResolution(val expectedReturnTypeRef: FirResolvedTypeRef?) : ResolutionMode() {
+        override fun toString(): String {
+            return "LambdaResolution: ${expectedReturnTypeRef.prettyString()}"
+        }
+    }
 
     class WithExpectedTypeFromCast(
         val expectedTypeRef: FirTypeRef,
-    ) : ResolutionMode()
+    ) : ResolutionMode() {
+        override fun toString(): String {
+            return "WithExpectedTypeFromCast: ${expectedTypeRef.prettyString()}"
+        }
+    }
 
     /**
      * This resolution mode is similar to
@@ -53,7 +80,19 @@ sealed class ResolutionMode {
      */
     class WithSuggestedType(
         val suggestedTypeRef: FirTypeRef,
-    ) : ResolutionMode()
+    ) : ResolutionMode() {
+        override fun toString(): String {
+            return "WithSuggestedType: ${suggestedTypeRef.prettyString()}"
+        }
+    }
+
+    private companion object {
+        private fun FirTypeRef?.prettyString(): String {
+            if (this == null) return "null"
+            val coneType = this.coneTypeSafe<ConeKotlinType>() ?: return this.render()
+            return coneType.renderForDebugging()
+        }
+    }
 }
 
 fun ResolutionMode.expectedType(components: BodyResolveComponents, allowFromCast: Boolean = false): FirTypeRef? = when (this) {