[FIR] In tree generator make it possible to add custom super types

...to element.

MR: KT-MR-12245

^KT-60671

compiler/fir/tree/tree-generator/src/org/jetbrains/kotlin/fir/tree/generator/NodeConfigurator.kt

@@ -707,6 +707,7 @@ object NodeConfigurator : AbstractFieldConfigurator<FirTreeBuilder>(FirTreeBuild
         resolvedTypeRef.configure {
             +field("type", coneKotlinTypeType)
             +field("delegatedTypeRef", typeRef, nullable = true)
+            // TODO: add supertype in subsequent commits
         }
 
         typeRefWithNullability.configure {

compiler/fir/tree/tree-generator/src/org/jetbrains/kotlin/fir/tree/generator/model/Element.kt

@@ -40,6 +40,7 @@ class Element(override val name: String, kind: Kind) : AbstractElement<Element,
     override val packageName: String = BASE_PACKAGE + kind.packageName.let { if (it.isBlank()) it else "." + it }
     override val fullQualifiedName: String get() = super.fullQualifiedName!!
     override val parentRefs = mutableListOf<ElementOrRef<Element, Field>>()
+    val additionalSupertypeInterfaces = mutableListOf<TypeRef>()
     var defaultImplementation: Implementation? = null
     val customImplementations = mutableListOf<Implementation>()
 

compiler/fir/tree/tree-generator/src/org/jetbrains/kotlin/fir/tree/generator/printer/element.kt

@@ -55,25 +55,24 @@ fun SmartPrinter.printElement(element: Element) {
         print("${kind!!.title} $type")
         print(typeParameters())
         val needPureAbstractElement = !isInterface && !allParents.any { it.kind == ImplementationKind.AbstractClass || it.kind == ImplementationKind.SealedClass }
+        val superTypesStrings = parentRefs.map {
+            // TODO: Factor out
+            var result = it.element.type
+            if (it.args.isNotEmpty()) {
+                result += it.args.values.joinToString(", ", "<", ">") { it.typeWithArguments }
+            }
+            result + it.element.kind.braces()
+        } + additionalSupertypeInterfaces.map { it.type }
 
-        if (parentRefs.isNotEmpty() || needPureAbstractElement) {
+        if (superTypesStrings.isNotEmpty() || needPureAbstractElement) {
             print(" : ")
             if (needPureAbstractElement) {
                 print("${pureAbstractElementType.type}()")
-                if (parentRefs.isNotEmpty()) {
+                if (superTypesStrings.isNotEmpty()) {
                     print(", ")
                 }
             }
-            print(
+            print(superTypesStrings.joinToString(", "),)
-                parentRefs.joinToString(", ") {
-                    // TODO: Factor out
-                    var result = it.element.type
-                    if (it.args.isNotEmpty()) {
-                        result += it.args.values.joinToString(", ", "<", ">") { it.typeWithArguments }
-                    }
-                    result + it.element.kind.braces()
-                },
-            )
         }
         print(multipleUpperBoundsList())
         println(" {")

compiler/fir/tree/tree-generator/src/org/jetbrains/kotlin/fir/tree/generator/printer/utils.kt

@@ -67,6 +67,7 @@ fun Element.collectImports(): List<String> {
     if (needPureAbstractElement) {
         baseTypes += pureAbstractElementType.fullQualifiedName
     }
+    baseTypes.addAll(additionalSupertypeInterfaces.map { it.fullQualifiedName!! })
     return collectImportsInternal(
         baseTypes,
         ImportKind.Element,