From 094287741c09cb1c2d4c83bebcef0f2d2f5fdccf Mon Sep 17 00:00:00 2001
From: Dmitriy Novozhilov <dmitriy.novozhilov@jetbrains.com>
Date: Tue, 16 Mar 2021 18:14:58 +0300
Subject: [PATCH] [FIR] Add `FirSourceElement.text` extension for debug
 purposes

---
 .../tree/src/org/jetbrains/kotlin/fir/FirSourceElement.kt  | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/compiler/fir/tree/src/org/jetbrains/kotlin/fir/FirSourceElement.kt b/compiler/fir/tree/src/org/jetbrains/kotlin/fir/FirSourceElement.kt
index e24d42d4043..cb26513641c 100644
--- a/compiler/fir/tree/src/org/jetbrains/kotlin/fir/FirSourceElement.kt
+++ b/compiler/fir/tree/src/org/jetbrains/kotlin/fir/FirSourceElement.kt
@@ -296,6 +296,13 @@ class FirLightSourceElement(
 
 val FirSourceElement?.psi: PsiElement? get() = (this as? FirPsiSourceElement<*>)?.psi
 
+val FirSourceElement?.text: CharSequence?
+    get() = when (this) {
+        is FirPsiSourceElement<*> -> psi.text
+        is FirLightSourceElement -> treeStructure.toString(lighterASTNode)
+        else -> null
+    }
+
 val FirElement.psi: PsiElement? get() = (source as? FirPsiSourceElement<*>)?.psi
 val FirElement.realPsi: PsiElement? get() = (source as? FirRealPsiSourceElement<*>)?.psi