--- /dev/null
+/*
+ * jSite2 - ClientPutComplexDir.java -
+ * Copyright © 2008 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 net.pterodactylus.util.fcp;
+
+import java.io.IOException;
+import java.io.InputStream;
+import java.io.OutputStream;
+import java.io.SequenceInputStream;
+import java.util.ArrayList;
+import java.util.Collections;
+import java.util.List;
+import java.util.Map;
+import java.util.Map.Entry;
+
+import net.pterodactylus.util.fcp.FileEntry.DirectFileEntry;
+
+/**
+ * The “ClientPutComplexDir” lets you upload a directory with different sources
+ * for each file.
+ *
+ * @see FileEntry
+ * @author David ‘Bombe’ Roden <bombe@freenetproject.org>
+ * @version $Id$
+ */
+public class ClientPutComplexDir extends FcpMessage {
+
+ /** The index for added file entries. */
+ private int fileIndex = 0;
+
+ /** The input streams from {@link DirectFileEntry}s. */
+ private final List<InputStream> directFileInputStreams = new ArrayList<InputStream>();
+
+ /**
+ * Creates a new “ClientPutComplexDir” with the given identifier and URI.
+ *
+ * @param identifier
+ * The identifier of the request
+ * @param uri
+ * The URI to insert the directory to
+ */
+ public ClientPutComplexDir(String identifier, String uri) {
+ super("ClientPutComplexDir");
+ setField("Identifier", identifier);
+ setField("URI", uri);
+ }
+
+ /**
+ * Sets the verbosity of the request.
+ *
+ * @param verbosity
+ * The verbosity of the request
+ */
+ public void setVerbosity(Verbosity verbosity) {
+ setField("Verbosity", String.valueOf(verbosity));
+ }
+
+ /**
+ * Sets the maximum number of retries for failed blocks.
+ *
+ * @param maxRetries
+ * The maximum number of retries for failed blocks, or
+ * <code>-1</code> to retry endlessly
+ */
+ public void setMaxRetries(int maxRetries) {
+ setField("MaxRetries", String.valueOf(maxRetries));
+ }
+
+ /**
+ * Sets the priority of the request.
+ *
+ * @param priority
+ * The priority of the request
+ */
+ public void setPriority(Priority priority) {
+ setField("PriorityClass", String.valueOf(priority));
+ }
+
+ /**
+ * Sets whether to generate the final URI only.
+ *
+ * @param getCHKOnly
+ * <code>true</code> to generate the final CHK only,
+ * <code>false</code> to complete the insert
+ */
+ public void setGetCHKOnly(boolean getCHKOnly) {
+ setField("GetCHKOnly", String.valueOf(getCHKOnly));
+ }
+
+ /**
+ * Sets whether the request is on the global queue.
+ *
+ * @param global
+ * <code>true</code> to put the request on the global queue,
+ * <code>false</code> to put it on the client-local queue
+ */
+ public void setGlobal(boolean global) {
+ setField("Global", String.valueOf(global));
+ }
+
+ /**
+ * Sets whether the node should not try to compress the data.
+ *
+ * @param dontCompress
+ * <code>true</code> to skip compression of the data,
+ * <code>false</code> to try and compress the data
+ */
+ public void setDontCompress(boolean dontCompress) {
+ setField("DontCompress", String.valueOf(dontCompress));
+ }
+
+ /**
+ * Sets the client token of the request.
+ *
+ * @param clientToken
+ * The client token of the request
+ */
+ public void setClientToken(String clientToken) {
+ setField("ClientToken", clientToken);
+ }
+
+ /**
+ * Sets the persistence of the request.
+ *
+ * @param persistence
+ * The persistence of the request
+ */
+ public void setPersistence(Persistence persistence) {
+ setField("Persistence", String.valueOf(persistence));
+ }
+
+ /**
+ * Sets the target filename of the request. This is useful for inserts that
+ * go to “CHK@” only and creates a manifest with a single file.
+ *
+ * @param targetFilename
+ * The target filename
+ */
+ public void setTargetFilename(String targetFilename) {
+ setField("TargetFilename", targetFilename);
+ }
+
+ /**
+ * Sets whether to encode the complete data early to generate the
+ * {@link URIGenerated} message early.
+ *
+ * @param earlyEncode
+ * <code>true</code> to encode the complete data early,
+ * <code>false</code> otherwise
+ */
+ public void setEarlyEncode(boolean earlyEncode) {
+ setField("EarlyEncode", String.valueOf(earlyEncode));
+ }
+
+ /**
+ * Sets the default name. This is the name of the file that should be shown
+ * if no file was specified.
+ *
+ * @param defaultName
+ * The default name
+ */
+ public void setDefaultName(String defaultName) {
+ setField("DefaultName", defaultName);
+ }
+
+ /**
+ * Adds an entry for a file.
+ *
+ * @param fileEntry
+ * The file entry to add
+ */
+ public void addFileEntry(FileEntry fileEntry) {
+ Map<String, String> fields = fileEntry.getFields();
+ for (Entry<String, String> fieldEntry: fields.entrySet()) {
+ setField("Files." + fileIndex + "." + fieldEntry.getKey(), fieldEntry.getValue());
+ }
+ fileIndex++;
+ if (fileEntry instanceof FileEntry.DirectFileEntry) {
+ directFileInputStreams.add(((DirectFileEntry) fileEntry).getInputStream());
+ }
+ }
+
+ /**
+ * {@inheritDoc}
+ * <p>
+ * Do not call this method to add input streams! The input streams, if any,
+ * will be taken directly from the {@link FileEntry}s and the stream you
+ * set here will be overridden!
+ */
+ @Override
+ public void setPayloadInputStream(InputStream payloadInputStream) {
+ /* do nothing. */
+ }
+
+ /**
+ * {@inheritDoc}
+ */
+ @Override
+ public void write(OutputStream outputStream) throws IOException {
+ /* create payload stream. */
+ setPayloadInputStream(new SequenceInputStream(Collections.enumeration(directFileInputStreams)));
+ /* write out all the fields. */
+ super.write(outputStream);
+ }
+
+}
--- /dev/null
+/*
+ * jSite2 - FileEntry.java -
+ * Copyright © 2008 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 net.pterodactylus.util.fcp;
+
+import java.io.InputStream;
+import java.util.HashMap;
+import java.util.Map;
+
+/**
+ * Container class for file entry data.
+ *
+ * @see ClientPutComplexDir#addFileEntry(FileEntry)
+ * @author David ‘Bombe’ Roden <bombe@freenetproject.org>
+ * @version $Id$
+ */
+public abstract class FileEntry {
+
+ /** The name of the file. */
+ protected final String name;
+
+ /** The upload source of the file. */
+ protected final UploadFrom uploadFrom;
+
+ /**
+ * Creates a new file entry with the given name and upload source.
+ *
+ * @param name
+ * The name of the file
+ * @param uploadFrom
+ * The upload source of the file
+ */
+ protected FileEntry(String name, UploadFrom uploadFrom) {
+ this.name = name;
+ this.uploadFrom = uploadFrom;
+ }
+
+ /**
+ * Creates a new file entry for a file that should be transmitted to the
+ * node in the payload of the message.
+ *
+ * @param name
+ * The name of the file
+ * @param contentType
+ * The content type of the file, or <code>null</code> to let
+ * the node auto-detect it
+ * @param length
+ * The length of the file
+ * @param dataInputStream
+ * The input stream of the file
+ * @return A file entry
+ */
+ public static FileEntry createDirectFileEntry(String name, String contentType, long length, InputStream dataInputStream) {
+ return new DirectFileEntry(name, contentType, length, dataInputStream);
+ }
+
+ /**
+ * Creates a new file entry for a file that should be uploaded from disk.
+ *
+ * @param name
+ * The name of the file
+ * @param filename
+ * The name of the file on disk
+ * @param contentType
+ * The content type of the file, or <code>null</code> to let
+ * the node auto-detect it
+ * @param length
+ * The length of the file, or <code>-1</code> to not specify a
+ * size
+ * @return A file entry
+ */
+ public static FileEntry createDiskFileEntry(String name, String filename, String contentType, long length) {
+ return new DiskFileEntry(name, filename, contentType, length);
+ }
+
+ /**
+ * Creates a new file entry for a file that redirects to another URI.
+ *
+ * @param name
+ * The name of the file
+ * @param targetURI
+ * The target URI of the redirect
+ * @return A file entry
+ */
+ public static FileEntry createRedirectFileEntry(String name, String targetURI) {
+ return new RedirectFileEntry(name, targetURI);
+ }
+
+ /**
+ * Returns the fields for this file entry.
+ *
+ * @return The fields for this file entry
+ */
+ abstract Map<String, String> getFields();
+
+ /**
+ * A file entry for a file that should be transmitted in the payload of the
+ * {@link ClientPutComplexDir} message.
+ *
+ * @author David ‘Bombe’ Roden <bombe@freenetproject.org>
+ * @version $Id$
+ */
+ static class DirectFileEntry extends FileEntry {
+
+ /** The content type of the data. */
+ private final String contentType;
+
+ /** The length of the data. */
+ private final long length;
+
+ /** The input stream of the data. */
+ private final InputStream inputStream;
+
+ /**
+ * Creates a new direct file entry with content type auto-detection.
+ *
+ * @param name
+ * The name of the file
+ * @param length
+ * The length of the file
+ * @param inputStream
+ * The input stream of the file
+ */
+ public DirectFileEntry(String name, long length, InputStream inputStream) {
+ this(name, null, length, inputStream);
+ }
+
+ /**
+ * Creates a new direct file entry.
+ *
+ * @param name
+ * The name of the file
+ * @param contentType
+ * The content type of the file, or <code>null</code> to
+ * let the node auto-detect it
+ * @param length
+ * The length of the file
+ * @param inputStream
+ * The input stream of the file
+ */
+ public DirectFileEntry(String name, String contentType, long length, InputStream inputStream) {
+ super(name, UploadFrom.direct);
+ this.contentType = contentType;
+ this.length = length;
+ this.inputStream = inputStream;
+ }
+
+ /**
+ * {@inheritDoc}
+ */
+ @Override
+ Map<String, String> getFields() {
+ Map<String, String> fields = new HashMap<String, String>();
+ fields.put("Name", name);
+ fields.put("UploadFrom", String.valueOf(uploadFrom));
+ fields.put("DataLength", String.valueOf(length));
+ if (contentType != null) {
+ fields.put("Metadata.ContentType", contentType);
+ }
+ return fields;
+ }
+
+ /**
+ * Returns the input stream of the file.
+ *
+ * @return The input stream of the file
+ */
+ InputStream getInputStream() {
+ return inputStream;
+ }
+
+ }
+
+ /**
+ * A file entry for a file that should be uploaded from the disk.
+ *
+ * @author David ‘Bombe’ Roden <bombe@freenetproject.org>
+ * @version $Id$
+ */
+ static class DiskFileEntry extends FileEntry {
+
+ /** The name of the on-disk file. */
+ private final String filename;
+
+ /** The content type of the file. */
+ private final String contentType;
+
+ /** The length of the file. */
+ private final long length;
+
+ /**
+ * Creates a new disk file entry.
+ *
+ * @param name
+ * The name of the file
+ * @param filename
+ * The name of the on-disk file
+ * @param length
+ * The length of the file
+ */
+ public DiskFileEntry(String name, String filename, long length) {
+ this(name, filename, null, length);
+ }
+
+ /**
+ * Creates a new disk file entry.
+ *
+ * @param name
+ * The name of the file
+ * @param filename
+ * The name of the on-disk file
+ * @param contentType
+ * The content type of the file, or <code>null</code> to
+ * let the node auto-detect it
+ * @param length
+ * The length of the file
+ */
+ public DiskFileEntry(String name, String filename, String contentType, long length) {
+ super(name, UploadFrom.disk);
+ this.filename = filename;
+ this.contentType = contentType;
+ this.length = length;
+ }
+
+ /**
+ * {@inheritDoc}
+ */
+ @Override
+ Map<String, String> getFields() {
+ Map<String, String> fields = new HashMap<String, String>();
+ fields.put("Name", name);
+ fields.put("UploadFrom", String.valueOf(uploadFrom));
+ fields.put("Filename", filename);
+ if (length > -1) {
+ fields.put("DataSize", String.valueOf(length));
+ }
+ if (contentType != null) {
+ fields.put("Metadata.ContentType", contentType);
+ }
+ return fields;
+ }
+
+ }
+
+ /**
+ * A file entry for a file that redirects to another URI.
+ *
+ * @author David ‘Bombe’ Roden <bombe@freenetproject.org>
+ * @version $Id$
+ */
+ static class RedirectFileEntry extends FileEntry {
+
+ /** The target URI of the redirect. */
+ private String targetURI;
+
+ /**
+ * Creates a new redirect file entry.
+ *
+ * @param name
+ * The name of the file
+ * @param targetURI
+ * The target URI of the redirect
+ */
+ public RedirectFileEntry(String name, String targetURI) {
+ super(name, UploadFrom.redirect);
+ this.targetURI = targetURI;
+ }
+
+ /**
+ * {@inheritDoc}
+ */
+ @Override
+ Map<String, String> getFields() {
+ Map<String, String> fields = new HashMap<String, String>();
+ fields.put("Name", name);
+ fields.put("UploadFrom", String.valueOf(uploadFrom));
+ fields.put("TargetURI", targetURI);
+ return fields;
+ }
+
+ }
+
+}
projects / jSite2.git / commitdiff