From ace2a645351b0c835c37463556daaa855da5ba37 Mon Sep 17 00:00:00 2001
From: Anna Kozlova <anna.kozlova@jetbrains.com>
Date: Wed, 3 Jul 2019 20:58:56 +0200
Subject: [PATCH] method chain inlays: don't repeat same types to reduce visual
 noise

GitOrigin-RevId: da0a4065ec33f4c2b2530c640c7d945dba5cefd7
---
 .../intellij/codeInsight/hints/MethodChainsInlayProvider.kt | 6 +++---
 .../java/codeInsight/daemon/inlays/MethodChainHintsTest.kt  | 2 +-
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/java/java-impl/src/com/intellij/codeInsight/hints/MethodChainsInlayProvider.kt b/java/java-impl/src/com/intellij/codeInsight/hints/MethodChainsInlayProvider.kt
index aaf9e20b8f7c..d1c7c03053b4 100644
--- a/java/java-impl/src/com/intellij/codeInsight/hints/MethodChainsInlayProvider.kt
+++ b/java/java-impl/src/com/intellij/codeInsight/hints/MethodChainsInlayProvider.kt
@@ -13,9 +13,6 @@ import com.intellij.util.ui.JBUI
 import com.siyeh.ig.psiutils.ExpressionUtils
 import org.intellij.lang.annotations.Language
 import javax.swing.JPanel
-import javax.swing.event.ChangeEvent
-import javax.swing.event.DocumentEvent
-import javax.swing.event.DocumentListener
 
 class MethodChainsInlayProvider : InlayHintsProvider<MethodChainsInlayProvider.Settings> {
   override fun getCollectorFor(file: PsiFile, editor: Editor, settings: Settings, sink: InlayHintsSink) =
@@ -41,8 +38,11 @@ class MethodChainsInlayProvider : InlayHintsProvider<MethodChainsInlayProvider.S
         }
         if (uniqueTypes.size < settings.uniqueTypeCount) return true // to hide hints for builders, where type is obvious
         val javaFactory = JavaTypeHintsPresentationFactory(factory, 3)
+        var lastType : PsiType? = null
         for ((index, currentCall) in chain.withIndex()) {
           val type = types[index]
+          if (lastType == type) continue
+          lastType = type
           val presentation = javaFactory.typeHint(type)
           val project = file.project
           val finalPresentation = InsetPresentation(MenuOnClickPresentation(presentation, project) {
diff --git a/java/java-tests/testSrc/com/intellij/java/codeInsight/daemon/inlays/MethodChainHintsTest.kt b/java/java-tests/testSrc/com/intellij/java/codeInsight/daemon/inlays/MethodChainHintsTest.kt
index 75569a9f183d..4f900c13adf1 100644
--- a/java/java-tests/testSrc/com/intellij/java/codeInsight/daemon/inlays/MethodChainHintsTest.kt
+++ b/java/java-tests/testSrc/com/intellij/java/codeInsight/daemon/inlays/MethodChainHintsTest.kt
@@ -74,7 +74,7 @@ public class Chains {
   public static void main(String[] args) {
     new A()
       .b()<hint text="[Chains . B]"/>
-      .c().b()<hint text="[Chains . B]"/>
+      .c().b()
       .a()<hint text="[Chains . A]"/>
       .c()<hint text="[Chains . C]"/>
       .b()<hint text="[Chains . B]"/>