New J2K: Add support of enum consts initializingClass

j2k/newSrc/org/jetbrains/kotlin/j2k/JavaToJKTreeBuilder.kt

@@ -321,6 +321,7 @@ class JavaToJKTreeBuilder(var symbolProvider: JKSymbolProvider) {
             JKEnumConstantImpl(
                 JKNameIdentifierImpl(name),
                 with(expressionTreeMapper) { argumentList.toJK() },
+                initializingClass?.createClassBody() ?: JKEmptyClassBodyImpl(),
                 JKTypeElementImpl(JKClassTypeImpl(symbolProvider.provideDirectSymbol(containingClass!!) as JKClassSymbol, emptyList()))
             )
 

j2k/newSrc/org/jetbrains/kotlin/j2k/NewCodeBuilder.kt

@@ -253,6 +253,9 @@ class NewCodeBuilder {
                     enumConstant.arguments.accept(this)
                 }
             }
+            if (enumConstant.body !is JKEmptyClassBody) {
+                enumConstant.body.accept(this)
+            }
         }
 
         override fun visitKtInitDeclaration(ktInitDeclaration: JKKtInitDeclaration) {

j2k/newSrc/org/jetbrains/kotlin/j2k/tree/impl/jk.kt

@@ -460,10 +460,12 @@ class JKTypeParameterTypeImpl(
 class JKEnumConstantImpl(
     name: JKNameIdentifier,
     arguments: JKExpressionList,
+    body: JKClassBody,
     type: JKTypeElement
 ) : JKEnumConstant, JKBranchElementBase(), PsiOwner by PsiOwnerImpl() {
     override var name: JKNameIdentifier by child(name)
     override val arguments: JKExpressionList by child(arguments)
+    override val body: JKClassBody by child(body)
     override var type: JKTypeElement by child(type)
     override var initializer: JKExpression by child(JKStubExpressionImpl())
 

j2k/newSrc/org/jetbrains/kotlin/j2k/tree/jk.kt

@@ -383,6 +383,7 @@ interface JKTypeParameterListOwner : JKTreeElement {
 
 interface JKEnumConstant : JKVariable {
     val arguments: JKExpressionList
+    val body: JKClassBody
 }
 
 interface JKForInStatement : JKStatement {