package org.eparapher.rcp.actions;
   
   import java.io.FileNotFoundException;
   import java.io.IOException;
   import java.security.NoSuchAlgorithmException;
   import java.security.NoSuchProviderException;
   import java.util.Iterator;
   
   import org.apache.log4j.Logger;
  import org.bouncycastle.cms.CMSException;
  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.swt.widgets.Display;
  import org.eclipse.ui.actions.ActionFactory.IWorkbenchAction;
  import org.eparapher.core.crypto.EPKeystoreManager;
  import org.eparapher.core.encryption.CMSEncryption;
  import org.eparapher.core.encryption.CMSEncryptionParameters;
  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;
  
  
  public class EncryptAction extends Action implements IWorkbenchAction {
  
  	private static Logger log = Logger.getLogger(EncryptAction.class);
  	
  	private TreeViewer viewer;
  	private boolean usewizard;
  	
  	public EncryptAction() {
  		this.setId("EncryptAction");
  		this.setText("Encrypt");
  		this.setToolTipText("Encrypt your files");
  		this.setImageDescriptor(GUIIcons.ENCRYPTION_ICON);
  	}
  
  	public EncryptAction(TreeViewer mviewer, boolean museWizard) {
  		this();
  		viewer = mviewer;
  		usewizard = museWizard;
  		if (usewizard) {
  			this.setText("Encrypt (wizard)");
  			this.setToolTipText("Encrypt your file(s) in CMS with a wizard");
  		}
  	}
  
  	public void run() {
  		if (viewer.getSelection()!=null) {
  			if (!viewer.getSelection().isEmpty()) {
  				CMSEncryptionJob job = new CMSEncryptionJob();
  				if (job.isEncryptionPrivateKeyLoaded) {
  					job.setUser(true);
  					job.schedule();
  				}
  			}
  		}
  	}
  	
  	private CMSEncryptionParameters getEncryptionParameters() {
  
  		CMSEncryptionParameters cmsencparams = new CMSEncryptionParameters();
  		cmsencparams.setCMSEncryptionParamsFromPreferences();
  		/* TODO : Encryption Wizard GUI
  		if (usewizard) {
  			//Launch Wizard to get PDF Signature Parameters
  			CMSEncryptWizard cmsEncWizard = new CMSEncryptWizard();
  			WizardDialog dialog = new WizardDialog(PlatformUI.getWorkbench().getActiveWorkbenchWindow().getShell(), cmsEncWizard);
  			if ( dialog.open() != Window.OK ) {
  				log.info("CMS encryption wizard cancelled");
  				return null;
  			}
  			//recover params from wizard
  			cmsencparams = cmsWizard.getCMSEncryptionParams();
  		}
  		*/
  		return cmsencparams;
  	}
  	
  	public void dispose() {}
  
  	class CMSEncryptionJob extends Job {
  
  		private CMSEncryptionParameters cmsencryptparams;
  		private CMSEncryption cmsencrypt;
  
  		private IStructuredSelection structselection;
  		protected boolean isEncryptionPrivateKeyLoaded;
  
  		protected CMSEncryptionJob() {
  
 			super("CMS Encryption");
 			// Use Wizard if needed
 			cmsencryptparams = getEncryptionParameters();
 			if (cmsencryptparams != null ) {
 				
 				//Ask PIN or Passphrase with CMSSigner Constructor
 				//TODO : use encryption private key
 				isEncryptionPrivateKeyLoaded = EPKeystoreManager.getInstance().getUserkeystore().loadPrivateKey();
 				
 				if (isEncryptionPrivateKeyLoaded) {
 					cmsencrypt = new CMSEncryption();
 					
 					// Iterate over selected file in view
 					ISelection selection = viewer.getSelection();
 					structselection = (IStructuredSelection) selection;
 					
 					//End Job Listener
 					this.addJobChangeListener(new JobChangeAdapter() {
 						public void done(IJobChangeEvent event) {
 							if (event.getResult().isOK())
 								log.info("CMS Encryption Job completed successfully");
 							else
 								log.warn("CMS Encryption Job did not complete successfully");
 						}
 					});
 				}
 			} else
 				log.warn("CMS Encryption cancelled because default successfully");
 		}
 		
 		@SuppressWarnings("unchecked")
 		@Override
 		protected IStatus run(IProgressMonitor monitor) {
 			
 			if (cmsencryptparams != null ) {
 				monitor.beginTask("CMS Encrypt", structselection.size() );
 				for (Iterator<Object> iterator = structselection.iterator(); iterator.hasNext();) {
 					Object obj =  iterator.next();
 					if (obj instanceof TreeParent)
 						eParapherTools.debugMessage("Cannot encrypt a directory!");
 					else {
 						TreeObject to = (TreeObject) obj;
 	
 						monitor.subTask("Encrypt "+to.getName());
 						try {
 							cmsencrypt.encrypt( to.getFilePath(), cmsencryptparams);
 						} catch (FileNotFoundException e) {
 							log.warn("File" + to.getFilePath() + " not found. Encrypting operation cancelled",e);
 						} catch (NoSuchAlgorithmException e) {
 							log.warn("Error while encrypting " + to.getFilePath() ,e);
 						} catch (NoSuchProviderException e) {
 							log.warn("Error while encrypting " + to.getFilePath() ,e);
 						} catch (IOException e) {
 							log.warn("Error while encrypting " + to.getFilePath() ,e);
 						} catch (CMSException e) {
 							log.warn("Error while encrypting " + to.getFilePath() ,e);
 						}
 						monitor.worked(1);
 						
 						//refresh the view
 						Display.getDefault().asyncExec(new Runnable() {
 							public void run() {	viewer.refresh(); }
 						});
 					}
 				}
 				monitor.done();
 			}
 			return Status.OK_STATUS;
 		}
 	}
 	
 }