src/de/todesbaum/util/freenet/fcp2/ClientPutComplexDir.java

   1 /*
   2  * todesbaum-lib -
   3  * Copyright (C) 2006 David Roden
   4  *
   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.
   9  *
  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.
  14  *
  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.
  18  */
  19
  20 package de.todesbaum.util.freenet.fcp2;
  21
  22 import java.io.File;
  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;
  31
  32 import de.todesbaum.util.io.Closer;
  33
  34 /**
  35  * Implementation of the <code>ClientPutComplexDir</code> command. This command
  36  * can be used to insert directories that do not exist on disk.
  37  *
  38  * @author David Roden &lt;droden@gmail.com&gt;
  39  * @version $Id$
  40  */
  41 public class ClientPutComplexDir extends ClientPutDir {
  42
  43         /** The file entries of this directory. */
  44         private List<FileEntry> fileEntries = new ArrayList<FileEntry>();
  45
  46         /** Whether this request has payload. */
  47         private boolean hasPayload = false;
  48
  49         /** The input streams for the payload. */
  50         private File payloadFile;
  51
  52         /** The total number of bytes of the payload. */
  53         private long payloadLength = 0;
  54
  55         /** The temp directory to use. */
  56         private final String tempDirectory;
  57
  58         /**
  59          * Creates a new <code>ClientPutComplexDir</code> command with the specified
  60          * identifier and URI.
  61          *
  62          * @param identifier
  63          *            The identifier of the command
  64          * @param uri
  65          *            The URI of the command
  66          */
  67         public ClientPutComplexDir(String identifier, String uri) {
  68                 this(identifier, uri, null);
  69         }
  70
  71         /**
  72          * Creates a new <code>ClientPutComplexDir</code> command with the specified
  73          * identifier and URI.
  74          *
  75          * @param identifier
  76          *            The identifier of the command
  77          * @param uri
  78          *            The URI of the command
  79          * @param tempDirectory
  80          *            The temp directory to use, or {@code null} to use the default
  81          *            temp directory
  82          */
  83         public ClientPutComplexDir(String identifier, String uri, String tempDirectory) {
  84                 super("ClientPutComplexDir", identifier, uri);
  85                 this.tempDirectory = tempDirectory;
  86         }
  87
  88         /**
  89          * Adds a file to the directory inserted by this request.
  90          *
  91          * @param fileEntry
  92          *            The file entry to add to the directory
  93          * @throws IOException
  94          *             if an I/O error occurs when creating the payload stream
  95          */
  96         public void addFileEntry(FileEntry fileEntry) throws IOException {
  97                 if (fileEntry instanceof DirectFileEntry) {
  98                         if (payloadFile == null) {
  99                                 try {
 100                                         payloadFile = File.createTempFile("payload", ".dat", (tempDirectory != null) ? new File(tempDirectory) : null);
 101                                         payloadFile.deleteOnExit();
 102                                 } catch (IOException e) {
 103                                         /* ignore. */
 104                                 }
 105                         }
 106                         if (payloadFile != null) {
 107                                 InputStream payloadInputStream = ((DirectFileEntry) fileEntry).getDataInputStream();
 108                                 FileOutputStream payloadOutputStream = null;
 109                                 try {
 110                                         payloadOutputStream = new FileOutputStream(payloadFile, true);
 111                                         byte[] buffer = new byte[65536];
 112                                         int read = 0;
 113                                         while ((read = payloadInputStream.read(buffer)) != -1) {
 114                                                 payloadOutputStream.write(buffer, 0, read);
 115                                         }
 116                                         payloadOutputStream.flush();
 117                                         fileEntries.add(fileEntry);
 118                                 } catch (IOException ioe1) {
 119                                         payloadFile.delete();
 120                                         throw ioe1;
 121                                 } finally {
 122                                         Closer.close(payloadOutputStream);
 123                                         Closer.close(payloadInputStream);
 124                                 }
 125                         }
 126                 } else {
 127                         fileEntries.add(fileEntry);
 128                 }
 129         }
 130
 131         /**
 132          * {@inheritDoc}
 133          */
 134         @Override
 135         protected void write(Writer writer) throws IOException {
 136                 super.write(writer);
 137                 int fileIndex = 0;
 138                 for (FileEntry fileEntry : fileEntries) {
 139                         writer.write("Files." + fileIndex + ".Name=" + fileEntry.getFilename() + LINEFEED);
 140                         if (fileEntry.getContentType() != null) {
 141                                 writer.write("Files." + fileIndex + ".Metadata.ContentType=" + fileEntry.getContentType() + LINEFEED);
 142                         }
 143                         writer.write("Files." + fileIndex + ".UploadFrom=" + fileEntry.getName() + LINEFEED);
 144                         if (fileEntry instanceof DirectFileEntry) {
 145                                 hasPayload = true;
 146                                 writer.write("Files." + fileIndex + ".DataLength=" + ((DirectFileEntry) fileEntry).getDataLength() + LINEFEED);
 147                                 payloadLength += ((DirectFileEntry) fileEntry).getDataLength();
 148                         } else if (fileEntry instanceof DiskFileEntry) {
 149                                 writer.write("Files." + fileIndex + ".Filename=" + ((DiskFileEntry) fileEntry).getFilename() + LINEFEED);
 150                         } else if (fileEntry instanceof RedirectFileEntry) {
 151                                 writer.write("Files." + fileIndex + ".TargetURI=" + ((RedirectFileEntry) fileEntry).getTargetURI() + LINEFEED);
 152                         }
 153                         fileIndex++;
 154                 }
 155         }
 156
 157         /**
 158          * {@inheritDoc}
 159          */
 160         @Override
 161         protected boolean hasPayload() {
 162                 return hasPayload;
 163         }
 164
 165         /**
 166          * {@inheritDoc}
 167          */
 168         @Override
 169         protected long getPayloadLength() {
 170                 return payloadLength;
 171         }
 172
 173         /**
 174          * {@inheritDoc}
 175          */
 176         @Override
 177         protected InputStream getPayload() {
 178                 if (payloadFile != null) {
 179                         try {
 180                                 return new FileInputStream(payloadFile);
 181                         } catch (FileNotFoundException e) {
 182                                 /* shouldn't occur. */
 183                         }
 184                 }
 185                 return null;
 186         }
 187
 188 }