[FIR] calculate java Target annotation on COMPILER_REQUIRED_ANNOTATIONS phase

^KT-57849 ^KT-57949
2023-04-11 18:34:33 +02:00
parent 1c456b216b
commit 7b91cd0ff3
11 changed files with 90 additions and 4 deletions
@@ -0,0 +1,25 @@
+/*
+ * Copyright 2010-2023 JetBrains s.r.o. and Kotlin Programming Language contributors.
+ * Use of this source code is governed by the Apache 2.0 license that can be found in the license/LICENSE.txt file.
+ */
+
+package org.jetbrains.kotlin.fir.java.enhancement
+
+import org.jetbrains.kotlin.fir.FirSession
+import org.jetbrains.kotlin.fir.declarations.FirRegularClass
+import org.jetbrains.kotlin.fir.resolve.transformers.plugin.CompilerRequiredAnnotationEnhancementProvider
+import org.jetbrains.kotlin.fir.symbols.impl.FirClassSymbol
+import org.jetbrains.kotlin.fir.symbols.impl.FirEnumEntrySymbol
+import org.jetbrains.kotlin.name.StandardClassIds
+
+object JavaCompilerRequiredAnnotationEnhancementProvider : CompilerRequiredAnnotationEnhancementProvider() {
+    override fun enhance(enumSymbol: FirClassSymbol<*>, enumEntrySymbol: FirEnumEntrySymbol, session: FirSession): FirEnumEntrySymbol {
+        val firRegularClass = enumSymbol.fir as? FirRegularClass ?: return enumEntrySymbol
+        if (enumSymbol.classId != StandardClassIds.Annotations.Java.ElementType) return enumEntrySymbol
+
+        return FirSignatureEnhancement(firRegularClass, session) { emptyList() }.enhancedProperty(
+            enumEntrySymbol,
+            enumEntrySymbol.name,
+        ) as FirEnumEntrySymbol
+    }
+}