New class wizard page

Signed-off-by: Sven Efftinge <sven.efftinge@itemis.de>

1 files changed, 199 insertions, 0 deletions

diff --git a/plugins/org.eclipse.xtext.xtend2.ui/src/org/eclipse/xtext/xtend2/ui/wizards/NewXtend2ClassWizardPage.java b/plugins/org.eclipse.xtext.xtend2.ui/src/org/eclipse/xtext/xtend2/ui/wizards/NewXtend2ClassWizardPage.java
new file mode 100644
index 0000000..d632ae3
--- a/dev/null
+++ b/plugins/org.eclipse.xtext.xtend2.ui/src/org/eclipse/xtext/xtend2/ui/wizards/NewXtend2ClassWizardPage.java
@@ -0,0 +1,199 @@
+/*******************************************************************************
+ * Copyright (c) 2011 itemis AG (http://www.itemis.eu) and others.
+ * 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
+ *******************************************************************************/
+package org.eclipse.xtext.xtend2.ui.wizards;
+
+import java.io.ByteArrayInputStream;
+import java.lang.reflect.InvocationTargetException;
+
+import org.eclipse.core.resources.IFile;
+import org.eclipse.core.resources.IFolder;
+import org.eclipse.core.resources.IResource;
+import org.eclipse.core.runtime.CoreException;
+import org.eclipse.core.runtime.IProgressMonitor;
+import org.eclipse.core.runtime.IStatus;
+import org.eclipse.core.runtime.NullProgressMonitor;
+import org.eclipse.jdt.core.IJavaElement;
+import org.eclipse.jdt.core.IPackageFragment;
+import org.eclipse.jdt.internal.ui.dialogs.StatusInfo;
+import org.eclipse.jdt.ui.wizards.NewTypeWizardPage;
+import org.eclipse.jface.dialogs.MessageDialog;
+import org.eclipse.jface.operation.IRunnableWithProgress;
+import org.eclipse.jface.viewers.IStructuredSelection;
+import org.eclipse.swt.SWT;
+import org.eclipse.swt.layout.GridLayout;
+import org.eclipse.swt.widgets.Composite;
+import org.eclipse.swt.widgets.Display;
+import org.eclipse.ui.actions.WorkspaceModifyOperation;
+
+import com.google.inject.Inject;
+
+/**
+ * @author Robert von Massow - Initial contribution and API
+ */
+@SuppressWarnings("restriction")
+public class NewXtend2ClassWizardPage extends NewTypeWizardPage {
+
+	private IResource fResource;
+	
+	@Inject
+	private FieldInitializerUtil util;
+
+	private int size;
+
+	public NewXtend2ClassWizardPage() {
+		super(0, NewXtend2ClassWizard.TITLE);
+		this.setTitle(NewXtend2ClassWizard.TITLE);
+		this.setDescription(Messages.XTEND_CLASS_WIZARD_DESCRIPTION);
+	}
+
+	public void createControl(Composite parent) {
+		initializeDialogUnits(parent);
+		Composite composite= new Composite(parent, SWT.NONE);
+		composite.setFont(parent.getFont());
+		int cols = 4;
+		GridLayout layout= new GridLayout();
+		layout.numColumns= cols;
+		composite.setLayout(layout);
+		createContainerControls(composite, cols);
+		createPackageControls(composite, cols);
+		createSeparator(composite, cols);
+		createTypeNameControls(composite, cols);
+		createSuperClassControls(composite, cols);
+		createSuperInterfacesControls(composite, cols);
+		setControl(composite);
+	}
+
+	@Override
+	public void setVisible(boolean visible) {
+		super.setVisible(visible);
+		if(visible) {
+			setFocus();
+		}
+	}
+
+	protected void init(IStructuredSelection selection){
+		IJavaElement elem = util.getSelectedResource(selection);
+		initContainerPage(elem);
+		initTypePage(elem);
+	}
+
+	@Override
+	public void createType(IProgressMonitor monitor) throws CoreException, InterruptedException {
+		throw new UnsupportedOperationException();
+	}
+
+	public int createType(){
+		IRunnableWithProgress op = new WorkspaceModifyOperation() {
+			@Override
+			protected void execute(IProgressMonitor monitor) throws CoreException, InvocationTargetException,
+						InterruptedException {
+				try {
+					createXtendClass(monitor);
+				} catch (Exception e) {
+					throw new InvocationTargetException(e);
+				}finally {
+					monitor.done();
+				}
+			}
+		};
+		try {
+			getContainer().run(true, false, op);
+		} catch (InterruptedException e) {
+			throw new RuntimeException(e);
+		} catch (InvocationTargetException e) {
+			Throwable realException = e.getTargetException();
+			MessageDialog
+					.openError(
+							getShell(),
+							Messages.ERROR_CREATING_CLASS,
+							realException.getMessage());
+		}
+		return size;
+	}
+
+	private void createXtendClass(IProgressMonitor monitor) {
+		if(monitor == null){
+			monitor = new NullProgressMonitor();
+		}
+		IResource res = getPackageFragment().getResource();
+		if(!res.exists()) {
+			try {
+				((IFolder)res).create(IResource.NONE, true, monitor);
+			} catch (final CoreException e) {
+				displayError(Messages.ERROR_CREATING_PACKAGE, e.getMessage());
+			}
+		}
+		IFile xtendClass = null;
+			xtendClass = ((IFolder) res).getFile(getTypeName() + ".xtend"); //$NON-NLS-1$
+		try {
+			String contents = XtendClassUtil.create(
+					getTypeName(),
+					getPackageFragment(),
+					getSuperClass(),
+					getSuperInterfaces(),
+					monitor);
+			size = contents.length();
+			xtendClass.create(new ByteArrayInputStream(contents.getBytes()), true, monitor);
+			this.fResource = xtendClass;
+		} catch (CoreException e) {
+			displayError(Messages.ERROR_CREATING_CLASS, e.getMessage());
+		}
+	}
+
+	private void displayError(final String title, final String message) {
+		Display.getDefault().syncExec(new Runnable() {
+			public void run() {
+				MessageDialog
+				.openError(getShell(),title, message);
+			}
+		});
+	}
+
+	private void doStatusUpdate() {
+		IStatus[] status= new IStatus[] {
+			fContainerStatus,
+			fPackageStatus,
+			fTypeNameStatus,
+			fSuperClassStatus,
+			fSuperInterfacesStatus
+		};
+		updateStatus(status);
+	}
+
+	/*
+	 * @see NewContainerWizardPage#handleFieldChanged
+	 */
+	@Override
+	protected void handleFieldChanged(String fieldName) {
+		super.handleFieldChanged(fieldName);
+		doStatusUpdate();
+	}
+
+	@Override
+	protected IStatus typeNameChanged() {
+		IPackageFragment packageFragment = getPackageFragment();
+		if(packageFragment != null) {
+			IResource resource = packageFragment.getResource();
+			if (resource instanceof IFolder) {
+				IFolder folder = (IFolder) resource;
+				if(folder.getFile(getTypeName()+".xtend").exists()) { //$NON-NLS-1$
+					String packageName = ""; //$NON-NLS-1$
+					if(!packageFragment.isDefaultPackage()) {
+						packageName = packageFragment.getElementName() + "."; //$NON-NLS-1$
+					}
+					return new StatusInfo(IStatus.ERROR, Messages.TYPE_EXISTS_0 + packageName + getTypeName() + Messages.TYPE_EXISTS_1);
+				}
+			}
+		}
+		return super.typeNameChanged();
+	}
+
+	public IResource getResource() {
+		return fResource;
+	}
+}