From 19bdf41655f8b36367786776f4cae302980a734f Mon Sep 17 00:00:00 2001
From: Branislav Stojkovic <Branislav Stojkovic@Bancika-PC>
Date: Sat, 9 Feb 2019 10:02:32 +0100
Subject: [PATCH] fix shortcuts

---
 .../diylc/swing/plugins/tree/TreePanel.java   | 23 +++++++++++--------
 1 file changed, 13 insertions(+), 10 deletions(-)

diff --git a/diylc/diylc-swing/src/org/diylc/swing/plugins/tree/TreePanel.java b/diylc/diylc-swing/src/org/diylc/swing/plugins/tree/TreePanel.java
index bc6547f4..0c91b107 100644
--- a/diylc/diylc-swing/src/org/diylc/swing/plugins/tree/TreePanel.java
+++ b/diylc/diylc-swing/src/org/diylc/swing/plugins/tree/TreePanel.java
@@ -659,7 +659,7 @@ public class TreePanel extends JPanel {
           }
         }
         
-        setText("<html>" + payload.toString() + shortCutHtml + variantsHtml + "</html>");
+        setText("<html>" + payload.forDisplay() + shortCutHtml + variantsHtml + "</html>");
       }
 
       return this;
@@ -704,20 +704,23 @@ public class TreePanel extends JPanel {
       return clickListener;
     }
     
-    private String categoryForDisplay() {
-      String display = category;
-      Matcher match = contributedPattern.matcher(display);
-      if (match.find()) {
-        String name = match.group(1);
-        String owner = match.group(2);
-        display = name + "<font color='gray'>[" + owner + "]</font>"; 
+    public String forDisplay() {
+      if (componentType == null) {
+        String display = category;
+        Matcher match = contributedPattern.matcher(display);
+        if (match.find()) {
+          String name = match.group(1);
+          String owner = match.group(2);
+          display = name + "<font color='gray'>[" + owner + "]</font>"; 
+        }
+        return display;
       }
-      return display;
+      return componentType.getName();
     }
 
     @Override
     public String toString() {
-      return componentType == null ? categoryForDisplay() : componentType.getName();
+      return componentType == null ? category : componentType.getName();
     }
   }