Edit JTree

Author: Shirouky

.idea/artifacts/lab2.xml

@@ -0,0 +1,3 @@
+Manifest-Version: 1.0
+Main-Class: main.Main
+

META-INF/MANIFEST.MF

@@ -1,7 +1,6 @@
 package factory;
 
 import com.github.javafaker.Faker;
-import gear.factory.MistyMountainsGearFactory;
 import gear.factory.MordorGearFactory;
 import ork.OrkBuilder;
 import ork.Tribe;

out/artifacts/lab2/lab2.jar

@@ -18,6 +18,7 @@ public class Tree extends JFrame {
     ArrayList<DefaultMutableTreeNode> nodes;
     private final Controller controller;
     private final HashMap<String, ArrayList<Ork>> orks = new HashMap<>();
+    JTree tree;
 
     public Tree(Controller controller) {
         var textColor = new Color(255, 0, 0);
@@ -31,7 +32,7 @@ public Tree(Controller controller) {
 
         TreeModel model = createTreeModel();
 
-        JTree tree = new JTree(model);
+        tree = new JTree(model);
         tree.addTreeSelectionListener(new SelectionListener());
 
         TreeSelectionModel selModel = new DefaultTreeSelectionModel();
@@ -78,6 +79,7 @@ public void update(Ork ork) {
         for (DefaultMutableTreeNode node : this.nodes) {
             if (Objects.equals(node.toString(), ork.getTribe())) {
                 node.add(new DefaultMutableTreeNode(ork.getName(), false));
+                tree.updateUI();
             }
         }
     }