plugins/org.eclipse.jst.j2ee.ejb.annotations.ui/src/org/eclipse/jst/j2ee/ejb/annotation/ui/internal/wizards/ChooseEjbTypeWizardPage.java

/**
 * 
 */
package org.eclipse.jst.j2ee.ejb.annotation.ui.internal.wizards;

import org.eclipse.jface.preference.IPreferenceNode;
import org.eclipse.jface.preference.PreferenceDialog;
import org.eclipse.jface.preference.PreferenceManager;
import org.eclipse.jface.window.Window;
import org.eclipse.jst.j2ee.ejb.annotation.internal.messages.IEJBAnnotationConstants;
import org.eclipse.jst.j2ee.ejb.annotation.internal.model.EnterpriseBeanClassDataModel;
import org.eclipse.jst.j2ee.ejb.annotation.internal.preferences.AnnotationPreferenceStore;
import org.eclipse.jst.j2ee.ejb.annotation.internal.provider.IAnnotationProvider;
import org.eclipse.jst.j2ee.ejb.annotation.internal.utility.AnnotationUtilities;
import org.eclipse.swt.SWT;
import org.eclipse.swt.custom.BusyIndicator;
import org.eclipse.swt.events.SelectionAdapter;
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.Combo;
import org.eclipse.swt.widgets.Composite;
import org.eclipse.swt.widgets.Label;
import org.eclipse.swt.widgets.Link;
import org.eclipse.ui.PlatformUI;
import org.eclipse.wst.common.frameworks.internal.operations.WTPOperationDataModel;
import org.eclipse.wst.common.frameworks.internal.ui.WTPWizardPage;

/**
 * @author naci
 */
public class ChooseEjbTypeWizardPage extends WTPWizardPage {

	protected Button sessionType;
	protected Button messageDrivenType;
	protected Combo annotationProvider;

	protected ChooseEjbTypeWizardPage(WTPOperationDataModel model, String pageName) {
		super(model, pageName);
		this.setDescription(IEJBAnnotationConstants.ADD_EJB_WIZARD_PAGE_DESC);
	}

	/*
	 * (non-Javadoc)
	 * 
	 * @see org.eclipse.wst.common.frameworks.internal.ui.WTPWizardPage#getValidationPropertyNames()
	 */
	protected String[] getValidationPropertyNames() {
		return new String[] { EnterpriseBeanClassDataModel.EJB_TYPE };
	}

	/*
	 * (non-Javadoc)
	 * 
	 * @see org.eclipse.wst.common.frameworks.internal.ui.WTPWizardPage#createTopLevelComposite(org.eclipse.swt.widgets.Composite)
	 */
	protected Composite createTopLevelComposite(Composite parent) {
		Composite aComposite = new Composite(parent, SWT.NULL);
		aComposite.setLayout(new GridLayout());
		aComposite.setLayoutData(new GridData(GridData.VERTICAL_ALIGN_FILL | GridData.HORIZONTAL_ALIGN_FILL));
		aComposite.setSize(aComposite.computeSize(SWT.DEFAULT, SWT.DEFAULT));
		aComposite.setFont(parent.getFont());
		createEjbTypeGroup(aComposite);
		createAnnotationProviderGroup(aComposite);
		addPreferenceLink(aComposite);

		setControl(aComposite);
		return aComposite;

	}

	protected void createEjbTypeGroup(Composite parent) {
		Composite ejbTypeGroup = new Composite(parent, SWT.NONE);
		GridLayout layout = new GridLayout();
		ejbTypeGroup.setLayout(layout);
		ejbTypeGroup.setLayoutData(new GridData(GridData.HORIZONTAL_ALIGN_FILL | GridData.GRAB_HORIZONTAL));

		sessionType = new Button(ejbTypeGroup, SWT.RADIO);
		sessionType.setText("SessionBean"); //$NON-NLS-1$

		messageDrivenType = new Button(ejbTypeGroup, SWT.RADIO);
		messageDrivenType.setText("MessageDrivenBean"); //$NON-NLS-1$

		sessionType.setSelection(true);
		messageDrivenType.setSelection(false);

		sessionType.addSelectionListener(new SelectionListener() {

			public void widgetSelected(SelectionEvent e) {
				if (sessionType.getSelection()) {
					ChooseEjbTypeWizardPage.this.model.setProperty(EnterpriseBeanClassDataModel.EJB_TYPE, "SessionBean");
				}
			}

			public void widgetDefaultSelected(SelectionEvent e) {
				this.widgetSelected(e);
			}
		});

		messageDrivenType.addSelectionListener(new SelectionListener() {

			public void widgetSelected(SelectionEvent e) {
				if (messageDrivenType.getSelection()) {
					ChooseEjbTypeWizardPage.this.model.setProperty(EnterpriseBeanClassDataModel.EJB_TYPE, "MessageDrivenBean");
				}
			}

			public void widgetDefaultSelected(SelectionEvent e) {
				this.widgetSelected(e);
			}
		});
	}

	protected void createAnnotationProviderGroup(Composite parent) {
		Composite annotationGroup = new Composite(parent, SWT.NONE);
		GridLayout layout = new GridLayout(2, false);
		annotationGroup.setLayout(layout);
		annotationGroup.setLayoutData(new GridData(GridData.HORIZONTAL_ALIGN_FILL | GridData.GRAB_HORIZONTAL));

		Label label = new Label(annotationGroup, SWT.WRAP);
		label.setText("Annotation Provider:");
		label.setToolTipText("Choose the annotation provider that will be used to create java classes and J2EE artifacts");

		annotationProvider = new Combo(annotationGroup, SWT.RADIO);
		String[] provider = AnnotationUtilities.getProviderNames();
		final String preferred = AnnotationPreferenceStore.getProperty(AnnotationPreferenceStore.ANNOTATIONPROVIDER);
		ChooseEjbTypeWizardPage.this.model.setProperty(EnterpriseBeanClassDataModel.ANNOTATIONPROVIDER, preferred);
		boolean selected = false;
		String providerS = null;
		for (int i = 0; i < provider.length; i++) {
			String name = provider[i];
			annotationProvider.add(name);
			if (preferred.equals(name)) {
				providerS = name;
				annotationProvider.select(i);
				selected = true;
			}

		}
		if (!selected) {
			providerS = provider[0];
			annotationProvider.select(0);
		}

		validateProvider();

		if (model != null)
			model.setProperty(EnterpriseBeanClassDataModel.ANNOTATIONPROVIDER, providerS);

		annotationProvider.addSelectionListener(new SelectionListener() {

			public void widgetSelected(SelectionEvent e) {
				validateProvider();

			}

			public void widgetDefaultSelected(SelectionEvent e) {
				this.widgetSelected(e);
			}
		});

	}

	public String getEJBType() {
		return model.getStringProperty(EnterpriseBeanClassDataModel.EJB_TYPE);
	}

	public boolean isPageComplete() {
		String provider = annotationProvider.getText();
		IAnnotationProvider annotationProvider = null;
		try {
			annotationProvider = AnnotationUtilities.findAnnotationProviderByName(provider);
		} catch (Exception e) {
			return false;
		}

		return (annotationProvider != null && annotationProvider.isValid());
	}

	private void addPreferenceLink(final Composite composite) {

		Link link = new Link(composite, SWT.NONE);
		link.setLayoutData(new GridData(SWT.FILL, SWT.NONE, true, false, 3, 1));
		link.setText("You can change your provider <a>preferences</a>");

		link.addSelectionListener(new SelectionAdapter() {
			public void widgetSelected(SelectionEvent e) {
				if (showPreferencePage(composite)) {
				}
				validateProvider();
			}
			public void widgetDefaultSelected(SelectionEvent e) {
					widgetSelected(e);
			}
		});

	}

	protected boolean showPreferencePage(Composite composite) {
		PreferenceManager manager = PlatformUI.getWorkbench().getPreferenceManager();
		IPreferenceNode node = manager.find("org.eclipse.jst.j2ee.ejb.annotation.ui.preferences");
		PreferenceManager manager2 = new PreferenceManager();
		manager2.addToRoot(node);
		final PreferenceDialog dialog = new PreferenceDialog(composite.getShell(), manager2);
		final boolean[] result = new boolean[] { false };
		BusyIndicator.showWhile(composite.getDisplay(), new Runnable() {
			public void run() {
				dialog.create();
				if (dialog.open() == Window.OK)
					result[0] = true;
			}
		});
		return result[0];
	}

	private void validateProvider() {
		String provider = annotationProvider.getText();
		ChooseEjbTypeWizardPage.this.model.setProperty(EnterpriseBeanClassDataModel.ANNOTATIONPROVIDER, provider);
		IAnnotationProvider annotationProvider = null;
		try {
			annotationProvider = AnnotationUtilities.findAnnotationProviderByName(provider);
		} catch (Exception ex) {
		}
		if (annotationProvider != null && annotationProvider.isValid())
			this.setErrorMessage(null);
		else
			this.setErrorMessage("Annotation provider definition is not valid, please check the preferences. ");
		getContainer().updateMessage();
	}

}