jSite: First commit : verion 4.0 (written by Bombe)

[jSite.git] / src / de / todesbaum / util / freenet / fcp2 / ClientPutComplexDir.java

diff --git a/src/de/todesbaum/util/freenet/fcp2/ClientPutComplexDir.java b/src/de/todesbaum/util/freenet/fcp2/ClientPutComplexDir.java
new file mode 100644 (file)
index 0000000..08c8a53
--- /dev/null
+++ b/src/de/todesbaum/util/freenet/fcp2/ClientPutComplexDir.java
@@ -0,0 +1,121 @@
+/*
+ * todesbaum-lib - 
+ * Copyright (C) 2006 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.IOException;
+import java.io.InputStream;
+import java.io.SequenceInputStream;
+import java.io.Writer;
+import java.util.ArrayList;
+import java.util.List;
+import java.util.Vector;
+
+/**
+ * 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: ClientPutComplexDir.java 356 2006-03-24 15:13:38Z bombe $
+ */
+public class ClientPutComplexDir extends ClientPutDir {
+
+       /** 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 List<InputStream> payloadInputStreams = new ArrayList<InputStream>();
+       
+       /** The total number of bytes of the payload. */
+       private long payloadLength = 0;
+
+       /**
+        * 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) {
+               super("ClientPutComplexDir", identifier, uri);
+       }
+
+       /**
+        * Adds a file to the directory inserted by this request.
+        * @param fileEntry The file entry to add to the directory
+        */
+       public void addFileEntry(FileEntry fileEntry) {
+               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();
+                               payloadInputStreams.add(((DirectFileEntry) fileEntry).getDataInputStream());
+                       } 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() {
+               /* grr. use Vector here because it returns an Enumeration. */
+               Vector<InputStream> inputStreams = new Vector<InputStream>(payloadInputStreams);
+               return new SequenceInputStream(inputStreams.elements());
+       }
+
+}