[FIR] Add opt-in to replaceIsTrailingLambda

#KT-66553

generators/tree-generator-common/src/org/jetbrains/kotlin/generators/tree/AbstractField.kt

@@ -27,6 +27,7 @@ abstract class AbstractField<Field : AbstractField<Field>> {
     open val arbitraryImportables: MutableList<Importable> = mutableListOf()
 
     open var optInAnnotation: ClassRef<*>? = null
+    open var replaceOptInAnnotation: ClassRef<*>? = null
 
     abstract var isMutable: Boolean
     open val withGetter: Boolean get() = false
@@ -102,6 +103,7 @@ abstract class AbstractField<Field : AbstractField<Field>> {
         copy.isLateinit = isLateinit
         copy.arbitraryImportables += arbitraryImportables
         copy.optInAnnotation = optInAnnotation
+        copy.replaceOptInAnnotation = replaceOptInAnnotation
         copy.isMutable = isMutable
         copy.deprecation = deprecation
         copy.visibility = visibility

generators/tree-generator-common/src/org/jetbrains/kotlin/generators/tree/printer/printUtils.kt

@@ -117,7 +117,12 @@ fun SmartPrinter.printFunctionDeclaration(
     override: Boolean = false,
     isInline: Boolean = false,
     allParametersOnSeparateLines: Boolean = false,
+    optInAnnotation: ClassRef<*>? = null,
 ) {
+    optInAnnotation?.let {
+        println("@", it.render())
+    }
+
     if (visibility != Visibility.PUBLIC) {
         print(visibility.name.toLowerCaseAsciiOnly(), " ")
     }