From c379fd7fa1e8d779a84b7865ffa2478c90f7f5b2 Mon Sep 17 00:00:00 2001
From: Sergej Jaskiewicz <sergej.jaskiewicz@jetbrains.com>
Date: Thu, 14 Mar 2024 18:24:57 +0100
Subject: [PATCH] [tree generator] Allow adding kDocs to implementation classes

---
 .../kotlin/generators/tree/AbstractImplementation.kt          | 2 ++
 .../kotlin/generators/tree/AbstractImplementationPrinter.kt   | 2 ++
 .../tree/config/AbstractImplementationConfigurator.kt         | 4 ++++
 3 files changed, 8 insertions(+)

diff --git a/generators/tree-generator-common/src/org/jetbrains/kotlin/generators/tree/AbstractImplementation.kt b/generators/tree-generator-common/src/org/jetbrains/kotlin/generators/tree/AbstractImplementation.kt
index 5cc65432883..acac023f788 100644
--- a/generators/tree-generator-common/src/org/jetbrains/kotlin/generators/tree/AbstractImplementation.kt
+++ b/generators/tree-generator-common/src/org/jetbrains/kotlin/generators/tree/AbstractImplementation.kt
@@ -48,6 +48,8 @@ abstract class AbstractImplementation<Implementation, Element, Field>(
      */
     val additionalImports = mutableListOf<Importable>()
 
+    var kDoc: String? = null
+
     init {
         @Suppress("UNCHECKED_CAST")
         element.implementations += this as Implementation
diff --git a/generators/tree-generator-common/src/org/jetbrains/kotlin/generators/tree/AbstractImplementationPrinter.kt b/generators/tree-generator-common/src/org/jetbrains/kotlin/generators/tree/AbstractImplementationPrinter.kt
index 8862ba6932b..54f2af938c0 100644
--- a/generators/tree-generator-common/src/org/jetbrains/kotlin/generators/tree/AbstractImplementationPrinter.kt
+++ b/generators/tree-generator-common/src/org/jetbrains/kotlin/generators/tree/AbstractImplementationPrinter.kt
@@ -8,6 +8,7 @@ package org.jetbrains.kotlin.generators.tree
 import org.jetbrains.kotlin.descriptors.Modality
 import org.jetbrains.kotlin.generators.tree.printer.braces
 import org.jetbrains.kotlin.generators.tree.printer.printBlock
+import org.jetbrains.kotlin.generators.tree.printer.printKDoc
 import org.jetbrains.kotlin.generators.tree.printer.typeParameters
 import org.jetbrains.kotlin.utils.SmartPrinter
 import org.jetbrains.kotlin.utils.addToStdlib.ifNotEmpty
@@ -37,6 +38,7 @@ abstract class AbstractImplementationPrinter<Implementation, Element, Implementa
     fun printImplementation(implementation: Implementation) {
         addAllImports(implementation.additionalImports)
         printer.run {
+            printKDoc(implementation.kDoc)
             buildSet {
                 if (implementation.requiresOptIn) {
                     add(implementationOptInAnnotation)
diff --git a/generators/tree-generator-common/src/org/jetbrains/kotlin/generators/tree/config/AbstractImplementationConfigurator.kt b/generators/tree-generator-common/src/org/jetbrains/kotlin/generators/tree/config/AbstractImplementationConfigurator.kt
index ca2cff0185b..7854ee438af 100644
--- a/generators/tree-generator-common/src/org/jetbrains/kotlin/generators/tree/config/AbstractImplementationConfigurator.kt
+++ b/generators/tree-generator-common/src/org/jetbrains/kotlin/generators/tree/config/AbstractImplementationConfigurator.kt
@@ -142,6 +142,10 @@ abstract class AbstractImplementationConfigurator<Implementation, Element, Imple
             implementation.isPublic = true
         }
 
+        fun kDoc(kDoc: String) {
+            implementation.kDoc = kDoc
+        }
+
         private fun getField(name: String): ImplementationField {
             return implementation.getField(name)
         }