SecureNetworkDriver

CaitlynMainer · CaitlynMainer · commit 07ebe04ab4e2 · 2016-04-17T14:25:41.000-05:00
diff --git a/src/main/java/pcl/opensecurity/drivers/SecureNetworkCardDriver.java b/src/main/java/pcl/opensecurity/drivers/SecureNetworkCardDriver.java
@@ -1,5 +1,7 @@
 package pcl.opensecurity.drivers;
 
+import java.util.Iterator;
+
 import pcl.opensecurity.OpenSecurity;
 import li.cil.oc.api.Network;
 import li.cil.oc.api.driver.EnvironmentHost;
@@ -14,12 +16,12 @@
 
 public class SecureNetworkCardDriver extends NetworkCard {
 
-	public final EnvironmentHost container;
+	public final li.cil.oc.api.network.EnvironmentHost container;
 	private ComponentConnector node;
 
-	public SecureNetworkCardDriver(EnvironmentHost container) {
-		super(container);
-		this.container = container;
+	public SecureNetworkCardDriver(li.cil.oc.api.network.EnvironmentHost arg1) {
+		super(arg1);
+		this.container = arg1;
 		this.setNode(Network.newNode(this, Visibility.Network)
 				.withComponent("modem", Visibility.Neighbors)
 				.withConnector(1)
@@ -29,20 +31,32 @@ public SecureNetworkCardDriver(EnvironmentHost container) {
 	@Callback(doc = "function() -- Randomises the UUID")
 	public Object[] generateUUID(Context context, Arguments args) {
 		//if(node.tryChangeBuffer(1)) {
-			this.node.remove();
-			this.setNode(Network.newNode(this, Visibility.Network)
-					.withComponent("modem", Visibility.Neighbors)
-					.withConnector(1)
-					.create());
-			OpenSecurity.logger.info(this.node.address());
-			return new Object[] { true };
+		//<@Sangar> well, in that case your best bet is to store its neighbors before disconnecting, then reconnect to them all
+		OpenSecurity.logger.info(this.node.address());
+		Iterable<Node> tempNodes = this.node().neighbors();
+
+		this.node.remove();
+		this.node = Network.newNode(this, Visibility.Network)
+				.withComponent("modem", Visibility.Neighbors)
+				.withConnector(1)
+				.create();
+		
+		Iterator<Node> meh = tempNodes.iterator();
+		
+		//OpenSecurity.logger.info(this.node());
+		while (meh.hasNext()) {
+			this.node().connect(meh.next());
+		}
+		
+		OpenSecurity.logger.info(this.node.address());
+		return new Object[] { true };
 		//} else {
-			//return new Object[] { false };
+		//return new Object[] { false };
 		//}
 	}
 
 	@Override
-	public EnvironmentHost host() {
+	public li.cil.oc.api.network.EnvironmentHost host() {
 		return this.container;
 	}
 
