From 9fd25af1a77c292a7d91078faa5950004d02cd96 Mon Sep 17 00:00:00 2001
From: Dmitriy Novozhilov <dmitriy.novozhilov@jetbrains.com>
Date: Tue, 25 May 2021 17:11:55 +0300
Subject: [PATCH] [FIR] Add pretty toString to FirSession and ModuleData

---
 .../fir/tree/src/org/jetbrains/kotlin/fir/FirModuleData.kt   | 4 ++++
 compiler/fir/tree/src/org/jetbrains/kotlin/fir/FirSession.kt | 5 +++++
 2 files changed, 9 insertions(+)

diff --git a/compiler/fir/tree/src/org/jetbrains/kotlin/fir/FirModuleData.kt b/compiler/fir/tree/src/org/jetbrains/kotlin/fir/FirModuleData.kt
index e0e40d6b5db..f827c306895 100644
--- a/compiler/fir/tree/src/org/jetbrains/kotlin/fir/FirModuleData.kt
+++ b/compiler/fir/tree/src/org/jetbrains/kotlin/fir/FirModuleData.kt
@@ -60,6 +60,10 @@ abstract class FirModuleData : FirSessionComponent {
         }
         _session = session
     }
+
+    override fun toString(): String {
+        return "Module $name"
+    }
 }
 
 class FirModuleDataImpl(
diff --git a/compiler/fir/tree/src/org/jetbrains/kotlin/fir/FirSession.kt b/compiler/fir/tree/src/org/jetbrains/kotlin/fir/FirSession.kt
index 2edf64a8098..7ec65a7c2b9 100644
--- a/compiler/fir/tree/src/org/jetbrains/kotlin/fir/FirSession.kt
+++ b/compiler/fir/tree/src/org/jetbrains/kotlin/fir/FirSession.kt
@@ -35,6 +35,11 @@ abstract class FirSession @PrivateSessionConstructor constructor(
     fun register(tClass: KClass<out FirSessionComponent>, value: FirSessionComponent) {
         registerComponent(tClass, value)
     }
+
+    override fun toString(): String {
+        val moduleData = nullableModuleData ?: return "Libraries session"
+        return "Source session for module ${moduleData.name}"
+    }
 }
 
 abstract class FirSessionProvider {