From: David ‘Bombe’ Roden <bombe@pterodactylus.net>
Date: Wed, 10 Apr 2013 04:46:27 +0000 (+0200)
Subject: Add DCC receiver.
X-Git-Url: https://git.pterodactylus.net/?a=commitdiff_plain;h=ff559fd198d827de652e57b64e41bc85eccd537b;p=xudocci.git

Add DCC receiver.
---

diff --git a/src/main/java/net/pterodactylus/irc/DccReceiver.java b/src/main/java/net/pterodactylus/irc/DccReceiver.java
new file mode 100644
index 0000000..39f236c
--- /dev/null
+++ b/src/main/java/net/pterodactylus/irc/DccReceiver.java
@@ -0,0 +1,144 @@
+/*
+ * XdccDownloader - DccReceiver.java - Copyright © 2013 David Roden
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ */
+
+package net.pterodactylus.irc;
+
+import java.io.IOException;
+import java.io.InputStream;
+import java.io.OutputStream;
+import java.net.InetAddress;
+import java.net.Socket;
+import java.util.logging.Level;
+import java.util.logging.Logger;
+
+import net.pterodactylus.irc.event.DccSendReceived;
+
+import com.google.common.util.concurrent.AbstractExecutionThreadService;
+
+/**
+ * Service that receives a file offered by a {@link DccSendReceived}.
+ *
+ * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
+ */
+public class DccReceiver extends AbstractExecutionThreadService {
+
+	/** The logger. */
+	private static final Logger logger = Logger.getLogger(DccReceiver.class.getName());
+
+	/** The address to connect to. */
+	private final InetAddress inetAddress;
+
+	/** The port number to connect to. */
+	private final int port;
+
+	/** The name of the file being offered. */
+	private final String filename;
+
+	/** The size of the file being offered. */
+	private final long size;
+
+	/** The output stream to write the file to. */
+	private final OutputStream outputStream;
+
+	/** The number of bytes already written. */
+	private long progress;
+
+	/**
+	 * Creates a new DCC receiver.
+	 *
+	 * @param inetAddress
+	 * 		The address to connect to
+	 * @param port
+	 * 		The port number to connect to
+	 * @param filename
+	 * 		The name of the file being downloaded
+	 * @param size
+	 * 		The size of the file being downloaded, or {@code -1} if the size is not
+	 * 		known
+	 * @param outputStream
+	 * 		The output stream to write the file to
+	 */
+	public DccReceiver(InetAddress inetAddress, int port, String filename, long size, OutputStream outputStream) {
+		this.inetAddress = inetAddress;
+		this.port = port;
+		this.filename = filename;
+		this.size = size;
+		this.outputStream = outputStream;
+	}
+
+	//
+	// ACCESSORS
+	//
+
+	/**
+	 * Returns the name of the file being downloaded. The name is not used by the
+	 * DCC receiver, it only serves as a kind of identifier.
+	 *
+	 * @return The name of the file being downloaded
+	 */
+	public String filename() {
+		return filename;
+	}
+
+	/**
+	 * Returns the size of the file being downloaded. If the size of the file is
+	 * not known, {@code -1} is returned.
+	 *
+	 * @return The size of the file being downloaded, or {@code -1} if the size is
+	 *         not known
+	 */
+	public long size() {
+		return size;
+	}
+
+	/**
+	 * Returns the number of bytes that have already been downloaded.
+	 *
+	 * @return The number of bytes that have already been downloaded
+	 */
+	public long progress() {
+		return progress;
+	}
+
+	//
+	// ABSTRACTEXECUTIONTHREADSERVICE METHODS
+	//
+
+	@Override
+	protected void run() throws Exception {
+		Socket socket = null;
+		try {
+			socket = new Socket(inetAddress, port);
+			InputStream socketInputStream = socket.getInputStream();
+			byte[] buffer = new byte[65536];
+			while (true) {
+				int r = socketInputStream.read(buffer);
+				if (r == -1) {
+					/* yay, eof! */
+					break;
+				}
+				outputStream.write(buffer, 0, r);
+				progress += r;
+			}
+		} catch (IOException ioe1) {
+			logger.log(Level.WARNING, "I/O error while receiving DCC!", ioe1);
+		} finally {
+			socket.close();
+		}
+	}
+
+}