Support interpretation of trimMargin and trimIndent

compiler/ir/ir.interpreter/src/org/jetbrains/kotlin/ir/interpreter/intrinsics/IntrinsicEvaluator.kt

@@ -15,7 +15,7 @@ internal object IntrinsicEvaluator {
     private val fqNameToHandler: Map<String, IntrinsicBase> = buildMap {
         listOf(
             EmptyArray, ArrayOf, ArrayOfNulls, ArrayConstructor, EnumValues, EnumValueOf,
-            JsPrimitives, SourceLocation, AssertIntrinsic, DataClassArrayToString
+            JsPrimitives, SourceLocation, AssertIntrinsic, DataClassArrayToString, Indent
         ).forEach { intrinsic -> intrinsic.getListOfAcceptableFunctions().forEach { put(it, intrinsic) } }
     }
 

compiler/ir/ir.interpreter/src/org/jetbrains/kotlin/ir/interpreter/intrinsics/IntrinsicImplementations.kt

@@ -12,6 +12,7 @@ import org.jetbrains.kotlin.ir.interpreter.createGetValue
 import org.jetbrains.kotlin.ir.interpreter.toIrConst
 import org.jetbrains.kotlin.ir.interpreter.exceptions.handleUserException
 import org.jetbrains.kotlin.ir.interpreter.exceptions.stop
+import org.jetbrains.kotlin.ir.interpreter.exceptions.withExceptionHandler
 import org.jetbrains.kotlin.ir.interpreter.state.*
 import org.jetbrains.kotlin.ir.interpreter.state.reflection.KFunctionState
 import org.jetbrains.kotlin.ir.interpreter.state.reflection.KPropertyState
@@ -31,8 +32,10 @@ internal sealed class IntrinsicBase {
 
     fun customEvaluateInstruction(irFunction: IrFunction, environment: IrInterpreterEnvironment): CustomInstruction {
         return CustomInstruction {
-            evaluate(irFunction, environment)
+            withExceptionHandler(environment) { // Exception handling is used only for indent actions; TODO: drop later
-            environment.callStack.dropFrameAndCopyResult()
+                evaluate(irFunction, environment)
+                environment.callStack.dropFrameAndCopyResult()
+            }
         }
     }
 }
@@ -340,4 +343,28 @@ internal object DataClassArrayToString : IntrinsicBase() {
         val array = environment.callStack.loadState(irFunction.valueParameters.single().symbol) as Primitive<*>
         environment.callStack.pushState(environment.convertToState(arrayToString(array.value), irFunction.returnType))
     }
-}
+}
+
+internal object Indent : IntrinsicBase() {
+    override fun getListOfAcceptableFunctions(): List<String> {
+        return listOf(
+            "kotlin.text.StringsKt.trimIndent", "kotlin.text.trimIndent",
+            "kotlin.text.StringsKt.trimMargin", "kotlin.text.trimMargin",
+            "kotlin.text.StringsKt.trimMargin\$default", "kotlin.text.trimMargin\$default",
+        )
+    }
+
+    override fun evaluate(irFunction: IrFunction, environment: IrInterpreterEnvironment) {
+        val str = environment.callStack.loadState(irFunction.getExtensionReceiver()!!).asString()
+        val trimmed = when (irFunction.fqName) {
+            "kotlin.text.StringsKt.trimIndent", "kotlin.text.trimIndent" -> str.trimIndent()
+            "kotlin.text.StringsKt.trimMargin", "kotlin.text.trimMargin" -> {
+                val marginPrefix = environment.callStack.loadState(irFunction.valueParameters.single().symbol).asString()
+                str.trimMargin(marginPrefix)
+            }
+            "kotlin.text.StringsKt.trimMargin\$default", "kotlin.text.trimMargin\$default" -> str.trimMargin()
+            else -> TODO("unknown trim function")
+        }
+        environment.callStack.pushState(environment.convertToState(trimmed, irFunction.returnType))
+    }
+}