[FIR] Add ENUM_JVM_RECORD diagnostic

compiler/fir/checkers/checkers-component-generator/src/org/jetbrains/kotlin/fir/checkers/generator/diagnostics/FirJvmDiagnosticsList.kt

@@ -64,6 +64,7 @@ object JVM_DIAGNOSTICS_LIST : DiagnosticList("FirJvmErrors") {
     val RECORDS by object : DiagnosticGroup("JVM Records") {
         val LOCAL_JVM_RECORD by error<PsiElement>()
         val NON_FINAL_JVM_RECORD by error<PsiElement>(PositioningStrategy.NON_FINAL_MODIFIER_OR_NAME)
+        val ENUM_JVM_RECORD by error<PsiElement>(PositioningStrategy.ENUM_MODIFIER)
     }
 
 }

compiler/fir/checkers/checkers-component-generator/src/org/jetbrains/kotlin/fir/checkers/generator/diagnostics/model/RegularDiagnosticData.kt

@@ -105,7 +105,7 @@ enum class PositioningStrategy(private val strategy: String? = null) {
     COMMAS,
     OPERATOR_MODIFIER,
     NON_FINAL_MODIFIER_OR_NAME,
-
+    ENUM_MODIFIER,
     ;
 
     val expressionToCreate get() = "SourceElementPositioningStrategies.${strategy ?: name}"

compiler/fir/checkers/checkers.jvm/gen/org/jetbrains/kotlin/fir/analysis/diagnostics/jvm/FirJvmErrors.kt

@@ -53,5 +53,6 @@ object FirJvmErrors {
     // JVM Records
     val LOCAL_JVM_RECORD by error0<PsiElement>()
     val NON_FINAL_JVM_RECORD by error0<PsiElement>(SourceElementPositioningStrategies.NON_FINAL_MODIFIER_OR_NAME)
+    val ENUM_JVM_RECORD by error0<PsiElement>(SourceElementPositioningStrategies.ENUM_MODIFIER)
 
 }

compiler/fir/checkers/checkers.jvm/src/org/jetbrains/kotlin/fir/analysis/jvm/checkers/declaration/FirJvmRecordChecker.kt

@@ -12,6 +12,7 @@ import org.jetbrains.kotlin.fir.analysis.diagnostics.jvm.FirJvmErrors
 import org.jetbrains.kotlin.fir.analysis.diagnostics.reportOn
 import org.jetbrains.kotlin.fir.declarations.FirRegularClass
 import org.jetbrains.kotlin.fir.declarations.getAnnotationByFqName
+import org.jetbrains.kotlin.fir.declarations.utils.isEnumClass
 import org.jetbrains.kotlin.fir.declarations.utils.isFinal
 import org.jetbrains.kotlin.fir.declarations.utils.isLocal
 import org.jetbrains.kotlin.name.FqName
@@ -30,6 +31,12 @@ object FirJvmRecordChecker : FirRegularClassChecker() {
 
         if (!declaration.isFinal) {
             reporter.reportOn(declaration.source, FirJvmErrors.NON_FINAL_JVM_RECORD, context)
+            return
+        }
+
+        if (declaration.isEnumClass) {
+            reporter.reportOn(declaration.source, FirJvmErrors.ENUM_JVM_RECORD, context)
+            return
         }
     }
 }

compiler/fir/checkers/src/org/jetbrains/kotlin/fir/analysis/diagnostics/LightTreePositioningStrategies.kt

@@ -376,6 +376,9 @@ object LightTreePositioningStrategies {
     val OPERATOR_MODIFIER: LightTreePositioningStrategy =
         ModifierSetBasedLightTreePositioningStrategy(TokenSet.create(KtTokens.OPERATOR_KEYWORD))
 
+    val ENUM_MODIFIER: LightTreePositioningStrategy =
+        ModifierSetBasedLightTreePositioningStrategy(TokenSet.create(KtTokens.ENUM_KEYWORD))
+
     val INLINE_PARAMETER_MODIFIER: LightTreePositioningStrategy =
         ModifierSetBasedLightTreePositioningStrategy(TokenSet.create(KtTokens.NOINLINE_KEYWORD, KtTokens.CROSSINLINE_KEYWORD))
 

compiler/fir/checkers/src/org/jetbrains/kotlin/fir/analysis/diagnostics/SourceElementPositioningStrategies.kt

@@ -314,4 +314,9 @@ object SourceElementPositioningStrategies {
         LightTreePositioningStrategies.NON_FINAL_MODIFIER_OR_NAME,
         PositioningStrategies.NON_FINAL_MODIFIER_OR_NAME
     )
+
+    val ENUM_MODIFIER = SourceElementPositioningStrategy(
+        LightTreePositioningStrategies.ENUM_MODIFIER,
+        PositioningStrategies.ENUM_MODIFIER
+    )
 }

compiler/frontend/src/org/jetbrains/kotlin/diagnostics/PositioningStrategies.kt

@@ -380,6 +380,9 @@ object PositioningStrategies {
     @JvmField
     val OPERATOR_MODIFIER: PositioningStrategy<KtModifierListOwner> = modifierSetPosition(KtTokens.OPERATOR_KEYWORD)
 
+    @JvmField
+    val ENUM_MODIFIER: PositioningStrategy<KtModifierListOwner> = modifierSetPosition(KtTokens.ENUM_KEYWORD)
+
     @JvmField
     val FOR_REDECLARATION: PositioningStrategy<PsiElement> = object : PositioningStrategy<PsiElement>() {
         override fun mark(element: PsiElement): List<TextRange> {

compiler/testData/diagnostics/tests/testsWithJava15/jvmRecord/diagnostics.fir.kt

@@ -34,7 +34,7 @@ class A5(vararg val name: String, y: Int)
 <!NON_FINAL_JVM_RECORD!>sealed<!> class A8(val x: String)
 
 @JvmRecord
-enum class A9(val x: String) {
+<!ENUM_JVM_RECORD!>enum<!> class A9(val x: String) {
     X("");
 }
 

idea/idea-frontend-fir/src/org/jetbrains/kotlin/idea/frontend/api/fir/diagnostics/KtFirDataClassConverters.kt

@@ -3550,4 +3550,10 @@ internal val KT_DIAGNOSTIC_CONVERTER = KtDiagnosticConverterBuilder.buildConvert
             token,
         )
     }
+    add(FirJvmErrors.ENUM_JVM_RECORD) { firDiagnostic ->
+        EnumJvmRecordImpl(
+            firDiagnostic as FirPsiDiagnostic,
+            token,
+        )
+    }
 }

idea/idea-frontend-fir/src/org/jetbrains/kotlin/idea/frontend/api/fir/diagnostics/KtFirDiagnostics.kt

@@ -2473,4 +2473,8 @@ sealed class KtFirDiagnostic<PSI : PsiElement> : KtDiagnosticWithPsi<PSI> {
         override val diagnosticClass get() = NonFinalJvmRecord::class
     }
 
+    abstract class EnumJvmRecord : KtFirDiagnostic<PsiElement>() {
+        override val diagnosticClass get() = EnumJvmRecord::class
+    }
+
 }

idea/idea-frontend-fir/src/org/jetbrains/kotlin/idea/frontend/api/fir/diagnostics/KtFirDiagnosticsImpl.kt

@@ -4007,3 +4007,10 @@ internal class NonFinalJvmRecordImpl(
     override val firDiagnostic: FirPsiDiagnostic by weakRef(firDiagnostic)
 }
 
+internal class EnumJvmRecordImpl(
+    firDiagnostic: FirPsiDiagnostic,
+    override val token: ValidityToken,
+) : KtFirDiagnostic.EnumJvmRecord(), KtAbstractFirDiagnostic<PsiElement> {
+    override val firDiagnostic: FirPsiDiagnostic by weakRef(firDiagnostic)
+}
+