package at.gv.egiz.pdfas.web.io;

import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.OutputStream;

import at.gv.egiz.pdfas.api.io.DataSink;

/**
 * A simple implementation of the {@link DataSink} interface writing to a byte[].
 * 
 * @author exthex
 *
 */
public class ByteArrayDataSink implements DataSink {
   
   private ByteArrayOutputStream baos;
   private String mimeType;
   private String characterEncoding;

   public OutputStream createOutputStream(String mimeType) throws IOException {
      this.mimeType = mimeType;
      baos = new ByteArrayOutputStream();
      return baos;
   }

   public OutputStream createOutputStream(String mimeType, String characterEncoding)
         throws IOException {
      this.mimeType = mimeType;
      this.characterEncoding = characterEncoding;
      baos = new ByteArrayOutputStream();
      return baos;
   }
   
   public byte[] getData() {
      return baos.toByteArray();
   }

   public String getMimeType() {
      return this.mimeType;
   }

   public String getCharacterEncoding() {
      return this.characterEncoding;
   }

   public void setMimeType(String mimeType) {
      this.mimeType = mimeType;
   }

   public void setCharacterEncoding(String characterEncoding) {
      this.characterEncoding = characterEncoding;
   }

}