X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fde%2Ftodesbaum%2Futil%2Ffreenet%2Ffcp2%2FClientPutComplexDir.java;fp=src%2Fmain%2Fjava%2Fde%2Ftodesbaum%2Futil%2Ffreenet%2Ffcp2%2FClientPutComplexDir.java;h=2c23f878e057ef8a97eed9d62909e580af9366d3;hb=38bdc433e50669e8244a63b5af59e597f88f1d29;hp=0000000000000000000000000000000000000000;hpb=f14b9fbe6d88e23920b10a75ebeba4d38390301b;p=jSite.git

diff --git a/src/main/java/de/todesbaum/util/freenet/fcp2/ClientPutComplexDir.java b/src/main/java/de/todesbaum/util/freenet/fcp2/ClientPutComplexDir.java
new file mode 100644
index 0000000..2c23f87
--- /dev/null
+++ b/src/main/java/de/todesbaum/util/freenet/fcp2/ClientPutComplexDir.java
@@ -0,0 +1,187 @@
+/*
+ * jSite - ClientPutComplexDir.java - Copyright © 2006–2012 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 2 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, write to the Free Software
+ * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
+ */
+
+package de.todesbaum.util.freenet.fcp2;
+
+import java.io.File;
+import java.io.FileInputStream;
+import java.io.FileNotFoundException;
+import java.io.FileOutputStream;
+import java.io.IOException;
+import java.io.InputStream;
+import java.io.Writer;
+import java.util.ArrayList;
+import java.util.List;
+
+import net.pterodactylus.util.io.Closer;
+
+/**
+ * Implementation of the <code>ClientPutComplexDir</code> command. This command
+ * can be used to insert directories that do not exist on disk.
+ *
+ * @author David Roden &lt;droden@gmail.com&gt;
+ * @version $Id$
+ */
+public class ClientPutComplexDir extends ClientPutDir<ClientPutComplexDir> {
+
+	/** The file entries of this directory. */
+	private List<FileEntry> fileEntries = new ArrayList<FileEntry>();
+
+	/** Whether this request has payload. */
+	private boolean hasPayload = false;
+
+	/** The input streams for the payload. */
+	private File payloadFile;
+
+	/** The total number of bytes of the payload. */
+	private long payloadLength = 0;
+
+	/** The temp directory to use. */
+	private final String tempDirectory;
+
+	/**
+	 * Creates a new <code>ClientPutComplexDir</code> command with the specified
+	 * identifier and URI.
+	 *
+	 * @param identifier
+	 *            The identifier of the command
+	 * @param uri
+	 *            The URI of the command
+	 */
+	public ClientPutComplexDir(String identifier, String uri) {
+		this(identifier, uri, null);
+	}
+
+	/**
+	 * Creates a new <code>ClientPutComplexDir</code> command with the specified
+	 * identifier and URI.
+	 *
+	 * @param identifier
+	 *            The identifier of the command
+	 * @param uri
+	 *            The URI of the command
+	 * @param tempDirectory
+	 *            The temp directory to use, or {@code null} to use the default
+	 *            temp directory
+	 */
+	public ClientPutComplexDir(String identifier, String uri, String tempDirectory) {
+		super("ClientPutComplexDir", identifier, uri);
+		this.tempDirectory = tempDirectory;
+	}
+
+	/**
+	 * Adds a file to the directory inserted by this request.
+	 *
+	 * @param fileEntry
+	 *            The file entry to add to the directory
+	 * @throws IOException
+	 *             if an I/O error occurs when creating the payload stream
+	 */
+	public void addFileEntry(FileEntry fileEntry) throws IOException {
+		if (fileEntry instanceof DirectFileEntry) {
+			if (payloadFile == null) {
+				try {
+					payloadFile = File.createTempFile("payload", ".dat", (tempDirectory != null) ? new File(tempDirectory) : null);
+					payloadFile.deleteOnExit();
+				} catch (IOException e) {
+					/* ignore. */
+				}
+			}
+			if (payloadFile != null) {
+				InputStream payloadInputStream = ((DirectFileEntry) fileEntry).getDataInputStream();
+				FileOutputStream payloadOutputStream = null;
+				try {
+					payloadOutputStream = new FileOutputStream(payloadFile, true);
+					byte[] buffer = new byte[65536];
+					int read = 0;
+					while ((read = payloadInputStream.read(buffer)) != -1) {
+						payloadOutputStream.write(buffer, 0, read);
+					}
+					payloadOutputStream.flush();
+					fileEntries.add(fileEntry);
+				} catch (IOException ioe1) {
+					payloadFile.delete();
+					throw ioe1;
+				} finally {
+					Closer.close(payloadOutputStream);
+					Closer.close(payloadInputStream);
+				}
+			}
+		} else {
+			fileEntries.add(fileEntry);
+		}
+	}
+
+	/**
+	 * {@inheritDoc}
+	 */
+	@Override
+	protected void write(Writer writer) throws IOException {
+		super.write(writer);
+		int fileIndex = 0;
+		for (FileEntry fileEntry : fileEntries) {
+			writer.write("Files." + fileIndex + ".Name=" + fileEntry.getFilename() + LINEFEED);
+			if (fileEntry.getContentType() != null) {
+				writer.write("Files." + fileIndex + ".Metadata.ContentType=" + fileEntry.getContentType() + LINEFEED);
+			}
+			writer.write("Files." + fileIndex + ".UploadFrom=" + fileEntry.getName() + LINEFEED);
+			if (fileEntry instanceof DirectFileEntry) {
+				hasPayload = true;
+				writer.write("Files." + fileIndex + ".DataLength=" + ((DirectFileEntry) fileEntry).getDataLength() + LINEFEED);
+				payloadLength += ((DirectFileEntry) fileEntry).getDataLength();
+			} else if (fileEntry instanceof DiskFileEntry) {
+				writer.write("Files." + fileIndex + ".Filename=" + ((DiskFileEntry) fileEntry).getFilename() + LINEFEED);
+			} else if (fileEntry instanceof RedirectFileEntry) {
+				writer.write("Files." + fileIndex + ".TargetURI=" + ((RedirectFileEntry) fileEntry).getTargetURI() + LINEFEED);
+			}
+			fileIndex++;
+		}
+	}
+
+	/**
+	 * {@inheritDoc}
+	 */
+	@Override
+	protected boolean hasPayload() {
+		return hasPayload;
+	}
+
+	/**
+	 * {@inheritDoc}
+	 */
+	@Override
+	protected long getPayloadLength() {
+		return payloadLength;
+	}
+
+	/**
+	 * {@inheritDoc}
+	 */
+	@Override
+	protected InputStream getPayload() {
+		if (payloadFile != null) {
+			try {
+				return new FileInputStream(payloadFile);
+			} catch (FileNotFoundException e) {
+				/* shouldn't occur. */
+			}
+		}
+		return null;
+	}
+
+}