|
| 1 | +/* |
| 2 | + * Copyright MapStruct Authors. |
| 3 | + * |
| 4 | + * Licensed under the Apache License version 2.0, available at http://www.apache.org/licenses/LICENSE-2.0 |
| 5 | + */ |
| 6 | +package org.mapstruct.intellij.codeinsight.references; |
| 7 | + |
| 8 | +import com.intellij.psi.PsiReferenceContributor; |
| 9 | +import com.intellij.psi.PsiReferenceRegistrar; |
| 10 | +import org.jetbrains.annotations.NotNull; |
| 11 | + |
| 12 | +import static org.mapstruct.intellij.util.MapstructKotlinElementUtils.mappingElementPattern; |
| 13 | +import static org.mapstruct.intellij.util.MapstructKotlinElementUtils.valueMappingElementPattern; |
| 14 | + |
| 15 | +/** |
| 16 | + * {@link PsiReferenceContributor} for MapStruct annotations in Kotlin code. |
| 17 | + * |
| 18 | + * @author Frank Wang |
| 19 | + * @author Filip Hrisafov |
| 20 | + */ |
| 21 | +public class MapstructKotlinReferenceContributor extends PsiReferenceContributor { |
| 22 | + |
| 23 | + @Override |
| 24 | + public void registerReferenceProviders(@NotNull PsiReferenceRegistrar registrar) { |
| 25 | + registrar.registerReferenceProvider( |
| 26 | + mappingElementPattern( "target" ), |
| 27 | + new MappingTargetReferenceProvider( MapstructTargetReference::create ) |
| 28 | + ); |
| 29 | + registrar.registerReferenceProvider( |
| 30 | + mappingElementPattern( "source" ), |
| 31 | + new MappingTargetReferenceProvider( MapstructSourceReference::create ) |
| 32 | + ); |
| 33 | + registrar.registerReferenceProvider( |
| 34 | + valueMappingElementPattern( "target" ), |
| 35 | + new MappingTargetReferenceProvider( ValueMappingSourceReference::create ) |
| 36 | + ); |
| 37 | + registrar.registerReferenceProvider( |
| 38 | + valueMappingElementPattern( "source" ), |
| 39 | + new MappingTargetReferenceProvider( ValueMappingTargetReference::create ) |
| 40 | + ); |
| 41 | + } |
| 42 | + |
| 43 | +} |
0 commit comments