/**
 * 
 */
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();
  }

}