diff --git a/analysis/symbol-light-classes/src/org/jetbrains/kotlin/light/classes/symbol/annotations/SimpleAdditionalAnnotationsProvider.kt b/analysis/symbol-light-classes/src/org/jetbrains/kotlin/light/classes/symbol/annotations/SimpleAdditionalAnnotationsProvider.kt new file mode 100644 index 00000000000..1bd24966af2 --- /dev/null +++ b/analysis/symbol-light-classes/src/org/jetbrains/kotlin/light/classes/symbol/annotations/SimpleAdditionalAnnotationsProvider.kt @@ -0,0 +1,35 @@ +/* + * 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.light.classes.symbol.annotations + +import com.intellij.psi.PsiAnnotation +import com.intellij.psi.PsiModifierList + +internal class SimpleAdditionalAnnotationsProvider( + private val additionalAnnotationQualifiers: Collection, +) : AdditionalAnnotationsProvider { + constructor(qualifiedName: String) : this(setOf(qualifiedName)) + + override fun addAllAnnotations( + currentRawAnnotations: MutableList, + foundQualifiers: MutableSet, + owner: PsiModifierList + ) { + for (qualifiedName in additionalAnnotationQualifiers) { + addSimpleAnnotationIfMissing(qualifiedName, currentRawAnnotations, foundQualifiers, owner) + } + } + + override fun findAdditionalAnnotation( + annotationsBox: LazyAnnotationsBox, + qualifiedName: String, + owner: PsiModifierList + ): PsiAnnotation? { + if (qualifiedName !in additionalAnnotationQualifiers) return null + + return createSimpleAnnotationIfMatches(qualifiedName, qualifiedName, owner) + } +} \ No newline at end of file