X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fxdcc%2Fcore%2FCore.java;h=d2531485a285b67e5961c64ae1757660fc9887db;hb=099e3500a7357ee556aaad819e06e7180ec29db0;hp=8eea508514b4687bafcd54d6543fde17ec678470;hpb=1550de41bcf0afcb011e32ab4d737e766ae31618;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 8eea508..d253148 100644
--- a/src/main/java/net/pterodactylus/xdcc/core/Core.java
+++ b/src/main/java/net/pterodactylus/xdcc/core/Core.java
@@ -17,7 +17,11 @@
 
 package net.pterodactylus.xdcc.core;
 
+import java.io.File;
+import java.io.FileNotFoundException;
+import java.io.FileOutputStream;
 import java.io.IOException;
+import java.io.OutputStream;
 import java.util.Collection;
 import java.util.Collections;
 import java.util.List;
@@ -29,8 +33,10 @@ import java.util.logging.Logger;
 import net.pterodactylus.irc.Connection;
 import net.pterodactylus.irc.ConnectionBuilder;
 import net.pterodactylus.irc.DccReceiver;
+import net.pterodactylus.irc.event.ChannelJoined;
 import net.pterodactylus.irc.event.ChannelMessageReceived;
 import net.pterodactylus.irc.event.ConnectionEstablished;
+import net.pterodactylus.irc.event.DccSendReceived;
 import net.pterodactylus.irc.util.MessageCleaner;
 import net.pterodactylus.irc.util.RandomNickname;
 import net.pterodactylus.xdcc.core.event.BotAdded;
@@ -43,6 +49,7 @@ import net.pterodactylus.xdcc.data.Server;
 
 import com.google.common.base.Optional;
 import com.google.common.collect.HashBasedTable;
+import com.google.common.collect.ImmutableSet;
 import com.google.common.collect.Lists;
 import com.google.common.collect.Maps;
 import com.google.common.collect.Sets;
@@ -68,6 +75,12 @@ public class Core extends AbstractIdleService {
 	/** The channels that should be monitored. */
 	private final Collection<Channel> channels = Sets.newHashSet();
 
+	/** The channels that are currentlymonitored. */
+	private final Collection<Channel> joinedChannels = Sets.newHashSet();
+
+	/** The channels that are joined but not configured. */
+	private final Collection<Channel> extraChannels = Sets.newHashSet();
+
 	/** The current network connections. */
 	private final Map<Network, Connection> networkConnections = Collections.synchronizedMap(Maps.<Network, Connection>newHashMap());
 
@@ -93,6 +106,34 @@ public class Core extends AbstractIdleService {
 	//
 
 	/**
+	 * Returns all configured channels. Due to various circumstances, configured
+	 * channels might not actually be joined.
+	 *
+	 * @return All configured channels
+	 */
+	public Collection<Channel> channels() {
+		return ImmutableSet.copyOf(channels);
+	}
+
+	/**
+	 * Returns all currently joined channels.
+	 *
+	 * @return All currently joined channels
+	 */
+	public Collection<Channel> joinedChannels() {
+		return ImmutableSet.copyOf(joinedChannels);
+	}
+
+	/**
+	 * Returns all currently joined channels that are not configured.
+	 *
+	 * @return All currently joined but not configured channels
+	 */
+	public Collection<Channel> extraChannels() {
+		return ImmutableSet.copyOf(extraChannels);
+	}
+
+	/**
 	 * Returns all currently known bots.
 	 *
 	 * @return All currently known bots
@@ -124,6 +165,27 @@ public class Core extends AbstractIdleService {
 		channels.add(channel);
 	}
 
+	/**
+	 * Fetches the given pack from the given bot.
+	 *
+	 * @param bot
+	 * 		The bot to fetch the pack from
+	 * @param pack
+	 * 		The pack to fetch
+	 */
+	public void fetch(Bot bot, Pack pack) {
+		Connection connection = networkConnections.get(bot.network());
+		if (connection == null) {
+			return;
+		}
+
+		try {
+			connection.sendMessage(bot.name(), "XDCC SEND " + pack.id());
+		} catch (IOException ioe1) {
+			logger.log(Level.WARNING, "Could not send message to bot!", ioe1);
+		}
+	}
+
 	//
 	// ABSTRACTIDLESERVICE METHODS
 	//
@@ -186,6 +248,33 @@ public class Core extends AbstractIdleService {
 	}
 
 	/**
+	 * Shows a message when a channel was joined by us.
+	 *
+	 * @param channelJoined
+	 * 		The channel joined event
+	 */
+	@Subscribe
+	public void channelJoined(ChannelJoined channelJoined) {
+		if (channelJoined.connection().isSource(channelJoined.client())) {
+			Optional<Network> network = getNetwork(channelJoined.connection());
+			if (!network.isPresent()) {
+				return;
+			}
+
+			Optional<Channel> channel = getChannel(network.get(), channelJoined.channel());
+			if (!channel.isPresent()) {
+				/* it’s an extra channel. */
+				extraChannels.add(new Channel(network.get(), channelJoined.channel()));
+				logger.info(String.format("Joined extra Channel %s on %s.", channelJoined.channel(), network.get().name()));
+				return;
+			}
+
+			joinedChannels.add(channel.get());
+			logger.info(String.format("Joined Channel %s on %s.", channelJoined.channel(), network.get().name()));
+		}
+	}
+
+	/**
 	 * If a message on a channel is received, it is parsed for pack information
 	 * with is then added to a bot.
 	 *
@@ -228,6 +317,25 @@ public class Core extends AbstractIdleService {
 		logger.fine(String.format("Bot %s now has %d packs.", bot, bot.packs().size()));
 	}
 
+	/**
+	 * Starts a DCC download.
+	 *
+	 * @param dccSendReceived
+	 * 		The DCC SEND event
+	 */
+	@Subscribe
+	public void dccSendReceived(DccSendReceived dccSendReceived) {
+		logger.info(String.format("Starting download of %s.", dccSendReceived.filename()));
+		try {
+			OutputStream fileOutputStream = new FileOutputStream(new File("/home/bombe/Temp", dccSendReceived.filename()));
+			DccReceiver dccReceiver = new DccReceiver(dccSendReceived.inetAddress(), dccSendReceived.port(), dccSendReceived.filename(), dccSendReceived.filesize(), fileOutputStream);
+			dccReceivers.add(dccReceiver);
+			dccReceiver.start();
+		} catch (FileNotFoundException fnfe1) {
+			logger.log(Level.WARNING, "Could not open file for download!", fnfe1);
+		}
+	}
+
 	//
 	// PRIVATE METHODS
 	//
@@ -251,6 +359,25 @@ public class Core extends AbstractIdleService {
 	}
 
 	/**
+	 * Returns the configured channel for the given network and name.
+	 *
+	 * @param network
+	 * 		The network the channel is located on
+	 * @param channelName
+	 * 		The name of the channel
+	 * @return The configured channel, or {@link Optional#absent()} if no
+	 *         configured channel matching the given network and name was found
+	 */
+	public Optional<Channel> getChannel(Network network, String channelName) {
+		for (Channel channel : channels) {
+			if (channel.network().equals(network) && (channel.name().equals(channelName))) {
+				return Optional.of(channel);
+			}
+		}
+		return Optional.absent();
+	}
+
+	/**
 	 * Parses {@link Pack} information from the given message.
 	 *
 	 * @param message