X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fxdcc%2Fcore%2FCore.java;h=30da4a88cb9b66cfff1dd572cf6b8a5376273b11;hb=e89d921415a0ae26ebb5bb7cee7a1e72fc0418b4;hp=d4e51558e3788838326e92f5721639ab1325875d;hpb=a1ad62aece998bef7d62cab8fa31a839d1d16ef8;p=xudocci.git

diff --git a/src/main/java/net/pterodactylus/xdcc/core/Core.java b/src/main/java/net/pterodactylus/xdcc/core/Core.java
index d4e5155..30da4a8 100644
--- a/src/main/java/net/pterodactylus/xdcc/core/Core.java
+++ b/src/main/java/net/pterodactylus/xdcc/core/Core.java
@@ -39,8 +39,9 @@ import java.util.concurrent.TimeUnit;
 import java.util.stream.Collectors;
 
 import net.pterodactylus.irc.Connection;
-import net.pterodactylus.irc.ConnectionBuilder;
+import net.pterodactylus.irc.ConnectionFactory;
 import net.pterodactylus.irc.DccReceiver;
+import net.pterodactylus.irc.DefaultConnection;
 import net.pterodactylus.irc.event.ChannelJoined;
 import net.pterodactylus.irc.event.ChannelLeft;
 import net.pterodactylus.irc.event.ChannelMessageReceived;
@@ -104,8 +105,10 @@ public class Core extends AbstractExecutionThreadService {
 	/** The logger. */
 	private static final Logger logger = Logger.getLogger(Core.class.getName());
 
+	private final Object syncObject = new Object();
 	/** The event bus. */
 	private final EventBus eventBus;
+	private final ConnectionFactory connectionFactory;
 	private final ChannelBanManager channelBanManager =
 			new ChannelBanManager();
 
@@ -147,8 +150,9 @@ public class Core extends AbstractExecutionThreadService {
 	 * 		The directory to move finished files to
 	 */
 	@Inject
-	public Core(EventBus eventBus, String temporaryDirectory, String finalDirectory) {
+	public Core(EventBus eventBus, ConnectionFactory connectionFactory, String temporaryDirectory, String finalDirectory) {
 		this.eventBus = eventBus;
+		this.connectionFactory = connectionFactory;
 		this.temporaryDirectory = temporaryDirectory;
 		this.finalDirectory = finalDirectory;
 	}
@@ -373,10 +377,15 @@ public class Core extends AbstractExecutionThreadService {
 	@Override
 	protected void run() throws Exception {
 		while (isRunning()) {
-			try {
-				Thread.sleep(TimeUnit.MINUTES.toMillis(1));
-			} catch (InterruptedException ie1) {
-				/* ignore. */
+			synchronized (syncObject) {
+				try {
+					syncObject.wait(TimeUnit.MINUTES.toMillis(1));
+				} catch (InterruptedException ie1) {
+					/* ignore. */
+				}
+			}
+			if (!isRunning()) {
+				break;
 			}
 
 			/* find channels that should be monitored but are not. */
@@ -405,7 +414,10 @@ public class Core extends AbstractExecutionThreadService {
 	}
 
 	@Override
-	protected void shutDown() {
+	protected void triggerShutdown() {
+		synchronized (syncObject) {
+			syncObject.notifyAll();
+		}
 	}
 
 	//
@@ -428,10 +440,11 @@ public class Core extends AbstractExecutionThreadService {
 				return;
 			}
 			Server server = servers.get((int) (Math.random() * servers.size()));
-			Connection connection = new ConnectionBuilder(eventBus).connect(server.hostname()).port(server.unencryptedPorts().iterator().next()).build();
+			Connection connection = connectionFactory.createConnection(server.hostname(),
+					server.unencryptedPorts().iterator().next());
 			connection.username(RandomNickname.get()).realName(RandomNickname.get());
 			networkConnections.put(network, connection);
-			connection.start();
+			connection.open();
 		}
 	}
 
@@ -577,6 +590,16 @@ public class Core extends AbstractExecutionThreadService {
 			return;
 		}
 
+		/* remove all bots for this channel, we might have been kicked. */
+		Collection<Bot> botsToRemove = networkBots.row(network.get())
+				.values().stream()
+				.filter(bot -> bot.channel()
+						.equalsIgnoreCase(channel.get().name()))
+				.collect(Collectors.toSet());
+		botsToRemove.stream()
+				.forEach(bot -> networkBots.row(network.get())
+						.remove(bot.name()));
+
 		if (channelNotJoined.reason() == registeredNicknamesOnly) {
 			channels.remove(channel.get());
 			eventBus.post(new GenericMessage(
@@ -654,9 +677,9 @@ public class Core extends AbstractExecutionThreadService {
 					return;
 				}
 
-				extraChannels.remove(channel);
+				extraChannels.remove(channel.get());
 			} else {
-				channels.remove(channel.get());
+				joinedChannels.remove(channel.get());
 			}
 			eventBus.post(new GenericMessage(format(
 					"Kicked from %s by %s: %s",