/*******************************************************************************
 * <copyright> Copyright 2014 by E-Government Innovation Center EGIZ, 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.cli.test;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;

import javax.activation.DataSource;

import org.apache.commons.io.IOUtils;

import at.gv.egiz.pdfas.common.settings.ISettings;
import at.gv.egiz.pdfas.lib.api.ByteArrayDataSource;
import at.gv.egiz.pdfas.lib.api.Configuration;
import at.gv.egiz.pdfas.lib.api.PdfAs;
import at.gv.egiz.pdfas.lib.api.PdfAsFactory;
import at.gv.egiz.pdfas.lib.api.sign.IPlainSigner;
import at.gv.egiz.pdfas.lib.api.sign.SignParameter;
import at.gv.egiz.pdfas.lib.api.sign.SignResult;
import at.gv.egiz.pdfas.sigs.pades.PAdESSignerKeystore;

public class SignaturProfileTest {

	public static final String sourcePDFA = "/home/afitzek/simple_pdfa.pdf";
	public static final String sourcePDF = "/home/afitzek/simple.pdf";
	public static final String targetFolder = "/home/afitzek/tmp/sigres/";

	public static final String KS_FILE = "/home/afitzek/devel/pdfas_neu/test.p12";
	public static final String KS_ALIAS = "ecc_test";
	public static final String KS_TYPE = "PKCS12";
	public static final String KS_PASS = "123456";
	public static final String KS_KEY_PASS = "123456";

	/*
	 * -p SIGNATURBLOCK_SMALL_DE -c ks -m sign -ksf
	 * /home/afitzek/devel/pdfas_neu/test.p12 -kst PKCS12 -ksa ecc_test -kskp
	 * 123456 -kssp 123456 /home/afitzek/simple2.pdf
	 */

	public static void main(String[] args) {
		String user_home = System.getProperty("user.home");
		String pdfas_dir = user_home + File.separator + ".pdfas";
		PdfAs pdfas = PdfAsFactory.createPdfAs(new File(pdfas_dir));
		try {
			Configuration config = pdfas.getConfiguration();
			ISettings settings = (ISettings) config;
			List<String> signatureProfiles = new ArrayList<String>();
			
			List<String> signaturePDFAProfiles = new ArrayList<String>();
			
			Iterator<String> itKeys = settings.getFirstLevelKeys(
					"sig_obj.types.").iterator();
			while (itKeys.hasNext()) {
				String key = itKeys.next();
				String profile = key.substring("sig_obj.types.".length());
				System.out.println("[" + profile + "]: "
						+ settings.getValue(key));
				if (settings.getValue(key).equals("on")) {
					signatureProfiles.add(profile);
					if(profile.contains("PDFA")) {
						signaturePDFAProfiles.add(profile);
					}
				}
			}

			byte[] input = IOUtils.toByteArray(new FileInputStream(sourcePDF));

			IPlainSigner signer = new PAdESSignerKeystore(KS_FILE, KS_ALIAS, KS_PASS, KS_KEY_PASS, KS_TYPE);
			
			Iterator<String> itProfiles = signatureProfiles.iterator();
			while (itProfiles.hasNext()) {
				String profile = itProfiles.next();
				System.out.println("Testing " + profile);

				DataSource source = new ByteArrayDataSource(input);

				FileOutputStream fos = new FileOutputStream(targetFolder + profile + ".pdf");
				SignParameter signParameter = PdfAsFactory.createSignParameter(
						config, source, fos);
				
				signParameter.setPlainSigner(signer);
				signParameter.setSignatureProfileId(profile);
				
				SignResult result = pdfas.sign(signParameter);
				
				fos.close();
			}
			
			byte[] inputPDFA = IOUtils.toByteArray(new FileInputStream(sourcePDFA));
			
			Iterator<String> itPDFAProfiles = signaturePDFAProfiles.iterator();
			while (itPDFAProfiles.hasNext()) {
				String profile = itPDFAProfiles.next();
				System.out.println("Testing " + profile);

				DataSource source = new ByteArrayDataSource(inputPDFA);
				FileOutputStream fos = new FileOutputStream(targetFolder + "PDFA_" + profile + ".pdf");
				SignParameter signParameter = PdfAsFactory.createSignParameter(
						config, source, fos);
				
				signParameter.setPlainSigner(signer);
				signParameter.setSignatureProfileId(profile);
				
				SignResult result = pdfas.sign(signParameter);
				
				fos.close();
			}
		} catch (Throwable e) {
			e.printStackTrace();
		}
	}

}