From 9088b673581c8f9ecf2fe55e0f8211f3615e77b6 Mon Sep 17 00:00:00 2001
From: weisj <weisj@arcor.de>
Date: Thu, 26 Mar 2020 17:53:04 +0100
Subject: [PATCH] Fixed tree not being able to start editing.

---
 .../ui/cell/DarkCellRendererToggleButton.java | 10 +++----
 .../ui/combobox/ComboBoxConstants.java        |  7 +++--
 .../weisj/darklaf/ui/list/DarkListUI.java     |  1 +
 .../darklaf/ui/spinner/SpinnerConstants.java  | 11 +++++---
 .../darklaf/ui/table/DarkTableCellEditor.java |  9 +++++--
 .../weisj/darklaf/ui/table/DarkTableUI.java   |  2 ++
 .../weisj/darklaf/ui/text/DarkTextBorder.java | 15 -----------
 .../weisj/darklaf/ui/text/DarkTextUI.java     | 11 +++++---
 .../togglebutton/ToggleButtonConstants.java   | 10 ++++---
 .../ui/tree/DarkDefaultTreeEditor.java        | 14 ++++++++++
 .../darklaf/ui/tree/DarkTreeCellEditor.java   | 16 +++++------
 .../weisj/darklaf/ui/tree/DarkTreeUI.java     | 27 ++++++++++---------
 12 files changed, 77 insertions(+), 56 deletions(-)

diff --git a/core/src/main/java/com/github/weisj/darklaf/ui/cell/DarkCellRendererToggleButton.java b/core/src/main/java/com/github/weisj/darklaf/ui/cell/DarkCellRendererToggleButton.java
index ddec3c78..3b8bb935 100644
--- a/core/src/main/java/com/github/weisj/darklaf/ui/cell/DarkCellRendererToggleButton.java
+++ b/core/src/main/java/com/github/weisj/darklaf/ui/cell/DarkCellRendererToggleButton.java
@@ -27,7 +27,7 @@ import com.github.weisj.darklaf.components.SelectableTreeNode;
 import com.github.weisj.darklaf.decorators.CellRenderer;
 import com.github.weisj.darklaf.ui.table.DarkTableCellFocusBorder;
 import com.github.weisj.darklaf.ui.table.DarkTableUI;
-import com.github.weisj.darklaf.ui.togglebutton.DarkToggleButtonUI;
+import com.github.weisj.darklaf.ui.togglebutton.ToggleButtonConstants;
 import com.github.weisj.darklaf.ui.tree.DarkTreeCellRenderer;
 import com.github.weisj.darklaf.util.DarkUIUtil;
 
@@ -104,8 +104,8 @@ public class DarkCellRendererToggleButton<T extends JToggleButton & CellEditorTo
 
         public CellEditorCheckBox(final boolean opaque) {
             setOpaque(opaque);
-            putClientProperty(DarkToggleButtonUI.KEY_IS_TREE_EDITOR, true);
-            putClientProperty(DarkToggleButtonUI.KEY_IS_TABLE_EDITOR, true);
+            putClientProperty(ToggleButtonConstants.KEY_IS_TREE_EDITOR, true);
+            putClientProperty(ToggleButtonConstants.KEY_IS_TABLE_EDITOR, true);
         }
 
         public void setHasFocus(final boolean hasFocus) {
@@ -129,8 +129,8 @@ public class DarkCellRendererToggleButton<T extends JToggleButton & CellEditorTo
 
         public CellEditorRadioButton(final boolean opaque) {
             setOpaque(opaque);
-            putClientProperty(DarkToggleButtonUI.KEY_IS_TREE_EDITOR, true);
-            putClientProperty(DarkToggleButtonUI.KEY_IS_TABLE_EDITOR, true);
+            putClientProperty(ToggleButtonConstants.KEY_IS_TREE_EDITOR, true);
+            putClientProperty(ToggleButtonConstants.KEY_IS_TABLE_EDITOR, true);
         }
 
         public void setHasFocus(final boolean hasFocus) {
diff --git a/core/src/main/java/com/github/weisj/darklaf/ui/combobox/ComboBoxConstants.java b/core/src/main/java/com/github/weisj/darklaf/ui/combobox/ComboBoxConstants.java
index f3bf0ad6..1518b145 100644
--- a/core/src/main/java/com/github/weisj/darklaf/ui/combobox/ComboBoxConstants.java
+++ b/core/src/main/java/com/github/weisj/darklaf/ui/combobox/ComboBoxConstants.java
@@ -23,12 +23,15 @@
  */
 package com.github.weisj.darklaf.ui.combobox;
 
+import com.github.weisj.darklaf.ui.table.DarkTableUI;
+import com.github.weisj.darklaf.ui.tree.DarkTreeUI;
+
 import javax.swing.*;
 import java.awt.*;
 
 public interface ComboBoxConstants {
-    String KEY_IS_TREE_EDITOR = "JComboBox.isTreeCellEditor";
-    String KEY_IS_TABLE_EDITOR = "JComboBox.isTableCellEditor";
+    String KEY_IS_TREE_EDITOR = DarkTreeUI.KEY_IS_TREE_EDITOR;
+    String KEY_IS_TABLE_EDITOR = DarkTableUI.KEY_IS_TABLE_EDITOR;
 
     static boolean isTreeOrTableCellEditor(final Component c) {
         return isTreeCellEditor(c) || isTableCellEditor(c);
diff --git a/core/src/main/java/com/github/weisj/darklaf/ui/list/DarkListUI.java b/core/src/main/java/com/github/weisj/darklaf/ui/list/DarkListUI.java
index e0181303..7ebefcce 100644
--- a/core/src/main/java/com/github/weisj/darklaf/ui/list/DarkListUI.java
+++ b/core/src/main/java/com/github/weisj/darklaf/ui/list/DarkListUI.java
@@ -45,6 +45,7 @@ public class DarkListUI extends DarkListUIBridge {
     public static final String KEY_IS_EDITING = KEY_PREFIX + "isEditing";
     public static final String RENDER_TYPE_CHECKBOX = "checkBox";
     public static final String RENDER_TYPE_RADIOBUTTON = "radioButton";
+    public static final String KEY_IS_LIST_RENDERER = "JComponent.listCellEditor";
 
     static {
         UIManager.put("List.cellRenderer", new DarkListCellRenderer());
diff --git a/core/src/main/java/com/github/weisj/darklaf/ui/spinner/SpinnerConstants.java b/core/src/main/java/com/github/weisj/darklaf/ui/spinner/SpinnerConstants.java
index 1c93b951..827b9897 100644
--- a/core/src/main/java/com/github/weisj/darklaf/ui/spinner/SpinnerConstants.java
+++ b/core/src/main/java/com/github/weisj/darklaf/ui/spinner/SpinnerConstants.java
@@ -23,15 +23,18 @@
  */
 package com.github.weisj.darklaf.ui.spinner;
 
+import com.github.weisj.darklaf.ui.table.DarkTableUI;
+import com.github.weisj.darklaf.ui.tree.DarkTreeUI;
+
 import javax.swing.*;
 import java.awt.*;
 
 public interface SpinnerConstants {
     String KEY_VARIANT = "JSpinner.variant";
-    String KEY_IS_TREE_EDITOR = "JSpinner.isTreeCellEditor";
-    String KEY_IS_TREE_RENDER = "JSpinner.isTreeCellRenderer";
-    String KEY_IS_TABLE_EDITOR = "JSpinner.isTableCellEditor";
-    String KEY_IS_TABLE_RENDERER = "JSpinner.isTableCellRenderer";
+    String KEY_IS_TREE_EDITOR = DarkTreeUI.KEY_IS_TREE_EDITOR;
+    String KEY_IS_TREE_RENDER = DarkTreeUI.KEY_IS_TREE_RENDERER;
+    String KEY_IS_TABLE_EDITOR = DarkTableUI.KEY_IS_TABLE_EDITOR;
+    String KEY_IS_TABLE_RENDERER = DarkTableUI.KEY_IS_TABLE_RENDERER;
     String KEY_EDITOR_ALIGNMENT = "JSpinner.cellEditorAlignment";
     String VARIANT_PLUS_MINUS = "plusMinus";
 
diff --git a/core/src/main/java/com/github/weisj/darklaf/ui/table/DarkTableCellEditor.java b/core/src/main/java/com/github/weisj/darklaf/ui/table/DarkTableCellEditor.java
index c9ef5207..b345f15f 100644
--- a/core/src/main/java/com/github/weisj/darklaf/ui/table/DarkTableCellEditor.java
+++ b/core/src/main/java/com/github/weisj/darklaf/ui/table/DarkTableCellEditor.java
@@ -24,9 +24,12 @@
 package com.github.weisj.darklaf.ui.table;
 
 import com.github.weisj.darklaf.ui.cell.CellUtil;
+import com.github.weisj.darklaf.ui.combobox.ComboBoxConstants;
 import com.github.weisj.darklaf.ui.combobox.DarkComboBoxUI;
 import com.github.weisj.darklaf.ui.spinner.DarkSpinnerUI;
+import com.github.weisj.darklaf.ui.spinner.SpinnerConstants;
 import com.github.weisj.darklaf.ui.text.DarkTextUI;
+import com.github.weisj.darklaf.ui.togglebutton.ToggleButtonConstants;
 import com.github.weisj.darklaf.util.PropertyValue;
 
 import javax.swing.*;
@@ -67,7 +70,7 @@ public class DarkTableCellEditor extends DefaultCellEditor {
     public DarkTableCellEditor(final JTextField textField) {
         super(textField);
         textField.setBorder(new TextTableCellEditorBorder());
-        textField.putClientProperty(DarkTextUI.KEY_IS_TABLE_EDITOR, Boolean.TRUE);
+        textField.putClientProperty(DarkTextUI.KEY_IS_TABLE_EDITOR, true);
         setClickCountToStart(2);
     }
 
@@ -89,13 +92,14 @@ public class DarkTableCellEditor extends DefaultCellEditor {
 
             }
         });
+        comboBox.putClientProperty(ComboBoxConstants.KEY_IS_TABLE_EDITOR, true);
         setClickCountToStart(2);
     }
 
     public DarkTableCellEditor(final JSpinner spinner) {
         super(dummyCheckBox);
         editorComponent = spinner;
-        spinner.putClientProperty("JSpinner.isTableCellEditor", Boolean.TRUE);
+        spinner.putClientProperty(SpinnerConstants.KEY_IS_TABLE_EDITOR, Boolean.TRUE);
         setClickCountToStart(2);
         delegate = new EditorDelegate() {
             public Object getCellEditorValue() {
@@ -150,6 +154,7 @@ public class DarkTableCellEditor extends DefaultCellEditor {
                 toggleButton.setSelected(selected);
             }
         };
+        toggleButton.putClientProperty(ToggleButtonConstants.KEY_IS_TABLE_EDITOR, true);
         toggleButton.addActionListener(delegate);
         toggleButton.setRequestFocusEnabled(false);
     }
diff --git a/core/src/main/java/com/github/weisj/darklaf/ui/table/DarkTableUI.java b/core/src/main/java/com/github/weisj/darklaf/ui/table/DarkTableUI.java
index 5d6d9dbb..541a7b76 100644
--- a/core/src/main/java/com/github/weisj/darklaf/ui/table/DarkTableUI.java
+++ b/core/src/main/java/com/github/weisj/darklaf/ui/table/DarkTableUI.java
@@ -46,6 +46,8 @@ import java.util.function.Supplier;
  */
 public class DarkTableUI extends DarkTableUIBridge implements FocusListener {
 
+    public static final String KEY_IS_TABLE_EDITOR = "JComponent.isTableEditor";
+    public static final String KEY_IS_TABLE_RENDERER = "JComponent.isTableRenderer";
     protected static final String KEY_PREFIX = "JTable.";
     public static final String KEY_ALTERNATE_ROW_COLOR = KEY_PREFIX + "alternateRowColor";
     public static final String KEY_RENDER_BOOLEAN_AS_CHECKBOX = KEY_PREFIX + "renderBooleanAsCheckBox";
diff --git a/core/src/main/java/com/github/weisj/darklaf/ui/text/DarkTextBorder.java b/core/src/main/java/com/github/weisj/darklaf/ui/text/DarkTextBorder.java
index ea560064..1dbc6d44 100644
--- a/core/src/main/java/com/github/weisj/darklaf/ui/text/DarkTextBorder.java
+++ b/core/src/main/java/com/github/weisj/darklaf/ui/text/DarkTextBorder.java
@@ -23,7 +23,6 @@
  */
 package com.github.weisj.darklaf.ui.text;
 
-import com.github.weisj.darklaf.ui.table.TextTableCellEditorBorder;
 import com.github.weisj.darklaf.util.DarkUIUtil;
 import com.github.weisj.darklaf.util.GraphicsContext;
 import com.github.weisj.darklaf.util.GraphicsUtil;
@@ -40,7 +39,6 @@ import java.awt.*;
  */
 public class DarkTextBorder implements Border, UIResource {
 
-    private static final Border editorBorder = new TextTableCellEditorBorder();
     protected final Color errorBorderColor;
     protected final Color focusErrorBorderColor;
     protected final Color focusBorderColor;
@@ -76,12 +74,6 @@ public class DarkTextBorder implements Border, UIResource {
                && Boolean.TRUE.equals(((JComponent) c).getClientProperty(DarkTextUI.KEY_HAS_ERROR));
     }
 
-
-    protected static boolean isCellEditor(final Component c) {
-        return c instanceof JComponent
-               && Boolean.TRUE.equals(((JComponent) c).getClientProperty(DarkTextUI.KEY_IS_CELL_EDITOR));
-    }
-
     protected int getArcSize(final Component c) {
         return DarkTextFieldUI.isSearchField(c) ? searchArc : arc;
     }
@@ -96,10 +88,6 @@ public class DarkTextBorder implements Border, UIResource {
 
     public void paintBorder(final Component c, final Graphics g2, final int x, final int y,
                             final int width, final int height) {
-        if (isCellEditor(c)) {
-            editorBorder.paintBorder(c, g2, x, y, width, height);
-            return;
-        }
         Graphics2D g = (Graphics2D) g2;
         g.translate(x, y);
         GraphicsContext config = GraphicsUtil.setupStrokePainting(g);
@@ -138,9 +126,6 @@ public class DarkTextBorder implements Border, UIResource {
 
     @Override
     public Insets getBorderInsets(final Component c) {
-        if (isCellEditor(c)) {
-            return editorBorder.getBorderInsets(c);
-        }
         Insets insets = new Insets(borderSize + padding.top, borderSize + padding.left,
                                    borderSize + padding.bottom, borderSize + padding.right);
         if (DarkTextFieldUI.isSearchField(c)) {
diff --git a/core/src/main/java/com/github/weisj/darklaf/ui/text/DarkTextUI.java b/core/src/main/java/com/github/weisj/darklaf/ui/text/DarkTextUI.java
index f862943e..09b44617 100644
--- a/core/src/main/java/com/github/weisj/darklaf/ui/text/DarkTextUI.java
+++ b/core/src/main/java/com/github/weisj/darklaf/ui/text/DarkTextUI.java
@@ -23,7 +23,10 @@
  */
 package com.github.weisj.darklaf.ui.text;
 
+import com.github.weisj.darklaf.ui.list.DarkListUI;
 import com.github.weisj.darklaf.ui.table.DarkTableCellBorder;
+import com.github.weisj.darklaf.ui.table.DarkTableUI;
+import com.github.weisj.darklaf.ui.tree.DarkTreeUI;
 import com.github.weisj.darklaf.util.DarkSwingUtil;
 import com.github.weisj.darklaf.util.DarkUIUtil;
 import com.github.weisj.darklaf.util.GraphicsContext;
@@ -53,9 +56,9 @@ public abstract class DarkTextUI extends BasicTextUI implements PropertyChangeLi
     protected static final String KEY_PREFIX = "JTextComponent.";
     public static final String KEY_ROUNDED_SELECTION = KEY_PREFIX + "roundedSelection";
     public static final String KEY_HAS_ERROR = KEY_PREFIX + "hasError";
-    public static final String KEY_IS_CELL_EDITOR = KEY_PREFIX + "cellEditor";
-    public static final String KEY_IS_TABLE_EDITOR = KEY_PREFIX + "isTableCellEditor";
-    public static final String KEY_IS_LIST_RENDER = KEY_PREFIX + "listCellEditor";
+    public static final String KEY_IS_TREE_EDITOR = DarkTreeUI.KEY_IS_TREE_EDITOR;
+    public static final String KEY_IS_TABLE_EDITOR = DarkTableUI.KEY_IS_TABLE_EDITOR;
+    public static final String KEY_IS_LIST_RENDER = DarkListUI.KEY_IS_LIST_RENDERER;
 
     protected JTextComponent editor;
     private FocusListener focusListener = new FocusListener() {
@@ -151,7 +154,7 @@ public abstract class DarkTextUI extends BasicTextUI implements PropertyChangeLi
             if (parent != null) {
                 g.setColor(parent.getBackground());
             }
-            if (DarkTextBorder.isCellEditor(editor) || DarkUIUtil.isInCell(editor)) {
+            if (DarkUIUtil.isInCell(editor)) {
                 g.setColor(getBackground(editor));
             }
             g.fillRect(0, 0, editor.getWidth(), editor.getHeight());
diff --git a/core/src/main/java/com/github/weisj/darklaf/ui/togglebutton/ToggleButtonConstants.java b/core/src/main/java/com/github/weisj/darklaf/ui/togglebutton/ToggleButtonConstants.java
index 0fd25868..e7598e44 100644
--- a/core/src/main/java/com/github/weisj/darklaf/ui/togglebutton/ToggleButtonConstants.java
+++ b/core/src/main/java/com/github/weisj/darklaf/ui/togglebutton/ToggleButtonConstants.java
@@ -23,6 +23,8 @@
  */
 package com.github.weisj.darklaf.ui.togglebutton;
 
+import com.github.weisj.darklaf.ui.table.DarkTableUI;
+import com.github.weisj.darklaf.ui.tree.DarkTreeUI;
 import com.github.weisj.darklaf.util.DarkUIUtil;
 
 import javax.swing.*;
@@ -30,10 +32,10 @@ import java.awt.*;
 
 public interface ToggleButtonConstants {
     String KEY_VARIANT = "JToggleButton.variant";
-    String KEY_IS_TREE_EDITOR = "JToggleButton.isTreeCellEditor";
-    String KEY_IS_TREE_RENDER = "JToggleButton.isTreeCellRenderer";
-    String KEY_IS_TABLE_EDITOR = "JToggleButton.isTableCellEditor";
-    String KEY_IS_TABLE_RENDERER = "JToggleButton.isTableCellRenderer";
+    String KEY_IS_TREE_EDITOR = DarkTreeUI.KEY_IS_TREE_EDITOR;
+    String KEY_IS_TREE_RENDERER = DarkTreeUI.KEY_IS_TREE_RENDERER;
+    String KEY_IS_TABLE_EDITOR = DarkTableUI.KEY_IS_TABLE_EDITOR;
+    String KEY_IS_TABLE_RENDERER = DarkTableUI.KEY_IS_TABLE_RENDERER;
     String KEY_CLEAR_HIT_AREA = "JToggleButton.clearHitArea";
     String VARIANT_SLIDER = "slider";
 
diff --git a/core/src/main/java/com/github/weisj/darklaf/ui/tree/DarkDefaultTreeEditor.java b/core/src/main/java/com/github/weisj/darklaf/ui/tree/DarkDefaultTreeEditor.java
index 8a71dd52..a82572ca 100644
--- a/core/src/main/java/com/github/weisj/darklaf/ui/tree/DarkDefaultTreeEditor.java
+++ b/core/src/main/java/com/github/weisj/darklaf/ui/tree/DarkDefaultTreeEditor.java
@@ -24,9 +24,12 @@
 package com.github.weisj.darklaf.ui.tree;
 
 import com.github.weisj.darklaf.ui.cell.DarkCellRendererToggleButton;
+import com.github.weisj.darklaf.ui.text.DarkTextUI;
 
 import javax.swing.*;
+import javax.swing.border.Border;
 import javax.swing.tree.DefaultTreeCellEditor;
+import javax.swing.tree.TreeCellEditor;
 import javax.swing.tree.TreeCellRenderer;
 import javax.swing.tree.TreePath;
 import java.awt.*;
@@ -67,6 +70,17 @@ public class DarkDefaultTreeEditor extends DefaultTreeCellEditor {
         };
     }
 
+    @Override
+    protected TreeCellEditor createTreeCellEditor() {
+        Border border = UIManager.getBorder("Tree.editorBorder");
+        JTextField textField = new JTextField();
+        textField.setBorder(border);
+        textField.putClientProperty(DarkTextUI.KEY_IS_TREE_EDITOR, true);
+        DefaultCellEditor editor = new DefaultCellEditor(textField);
+        editor.setClickCountToStart(1);
+        return editor;
+    }
+
     @Override
     public Component getTreeCellEditorComponent(final JTree tree, final Object value, final boolean isSelected,
                                                 final boolean expanded, final boolean leaf, final int row) {
diff --git a/core/src/main/java/com/github/weisj/darklaf/ui/tree/DarkTreeCellEditor.java b/core/src/main/java/com/github/weisj/darklaf/ui/tree/DarkTreeCellEditor.java
index 9d468dea..e3f314e5 100644
--- a/core/src/main/java/com/github/weisj/darklaf/ui/tree/DarkTreeCellEditor.java
+++ b/core/src/main/java/com/github/weisj/darklaf/ui/tree/DarkTreeCellEditor.java
@@ -24,7 +24,10 @@
 package com.github.weisj.darklaf.ui.tree;
 
 import com.github.weisj.darklaf.components.SelectableTreeNode;
-import com.github.weisj.darklaf.ui.togglebutton.DarkToggleButtonUI;
+import com.github.weisj.darklaf.ui.combobox.ComboBoxConstants;
+import com.github.weisj.darklaf.ui.spinner.SpinnerConstants;
+import com.github.weisj.darklaf.ui.text.DarkTextUI;
+import com.github.weisj.darklaf.ui.togglebutton.ToggleButtonConstants;
 import com.github.weisj.darklaf.util.DarkUIUtil;
 
 import javax.swing.*;
@@ -52,6 +55,7 @@ public class DarkTreeCellEditor extends DefaultCellEditor implements TreeCellEdi
         super(textField);
         textField.setBorder(UIManager.getBorder("Tree.editorBorder"));
         textField.addFocusListener(this);
+        textField.putClientProperty(DarkTextUI.KEY_IS_TREE_EDITOR, true);
     }
 
     public DarkTreeCellEditor(final JCheckBox checkBox) {
@@ -75,7 +79,7 @@ public class DarkTreeCellEditor extends DefaultCellEditor implements TreeCellEdi
             }
         };
         toggleButton.setFocusPainted(false);
-        toggleButton.putClientProperty(DarkToggleButtonUI.KEY_IS_TREE_EDITOR, Boolean.TRUE);
+        toggleButton.putClientProperty(ToggleButtonConstants.KEY_IS_TREE_EDITOR, Boolean.TRUE);
         toggleButton.addActionListener(delegate);
         toggleButton.setRequestFocusEnabled(false);
         toggleButton.addFocusListener(this);
@@ -100,6 +104,7 @@ public class DarkTreeCellEditor extends DefaultCellEditor implements TreeCellEdi
 
             }
         });
+        comboBox.putClientProperty(ComboBoxConstants.KEY_IS_TREE_EDITOR, Boolean.TRUE);
         comboBox.addFocusListener(this);
         setClickCountToStart(2);
     }
@@ -107,8 +112,8 @@ public class DarkTreeCellEditor extends DefaultCellEditor implements TreeCellEdi
     public DarkTreeCellEditor(final JSpinner spinner) {
         super(dummyCheckBox);
         editorComponent = spinner;
+        editorComponent.putClientProperty(SpinnerConstants.KEY_IS_TREE_EDITOR, Boolean.TRUE);
         editorComponent.addFocusListener(this);
-        spinner.putClientProperty("JSpinner.isTreeCellEditor", Boolean.TRUE);
         setClickCountToStart(2);
         delegate = new EditorDelegate() {
             public Object getCellEditorValue() {
@@ -144,11 +149,6 @@ public class DarkTreeCellEditor extends DefaultCellEditor implements TreeCellEdi
         return editorComponent instanceof JToggleButton && DarkTreeCellRenderer.isBooleanRenderingEnabled(tree);
     }
 
-    @Override
-    public boolean isCellEditable(final EventObject anEvent) {
-        return super.isCellEditable(anEvent);
-    }
-
     @SuppressWarnings("unchecked")
     @Override
     public Component getTreeCellEditorComponent(final JTree tree, final Object value,
diff --git a/core/src/main/java/com/github/weisj/darklaf/ui/tree/DarkTreeUI.java b/core/src/main/java/com/github/weisj/darklaf/ui/tree/DarkTreeUI.java
index bbe9464b..18b457e0 100644
--- a/core/src/main/java/com/github/weisj/darklaf/ui/tree/DarkTreeUI.java
+++ b/core/src/main/java/com/github/weisj/darklaf/ui/tree/DarkTreeUI.java
@@ -23,7 +23,6 @@
  */
 package com.github.weisj.darklaf.ui.tree;
 
-import com.github.weisj.darklaf.ui.togglebutton.DarkToggleButtonUI;
 import com.github.weisj.darklaf.util.DarkUIUtil;
 import com.github.weisj.darklaf.util.SystemInfo;
 
@@ -59,6 +58,8 @@ public class DarkTreeUI extends BasicTreeUI implements PropertyChangeListener {
     public static final String STYLE_LINE = "line";
     public static final String STYLE_DASHED = "dashed";
     public static final String STYLE_NONE = "none";
+    public static final String KEY_IS_TREE_EDITOR = "JComponent.isTreeEditor";
+    public static final String KEY_IS_TREE_RENDERER = "JComponent.isTreeRenderer";
 
     private final MouseListener selectionListener = new MouseAdapter() {
         boolean handled = false;
@@ -308,21 +309,16 @@ public class DarkTreeUI extends BasicTreeUI implements PropertyChangeListener {
     protected FocusListener createFocusListener() {
         return new FocusListener() {
 
-            boolean focused = false;
-
             @Override
             public void focusGained(final FocusEvent e) {
-                if (!focused) {
-                    tree.repaint();
-                }
-                focused = true;
+                tree.repaint();
             }
 
             @Override
             public void focusLost(final FocusEvent e) {
-                tree.stopEditing();
-                focused = hasFocus(e != null ? e.getOppositeComponent() : null);
+                boolean focused = hasFocus(e != null ? e.getOppositeComponent() : null);
                 if (!focused) {
+                    tree.stopEditing();
                     tree.repaint();
                 }
             }
@@ -341,10 +337,10 @@ public class DarkTreeUI extends BasicTreeUI implements PropertyChangeListener {
             }
             boolean treeEditor = owner instanceof JComponent
                                  && Boolean.TRUE.equals(
-                ((JComponent) owner).getClientProperty(DarkToggleButtonUI.KEY_IS_TREE_EDITOR));
-            boolean treeRenderer = owner instanceof JComponent
+                ((JComponent) owner).getClientProperty(DarkTreeUI.KEY_IS_TREE_EDITOR));
+            boolean treeRenderer = !treeEditor && owner instanceof JComponent
                                    && Boolean.TRUE.equals(
-                ((JComponent) owner).getClientProperty(DarkToggleButtonUI.KEY_IS_TREE_RENDER));
+                ((JComponent) owner).getClientProperty(DarkTreeUI.KEY_IS_TREE_RENDERER));
             return treeEditor || treeRenderer;
         }
         return true;
@@ -376,6 +372,13 @@ public class DarkTreeUI extends BasicTreeUI implements PropertyChangeListener {
         tree.removePropertyChangeListener(this);
     }
 
+    @Override
+    protected boolean startEditing(final TreePath path, final MouseEvent event) {
+        boolean editing = super.startEditing(path, event);
+        if (editing) tree.repaint();
+        return editing;
+    }
+
     @Override
     public void paint(final Graphics g, final JComponent c) {
         if (tree != c) {