/******************************************************************************* 
 * Copyright (c) 2010 Red Hat, Inc. 
 * Distributed under license by Red Hat, Inc. All rights reserved. 
 * This program is made available under the terms of the 
 * Eclipse Public License v1.0 which accompanies this distribution, 
 * and is available at http://www.eclipse.org/legal/epl-v10.html 
 * 
 * Contributors: 
 * Red Hat, Inc. - initial API and implementation 
 ******************************************************************************/
package org.jboss.tools.ws.ui.wizards;

import org.eclipse.core.commands.ExecutionException;
import org.eclipse.core.resources.IFile;
import org.eclipse.core.resources.IProject;
import org.eclipse.core.resources.IResource;
import org.eclipse.core.resources.IncrementalProjectBuilder;
import org.eclipse.core.resources.ResourcesPlugin;
import org.eclipse.core.runtime.CoreException;
import org.eclipse.core.runtime.IStatus;
import org.eclipse.core.runtime.NullProgressMonitor;
import org.eclipse.core.runtime.Status;
import org.eclipse.jdt.core.IJavaProject;
import org.eclipse.jdt.core.JavaCore;
import org.eclipse.jface.dialogs.MessageDialog;
import org.eclipse.jface.viewers.IStructuredSelection;
import org.eclipse.jface.wizard.Wizard;
import org.eclipse.jst.j2ee.project.JavaEEProjectUtilities;
import org.eclipse.swt.widgets.Display;
import org.eclipse.ui.INewWizard;
import org.eclipse.ui.IWorkbench;
import org.eclipse.ui.IWorkbenchPage;
import org.eclipse.ui.IWorkbenchWindow;
import org.eclipse.ui.PartInitException;
import org.eclipse.ui.ide.IDE;
import org.eclipse.ui.wizards.newresource.BasicNewResourceWizard;
import org.eclipse.wst.common.frameworks.datamodel.AbstractDataModelOperation;
import org.jboss.tools.common.ui.CommonUIImages;
import org.jboss.tools.ws.creation.core.commands.AddRestEasyJarsCommand;
import org.jboss.tools.ws.creation.core.commands.MergeWebXMLCommand;
import org.jboss.tools.ws.creation.core.commands.RSMergeWebXMLCommand;
import org.jboss.tools.ws.creation.core.commands.RSServiceCreationCommand;
import org.jboss.tools.ws.creation.core.commands.ServiceCreationCommand;
import org.jboss.tools.ws.creation.core.data.ServiceModel;
import org.jboss.tools.ws.creation.core.utils.JBossWSCreationUtils;
import org.jboss.tools.ws.ui.JBossWSUIPlugin;
import org.jboss.tools.ws.ui.messages.JBossWSUIMessages;

/**
 * @author Brian Fitzpatrick
 *
 */
public class JBossWSAnnotatedClassWizard extends Wizard implements INewWizard {

	private static final String JDT_EDITOR = 
		"org.eclipse.jdt.ui.CompilationUnitEditor"; //$NON-NLS-1$

	public static String WSNAMEDEFAULT = "HelloWorld"; //$NON-NLS-1$
	public static String PACKAGEDEFAULT = "org.jboss.samples.webservices"; //$NON-NLS-1$
	public static String WSCLASSDEFAULT = "HelloWorld"; //$NON-NLS-1$

	public static String RSNAMEDEFAULT = "MyRESTApplication"; //$NON-NLS-1$
	public static String RSCLASSDEFAULT = "HelloWorldResource"; //$NON-NLS-1$
	public static String RSAPPCLASSDEFAULT = "MyRESTApplication"; //$NON-NLS-1$

	private String serviceName = WSNAMEDEFAULT;
	private String packageName = PACKAGEDEFAULT;
	private String className = WSCLASSDEFAULT;
	private String appClassName = ""; //$NON-NLS-1$
	private boolean useDefaultServiceName = true;
	private boolean useDefaultClassName = true;
	private boolean updateWebXML = true;
	private boolean isJAXWS = true;
	private boolean addJarsFromRootRuntime = false;

	private IStructuredSelection selection;
	private IProject project;

	private static String WEB = "web.xml"; //$NON-NLS-1$
	private static String WEBINF = "WEB-INF"; //$NON-NLS-1$
	private IFile webFile;

	public JBossWSAnnotatedClassWizard() {
		super();
		super.setWindowTitle(JBossWSUIMessages.JBossWSAnnotatedClassWizard_Annotated_Class_WS_Wizard_Title);
		super.setHelpAvailable(false);
		setDefaultPageImageDescriptor(CommonUIImages.getInstance().getOrCreateImageDescriptor(CommonUIImages.WEB_SERVICE_IMAGE));
	}

	public void addPages() {
		super.addPages();
		JBossWSAnnotatedClassWizardPage onePage =
			new JBossWSAnnotatedClassWizardPage("onePage"); //$NON-NLS-1$
		addPage(onePage);
	}

	@Override
	public boolean performFinish() {
		if (canFinish()) {
			ServiceModel model = new ServiceModel();
			model.setWebProjectName(project.getName());
			IJavaProject javaProject = JavaCore.create(project);
			model.setJavaProject(javaProject);
			model.addServiceClasses(new StringBuffer().append(getPackageName())
					.append(".").append(getClassName()).toString()); //$NON-NLS-1$
			model.setServiceName(getServiceName());
			model.setUpdateWebxml(getUpdateWebXML());
			model.setCustomPackage(getPackageName());
			model.setApplicationClassName( getAppClassName());

			AbstractDataModelOperation mergeCommand = null;
			if (isJAXWS()) {
				mergeCommand = new MergeWebXMLCommand(model);
			} else {
				mergeCommand = new RSMergeWebXMLCommand(model);
			}
			
			IStatus status = null;
			if (getUpdateWebXML()) {
				try {
					status = mergeCommand.execute(null, null);
				} catch (ExecutionException e) {
					JBossWSUIPlugin.log(e);
				}
				if (status != null && status.getSeverity() == Status.ERROR) {
					MessageDialog
							.openError(
									this.getShell(),
									JBossWSUIMessages.JBossWS_GenerateWizard_MessageDialog_Title,
									status.getMessage());
					return false;
				}
			}
			
			AbstractDataModelOperation addJarsCommand = null;
			AbstractDataModelOperation addClassesCommand = null;
			if (!isJAXWS()) {
				if (getAddJarsFromRootRuntime())
					addJarsCommand = new AddRestEasyJarsCommand(model);
				addClassesCommand = new RSServiceCreationCommand(model);
			} else {
				addClassesCommand = new ServiceCreationCommand(model);
			}
			try {
				boolean addedJars = false;
				if (addJarsCommand != null) {
					addJarsCommand.execute(null, null);
					addedJars = true;
				}
				if (addClassesCommand != null) {
					addClassesCommand.execute(null, null);
				}
				getProject().refreshLocal(IProject.DEPTH_INFINITE, new NullProgressMonitor());
				IFile openFile1 = null;
				IFile openFile2 = null;
				if (addClassesCommand instanceof ServiceCreationCommand) {
					ServiceCreationCommand cmd = (ServiceCreationCommand) addClassesCommand;
					if (cmd.getResource() != null && cmd.getResource() instanceof IFile) {
						openFile1 = (IFile) cmd.getResource();
					}
				} else if (addClassesCommand instanceof RSServiceCreationCommand) {
					if (addedJars)
						getProject().build(IncrementalProjectBuilder.CLEAN_BUILD, null);
					RSServiceCreationCommand cmd = (RSServiceCreationCommand) addClassesCommand;
					if (cmd.getAnnotatedClassResource() != null && cmd.getAnnotatedClassResource() instanceof IFile) {
						openFile1 = (IFile) cmd.getAnnotatedClassResource();
					}
					if (cmd.getApplicationClassResource() != null && cmd.getApplicationClassResource() instanceof IFile) {
						openFile2 = (IFile) cmd.getApplicationClassResource();
					}
				}
				if (openFile1 != null) {
					openResource(openFile1);
				}
				if (openFile2 != null) {
					openResource(openFile2);
				}
			} catch (ExecutionException e) {
				JBossWSUIPlugin.log(e);
				MessageDialog
					.openError(
						this.getShell(),
						JBossWSUIMessages.JBossWS_GenerateWizard_MessageDialog_Title,
						e.getMessage());
			} catch (CoreException e) {
				JBossWSUIPlugin.log(e);
				MessageDialog
					.openError(
						this.getShell(),
						JBossWSUIMessages.JBossWS_GenerateWizard_MessageDialog_Title,
						e.getMessage());
			}
		}
		return true;
	}

	public void init(IWorkbench workbench, IStructuredSelection selection) {
		this.selection = selection;
		if (this.selection.getFirstElement() instanceof IProject) {
			project = (IProject) this.selection.getFirstElement();
		}
		if (project != null
				&& JavaEEProjectUtilities.isDynamicWebProject(project)) {
			webFile = project.getParent().getFolder(
					JBossWSCreationUtils.getWebContentRootPath(project).append(WEBINF))
					.getFile(WEB);
		}
	}

	@Override
	public boolean canFinish() {
		return super.canFinish();
	}

	public String getServiceName() {
		return serviceName;
	}

	public void setServiceName(String serviceName) {
		this.serviceName = serviceName;
	}

	public String getPackageName() {
		return packageName;
	}

	public void setPackageName(String packageName) {
		this.packageName = packageName;
	}

	public String getClassName() {
		return className;
	}

	public void setClassName(String className) {
		this.className = className;
	}

	public String getAppClassName() {
		return appClassName;
	}

	public void setAppClassName(String className) {
		this.appClassName = className;
	}

	public boolean isUseDefaultServiceName() {
		return useDefaultServiceName;
	}

	public void setUseDefaultServiceName(boolean useDefaultServiceName) {
		this.useDefaultServiceName = useDefaultServiceName;
	}

	public boolean isUseDefaultClassName() {
		return useDefaultClassName;
	}

	public void setUseDefaultClassName(boolean useDefaultClassName) {
		this.useDefaultClassName = useDefaultClassName;
	}

	public void setUpdateWebXML(boolean updateWebXML) {
		this.updateWebXML = updateWebXML;
	}

	public boolean getUpdateWebXML() {
		return updateWebXML;
	}

	public boolean getAddJarsFromRootRuntime() {
		return addJarsFromRootRuntime;
	}

	public void setAddJarsFromRootRuntime(boolean addJarsFromRootRuntime) {
		this.addJarsFromRootRuntime = addJarsFromRootRuntime;
	}

	public void setJAXWS(boolean isJAXWS) {
		this.isJAXWS = isJAXWS;
	}

	public boolean isJAXWS() {
		return isJAXWS;
	}

	public IProject getProject() {
		return project;
	}
	
	public ServiceModel getServiceModel() {
		ServiceModel model = new ServiceModel();
		if (project != null) {
			model.setWebProjectName(project.getName());
		}
		if (getPackageName() != null) {
			model.addServiceClasses(new StringBuffer().append(getPackageName())
				.append(".").append(getClassName()).toString()); //$NON-NLS-1$
		}
		model.setServiceName(getServiceName());
		model.setUpdateWebxml(true);
		model.setCustomPackage(getPackageName());
		model.setCustomClassName(getClassName());
		return model;
	}
	
	public void setProject (String projectName) {
		if (projectName != null && projectName.trim().length() > 0) {
			IProject test =
				ResourcesPlugin.getWorkspace().getRoot().getProject(projectName);
			if (test != null) {
				this.project = test;
				if (project != null
						&& JavaEEProjectUtilities.isDynamicWebProject(project)) {
					webFile = project.getParent().getFolder(
							JBossWSCreationUtils.getWebContentRootPath(project).append(WEBINF))
							.getFile(WEB);
				}
			}
		}
	}
	

	public IFile getWebFile() {
		return webFile;
	}
	
	protected void openResource(final IFile resource) {
		if (resource.getType() != IResource.FILE) {
			return;
		}

		IWorkbenchWindow window = JBossWSUIPlugin.getActiveWorkbenchWindow();
		if (window == null) {
			return;
		}

		final IWorkbenchPage activePage = window.getActivePage();
		if (activePage != null) {
			final Display display = getShell().getDisplay();
			display.asyncExec(new Runnable() {
				public void run() {
					try {
						IDE.openEditor(activePage, resource, JDT_EDITOR, true);
					} catch (PartInitException e) {
						JBossWSUIPlugin.log(e);
					}
				}
			});
			BasicNewResourceWizard.selectAndReveal(resource, activePage
					.getWorkbenchWindow());
		}
	}

}
