fix: ghost network block (#118)

SlimefunGuguProject · Jun 13, 2024 · 1514bf7 · 1514bf7
1 parent 3c8ba40
commit 1514bf7
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/src/main/java/io/github/sefiraat/networks/listeners/BlockListener.java b/src/main/java/io/github/sefiraat/networks/listeners/BlockListener.java
@@ -1,6 +1,7 @@
 package io.github.sefiraat.networks.listeners;
 
 import io.github.sefiraat.networks.NetworkStorage;
+import io.github.sefiraat.networks.network.NetworkNode;
 import io.github.sefiraat.networks.network.NodeDefinition;
 import io.github.sefiraat.networks.network.NodeType;
 import io.github.sefiraat.networks.slimefun.network.NetworkController;
@@ -25,11 +26,10 @@ public void onBlockPlace(BlockPlaceEvent e) {
     private void removeNetwork(Location location) {
         NodeDefinition definition = NetworkStorage.getAllNetworkObjects().get(location);
         if (definition == null) return;
-        if (definition.getNode() == null) return;
-        NetworkStorage.removeNode(location);
-
-        if (definition.getNode().getNodeType() == NodeType.CONTROLLER) {
+        NetworkNode node = definition.getNode();
+        if (node != null && node.getNodeType() == NodeType.CONTROLLER) {
             NetworkController.wipeNetwork(location);
         }
+        NetworkStorage.removeNode(location);
     }
 }
diff --git a/src/main/java/io/github/sefiraat/networks/network/NetworkRoot.java b/src/main/java/io/github/sefiraat/networks/network/NetworkRoot.java
@@ -68,6 +68,8 @@ public NetworkRoot(@Nonnull Location location, @Nonnull NodeType type, int maxNo
         super(location, type);
         this.maxNodes = maxNodes;
         this.root = this;
+        NetworkNode node = new NetworkNode(location, NodeType.CONTROLLER);
+        io.github.sefiraat.networks.NetworkStorage.getAllNetworkObjects().get(location).setNode(node);
     }
 
     public void registerNode(@Nonnull Location location, @Nonnull NodeType type) {