jpa/plugins/org.eclipse.jpt.ui/src/org/eclipse/jpt/ui/internal/wizards/orm/MappingFileWizard.java

/*******************************************************************************
 *  Copyright (c) 2008  Oracle. 
 *  All rights reserved.  This program and the accompanying materials are 
 *  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: 
 *  	Oracle - initial API and implementation
 *******************************************************************************/
package org.eclipse.jpt.ui.internal.wizards.orm;

import java.lang.reflect.InvocationTargetException;
import org.eclipse.core.resources.IFile;
import org.eclipse.core.resources.IFolder;
import org.eclipse.core.resources.IProject;
import org.eclipse.core.resources.IResource;
import org.eclipse.core.runtime.IAdaptable;
import org.eclipse.core.runtime.Path;
import org.eclipse.jdt.core.IJavaElement;
import org.eclipse.jdt.core.IJavaProject;
import org.eclipse.jdt.core.IPackageFragment;
import org.eclipse.jdt.core.IPackageFragmentRoot;
import org.eclipse.jdt.core.JavaCore;
import org.eclipse.jdt.core.JavaModelException;
import org.eclipse.jem.util.emf.workbench.ProjectUtilities;
import org.eclipse.jface.viewers.IStructuredSelection;
import org.eclipse.jpt.core.JptCorePlugin;
import org.eclipse.jpt.core.context.JpaContextNode;
import org.eclipse.jpt.core.context.persistence.PersistenceUnit;
import org.eclipse.jpt.core.internal.operations.OrmFileCreationDataModelProperties;
import org.eclipse.jpt.core.internal.operations.OrmFileCreationDataModelProvider;
import org.eclipse.jpt.ui.JptUiPlugin;
import org.eclipse.jpt.ui.internal.JptUiIcons;
import org.eclipse.jpt.ui.internal.JptUiMessages;
import org.eclipse.ui.INewWizard;
import org.eclipse.ui.IWorkbench;
import org.eclipse.ui.IWorkbenchPage;
import org.eclipse.ui.PartInitException;
import org.eclipse.ui.PlatformUI;
import org.eclipse.ui.ide.IDE;
import org.eclipse.wst.common.frameworks.datamodel.IDataModel;
import org.eclipse.wst.common.frameworks.datamodel.IDataModelProvider;
import org.eclipse.wst.common.frameworks.internal.datamodel.ui.DataModelWizard;

public class MappingFileWizard extends DataModelWizard 
	implements INewWizard 
{
	private MappingFileWizardPage page;
	
	
	public MappingFileWizard() {
		this(null);
	}
	
	public MappingFileWizard(IDataModel dataModel) {
		super(dataModel);
		setWindowTitle(JptUiMessages.MappingFileWizard_title);
		setDefaultPageImageDescriptor(JptUiPlugin.getImageDescriptor(JptUiIcons.JPA_FILE_WIZ_BANNER));
	}
	
	
	@Override
	protected void doAddPages() {
		super.doAddPages();
		page = buildMappingFileWizardPage();
		addPage(page);
	}
	
	protected MappingFileWizardPage buildMappingFileWizardPage() {
		return new MappingFileWizardPage(getDataModel(), "Page_1");
	}
	
	@Override
	protected IDataModelProvider getDefaultProvider() {
		return new OrmFileCreationDataModelProvider();
	}
	
	public void init(IWorkbench workbench, IStructuredSelection selection) {
		// check for project, source folder, persistence unit?
		if (selection == null || selection.isEmpty()) {
			return;
		}
		
		Object firstSelection = selection.getFirstElement();
		
		PersistenceUnit pUnit = extractPersistenceUnit(firstSelection);
		IFolder sourceFolder = extractSourceFolder(pUnit, firstSelection);
		IProject project = extractProject(pUnit, sourceFolder, firstSelection);
		
		if (project != null) {
			getDataModel().setStringProperty(OrmFileCreationDataModelProperties.PROJECT_NAME, project.getName());
		}
		if (sourceFolder != null) {
			getDataModel().setStringProperty(OrmFileCreationDataModelProperties.SOURCE_FOLDER, sourceFolder.getFullPath().toPortableString());
		}
		if (pUnit != null) {
			getDataModel().setBooleanProperty(OrmFileCreationDataModelProperties.ADD_TO_PERSISTENCE_UNIT, true);
			getDataModel().setStringProperty(OrmFileCreationDataModelProperties.PERSISTENCE_UNIT, pUnit.getName());
		}
	}
	
	private PersistenceUnit extractPersistenceUnit(Object selection) {
		if (selection instanceof PersistenceUnit) {
			return (PersistenceUnit) selection;
		}
		PersistenceUnit pUnit = null;
		if (selection instanceof JpaContextNode) {
			try {
				pUnit = ((JpaContextNode) selection).getPersistenceUnit();
			}
			catch (Exception e) { /* do nothing, just continue */ }
		}
		if (pUnit == null && selection instanceof IAdaptable) {
			pUnit = (PersistenceUnit) ((IAdaptable) selection).getAdapter(PersistenceUnit.class);
			
		}
		return pUnit;
	}
	
	private IFolder extractSourceFolder(PersistenceUnit pUnit, Object selection) {
		IJavaProject javaProject = null;
		IFolder srcFolder = null;
		if (pUnit != null) {
			javaProject = pUnit.getJpaProject().getJavaProject();
			srcFolder = findSourceFolder(javaProject, pUnit.getResource());
			if (srcFolder != null) {
				return srcFolder;
			}
			
		}
		if (selection instanceof IResource) {
			javaProject = JavaCore.create(((IResource) selection).getProject());
			if (javaProject.exists()) {
				srcFolder = findSourceFolder(javaProject, (IResource) selection);
				if (srcFolder != null) {
					return srcFolder;
				}
			}
		}
		
		if (selection instanceof IAdaptable) {
			IResource resource = (IResource) ((IAdaptable) selection).getAdapter(IResource.class);
			if (resource != null) {
				javaProject = JavaCore.create((resource).getProject());
				if (javaProject.exists()) {
					srcFolder = findSourceFolder(javaProject, resource);	
					if (srcFolder != null) {
						return srcFolder;
					}
				}
			}
		}
		return null;
	}
	
	private IFolder findSourceFolder(IJavaProject javaProject, IResource resource) {
		if (JptCorePlugin.getJpaProject(javaProject.getProject()) == null) {
			// not even a jpa project
			return null;
		}
		while (resource != null && ! (resource instanceof IFolder)) {
			resource = resource.getParent();
		}
		if (resource == null) {
			return null;
		}
		IFolder folder = (IFolder) resource;
		try {
			IPackageFragmentRoot packageFragmentRoot = null;
			while (packageFragmentRoot == null && folder != null) {
				packageFragmentRoot = javaProject.findPackageFragmentRoot(folder.getFullPath());
				if (packageFragmentRoot == null) {
					IPackageFragment packageFragment = javaProject.findPackageFragment(folder.getFullPath());
					if (packageFragment != null) {
						packageFragmentRoot = (IPackageFragmentRoot) packageFragment.getAncestor(IJavaElement.PACKAGE_FRAGMENT_ROOT);
					}
				}
				if (packageFragmentRoot == null) {
					try {
						folder = (IFolder) folder.getParent();
					}
					catch (ClassCastException cce) {
						folder = null;
					}
				}
			}
			if (packageFragmentRoot == null) {
				return null;
			}
			if (packageFragmentRoot.getKind() == IPackageFragmentRoot.K_SOURCE) {
				return (IFolder) packageFragmentRoot.getResource();
			}
		}
		catch (JavaModelException jme) { /* do nothing, return null */ }
		return null;
	}
	
	private IProject extractProject(PersistenceUnit pUnit, IFolder sourceFolder, Object selection) {
		if (pUnit != null) {
			return pUnit.getJpaProject().getProject();
		}
		if (sourceFolder != null) {
			return sourceFolder.getProject();
		}
		
		IProject project = null;
		if (selection instanceof IResource) {
			project = ((IResource) selection).getProject();
		}
		if (project == null && selection instanceof IJavaElement) {
			project = ((IJavaElement) selection).getJavaProject().getProject();
		}
		if (project == null && selection instanceof JpaContextNode) {
			project = ((JpaContextNode) selection).getJpaProject().getProject();
		}
		if (project == null && selection instanceof IAdaptable) {
			project = (IProject) ((IAdaptable) selection).getAdapter(IProject.class);
			if (project == null) {
				IResource resource = (IResource) ((IAdaptable) selection).getAdapter(IResource.class);
				if (resource != null) {
					project = resource.getProject();
				}
			}
			if (project == null) {
				IJavaElement javaElement = (IJavaElement) ((IAdaptable) selection).getAdapter(IJavaElement.class);
				if (javaElement != null) {
					project = javaElement.getJavaProject().getProject();
				}
			}
		}
		
		if (project != null && JptCorePlugin.getJpaProject(project) != null) {
			return project;
		}
		
		return null;
	}
	
	protected void postPerformFinish() throws InvocationTargetException {      
        try {
        	String projectName = (String) getDataModel().getProperty(OrmFileCreationDataModelProperties.PROJECT_NAME);
        	IProject project = ProjectUtilities.getProject(projectName);
        	String sourceFolder = getDataModel().getStringProperty(OrmFileCreationDataModelProperties.SOURCE_FOLDER);
        	String filePath = getDataModel().getStringProperty(OrmFileCreationDataModelProperties.FILE_PATH);
        	
        	IFile file = project.getWorkspace().getRoot().getFile(new Path(sourceFolder).append(filePath));
            openEditor(file);
        } 
        catch (Exception cantOpen) {
        	throw new InvocationTargetException(cantOpen);
        } 
    }
	
	private void openEditor(final IFile file) {
        if (file != null) {
            getShell().getDisplay().asyncExec(new Runnable() {
                public void run() {
                    try {
                        IWorkbenchPage page = PlatformUI.getWorkbench().getActiveWorkbenchWindow().getActivePage();
                        IDE.openEditor(page, file, true);
                    }
                    catch (PartInitException e) {
                    	JptUiPlugin.log(e);
                    }
                }
            });
        }
    }
}