jpa/plugins/org.eclipse.jpt.ui/src/org/eclipse/jpt/ui/internal/wizards/gen/GenerateEntitiesFromSchemaWizard.java - dali/webtools.dali - Git at Google

 /*******************************************************************************
  * Copyright (c) 2007, 2010 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.gen;

 import java.io.File;
 import java.io.IOException;

 import org.eclipse.core.resources.IProject;
 import org.eclipse.core.resources.IResource;
 import org.eclipse.core.resources.IResourceRuleFactory;
 import org.eclipse.core.resources.ResourcesPlugin;
 import org.eclipse.core.resources.WorkspaceJob;
 import org.eclipse.core.runtime.CoreException;
 import org.eclipse.core.runtime.IProgressMonitor;
 import org.eclipse.core.runtime.IStatus;
 import org.eclipse.core.runtime.OperationCanceledException;
 import org.eclipse.core.runtime.Platform;
 import org.eclipse.core.runtime.Status;
 import org.eclipse.core.runtime.preferences.IEclipsePreferences;
 import org.eclipse.core.runtime.preferences.InstanceScope;
 import org.eclipse.jdt.core.IJavaElement;
 import org.eclipse.jdt.core.IPackageFragment;
 import org.eclipse.jdt.core.IPackageFragmentRoot;
 import org.eclipse.jface.dialogs.Dialog;
 import org.eclipse.jface.dialogs.IDialogConstants;
 import org.eclipse.jface.resource.JFaceResources;
 import org.eclipse.jface.resource.LocalResourceManager;
 import org.eclipse.jface.resource.ResourceManager;
 import org.eclipse.jface.viewers.IStructuredSelection;
 import org.eclipse.jface.window.Window;
 import org.eclipse.jface.wizard.IWizardPage;
 import org.eclipse.jface.wizard.Wizard;
 import org.eclipse.jpt.core.EntityGeneratorDatabaseAnnotationNameBuilder;
 import org.eclipse.jpt.core.JpaPlatform;
 import org.eclipse.jpt.core.JpaProject;
 import org.eclipse.jpt.core.JptCorePlugin;
 import org.eclipse.jpt.core.internal.utility.jdt.JDTTools;
 import org.eclipse.jpt.db.Column;
 import org.eclipse.jpt.db.ConnectionProfile;
 import org.eclipse.jpt.db.ForeignKey;
 import org.eclipse.jpt.db.Schema;
 import org.eclipse.jpt.db.Table;
 import org.eclipse.jpt.gen.internal.BaseEntityGenCustomizer;
 import org.eclipse.jpt.gen.internal.DatabaseAnnotationNameBuilder;
 import org.eclipse.jpt.gen.internal.ORMGenCustomizer;
 import org.eclipse.jpt.gen.internal.ORMGenTable;
 import org.eclipse.jpt.gen.internal.PackageGenerator;
 import org.eclipse.jpt.ui.JptUiPlugin;
 import org.eclipse.jpt.ui.internal.JptUiIcons;
 import org.eclipse.jpt.ui.internal.JptUiMessages;
 import org.eclipse.osgi.util.NLS;
 import org.eclipse.swt.SWT;
 import org.eclipse.swt.events.SelectionEvent;
 import org.eclipse.swt.events.SelectionListener;
 import org.eclipse.swt.layout.GridData;
 import org.eclipse.swt.layout.GridLayout;
 import org.eclipse.swt.widgets.Button;
 import org.eclipse.swt.widgets.Composite;
 import org.eclipse.swt.widgets.Control;
 import org.eclipse.swt.widgets.Display;
 import org.eclipse.swt.widgets.Label;
 import org.eclipse.swt.widgets.Shell;
 import org.eclipse.ui.INewWizard;
 import org.eclipse.ui.IWorkbench;

 public class GenerateEntitiesFromSchemaWizard extends Wizard
 	implements INewWizard  {

 	public static final String HELP_CONTEXT_ID = JptUiPlugin.PLUGIN_ID + ".GenerateEntitiesFromSchemaWizard"; //$NON-NLS-1$

 	private static final String DONT_SHOW_OVERWRITE_WARNING_DIALOG = "DONT_SHOW_OVERWRITE_WARNING_DIALOG"; //$NON-NLS-1$

 	private JpaProject jpaProject;

 	private IStructuredSelection selection;

 	private ORMGenCustomizer customizer = null;

 	private PromptJPAProjectWizardPage projectPage;

 	private TablesSelectorWizardPage tablesSelectorPage;

 	private TableAssociationsWizardPage tableAssociationsPage;

 	private DefaultTableGenerationWizardPage defaultTableGenerationPage;

 	private TablesAndColumnsCustomizationWizardPage tablesAndColumnsCustomizationPage;

 	protected final ResourceManager resourceManager;

 	public GenerateEntitiesFromSchemaWizard() {
 		this.resourceManager = new LocalResourceManager(JFaceResources.getResources());
 		this.setWindowTitle( JptUiEntityGenMessages.GenerateEntitiesWizard_generateEntities);
 	}

 	public GenerateEntitiesFromSchemaWizard( JpaProject jpaProject, IStructuredSelection selection) {
 		super();
 		this.jpaProject = jpaProject;
 		this.selection = selection;
 		this.resourceManager = new LocalResourceManager(JFaceResources.getResources());
 		this.setWindowTitle( JptUiEntityGenMessages.GenerateEntitiesWizard_generateEntities);
 		this.setDefaultPageImageDescriptor(JptUiPlugin.getImageDescriptor(JptUiIcons.ENTITY_WIZ_BANNER));
 	}

 	@Override
 	public void addPages() {
 		setForcePreviousAndNextButtons(true);

 		//If this.jpaProject is not initialized because user didn't select a JPA project
 		if( this.jpaProject == null ){
 			this.projectPage = new PromptJPAProjectWizardPage(HELP_CONTEXT_ID);
 			this.addPage(this.projectPage);
 			return;
 		}
 		addMainPages();
 	}

 	private void addMainPages() {
 		this.tablesSelectorPage = new TablesSelectorWizardPage(this.jpaProject, this.resourceManager);
 		this.addPage(this.tablesSelectorPage);

 		this.tableAssociationsPage = new TableAssociationsWizardPage(this.jpaProject, this.resourceManager);
 		this.addPage(this.tableAssociationsPage);

 		this.defaultTableGenerationPage = new DefaultTableGenerationWizardPage(this.jpaProject);
 		this.addPage(this.defaultTableGenerationPage);
 		this.defaultTableGenerationPage.init(this.selection);

 		this.tablesAndColumnsCustomizationPage = new TablesAndColumnsCustomizationWizardPage(this.jpaProject, this.resourceManager);
 		this.addPage(this.tablesAndColumnsCustomizationPage);
 		this.tablesAndColumnsCustomizationPage.init(this.selection);
 	}

 	public ORMGenCustomizer getORMGenCustomizer(){
 		return this.customizer;
 	}

 	/**
 	 * Create the ORMGenCustomizer when user selects a new connection profile and schema
 	 *
 	 * JpaPlatform implementor can provide a custom ORMGenCustomizer specific to a platform
 	 * with AdapterFactory through Eclipse org.eclipse.core.runtime.adapters extension point:
 	 * <pre>
 	 *
 	 *<extension
      *    point="org.eclipse.core.runtime.adapters">
      * <factory
      *       adaptableType="org.eclipse.jpt.eclipselink.core.internal.EclipseLinkPlatform"
      *       class="oracle.eclipse.tools.orm.internal.EclipseLinkORMGenCustomizerAdapterFactory">
      *    <adapter
      *          type="oracle.eclipse.tools.orm.internal.ORMGenCustomizer">
      *    </adapter>
      * </factory>
      *</extension>
 	 *</pre>
 	 *
 	 * @param schema
 	 */
 	public ORMGenCustomizer createORMGenCustomizer(Schema schema){
 		JpaPlatform jpaPlatform = this.jpaProject.getJpaPlatform();
 		Object obj = Platform.getAdapterManager().getAdapter( jpaPlatform, ORMGenCustomizer.class );
 		if (obj != null  && obj instanceof ORMGenCustomizer) {
 			this.customizer = (ORMGenCustomizer) obj;
 			this.customizer.init(getCustomizationFile(), schema);
 		} else{
 			this.customizer = new BaseEntityGenCustomizer( );
 			this.customizer.init(getCustomizationFile(), schema);
 		}

 		ORMGenTable newDefaultTable = getCustomizer().createGenTable(null);
 		if ( selection!=null && selection.getFirstElement() instanceof IPackageFragment ) {
 			IPackageFragment packageFrag = (IPackageFragment)selection.getFirstElement();
 			newDefaultTable.setPackage( packageFrag.getElementName() );
 			for (IPackageFragmentRoot root : JDTTools.getJavaSourceFolders(this.jpaProject.getJavaProject())) {
 				String srcFolder = root.getPath().toPortableString();
 				if( packageFrag.getPath().toPortableString().startsWith( srcFolder +'/' )){
 					newDefaultTable.setSourceFolder(srcFolder.substring(1));
 				}
 			}
 		}
 		return this.customizer;
 	}

 	protected String getCustomizationFileName() {
 		ConnectionProfile profile = getProjectConnectionProfile();
 		String connection = profile == null ? "" : profile.getName();
 		String name = "org.eclipse.jpt.entitygen." + (connection == null ? "" :connection.replace(' ', '-'));  //$NON-NLS-1$
 		Schema schema = getDefaultSchema();
 		if ( schema!= null  ) {
 			name += "." + schema.getName();//$NON-NLS-1$
 		}
 		return name.toLowerCase();
 	}

 	/**
 	 * Returns the nodes state file.
 	 */
 	private File getCustomizationFile() {
 		String projectPath = this.jpaProject.getProject().getLocation().toPortableString();
 		File genDir = new File(projectPath + "/.settings");//$NON-NLS-1$
         genDir.mkdirs();
 		return new File(genDir, getCustomizationFileName());
 	}

 	@Override
 	public boolean performFinish() {
 		if (this.jpaProject == null) {
 			return true;
 		}
 		try {
 			this.customizer.setDatabaseAnnotationNameBuilder( buildDatabaseAnnotationNameBuilder() );
 			this.customizer.save();
 		} catch (IOException e) {
 			JptUiPlugin.log(e);
 		}
 		if (shouldShowOverwriteWarning()) {
 			PackageGenerator.setOverwriteConfirmer(new OverwriteConfirmer());
 		}

 		WorkspaceJob genEntitiesJob = new GenerateEntitiesJob(this.jpaProject, getCustomizer());
 		genEntitiesJob.schedule();
 		return true;
 	}

 	// ********** generate entities job **********

 	static class GenerateEntitiesJob extends WorkspaceJob {
 		JpaProject jpaProject;
 		ORMGenCustomizer customizer;
 		GenerateEntitiesJob(JpaProject jpaProject, ORMGenCustomizer customizer) {
 			super(JptUiMessages.EntitiesGenerator_jobName);
 			this.customizer = customizer;
 			this.jpaProject = jpaProject ;
 			IResourceRuleFactory ruleFactory = ResourcesPlugin.getWorkspace().getRuleFactory();
 			this.setRule(ruleFactory.modifyRule(jpaProject.getProject()));
 		}

 		@Override
 		public IStatus runInWorkspace(IProgressMonitor monitor) throws CoreException {
 			try{
 				PackageGenerator.generate(this.jpaProject,this.customizer, monitor);
 			}catch(OperationCanceledException e){
 				//user canceled generation
 			}
 			return Status.OK_STATUS;
 		}

 	}
     public static boolean shouldShowOverwriteWarning(){
     	IEclipsePreferences pref = new InstanceScope().getNode(JptUiPlugin.PLUGIN_ID);
     	boolean ret = ! pref.getBoolean( DONT_SHOW_OVERWRITE_WARNING_DIALOG, false) ;
     	return ret;
     }

     // ********** overwrite confirmer **********

 	static class OverwriteConfirmer implements org.eclipse.jpt.gen.internal.OverwriteConfirmer {
 		private boolean overwriteAll = false;
 		private boolean skipAll = false;

 		OverwriteConfirmer() {
 		}

 		public boolean overwrite(final String className) {
 			if (this.overwriteAll) {
 				return true;
 			}
 			if (this.skipAll) {
 				return false;
 			}
 			return this.promptUser(className);
 		}

 		private boolean promptUser(final String className) {
 			// get on the UI thread synchronously, need feedback before continuing
 			final boolean ret[]=new boolean[1];
 			Display.getDefault().syncExec(new Runnable() {
 				public void run() {
 					final OverwriteConfirmerDialog dialog = new OverwriteConfirmerDialog(Display.getCurrent().getActiveShell(), className);
 					dialog.open();
 					if (dialog.getReturnCode() == Window.CANCEL) {
 						//throw new OperationCanceledException();
 						skipAll = true;
 						ret[0] = false;
 						return;
 					}
 					if (dialog.yes()) {
 						ret[0] = true;
 					}
 					if (dialog.yesToAll()) {
 						overwriteAll = true;
 						ret[0] = true;
 					}
 					if (dialog.no()) {
 						ret[0] = false;
 					}
 					if (dialog.noToAll()) {
 						skipAll = true;
 						ret[0] = false;
 					}
 				}
 			});
 			return ret[0];
 		}

 	}


 	// ********** overwrite dialog **********

 	static class OverwriteConfirmerDialog extends Dialog {
 		private final String className;
 		private boolean yes = false;
 		private boolean yesToAll = false;
 		private boolean no = false;
 		private boolean noToAll = false;

 		OverwriteConfirmerDialog(Shell parent, String className) {
 			super(parent);
 			this.className = className;
 		}

 		@Override
 		protected void configureShell(Shell shell) {
 			super.configureShell(shell);
 			shell.setText(JptUiMessages.OverwriteConfirmerDialog_title);
 		}

 		@Override
 		protected Control createDialogArea(Composite parent) {
 			Composite composite = (Composite) super.createDialogArea(parent);
 			GridLayout gridLayout = (GridLayout) composite.getLayout();
 			gridLayout.numColumns = 1;

 			Label text = new Label(composite, SWT.LEFT);
 			text.setText(NLS.bind(JptUiMessages.OverwriteConfirmerDialog_text, this.className));
 			text.setLayoutData(new GridData());

 			createDontShowControl(composite);

 			return composite;
 		}

 	    protected Control createDontShowControl(Composite composite) {
 	    	final Button checkbox = new Button( composite, SWT.CHECK );
 	    	checkbox.setText( JptUiEntityGenMessages.GenerateEntitiesWizard_doNotShowWarning );
 	    	checkbox.setSelection(false);
 	    	final IEclipsePreferences pref = new InstanceScope().getNode( JptUiPlugin.PLUGIN_ID);
 	    	checkbox.setLayoutData( new GridData(GridData.FILL_BOTH) );
 	    	checkbox.addSelectionListener(new SelectionListener (){
 	    		public void widgetDefaultSelected(SelectionEvent e) {}
 				public void widgetSelected(SelectionEvent e) {
 					boolean b = checkbox.getSelection();
 	                pref.putBoolean( DONT_SHOW_OVERWRITE_WARNING_DIALOG, b);
 				}
 	    	});
 	    	return checkbox;
 	    }

 		@Override
 		protected void createButtonsForButtonBar(Composite parent) {
 			this.createButton(parent, IDialogConstants.YES_ID, IDialogConstants.YES_LABEL, false);
 			this.createButton(parent, IDialogConstants.YES_TO_ALL_ID, IDialogConstants.YES_TO_ALL_LABEL, false);
 			this.createButton(parent, IDialogConstants.NO_ID, IDialogConstants.NO_LABEL, true);
 			this.createButton(parent, IDialogConstants.NO_TO_ALL_ID, IDialogConstants.NO_TO_ALL_LABEL, false);
 			this.createButton(parent, IDialogConstants.CANCEL_ID, IDialogConstants.CANCEL_LABEL, false);
 		}

 		@Override
 		protected void buttonPressed(int buttonId) {
 			switch (buttonId) {
 				case IDialogConstants.YES_ID :
 					this.yesPressed();
 					break;
 				case IDialogConstants.YES_TO_ALL_ID :
 					this.yesToAllPressed();
 					break;
 				case IDialogConstants.NO_ID :
 					this.noPressed();
 					break;
 				case IDialogConstants.NO_TO_ALL_ID :
 					this.noToAllPressed();
 					break;
 				case IDialogConstants.CANCEL_ID :
 					this.cancelPressed();
 					break;
 				default :
 					break;
 			}
 		}

 		private void yesPressed() {
 			this.yes = true;
 			this.setReturnCode(OK);
 			this.close();
 		}

 		private void yesToAllPressed() {
 			this.yesToAll = true;
 			this.setReturnCode(OK);
 			this.close();
 		}

 		private void noPressed() {
 			this.no = true;
 			this.setReturnCode(OK);
 			this.close();
 		}

 		private void noToAllPressed() {
 			this.noToAll = true;
 			this.setReturnCode(OK);
 			this.close();
 		}

 		boolean yes() {
 			return this.yes;
 		}

 		boolean yesToAll() {
 			return this.yesToAll;
 		}

 		boolean no() {
 			return this.no;
 		}

 		boolean noToAll() {
 			return this.noToAll;
 		}
 	}


 	private DatabaseAnnotationNameBuilder buildDatabaseAnnotationNameBuilder() {
 		return new LocalDatabaseAnnotationNameBuilder(this.jpaProject.getJpaPlatform().getEntityGeneratorDatabaseAnnotationNameBuilder());
 	}

 	// ********** name builder adapter **********

 	/**
 	 * adapt the JPA platform-supplied builder to the builder interface
 	 * expected by the entity generator
 	 */
 	static class LocalDatabaseAnnotationNameBuilder implements DatabaseAnnotationNameBuilder {
 		private EntityGeneratorDatabaseAnnotationNameBuilder builder;
 		LocalDatabaseAnnotationNameBuilder(EntityGeneratorDatabaseAnnotationNameBuilder builder) {
 			super();
 			this.builder = builder;
 		}
 		public String buildTableAnnotationName(String entityName, Table table) {
 			return this.builder.buildTableAnnotationName(entityName, table);
 		}
 		public String buildColumnAnnotationName(String attributeName, Column column) {
 			return this.builder.buildColumnAnnotationName(attributeName, column);
 		}
 		public String buildJoinColumnAnnotationName(String attributeName, ForeignKey foreignKey) {
 			return this.builder.buildJoinColumnAnnotationName(attributeName, foreignKey);
 		}
 		public String buildJoinColumnAnnotationName(Column column) {
 			return this.builder.buildJoinColumnAnnotationName(column);
 		}
 		public String buildJoinTableAnnotationName(Table table) {
 			return this.builder.buildJoinTableAnnotationName(table);
 		}
 	}

 	@Override
     public IWizardPage getStartingPage() {
 		if (this.projectPage != null) {
 			if (this.tablesSelectorPage != null) {
 				return this.tablesSelectorPage;
 			}
 			return this.projectPage;
 		}
 		return super.getStartingPage();
     }

 	public ORMGenCustomizer getCustomizer (){
 		return customizer;
 	}
 //	Collection<Table> getPossibleTables() {
 //		if ( this.tablesSelectorPage != null) {
 //			return this.tablesSelectorPage.getTables();
 //		}
 //		return ( this.projectDefaultSchemaExists()) ? CollectionTools.collection( this.getDefaultSchema().tables()) : Collections.<Table>emptyList();
 //	}

 	public ConnectionProfile getProjectConnectionProfile() {
 		return this.jpaProject.getConnectionProfile();
 	}

 	public JpaProject getJpaProject(){
 		return this.jpaProject;
 	}

 	public void setJpaProject(JpaProject jpaProject) {
 		if (this.jpaProject == null) {
 			this.jpaProject = jpaProject;
 			IWizardPage currentPage = getContainer().getCurrentPage();
 			if (this.projectPage != null && currentPage.equals(this.projectPage)) {
 				addMainPages();
 			}
 		}
 	}

 	public Schema getDefaultSchema() {
 		return getJpaProject().getDefaultDbSchema();
 	}

 	public void init(IWorkbench workbench, IStructuredSelection selection) {
 		Object sel = selection.getFirstElement();
 		if ( sel instanceof IResource ) {
 			IProject proj = ((IResource) sel).getProject();
 			JpaProject jpaProj = JptCorePlugin.getJpaProject(proj);
 			this.jpaProject = jpaProj;
 		} else if( sel instanceof org.eclipse.jdt.core.IPackageFragmentRoot ) {
 			org.eclipse.jdt.core.IPackageFragmentRoot root = (org.eclipse.jdt.core.IPackageFragmentRoot) sel;
 			IProject proj = root.getJavaProject().getProject();
 			JpaProject jpaProj = JptCorePlugin.getJpaProject(proj);
 			this.jpaProject = jpaProj;
 		} else if( sel instanceof org.eclipse.jdt.core.IPackageFragment) {
 			org.eclipse.jdt.core.IPackageFragment frag = (org.eclipse.jdt.core.IPackageFragment) sel;
 			IProject proj = frag.getJavaProject().getProject();
 			JpaProject jpaProj = JptCorePlugin.getJpaProject(proj);
 			this.jpaProject = jpaProj;
 		}

 		this.selection = selection;
 		this.setWindowTitle(JptUiEntityGenMessages.GenerateEntitiesWizard_generateEntities);
 	}

 	@Override
 	public void dispose() {
 		this.resourceManager.dispose();
 		super.dispose();
 	}
 }
	/*******************************************************************************
	* Copyright (c) 2007, 2010 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.gen;

	import java.io.File;
	import java.io.IOException;

	import org.eclipse.core.resources.IProject;
	import org.eclipse.core.resources.IResource;
	import org.eclipse.core.resources.IResourceRuleFactory;
	import org.eclipse.core.resources.ResourcesPlugin;
	import org.eclipse.core.resources.WorkspaceJob;
	import org.eclipse.core.runtime.CoreException;
	import org.eclipse.core.runtime.IProgressMonitor;
	import org.eclipse.core.runtime.IStatus;
	import org.eclipse.core.runtime.OperationCanceledException;
	import org.eclipse.core.runtime.Platform;
	import org.eclipse.core.runtime.Status;
	import org.eclipse.core.runtime.preferences.IEclipsePreferences;
	import org.eclipse.core.runtime.preferences.InstanceScope;
	import org.eclipse.jdt.core.IJavaElement;
	import org.eclipse.jdt.core.IPackageFragment;
	import org.eclipse.jdt.core.IPackageFragmentRoot;
	import org.eclipse.jface.dialogs.Dialog;
	import org.eclipse.jface.dialogs.IDialogConstants;
	import org.eclipse.jface.resource.JFaceResources;
	import org.eclipse.jface.resource.LocalResourceManager;
	import org.eclipse.jface.resource.ResourceManager;
	import org.eclipse.jface.viewers.IStructuredSelection;
	import org.eclipse.jface.window.Window;
	import org.eclipse.jface.wizard.IWizardPage;
	import org.eclipse.jface.wizard.Wizard;
	import org.eclipse.jpt.core.EntityGeneratorDatabaseAnnotationNameBuilder;
	import org.eclipse.jpt.core.JpaPlatform;
	import org.eclipse.jpt.core.JpaProject;
	import org.eclipse.jpt.core.JptCorePlugin;
	import org.eclipse.jpt.core.internal.utility.jdt.JDTTools;
	import org.eclipse.jpt.db.Column;
	import org.eclipse.jpt.db.ConnectionProfile;
	import org.eclipse.jpt.db.ForeignKey;
	import org.eclipse.jpt.db.Schema;
	import org.eclipse.jpt.db.Table;
	import org.eclipse.jpt.gen.internal.BaseEntityGenCustomizer;
	import org.eclipse.jpt.gen.internal.DatabaseAnnotationNameBuilder;
	import org.eclipse.jpt.gen.internal.ORMGenCustomizer;
	import org.eclipse.jpt.gen.internal.ORMGenTable;
	import org.eclipse.jpt.gen.internal.PackageGenerator;
	import org.eclipse.jpt.ui.JptUiPlugin;
	import org.eclipse.jpt.ui.internal.JptUiIcons;
	import org.eclipse.jpt.ui.internal.JptUiMessages;
	import org.eclipse.osgi.util.NLS;
	import org.eclipse.swt.SWT;
	import org.eclipse.swt.events.SelectionEvent;
	import org.eclipse.swt.events.SelectionListener;
	import org.eclipse.swt.layout.GridData;
	import org.eclipse.swt.layout.GridLayout;
	import org.eclipse.swt.widgets.Button;
	import org.eclipse.swt.widgets.Composite;
	import org.eclipse.swt.widgets.Control;
	import org.eclipse.swt.widgets.Display;
	import org.eclipse.swt.widgets.Label;
	import org.eclipse.swt.widgets.Shell;
	import org.eclipse.ui.INewWizard;
	import org.eclipse.ui.IWorkbench;

	public class GenerateEntitiesFromSchemaWizard extends Wizard
	implements INewWizard {

	public static final String HELP_CONTEXT_ID = JptUiPlugin.PLUGIN_ID + ".GenerateEntitiesFromSchemaWizard"; //$NON-NLS-1$

	private static final String DONT_SHOW_OVERWRITE_WARNING_DIALOG = "DONT_SHOW_OVERWRITE_WARNING_DIALOG"; //$NON-NLS-1$

	private JpaProject jpaProject;

	private IStructuredSelection selection;

	private ORMGenCustomizer customizer = null;

	private PromptJPAProjectWizardPage projectPage;

	private TablesSelectorWizardPage tablesSelectorPage;

	private TableAssociationsWizardPage tableAssociationsPage;

	private DefaultTableGenerationWizardPage defaultTableGenerationPage;

	private TablesAndColumnsCustomizationWizardPage tablesAndColumnsCustomizationPage;

	protected final ResourceManager resourceManager;

	public GenerateEntitiesFromSchemaWizard() {
	this.resourceManager = new LocalResourceManager(JFaceResources.getResources());
	this.setWindowTitle( JptUiEntityGenMessages.GenerateEntitiesWizard_generateEntities);
	}

	public GenerateEntitiesFromSchemaWizard( JpaProject jpaProject, IStructuredSelection selection) {
	super();
	this.jpaProject = jpaProject;
	this.selection = selection;
	this.resourceManager = new LocalResourceManager(JFaceResources.getResources());
	this.setWindowTitle( JptUiEntityGenMessages.GenerateEntitiesWizard_generateEntities);
	this.setDefaultPageImageDescriptor(JptUiPlugin.getImageDescriptor(JptUiIcons.ENTITY_WIZ_BANNER));
	}

	@Override
	public void addPages() {
	setForcePreviousAndNextButtons(true);

	//If this.jpaProject is not initialized because user didn't select a JPA project
	if( this.jpaProject == null ){
	this.projectPage = new PromptJPAProjectWizardPage(HELP_CONTEXT_ID);
	this.addPage(this.projectPage);
	return;
	}
	addMainPages();
	}

	private void addMainPages() {
	this.tablesSelectorPage = new TablesSelectorWizardPage(this.jpaProject, this.resourceManager);
	this.addPage(this.tablesSelectorPage);

	this.tableAssociationsPage = new TableAssociationsWizardPage(this.jpaProject, this.resourceManager);
	this.addPage(this.tableAssociationsPage);

	this.defaultTableGenerationPage = new DefaultTableGenerationWizardPage(this.jpaProject);
	this.addPage(this.defaultTableGenerationPage);
	this.defaultTableGenerationPage.init(this.selection);

	this.tablesAndColumnsCustomizationPage = new TablesAndColumnsCustomizationWizardPage(this.jpaProject, this.resourceManager);
	this.addPage(this.tablesAndColumnsCustomizationPage);
	this.tablesAndColumnsCustomizationPage.init(this.selection);
	}

	public ORMGenCustomizer getORMGenCustomizer(){
	return this.customizer;
	}

	/**
	* Create the ORMGenCustomizer when user selects a new connection profile and schema
	*
	* JpaPlatform implementor can provide a custom ORMGenCustomizer specific to a platform
	* with AdapterFactory through Eclipse org.eclipse.core.runtime.adapters extension point:
	* <pre>
	*
	*<extension
	* point="org.eclipse.core.runtime.adapters">
	* <factory
	* adaptableType="org.eclipse.jpt.eclipselink.core.internal.EclipseLinkPlatform"
	* class="oracle.eclipse.tools.orm.internal.EclipseLinkORMGenCustomizerAdapterFactory">
	* <adapter
	* type="oracle.eclipse.tools.orm.internal.ORMGenCustomizer">
	* </adapter>
	* </factory>
	*</extension>
	*</pre>
	*
	* @param schema
	*/
	public ORMGenCustomizer createORMGenCustomizer(Schema schema){
	JpaPlatform jpaPlatform = this.jpaProject.getJpaPlatform();
	Object obj = Platform.getAdapterManager().getAdapter( jpaPlatform, ORMGenCustomizer.class );
	if (obj != null && obj instanceof ORMGenCustomizer) {
	this.customizer = (ORMGenCustomizer) obj;
	this.customizer.init(getCustomizationFile(), schema);
	} else{
	this.customizer = new BaseEntityGenCustomizer( );
	this.customizer.init(getCustomizationFile(), schema);
	}

	ORMGenTable newDefaultTable = getCustomizer().createGenTable(null);
	if ( selection!=null && selection.getFirstElement() instanceof IPackageFragment ) {
	IPackageFragment packageFrag = (IPackageFragment)selection.getFirstElement();
	newDefaultTable.setPackage( packageFrag.getElementName() );
	for (IPackageFragmentRoot root : JDTTools.getJavaSourceFolders(this.jpaProject.getJavaProject())) {
	String srcFolder = root.getPath().toPortableString();
	if( packageFrag.getPath().toPortableString().startsWith( srcFolder +'/' )){
	newDefaultTable.setSourceFolder(srcFolder.substring(1));
	}
	}
	}
	return this.customizer;
	}

	protected String getCustomizationFileName() {
	ConnectionProfile profile = getProjectConnectionProfile();
	String connection = profile == null ? "" : profile.getName();
	String name = "org.eclipse.jpt.entitygen." + (connection == null ? "" :connection.replace(' ', '-')); //$NON-NLS-1$
	Schema schema = getDefaultSchema();
	if ( schema!= null ) {
	name += "." + schema.getName();//$NON-NLS-1$
	}
	return name.toLowerCase();
	}

	/**
	* Returns the nodes state file.
	*/
	private File getCustomizationFile() {
	String projectPath = this.jpaProject.getProject().getLocation().toPortableString();
	File genDir = new File(projectPath + "/.settings");//$NON-NLS-1$
	genDir.mkdirs();
	return new File(genDir, getCustomizationFileName());
	}

	@Override
	public boolean performFinish() {
	if (this.jpaProject == null) {
	return true;
	}
	try {
	this.customizer.setDatabaseAnnotationNameBuilder( buildDatabaseAnnotationNameBuilder() );
	this.customizer.save();
	} catch (IOException e) {
	JptUiPlugin.log(e);
	}
	if (shouldShowOverwriteWarning()) {
	PackageGenerator.setOverwriteConfirmer(new OverwriteConfirmer());
	}

	WorkspaceJob genEntitiesJob = new GenerateEntitiesJob(this.jpaProject, getCustomizer());
	genEntitiesJob.schedule();
	return true;
	}

	// ******** generate entities job ********

	static class GenerateEntitiesJob extends WorkspaceJob {
	JpaProject jpaProject;
	ORMGenCustomizer customizer;
	GenerateEntitiesJob(JpaProject jpaProject, ORMGenCustomizer customizer) {
	super(JptUiMessages.EntitiesGenerator_jobName);
	this.customizer = customizer;
	this.jpaProject = jpaProject ;
	IResourceRuleFactory ruleFactory = ResourcesPlugin.getWorkspace().getRuleFactory();
	this.setRule(ruleFactory.modifyRule(jpaProject.getProject()));
	}

	@Override
	public IStatus runInWorkspace(IProgressMonitor monitor) throws CoreException {
	try{
	PackageGenerator.generate(this.jpaProject,this.customizer, monitor);
	}catch(OperationCanceledException e){
	//user canceled generation
	}
	return Status.OK_STATUS;
	}

	}
	public static boolean shouldShowOverwriteWarning(){
	IEclipsePreferences pref = new InstanceScope().getNode(JptUiPlugin.PLUGIN_ID);
	boolean ret = ! pref.getBoolean( DONT_SHOW_OVERWRITE_WARNING_DIALOG, false) ;
	return ret;
	}

	// ******** overwrite confirmer ********

	static class OverwriteConfirmer implements org.eclipse.jpt.gen.internal.OverwriteConfirmer {
	private boolean overwriteAll = false;
	private boolean skipAll = false;

	OverwriteConfirmer() {
	}

	public boolean overwrite(final String className) {
	if (this.overwriteAll) {
	return true;
	}
	if (this.skipAll) {
	return false;
	}
	return this.promptUser(className);
	}

	private boolean promptUser(final String className) {
	// get on the UI thread synchronously, need feedback before continuing
	final boolean ret[]=new boolean[1];
	Display.getDefault().syncExec(new Runnable() {
	public void run() {
	final OverwriteConfirmerDialog dialog = new OverwriteConfirmerDialog(Display.getCurrent().getActiveShell(), className);
	dialog.open();
	if (dialog.getReturnCode() == Window.CANCEL) {
	//throw new OperationCanceledException();
	skipAll = true;
	ret[0] = false;
	return;
	}
	if (dialog.yes()) {
	ret[0] = true;
	}
	if (dialog.yesToAll()) {
	overwriteAll = true;
	ret[0] = true;
	}
	if (dialog.no()) {
	ret[0] = false;
	}
	if (dialog.noToAll()) {
	skipAll = true;
	ret[0] = false;
	}
	}
	});
	return ret[0];
	}

	}


	// ******** overwrite dialog ********

	static class OverwriteConfirmerDialog extends Dialog {
	private final String className;
	private boolean yes = false;
	private boolean yesToAll = false;
	private boolean no = false;
	private boolean noToAll = false;

	OverwriteConfirmerDialog(Shell parent, String className) {
	super(parent);
	this.className = className;
	}

	@Override
	protected void configureShell(Shell shell) {
	super.configureShell(shell);
	shell.setText(JptUiMessages.OverwriteConfirmerDialog_title);
	}

	@Override
	protected Control createDialogArea(Composite parent) {
	Composite composite = (Composite) super.createDialogArea(parent);
	GridLayout gridLayout = (GridLayout) composite.getLayout();
	gridLayout.numColumns = 1;

	Label text = new Label(composite, SWT.LEFT);
	text.setText(NLS.bind(JptUiMessages.OverwriteConfirmerDialog_text, this.className));
	text.setLayoutData(new GridData());

	createDontShowControl(composite);

	return composite;
	}

	protected Control createDontShowControl(Composite composite) {
	final Button checkbox = new Button( composite, SWT.CHECK );
	checkbox.setText( JptUiEntityGenMessages.GenerateEntitiesWizard_doNotShowWarning );
	checkbox.setSelection(false);
	final IEclipsePreferences pref = new InstanceScope().getNode( JptUiPlugin.PLUGIN_ID);
	checkbox.setLayoutData( new GridData(GridData.FILL_BOTH) );
	checkbox.addSelectionListener(new SelectionListener (){
	public void widgetDefaultSelected(SelectionEvent e) {}
	public void widgetSelected(SelectionEvent e) {
	boolean b = checkbox.getSelection();
	pref.putBoolean( DONT_SHOW_OVERWRITE_WARNING_DIALOG, b);
	}
	});
	return checkbox;
	}

	@Override
	protected void createButtonsForButtonBar(Composite parent) {
	this.createButton(parent, IDialogConstants.YES_ID, IDialogConstants.YES_LABEL, false);
	this.createButton(parent, IDialogConstants.YES_TO_ALL_ID, IDialogConstants.YES_TO_ALL_LABEL, false);
	this.createButton(parent, IDialogConstants.NO_ID, IDialogConstants.NO_LABEL, true);
	this.createButton(parent, IDialogConstants.NO_TO_ALL_ID, IDialogConstants.NO_TO_ALL_LABEL, false);
	this.createButton(parent, IDialogConstants.CANCEL_ID, IDialogConstants.CANCEL_LABEL, false);
	}

	@Override
	protected void buttonPressed(int buttonId) {
	switch (buttonId) {
	case IDialogConstants.YES_ID :
	this.yesPressed();
	break;
	case IDialogConstants.YES_TO_ALL_ID :
	this.yesToAllPressed();
	break;
	case IDialogConstants.NO_ID :
	this.noPressed();
	break;
	case IDialogConstants.NO_TO_ALL_ID :
	this.noToAllPressed();
	break;
	case IDialogConstants.CANCEL_ID :
	this.cancelPressed();
	break;
	default :
	break;
	}
	}

	private void yesPressed() {
	this.yes = true;
	this.setReturnCode(OK);
	this.close();
	}

	private void yesToAllPressed() {
	this.yesToAll = true;
	this.setReturnCode(OK);
	this.close();
	}

	private void noPressed() {
	this.no = true;
	this.setReturnCode(OK);
	this.close();
	}

	private void noToAllPressed() {
	this.noToAll = true;
	this.setReturnCode(OK);
	this.close();
	}

	boolean yes() {
	return this.yes;
	}

	boolean yesToAll() {
	return this.yesToAll;
	}

	boolean no() {
	return this.no;
	}

	boolean noToAll() {
	return this.noToAll;
	}
	}


	private DatabaseAnnotationNameBuilder buildDatabaseAnnotationNameBuilder() {
	return new LocalDatabaseAnnotationNameBuilder(this.jpaProject.getJpaPlatform().getEntityGeneratorDatabaseAnnotationNameBuilder());
	}

	// ******** name builder adapter ********

	/**
	* adapt the JPA platform-supplied builder to the builder interface
	* expected by the entity generator
	*/
	static class LocalDatabaseAnnotationNameBuilder implements DatabaseAnnotationNameBuilder {
	private EntityGeneratorDatabaseAnnotationNameBuilder builder;
	LocalDatabaseAnnotationNameBuilder(EntityGeneratorDatabaseAnnotationNameBuilder builder) {
	super();
	this.builder = builder;
	}
	public String buildTableAnnotationName(String entityName, Table table) {
	return this.builder.buildTableAnnotationName(entityName, table);
	}
	public String buildColumnAnnotationName(String attributeName, Column column) {
	return this.builder.buildColumnAnnotationName(attributeName, column);
	}
	public String buildJoinColumnAnnotationName(String attributeName, ForeignKey foreignKey) {
	return this.builder.buildJoinColumnAnnotationName(attributeName, foreignKey);
	}
	public String buildJoinColumnAnnotationName(Column column) {
	return this.builder.buildJoinColumnAnnotationName(column);
	}
	public String buildJoinTableAnnotationName(Table table) {
	return this.builder.buildJoinTableAnnotationName(table);
	}
	}

	@Override
	public IWizardPage getStartingPage() {
	if (this.projectPage != null) {
	if (this.tablesSelectorPage != null) {
	return this.tablesSelectorPage;
	}
	return this.projectPage;
	}
	return super.getStartingPage();
	}

	public ORMGenCustomizer getCustomizer (){
	return customizer;
	}
	// Collection<Table> getPossibleTables() {
	// if ( this.tablesSelectorPage != null) {
	// return this.tablesSelectorPage.getTables();
	// }
	// return ( this.projectDefaultSchemaExists()) ? CollectionTools.collection( this.getDefaultSchema().tables()) : Collections.<Table>emptyList();
	// }

	public ConnectionProfile getProjectConnectionProfile() {
	return this.jpaProject.getConnectionProfile();
	}

	public JpaProject getJpaProject(){
	return this.jpaProject;
	}

	public void setJpaProject(JpaProject jpaProject) {
	if (this.jpaProject == null) {
	this.jpaProject = jpaProject;
	IWizardPage currentPage = getContainer().getCurrentPage();
	if (this.projectPage != null && currentPage.equals(this.projectPage)) {
	addMainPages();
	}
	}
	}

	public Schema getDefaultSchema() {
	return getJpaProject().getDefaultDbSchema();
	}

	public void init(IWorkbench workbench, IStructuredSelection selection) {
	Object sel = selection.getFirstElement();
	if ( sel instanceof IResource ) {
	IProject proj = ((IResource) sel).getProject();
	JpaProject jpaProj = JptCorePlugin.getJpaProject(proj);
	this.jpaProject = jpaProj;
	} else if( sel instanceof org.eclipse.jdt.core.IPackageFragmentRoot ) {
	org.eclipse.jdt.core.IPackageFragmentRoot root = (org.eclipse.jdt.core.IPackageFragmentRoot) sel;
	IProject proj = root.getJavaProject().getProject();
	JpaProject jpaProj = JptCorePlugin.getJpaProject(proj);
	this.jpaProject = jpaProj;
	} else if( sel instanceof org.eclipse.jdt.core.IPackageFragment) {
	org.eclipse.jdt.core.IPackageFragment frag = (org.eclipse.jdt.core.IPackageFragment) sel;
	IProject proj = frag.getJavaProject().getProject();
	JpaProject jpaProj = JptCorePlugin.getJpaProject(proj);
	this.jpaProject = jpaProj;
	}

	this.selection = selection;
	this.setWindowTitle(JptUiEntityGenMessages.GenerateEntitiesWizard_generateEntities);
	}

	@Override
	public void dispose() {
	this.resourceManager.dispose();
	super.dispose();
	}
	}