From 1ce2fdbcb5195732b98d94dace64a8db2ba18e27 Mon Sep 17 00:00:00 2001
From: Bas Leijdekkers <basleijdekkers@gmail.com>
Date: Mon, 7 Oct 2024 15:34:17 +0200
Subject: [PATCH] Java: presentation provider for record components
 (IDEA-360262)

GitOrigin-RevId: b3fef1fe5d58659efea5b21eb458d7c2880ba9a7
---
 .../src/META-INF/JavaPsiPlugin.xml            |  2 ++
 .../impl/source/PsiRecordComponentImpl.java   |  9 ++++-
 .../java/JavaPresentationUtil.java            | 36 +++++++++++++------
 .../RecordComponentPresentationProvider.java  | 15 ++++++++
 4 files changed, 51 insertions(+), 11 deletions(-)
 create mode 100644 java/java-psi-impl/src/com/intellij/psi/presentation/java/RecordComponentPresentationProvider.java

diff --git a/java/java-psi-impl/src/META-INF/JavaPsiPlugin.xml b/java/java-psi-impl/src/META-INF/JavaPsiPlugin.xml
index 689285b36292..c289322ff862 100644
--- a/java/java-psi-impl/src/META-INF/JavaPsiPlugin.xml
+++ b/java/java-psi-impl/src/META-INF/JavaPsiPlugin.xml
@@ -110,6 +110,8 @@
                               implementationClass="com.intellij.psi.presentation.java.VariablePresentationProvider"/>
     <itemPresentationProvider forClass="com.intellij.psi.PsiField"
                               implementationClass="com.intellij.psi.presentation.java.FieldPresentationProvider"/>
+    <itemPresentationProvider forClass="com.intellij.psi.PsiRecordComponent"
+                              implementationClass="com.intellij.psi.presentation.java.RecordComponentPresentationProvider"/>
     <itemPresentationProvider forClass="com.intellij.psi.PsiMethod"
                               implementationClass="com.intellij.psi.presentation.java.MethodPresentationProvider"/>
     <itemPresentationProvider forClass="com.intellij.psi.PsiClass"
diff --git a/java/java-psi-impl/src/com/intellij/psi/impl/source/PsiRecordComponentImpl.java b/java/java-psi-impl/src/com/intellij/psi/impl/source/PsiRecordComponentImpl.java
index 5cffc724bdd0..ba18ae4540c4 100644
--- a/java/java-psi-impl/src/com/intellij/psi/impl/source/PsiRecordComponentImpl.java
+++ b/java/java-psi-impl/src/com/intellij/psi/impl/source/PsiRecordComponentImpl.java
@@ -1,7 +1,9 @@
-// Copyright 2000-2023 JetBrains s.r.o. and contributors. Use of this source code is governed by the Apache 2.0 license.
+// Copyright 2000-2024 JetBrains s.r.o. and contributors. Use of this source code is governed by the Apache 2.0 license.
 package com.intellij.psi.impl.source;
 
 import com.intellij.lang.ASTNode;
+import com.intellij.navigation.ItemPresentation;
+import com.intellij.navigation.ItemPresentationProviders;
 import com.intellij.psi.*;
 import com.intellij.psi.impl.CheckUtil;
 import com.intellij.psi.impl.ElementPresentationUtil;
@@ -159,6 +161,11 @@ public final class PsiRecordComponentImpl extends JavaStubPsiElement<PsiRecordCo
       .createLayeredIcon(this, IconManager.getInstance().getPlatformIcon(PlatformIcons.Field), ElementPresentationUtil.getFlags(this, false));
   }
 
+  @Override
+  public ItemPresentation getPresentation() {
+    return ItemPresentationProviders.getItemPresentation(this);
+  }
+
   @Override
   public @NotNull SearchScope getUseScope() {
     return PsiImplUtil.getMemberUseScope(this);
diff --git a/java/java-psi-impl/src/com/intellij/psi/presentation/java/JavaPresentationUtil.java b/java/java-psi-impl/src/com/intellij/psi/presentation/java/JavaPresentationUtil.java
index 84b307690576..4e8d2d2d0b70 100644
--- a/java/java-psi-impl/src/com/intellij/psi/presentation/java/JavaPresentationUtil.java
+++ b/java/java-psi-impl/src/com/intellij/psi/presentation/java/JavaPresentationUtil.java
@@ -88,18 +88,37 @@ public final class JavaPresentationUtil {
     };
   }
 
+  public static @NotNull ItemPresentation getRecordComponentPresentation(final @NotNull PsiRecordComponent component) {
+    return new ColoredItemPresentation() {
+      @Override
+      public String getPresentableText() {
+        return component.getName();
+      }
+
+      @Override
+      public TextAttributesKey getTextAttributesKey() {
+        return null;
+      }
+
+      @Override
+      public String getLocationString() {
+        return getJavaSymbolContainerText(component);
+      }
+
+      @Override
+      public Icon getIcon(boolean open) {
+        return component.getIcon(Iconable.ICON_FLAG_VISIBILITY);
+      }
+    };
+  }
+
   private static @Nullable String getJavaSymbolContainerText(final @NotNull PsiElement element) {
     final String result;
     PsiElement container = PsiTreeUtil.getParentOfType(element, PsiMember.class, PsiFile.class);
 
     if (container instanceof PsiClass) {
       String qName = ((PsiClass)container).getQualifiedName();
-      if (qName != null) {
-        result = qName;
-      }
-      else {
-        result = ((PsiClass)container).getName();
-      }
+      result = (qName != null) ? qName : ((PsiClass)container).getName();
     }
     else if (container instanceof PsiJavaFile) {
       result = ((PsiJavaFile)container).getPackageName();
@@ -107,9 +126,6 @@ public final class JavaPresentationUtil {
     else {//TODO: local classes
       result = null;
     }
-    if (result != null) {
-      return JavaPsiBundle.message("aux.context.display", result);
-    }
-    return null;
+    return result == null ? null : JavaPsiBundle.message("aux.context.display", result);
   }
 }
diff --git a/java/java-psi-impl/src/com/intellij/psi/presentation/java/RecordComponentPresentationProvider.java b/java/java-psi-impl/src/com/intellij/psi/presentation/java/RecordComponentPresentationProvider.java
new file mode 100644
index 000000000000..45499a6410e7
--- /dev/null
+++ b/java/java-psi-impl/src/com/intellij/psi/presentation/java/RecordComponentPresentationProvider.java
@@ -0,0 +1,15 @@
+// Copyright 2000-2024 JetBrains s.r.o. and contributors. Use of this source code is governed by the Apache 2.0 license.
+package com.intellij.psi.presentation.java;
+
+import com.intellij.navigation.ItemPresentation;
+import com.intellij.navigation.ItemPresentationProvider;
+import com.intellij.psi.PsiRecordComponent;
+import org.jetbrains.annotations.NotNull;
+
+
+public final class RecordComponentPresentationProvider implements ItemPresentationProvider<PsiRecordComponent> {
+  @Override
+  public ItemPresentation getPresentation(@NotNull PsiRecordComponent item) {
+    return JavaPresentationUtil.getRecordComponentPresentation(item);
+  }
+}