ReplacementPerformer: fix potential PIEAE

This commit is contained in:
Dmitry Gridin
2020-04-09 17:04:08 +07:00
parent 76a573921e
commit 91700c9d2c
@@ -63,11 +63,12 @@ internal class AnnotationEntryReplacementPerformer(
codeToInline: MutableCodeToInline,
elementToBeReplaced: KtAnnotationEntry
) : AbstractSimpleReplacementPerformer<KtAnnotationEntry>(codeToInline, elementToBeReplaced) {
private val useSiteTarget = elementToBeReplaced.useSiteTarget?.getAnnotationUseSiteTarget()
override fun createDummyElement(mainExpression: KtExpression): KtAnnotationEntry =
createByPattern("@Dummy($0)", mainExpression) { psiFactory.createAnnotationEntry(it) }
override fun rangeToElement(range: PsiChildRange): KtAnnotationEntry {
val useSiteTarget = elementToBeReplaced.useSiteTarget?.getAnnotationUseSiteTarget()
val useSiteTargetText = useSiteTarget?.renderName?.let { "$it:" } ?: ""
val isFileUseSiteTarget = useSiteTarget == AnnotationUseSiteTarget.FILE