"Add annotation target" quickfix: add SOURCE retention when target is EXPRESSION #KT-26306 Fixed

idea/src/org/jetbrains/kotlin/idea/quickfix/AddAnnotationTargetFix.kt

@@ -119,9 +119,9 @@ private fun PsiAnnotation.getActualTargetList(): List<KotlinTarget> {
 
 private fun KtAnnotationEntry.getActualTargetList(): List<KotlinTarget> {
     val annotatedElement = getStrictParentOfType<KtModifierList>()?.owner as? KtElement
-            ?: getStrictParentOfType<KtAnnotatedExpression>()?.baseExpression
-            ?: getStrictParentOfType<KtFile>()
-            ?: return emptyList()
+        ?: getStrictParentOfType<KtAnnotatedExpression>()?.baseExpression
+        ?: getStrictParentOfType<KtFile>()
+        ?: return emptyList()
 
     val targetList = AnnotationChecker.getActualTargetList(annotatedElement, null, BindingTraceContext().bindingContext)
 
@@ -147,8 +147,17 @@ private fun KtAnnotationEntry.getActualTargetList(): List<KotlinTarget> {
 }
 
 private fun KtClass.addAnnotationTargets(annotationTargets: List<KotlinTarget>, psiFactory: KtPsiFactory) {
-    val targetAnnotationName = KotlinBuiltIns.FQ_NAMES.target.shortName().asString()
+    val retentionAnnotationName = KotlinBuiltIns.FQ_NAMES.retention.shortName().asString()
+    if (annotationTargets.any { it == KotlinTarget.EXPRESSION }
+        && annotationEntries.none { it.typeReference?.text == retentionAnnotationName }) {
+        addAnnotationEntry(
+            psiFactory.createAnnotationEntry(
+                "@$retentionAnnotationName(${KotlinBuiltIns.FQ_NAMES.annotationRetention.shortName()}.${AnnotationRetention.SOURCE.name})"
+            )
+        )
+    }
 
+    val targetAnnotationName = KotlinBuiltIns.FQ_NAMES.target.shortName().asString()
     val targetAnnotationEntry = annotationEntries.find { it.typeReference?.text == targetAnnotationName } ?: run {
         val text = "@$targetAnnotationName${annotationTargets.toArgumentListString()}"
         addAnnotationEntry(psiFactory.createAnnotationEntry(text))