From 980a24e95527b69a8d37cb39c5e412908b7a8898 Mon Sep 17 00:00:00 2001
From: Karol Lewandowski <karol.lewandowski@jetbrains.com>
Date: Tue, 17 Dec 2024 13:44:31 +0100
Subject: [PATCH] reference_contributor.md: Use PsiPolyVariantReferenceBase

---
 .../sdk/language/SimpleReference.java         | 22 ++++++++-----------
 1 file changed, 9 insertions(+), 13 deletions(-)

diff --git a/simple_language_plugin/src/main/java/org/intellij/sdk/language/SimpleReference.java b/simple_language_plugin/src/main/java/org/intellij/sdk/language/SimpleReference.java
index ec49b96e6..6c64acae8 100644
--- a/simple_language_plugin/src/main/java/org/intellij/sdk/language/SimpleReference.java
+++ b/simple_language_plugin/src/main/java/org/intellij/sdk/language/SimpleReference.java
@@ -6,27 +6,30 @@ import com.intellij.codeInsight.lookup.LookupElement;
 import com.intellij.codeInsight.lookup.LookupElementBuilder;
 import com.intellij.openapi.project.Project;
 import com.intellij.openapi.util.TextRange;
-import com.intellij.psi.*;
+import com.intellij.psi.PsiElement;
+import com.intellij.psi.PsiElementResolveResult;
+import com.intellij.psi.PsiPolyVariantReferenceBase;
+import com.intellij.psi.ResolveResult;
 import org.intellij.sdk.language.psi.SimpleProperty;
 import org.jetbrains.annotations.NotNull;
-import org.jetbrains.annotations.Nullable;
 
 import java.util.ArrayList;
 import java.util.List;
 
-final class SimpleReference extends PsiReferenceBase<PsiElement> implements PsiPolyVariantReference {
+final class SimpleReference extends PsiPolyVariantReferenceBase<PsiElement> {
 
   private final String key;
 
   SimpleReference(@NotNull PsiElement element, TextRange textRange) {
     super(element, textRange);
-    key = element.getText().substring(textRange.getStartOffset(), textRange.getEndOffset());
+    key = element.getText()
+        .substring(textRange.getStartOffset(), textRange.getEndOffset());
   }
 
   @Override
   public ResolveResult @NotNull [] multiResolve(boolean incompleteCode) {
     Project project = myElement.getProject();
-    final List<SimpleProperty> properties = SimpleUtil.findProperties(project, key);
+    List<SimpleProperty> properties = SimpleUtil.findProperties(project, key);
     List<ResolveResult> results = new ArrayList<>();
     for (SimpleProperty property : properties) {
       results.add(new PsiElementResolveResult(property));
@@ -34,19 +37,12 @@ final class SimpleReference extends PsiReferenceBase<PsiElement> implements PsiP
     return results.toArray(new ResolveResult[0]);
   }
 
-  @Nullable
-  @Override
-  public PsiElement resolve() {
-    ResolveResult[] resolveResults = multiResolve(false);
-    return resolveResults.length == 1 ? resolveResults[0].getElement() : null;
-  }
-
   @Override
   public Object @NotNull [] getVariants() {
     Project project = myElement.getProject();
     List<SimpleProperty> properties = SimpleUtil.findProperties(project);
     List<LookupElement> variants = new ArrayList<>();
-    for (final SimpleProperty property : properties) {
+    for (SimpleProperty property : properties) {
       if (property.getKey() != null && !property.getKey().isEmpty()) {
         variants.add(LookupElementBuilder
             .create(property).withIcon(SimpleIcons.FILE)