/**
 * <copyright> Copyright 2006 by Know-Center, Graz, Austria </copyright>
 * PDF-AS has been contracted by the E-Government Innovation Center EGIZ, a
 * joint initiative of the Federal Chancellery Austria and Graz University of
 * Technology.
 *
 * Licensed under the EUPL, Version 1.1 or - as soon they will be approved by
 * the European Commission - subsequent versions of the EUPL (the "Licence");
 * You may not use this work except in compliance with the Licence.
 * You may obtain a copy of the Licence at:
 * http://www.osor.eu/eupl/
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the Licence is distributed on an "AS IS" basis,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the Licence for the specific language governing permissions and
 * limitations under the Licence.
 *
 * This product combines work with different licenses. See the "NOTICE" text
 * file for details on the various modules and licenses.
 * The "NOTICE" text file is part of the distribution. Any derivative works
 * that you distribute must include a readable copy of the "NOTICE" text file.
 */
package at.gv.egiz.pdfas.impl.input.correction;

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

import at.gv.egiz.pdfas.api.io.DataSource;
import at.gv.egiz.pdfas.exceptions.ErrorCode;
import at.gv.egiz.pdfas.exceptions.framework.CorrectorException;
import at.gv.egiz.pdfas.framework.input.PdfDataSource;
import at.gv.egiz.pdfas.framework.input.correction.Corrector;
import at.gv.egiz.pdfas.impl.input.ByteArrayPdfDataSourceImpl;
import at.gv.egiz.pdfas.utils.PDFASUtils;
import at.knowcenter.wag.egov.egiz.exceptions.PDFDocumentException;

import com.lowagie.text.DocumentException;
import com.lowagie.text.pdf.PdfReader;
import com.lowagie.text.pdf.PdfStamper;

/**
 * Corrects a document using iText.
 *
 * @author wprinz
 */
public class InternalCorrector implements Corrector
{

  /**
   * @see at.gv.egiz.pdfas.framework.input.correction.Corrector#correctDocument(at.gv.egiz.pdfas.framework.input.PdfDataSource)
   */
  public PdfDataSource correctDocument(PdfDataSource document) throws CorrectorException
  {
    try
    {
      PdfReader reader = PDFASUtils.createPdfReaderCheckingPermissions(document);

      ByteArrayOutputStream baos = new ByteArrayOutputStream(document.getLength());

      PdfStamper stamper = new PdfStamper(reader, baos, '\0', false);
      stamper.close();

      baos.close();
      byte[] corrected_pdf = baos.toByteArray();

      return new ByteArrayPdfDataSourceImpl(corrected_pdf);
    }
    catch (DocumentException e)
    {
      throw new CorrectorException(ErrorCode.CORRECTOR_EXCEPTION, e);
    }
    catch (IOException e)
    {
      throw new CorrectorException(ErrorCode.CORRECTOR_EXCEPTION, e);
    } catch (PDFDocumentException e) {
      throw new CorrectorException(e.getErrorCode(), e);
   }
  }

}