From 4dafaab5d9a7925dbb57b7dd22fba54e3e29819d Mon Sep 17 00:00:00 2001
From: yoyosource <yoyosource@nidido.de>
Date: Sun, 30 Apr 2023 12:39:11 +0200
Subject: [PATCH] Update TNTSimulator.edit to better reflect interactability

Signed-off-by: yoyosource <yoyosource@nidido.de>
---
 .../features/simulator/TNTSimulator.java      | 25 ++++++++++++++++++-
 1 file changed, 24 insertions(+), 1 deletion(-)

diff --git a/BauSystem_Main/src/de/steamwar/bausystem/features/simulator/TNTSimulator.java b/BauSystem_Main/src/de/steamwar/bausystem/features/simulator/TNTSimulator.java
index f9513cd4..46d9779a 100644
--- a/BauSystem_Main/src/de/steamwar/bausystem/features/simulator/TNTSimulator.java
+++ b/BauSystem_Main/src/de/steamwar/bausystem/features/simulator/TNTSimulator.java
@@ -167,7 +167,30 @@ public class TNTSimulator {
             if (elements.size() == 1) {
                 TNTElementGUI.open(player, (TNTElement) elements.get(0), null);
             } else {
-                TNTSimulatorGui.open(player, null, null, () -> elements, null);
+                List<TNTGroup> tntGroups = tntElementList.stream().filter(TNTGroup.class::isInstance).map(TNTGroup.class::cast).collect(Collectors.toList());
+                List<SimulatorElement> newElementList = new ArrayList<>();
+                for (TNTGroup tntGroup : tntGroups) {
+                    if (new HashSet<>(elements).containsAll(tntGroup.getElements())) {
+                        newElementList.add(tntGroup);
+                        elements.removeAll(tntGroup.getElements());
+                    }
+                }
+                newElementList.addAll(elements);
+                if (newElementList.size() == 1) {
+                    SimulatorElement element = newElementList.get(0);
+                    if (element instanceof TNTGroup) {
+                        TNTGroup tntGroup = (TNTGroup) element;
+                        TNTSimulatorGui.open(player, null, tntGroup, () -> {
+                            List<SimulatorElement> elementList = new ArrayList<>();
+                            elementList.addAll(tntGroup.getElements());
+                            return elementList;
+                        }, null);
+                    } else {
+                        TNTElementGUI.open(player, (TNTElement) elements.get(0), null);
+                    }
+                } else {
+                    TNTSimulatorGui.open(player, null, null, () -> newElementList, null);
+                }
             }
             return;
         }