diff --git a/src/main/java/org/mapstruct/intellij/util/patterns/KotlinElementPattern.java b/src/main/java/org/mapstruct/intellij/util/patterns/KotlinElementPattern.java index 9c2a1112..7102e2ee 100644 --- a/src/main/java/org/mapstruct/intellij/util/patterns/KotlinElementPattern.java +++ b/src/main/java/org/mapstruct/intellij/util/patterns/KotlinElementPattern.java @@ -6,9 +6,10 @@ package org.mapstruct.intellij.util.patterns; import com.intellij.patterns.ElementPattern; +import com.intellij.patterns.PlatformPatterns; import com.intellij.patterns.PsiElementPattern; import com.intellij.psi.PsiElement; -import org.jetbrains.kotlin.psi.stubs.elements.KtStubElementTypes; +import org.jetbrains.kotlin.psi.KtStringTemplateExpression; import static org.mapstruct.intellij.util.patterns.MapStructKotlinPatterns.ktAnnotation; import static org.mapstruct.intellij.util.patterns.MapStructKotlinPatterns.ktValueArgument; @@ -36,7 +37,7 @@ public Self insideRepeatableAnnotationParam( // @Mapping(target = "name") KtValueArgumentPattern ktValueArgumentPattern = ktValueArgument().withName( parameterName ); - return withElementType( KtStubElementTypes.STRING_TEMPLATE ).andOr( + return and( PlatformPatterns.psiElement( KtStringTemplateExpression.class ) ).andOr( withParent( ktValueArgumentPattern .withAncestor( 5, ktAnnotation().qName( annotationHolderQualifiedName ) ) @@ -54,7 +55,7 @@ public Self insideAnnotationParam( String parameterName) { KtValueArgumentPattern ktValueArgumentPattern = ktValueArgument().withName( parameterName ); - return withElementType( KtStubElementTypes.STRING_TEMPLATE ).andOr( + return and( PlatformPatterns.psiElement( KtStringTemplateExpression.class ) ).andOr( withAncestor( 2, ktValueArgumentPattern