3 * Copyright (C) 2006 David Roden
5 * This program is free software; you can redistribute it and/or modify
6 * it under the terms of the GNU General Public License as published by
7 * the Free Software Foundation; either version 2 of the License, or
8 * (at your option) any later version.
10 * This program is distributed in the hope that it will be useful,
11 * but WITHOUT ANY WARRANTY; without even the implied warranty of
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 * GNU General Public License for more details.
15 * You should have received a copy of the GNU General Public License
16 * along with this program; if not, write to the Free Software
17 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
20 package de.todesbaum.util.freenet.fcp2;
23 import java.io.FileInputStream;
24 import java.io.FileNotFoundException;
25 import java.io.FileOutputStream;
26 import java.io.IOException;
27 import java.io.InputStream;
28 import java.io.Writer;
29 import java.util.ArrayList;
30 import java.util.List;
32 import de.todesbaum.util.io.Closer;
35 * Implementation of the <code>ClientPutComplexDir</code> command. This
36 * command can be used to insert directories that do not exist on disk.
38 * @author David Roden <droden@gmail.com>
41 public class ClientPutComplexDir extends ClientPutDir {
43 /** The file entries of this directory. */
44 private List<FileEntry> fileEntries = new ArrayList<FileEntry>();
46 /** Whether this request has payload. */
47 private boolean hasPayload = false;
49 /** The input streams for the payload. */
50 private File payloadFile;
52 /** The total number of bytes of the payload. */
53 private long payloadLength = 0;
55 /** The temp directory to use. */
56 private final String tempDirectory;
59 * Creates a new <code>ClientPutComplexDir</code> command with the specified identifier and URI.
60 * @param identifier The identifier of the command
61 * @param uri The URI of the command
63 public ClientPutComplexDir(String identifier, String uri) {
64 this(identifier, uri, null);
68 * Creates a new <code>ClientPutComplexDir</code> command with the specified
72 * The identifier of the command
74 * The URI of the command
75 * @param tempDirectory
76 * The temp directory to use, or {@code null} to use the default
79 public ClientPutComplexDir(String identifier, String uri, String tempDirectory) {
80 super("ClientPutComplexDir", identifier, uri);
81 this.tempDirectory = tempDirectory;
85 * Adds a file to the directory inserted by this request.
86 * @param fileEntry The file entry to add to the directory
88 public void addFileEntry(FileEntry fileEntry) {
89 if (fileEntry instanceof DirectFileEntry) {
90 if (payloadFile == null){
92 payloadFile = File.createTempFile("payload", ".dat", (tempDirectory != null) ? new File(tempDirectory) : null);
93 payloadFile.deleteOnExit();
94 } catch (IOException e) {
97 if (payloadFile != null) {
98 InputStream payloadInputStream = ((DirectFileEntry) fileEntry).getDataInputStream();
99 FileOutputStream payloadOutputStream = null;
101 payloadOutputStream = new FileOutputStream(payloadFile, true);
102 byte[] buffer = new byte[65536];
104 while ((read = payloadInputStream.read(buffer)) != -1) {
105 payloadOutputStream.write(buffer, 0, read);
107 payloadOutputStream.flush();
108 fileEntries.add(fileEntry);
109 } catch (IOException ioe1) {
112 Closer.close(payloadOutputStream);
113 Closer.close(payloadInputStream);
117 fileEntries.add(fileEntry);
125 protected void write(Writer writer) throws IOException {
128 for (FileEntry fileEntry: fileEntries) {
129 writer.write("Files." + fileIndex + ".Name=" + fileEntry.getFilename() + LINEFEED);
130 if (fileEntry.getContentType() != null) {
131 writer.write("Files." + fileIndex + ".Metadata.ContentType=" + fileEntry.getContentType() + LINEFEED);
133 writer.write("Files." + fileIndex + ".UploadFrom=" + fileEntry.getName() + LINEFEED);
134 if (fileEntry instanceof DirectFileEntry) {
136 writer.write("Files." + fileIndex + ".DataLength=" + ((DirectFileEntry) fileEntry).getDataLength() + LINEFEED);
137 payloadLength += ((DirectFileEntry) fileEntry).getDataLength();
138 } else if (fileEntry instanceof DiskFileEntry) {
139 writer.write("Files." + fileIndex + ".Filename=" + ((DiskFileEntry) fileEntry).getFilename() + LINEFEED);
140 } else if (fileEntry instanceof RedirectFileEntry) {
141 writer.write("Files." + fileIndex + ".TargetURI=" + ((RedirectFileEntry) fileEntry).getTargetURI() + LINEFEED);
151 protected boolean hasPayload() {
159 protected long getPayloadLength() {
160 return payloadLength;
167 protected InputStream getPayload() {
168 if (payloadFile != null) {
170 return new FileInputStream(payloadFile);
171 } catch (FileNotFoundException e) {
172 /* shouldn't occur. */