ClientPutComplexDir.java

/*
 * 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 <droden@gmail.com>
 * @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;
	}

}