package org.eparapher.rcp.actions;
   
   import java.io.File;
   import java.util.ArrayList;
   import java.util.Iterator;
   
   import org.apache.log4j.Logger;
   import org.eclipse.core.runtime.IProgressMonitor;
   import org.eclipse.core.runtime.IStatus;
  import org.eclipse.core.runtime.Status;
  import org.eclipse.core.runtime.jobs.IJobChangeEvent;
  import org.eclipse.core.runtime.jobs.Job;
  import org.eclipse.core.runtime.jobs.JobChangeAdapter;
  import org.eclipse.jface.action.Action;
  import org.eclipse.jface.viewers.ISelection;
  import org.eclipse.jface.viewers.IStructuredSelection;
  import org.eclipse.jface.viewers.TreeViewer;
  import org.eclipse.jface.window.Window;
  import org.eclipse.jface.wizard.WizardDialog;
  import org.eclipse.swt.SWT;
  import org.eclipse.swt.widgets.Display;
  import org.eclipse.ui.IWorkbenchWindow;
  import org.eclipse.ui.PlatformUI;
  import org.eclipse.ui.actions.ActionFactory.IWorkbenchAction;
  import org.eparapher.core.crypto.EPKeystoreManager;
  import org.eparapher.core.crypto.keystore.IUserKeystore;
  import org.eparapher.rcp.tools.GUIIcons;
  import org.eparapher.rcp.tools.eParapherTools;
  import org.eparapher.rcp.views.documents.SecuredDocumentsView.TreeObject;
  import org.eparapher.rcp.views.documents.SecuredDocumentsView.TreeParent;
  import org.eparapher.rcp.wizards.CMSSigningWizard;
  import org.eparapher.core.signature.CMSSignatureParameters;
  import org.eparapher.core.signature.CMSSigner;
  
  public class CMSSignDocumentAction extends Action implements IWorkbenchAction {
  
  	private static Logger log = Logger.getLogger(CMSSignDocumentAction.class);
  	
  	private TreeViewer viewer;
  	private boolean usewizard;
  	private IWorkbenchWindow window;
  	
  	public CMSSignDocumentAction() {
  		this(null,true);
  	}
  
  	public CMSSignDocumentAction(TreeViewer mviewer, boolean museWizard) {
  		super();
  		viewer = mviewer;
  		usewizard = museWizard;
  		window=null;
  		this.setId("CMSSignDocumentAction");
  		this.setImageDescriptor(GUIIcons.SIGNATURE_ICON);
  		if (!usewizard) {
  			this.setText("CMS Digital Signature");
  			this.setToolTipText("Sign your document(s) in CMS format with default parameters");
  			this.setAccelerator(SWT.CTRL | 'R');
  		} else {
  			this.setText("CMS Digital Signature (wizard)");
  			this.setToolTipText("Sign your document(s) in CMS format with a wizard");
  			this.setAccelerator(SWT.CTRL | 'T');
  		}
  	}
  
  	public CMSSignDocumentAction(IWorkbenchWindow mwindow) {
  		this();
  		window=mwindow;
  	}
  
  	public void run() {
  		CMSSigningJob job = new CMSSigningJob();
  		if (job.isSigningPrivateKeyLoaded) {
  			job.setUser(true);
  			job.schedule();
  		}
  	}
  
  	private CMSSignatureParameters getSignatureParameters() {
  
  		CMSSignatureParameters cmssignparams = new CMSSignatureParameters();
  		cmssignparams.setCMSSignatureParamsFromPreferences();
  		if (viewer != null)
  			cmssignparams.setFileSelectiontoSign(getFileSelectionFromView());
  		
  		if (usewizard) {
  			//Launch Wizard to get CMS Signature Parameters
  			CMSSigningWizard cmsWizard = new CMSSigningWizard(cmssignparams);
  			if (window==null)
  				window=PlatformUI.getWorkbench().getActiveWorkbenchWindow();
  			WizardDialog dialog = new WizardDialog(window.getShell(), cmsWizard);
  			try {
  			if ( dialog.open() != Window.OK ) {
  				log.warn("CMS signing wizard cancelled");
  				return null;
  			}
  			} catch (Exception e) {
  				log.debug("Problem while invoking the CMS Settings Signature Wizard",e);
  			}
  			//recover params from wizard
 			cmssignparams = cmsWizard.getCMSSignatureParams();
 		}
 		return cmssignparams;
 	}
 
 	private File[] getFileSelectionFromView() {
 
 		ArrayList<File> fileList = new ArrayList<File>();
 
 		if (viewer.getSelection()!=null && !viewer.getSelection().isEmpty()) {
 			// Iterate over selected file in view
 			ISelection selection = viewer.getSelection();
 			IStructuredSelection structselection = (IStructuredSelection) selection;
 			
 			//Build File Array from Table Selection
 			for (Iterator<Object> iterator = structselection.iterator(); iterator.hasNext();) {
 				Object obj =  iterator.next();
 				if (obj instanceof TreeParent)
 					//TODO : sign all files in a directory (if confirm?)
 					eParapherTools.debugMessage("Cannot sign a directory!");
 				else {
 					TreeObject to = (TreeObject) obj;
 					File f = new File(to.getFilePath());
 					if (!f.exists())
 						log.error("file " + to.getFilePath() + " has been deleted");
 					else
 						fileList.add(f);
 				}
 			}
 		}
 		return fileList.toArray(new File[] {});
 	}
 
 	public void dispose() {
 		
 	}
 
 	class CMSSigningJob extends Job {
 
 		private CMSSignatureParameters cmssignparams;
 		
 		private CMSSigner cmssigner;
 
 		private String originalAlias;
 		protected boolean isSigningPrivateKeyLoaded;
 
 		protected CMSSigningJob() {
 
 			super("CMS Signer job");
 			
 			//this.setUser(true);
 			//this.setSystem(true);
 			// Getting signature settings (with wizard if needed)
 			cmssignparams = getSignatureParameters();
 			if (cmssignparams.getFileSelectiontoSign().length==0) {
 				log.info("Nothing to sign");
 				return;
 			}
 			
 			IUserKeystore userpkandcert = EPKeystoreManager.getInstance().getUserkeystore();
 			
 			if (cmssignparams != null ) {
 				
 				// Set the alias keystore to use
 				originalAlias = userpkandcert.getDefaultAlias();
 				if (cmssignparams.getSignatureAlias()!=null
 					&& userpkandcert.containsAlias(cmssignparams.getSignatureAlias()) )
 					userpkandcert.setDefaultAlias(cmssignparams.getSignatureAlias());
 				
 				//Load the private key (Ask PIN or Passphrase)
 				isSigningPrivateKeyLoaded = userpkandcert.loadPrivateKey();
 				
 				if (isSigningPrivateKeyLoaded) {
 					cmssigner = new CMSSigner();
 					
 					//End Job Listener
 					this.addJobChangeListener(new JobChangeAdapter() {
 						public void done(IJobChangeEvent event) {
 							if (event.getResult().isOK())
 								log.info("CMS Signing Job completed successfully");
 							else
 								log.warn("CMS Signing Job did not complete successfully");
 						}
 					});
 				}
 			}
 		}
 		
 		@SuppressWarnings("unchecked")
 		@Override
 		protected IStatus run(IProgressMonitor monitor) {
 			
 			if ( cmssignparams != null && isSigningPrivateKeyLoaded) {
 				monitor.beginTask("CMS Signing", cmssignparams.getFileSelectiontoSign().length );
 				File[] f2s = cmssignparams.getFileSelectiontoSign();
 				for (int i = 0; i < f2s.length; i++) {
 					
 					monitor.subTask("Signing "+f2s[i].getName());
 					cmssigner.sign( f2s[i].getAbsolutePath(), cmssignparams);
 					monitor.worked(1);
 
 					//refresh the view
 					Display.getDefault().asyncExec(new Runnable() {
 						public void run() {	if (viewer != null) viewer.refresh(); }
 					});
 				}
 				monitor.done();
 			}
 			if (originalAlias!=null)
 				EPKeystoreManager.getInstance().getUserkeystore().setDefaultAlias(originalAlias);
 
 			return Status.OK_STATUS;
 		}
 	}
 
 }