/*
 * Copyright 2012 by A-SIT, Secure Information Technology Center Austria
 *
 * 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://joinup.ec.europa.eu/software/page/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.
 */
package at.asit.pdfover.gui.composites;

// Imports
import java.awt.Desktop;
import java.io.File;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;

import org.apache.commons.io.FilenameUtils;
import org.eclipse.swt.SWT;
import org.eclipse.swt.events.SelectionAdapter;
import org.eclipse.swt.events.SelectionEvent;
import org.eclipse.swt.graphics.Font;
import org.eclipse.swt.graphics.FontData;
import org.eclipse.swt.layout.FormAttachment;
import org.eclipse.swt.layout.FormData;
import org.eclipse.swt.layout.FormLayout;
import org.eclipse.swt.widgets.Button;
import org.eclipse.swt.widgets.Composite;
import org.eclipse.swt.widgets.Display;
import org.eclipse.swt.widgets.FileDialog;
import org.eclipse.swt.widgets.Label;
import org.eclipse.swt.widgets.Link;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import at.asit.pdfover.gui.Constants;
import at.asit.pdfover.gui.utils.Messages;
import at.asit.pdfover.gui.workflow.states.State;
import at.asit.pdfover.signator.DocumentSource;

/**
 * GUI component for Output State
 */
public class OutputComposite extends StateComposite {

	/**
	 * SLF4J Logger instance
	 **/
	static final Logger log = LoggerFactory.getLogger(OutputComposite.class);

	private File inputFile;

	File outputFile = null;

	String outputDir = null;

	/**
	 * @return the outputDir
	 */
	public String getOutputDir() {
		return this.outputDir;
	}

	/**
	 * @param outputDir
	 *            the outputDir to set
	 */
	public void setOutputDir(String outputDir) {
		this.outputDir = outputDir;
	}

	/**
	 * Sets the input file
	 * 
	 * @param inputFile
	 *            the input file
	 */
	public void setInputFile(File inputFile) {
		this.inputFile = inputFile;
	}

	/**
	 * Gets the input file
	 * 
	 * @return the input file
	 */
	public File getInputFile() {
		return this.inputFile;
	}

	/**
	 * Saves the file
	 * 
	 * @throws IOException
	 */
	void saveFile() throws IOException {
		FileDialog save = new FileDialog(OutputComposite.this.getShell(),
				SWT.SAVE | SWT.NATIVE);
		save.setFilterExtensions(new String[] { "*.pdf", "*" }); //$NON-NLS-1$ //$NON-NLS-2$
		save.setFilterNames(new String[] { 
				Messages.getString("common.PDFExtension_Description"), //$NON-NLS-1$
				Messages.getString("common.AllExtension_Description")}); //$NON-NLS-1$

		String proposed = OutputComposite.this.getInputFile().getAbsolutePath();

		String path = this.getOutputDir();

		if (path == null || path.equals("")) { //$NON-NLS-1$
			path = FilenameUtils.getFullPath(proposed);
		}

		if(!path.endsWith(File.separator)) {
			path += File.separator;
		}
		
		String name = FilenameUtils.getName(proposed);

		String extension = FilenameUtils.getExtension(proposed);

		name = FilenameUtils.removeExtension(name);

		proposed = path + name + Constants.SIGNED_SUFFIX + "." + extension; //$NON-NLS-1$

		if (this.getOutputDir() != null && !this.getOutputDir().equals("")) //$NON-NLS-1$
		{
			// Output directory is set save there without asking user...
			
			saveResultAsFile(proposed);
		} else {

			save.setFileName(proposed);

			String target = save.open();

			if (target != null) {
				saveResultAsFile(target);
			} else {
				// Show save message ...
				this.lnk_saved_file.setText(Messages
						.getString("output.link_save_message")); //$NON-NLS-1$
				this.btn_save.setVisible(true);
			}
		}
	}

	/**
	 * @param target
	 * @throws FileNotFoundException
	 * @throws IOException
	 */
	private void saveResultAsFile(String target) throws FileNotFoundException,
			IOException {
		File targetFile = new File(target);

		DocumentSource source = OutputComposite.this
				.getSignedDocument();

		FileOutputStream outstream = new FileOutputStream(targetFile);
		outstream.write(source.getByteArray(), 0,
				source.getByteArray().length);
		outstream.close();

		OutputComposite.this.savedFile = targetFile;

		this.outputFile = targetFile;
		// Show open message ...
		this.lnk_saved_file.setText(Messages
				.getString("output.link_open_message")); //$NON-NLS-1$
		this.layout(true);
		this.btn_save.setVisible(false);
	}

	/**
	 * SelectionListener for save button
	 */
	private final class SaveSelectionListener extends SelectionAdapter {
		/**
		 * Empty constructor
		 */
		public SaveSelectionListener() {
		}

		@Override
		public void widgetSelected(SelectionEvent e) {
			try {
				OutputComposite.this.saveFile();
			} catch (Exception ex) {
				log.error("SaveSelectionListener: ", ex); //$NON-NLS-1$
			}
		}
	}

	/**
	 * Selection Listener for open button
	 */
	private final class OpenSelectionListener extends SelectionAdapter {
		/**
		 * Empty constructor
		 */
		public OpenSelectionListener() {
		}

		@Override
		public void widgetSelected(SelectionEvent e) {
			try {

				if (OutputComposite.this.outputFile != null) {

					if (OutputComposite.this.outputFile.exists()) {
						// Desktop supported check already done in constructor
						if (Desktop.isDesktopSupported()) {
							Desktop.getDesktop().open(
									OutputComposite.this.outputFile);
						} else {
							log.error("SWT Desktop is not supported on this platform!"); //$NON-NLS-1$
						}
						return;
					}
				}
			} catch (Exception ex) {
				log.error("OpenSelectionListener: ", ex); //$NON-NLS-1$
			}
		}
	}

	File savedFile = null;

	private DocumentSource signedDocument;

	/**
	 * Create the composite.
	 * 
	 * @param parent
	 *            The parent composite
	 * @param style
	 *            The swt style
	 * @param state
	 *            The owning state
	 */
	public OutputComposite(Composite parent, int style, State state) {
		super(parent, style, state);

		this.setLayout(new FormLayout());

		this.lbl_success_message = new Label(this, SWT.NATIVE | SWT.RESIZE);
		FormData fd_lbl_success_message = new FormData();
		fd_lbl_success_message.top = new FormAttachment(40, 0);
		fd_lbl_success_message.left = new FormAttachment(0);
		fd_lbl_success_message.right = new FormAttachment(100);
		this.lbl_success_message.setLayoutData(fd_lbl_success_message);
		this.lbl_success_message.setAlignment(SWT.CENTER);
		this.lbl_success_message.setText(Messages
				.getString("output.success_message")); //$NON-NLS-1$

		FontData[] fD1 = this.lbl_success_message.getFont().getFontData();
		fD1[0].setHeight(Constants.TEXT_SIZE_BIG);
		this.lbl_success_message.setFont(new Font(Display.getCurrent(), fD1[0]));

		this.lnk_saved_file = new Link(this, SWT.NATIVE | SWT.RESIZE);
		this.lnk_saved_file.setText(Messages
				.getString("output.link_save_message")); //$NON-NLS-1$
		FormData fd_lnk_saved_file = new FormData();
		fd_lnk_saved_file.top = new FormAttachment(this.lbl_success_message, 10);
		fd_lnk_saved_file.left = new FormAttachment(this.lbl_success_message, 0,
				SWT.CENTER);
		// fd_lnk_saved_file.right = new FormAttachment(100);
		this.lnk_saved_file.setLayoutData(fd_lnk_saved_file);

		this.lnk_saved_file.addSelectionListener(new OpenSelectionListener());

		FontData[] fD2 = this.lnk_saved_file.getFont().getFontData();
		fD2[0].setHeight(Constants.TEXT_SIZE_NORMAL);
		this.lnk_saved_file.setFont(new Font(Display.getCurrent(), fD2[0]));

		this.btn_save = new Button(this, SWT.NATIVE | SWT.RESIZE);
		this.btn_save.setText(Messages.getString("common.Save")); //$NON-NLS-1$

		FontData[] fD_btn_save = this.btn_save.getFont().getFontData();
		fD_btn_save[0].setHeight(Constants.TEXT_SIZE_BUTTON);
		this.btn_save.setFont(new Font(Display.getCurrent(), fD_btn_save[0]));

		FormData fd_btn_save = new FormData();
		fd_btn_save.top = new FormAttachment(this.lnk_saved_file, 10);
		fd_btn_save.left = new FormAttachment(this.lnk_saved_file, 0,
				SWT.CENTER);
		this.btn_save.setLayoutData(fd_btn_save);

		this.btn_save.setVisible(false);
		this.btn_save.addSelectionListener(new SaveSelectionListener());

	}

	String tempDirectory;

	/**
	 * @param tempDirectory
	 */
	public void setTempDirectory(String tempDirectory) {
		this.tempDirectory = tempDirectory;
	}

	/**
	 * Gets the signed document
	 * 
	 * @return the signed document
	 */
	public DocumentSource getSignedDocument() {
		return this.signedDocument;
	}

	/**
	 * Sets the signed document
	 * 
	 * @param signedDocument
	 *            the signed document
	 */
	public void setSignedDocument(final DocumentSource signedDocument) {
		this.signedDocument = signedDocument;
	}

	@Override
	protected void checkSubclass() {
		// Disable the check that prevents subclassing of SWT components
	}

	private boolean save_showed = false;

	private Link lnk_saved_file;

	private Button btn_save;

	private Label lbl_success_message;

	/*
	 * (non-Javadoc)
	 * 
	 * @see at.asit.pdfover.gui.composites.StateComposite#doLayout()
	 */
	@Override
	public void doLayout() {
		// Nothing to do
		this.layout(true);
		try {
			if (!this.save_showed) {
				OutputComposite.this.saveFile();
				this.save_showed = true;
			}
		} catch (Exception ex) {
			log.error("SaveSelectionListener: ", ex); //$NON-NLS-1$
		}
	}

	/* (non-Javadoc)
	 * @see at.asit.pdfover.gui.composites.StateComposite#reloadResources()
	 */
	@Override
	public void reloadResources() {
		this.lbl_success_message.setText(Messages
				.getString("output.success_message")); //$NON-NLS-1$
	}

}