1 files changed, 87 insertions, 0 deletions
diff --git a/src/main/java/at/gv/egiz/pdfas/io/ByteArrayDataSink.java b/src/main/java/at/gv/egiz/pdfas/io/ByteArrayDataSink.java
new file mode 100644
index 0000000..bf74ed5
--- /dev/null
+++ b/src/main/java/at/gv/egiz/pdfas/io/ByteArrayDataSink.java
@@ -0,0 +1,87 @@
+/**
+ * 
+ */
+package at.gv.egiz.pdfas.io;
+
+import java.io.ByteArrayOutputStream;
+import java.io.IOException;
+import java.io.OutputStream;
+
+import at.gv.egiz.pdfas.api.io.DataSink;
+
+/**
+ * @author wprinz
+ * 
+ */
+public class ByteArrayDataSink implements DataSink
+{
+  /**
+   * The byte output stream.
+   */
+  ByteArrayOutputStream baos = null;
+
+  /**
+   * The mime type.
+   */
+  protected String mimeType = null;
+
+  /**
+   * The character encoding.
+   */
+  protected String characterEncoding = null;
+
+  /**
+   * @see at.gv.egiz.pdfas.api.io.DataSink#createOutputStream(java.lang.String)
+   */
+  public OutputStream createOutputStream(String mimeType) throws IOException
+  {
+    return createOutputStream(mimeType, null);
+  }
+
+  /**
+   * @see at.gv.egiz.pdfas.api.io.DataSink#createOutputStream(java.lang.String,
+   *      java.lang.String)
+   */
+  public OutputStream createOutputStream(String mimeType, String characterEncoding) throws IOException
+  {
+    this.mimeType = mimeType;
+    this.characterEncoding = characterEncoding;
+    this.baos = new ByteArrayOutputStream();
+    return this.baos;
+  }
+
+  /**
+   * @see at.gv.egiz.pdfas.api.io.DataSink#getMimeType()
+   */
+  public String getMimeType()
+  {
+    return this.mimeType;
+  }
+
+  /**
+   * @see at.gv.egiz.pdfas.api.io.DataSink#getCharacterEncoding()
+   */
+  public String getCharacterEncoding()
+  {
+    return this.characterEncoding;
+  }
+
+  /**
+   * Returns the byte data, or null if none available.
+   * 
+   * <p>
+   * Note that internally, this just calls the {@link ByteArrayOutputStream#toByteArray()} method.
+   * </p>
+   * 
+   * @return Returns the byte data, or null if none available.
+   */
+  public byte [] getData()
+  {
+    if (this.baos == null)
+    {
+      return null;
+    }
+    return this.baos.toByteArray();
+  }
+
+}