AbstractNewObjectAction.java

/*
 * ====================================================================
 * 
 * The ObjectStyle Group Software License, Version 1.0
 * 
 * Copyright (c) 2006 The ObjectStyle Group and individual authors of the
 * software. All rights reserved.
 * 
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 * 
 * 1. Redistributions of source code must retain the above copyright notice,
 * this list of conditions and the following disclaimer.
 * 
 * 2. Redistributions in binary form must reproduce the above copyright notice,
 * this list of conditions and the following disclaimer in the documentation
 * and/or other materials provided with the distribution.
 * 
 * 3. The end-user documentation included with the redistribution, if any, must
 * include the following acknowlegement: "This product includes software
 * developed by the ObjectStyle Group (http://objectstyle.org/)." Alternately,
 * this acknowlegement may appear in the software itself, if and wherever such
 * third-party acknowlegements normally appear.
 * 
 * 4. The names "ObjectStyle Group" and "Cayenne" must not be used to endorse or
 * promote products derived from this software without prior written permission.
 * For written permission, please contact andrus@objectstyle.org.
 * 
 * 5. Products derived from this software may not be called "ObjectStyle" nor
 * may "ObjectStyle" appear in their names without prior written permission of
 * the ObjectStyle Group.
 * 
 * THIS SOFTWARE IS PROVIDED ``AS IS'' AND ANY EXPRESSED OR IMPLIED WARRANTIES,
 * INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND
 * FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
 * OBJECTSTYLE GROUP OR ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA,
 * OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
 * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
 * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE,
 * EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 * ====================================================================
 * 
 * This software consists of voluntary contributions made by many individuals on
 * behalf of the ObjectStyle Group. For more information on the ObjectStyle
 * Group, please see <http://objectstyle.org/>.
 *  
 */
package org.objectstyle.wolips.eomodeler.actions;

import java.util.HashSet;
import java.util.Set;

import org.eclipse.core.commands.ExecutionException;
import org.eclipse.core.commands.operations.AbstractOperation;
import org.eclipse.core.commands.operations.IOperationHistory;
import org.eclipse.core.runtime.IAdaptable;
import org.eclipse.core.runtime.IProgressMonitor;
import org.eclipse.core.runtime.IStatus;
import org.eclipse.core.runtime.Status;
import org.eclipse.jface.action.IAction;
import org.eclipse.jface.dialogs.MessageDialog;
import org.eclipse.jface.viewers.IStructuredSelection;
import org.eclipse.swt.widgets.Display;
import org.eclipse.ui.PlatformUI;
import org.objectstyle.wolips.baseforuiplugins.utils.ErrorUtils;
import org.objectstyle.wolips.eomodeler.core.model.EOModelException;
import org.objectstyle.wolips.eomodeler.core.model.EOModelObject;
import org.objectstyle.wolips.eomodeler.core.model.EOModelVerificationFailure;
import org.objectstyle.wolips.eomodeler.core.utils.EOModelUtils;

public abstract class AbstractNewObjectAction<T extends EOModelObject, U extends EOModelObject> extends EMAction {
	private Class<T> _parentType;

	private String _label;

	public AbstractNewObjectAction(Class<T> parentType, String label) {
		_label = label;
		_parentType = parentType;
	}

	public String getLabel() {
		return _label;
	}

	public void run(IAction action) {
		try {
			T selectedObject = null;
			IStructuredSelection selection = getSelection();
			if (selection != null) {
				Object firstElement = selection.getFirstElement();
				if (firstElement instanceof EOModelObject) {
					selectedObject = (T) EOModelUtils.getRelated(_parentType, (EOModelObject) firstElement);
				}
			}
			if (selectedObject != null) {
				NewOperation operation = new NewOperation(selectedObject);
				operation.addContext(EOModelUtils.getUndoContext(selectedObject));
				IOperationHistory operationHistory = PlatformUI.getWorkbench().getOperationSupport().getOperationHistory();
				operationHistory.execute(operation, null, null);
			} else {
				MessageDialog.openError(getWindow().getShell(), getNoSelectionTitle(), getNoSelectionMessage());
			}
		} catch (Throwable e) {
			ErrorUtils.openErrorDialog(Display.getDefault().getActiveShell(), e);
		}
	}

	protected abstract String getNoSelectionTitle();

	protected abstract String getNoSelectionMessage();

	protected abstract U createChild(T parent, Set<EOModelVerificationFailure> failures) throws EOModelException;

	protected class NewOperation extends AbstractOperation {
		private T _parent;

		private U _child;

		public NewOperation(T parent) {
			super(AbstractNewObjectAction.this.getLabel());
			_parent = parent;
		}

		@Override
		public IStatus execute(IProgressMonitor monitor, IAdaptable info) throws ExecutionException {
			try {
				Set<EOModelVerificationFailure> failures = new HashSet<EOModelVerificationFailure>();
				_child = AbstractNewObjectAction.this.createChild(_parent, failures);
				//System.out.println("NewOperation.execute: Added " + _child + " to " + _parent);
				return Status.OK_STATUS;
			} catch (EOModelException e) {
				throw new ExecutionException("Failed to add new object.", e);
			}
		}

		@Override
		public IStatus redo(IProgressMonitor monitor, IAdaptable info) throws ExecutionException {
			return execute(monitor, info);
		}

		@Override
		public IStatus undo(IProgressMonitor monitor, IAdaptable info) throws ExecutionException {
			try {
				//System.out.println("NewOperation.undo: undo adding " + _child + " to " + _parent);
				Set<EOModelVerificationFailure> failures = new HashSet<EOModelVerificationFailure>();
				_child._removeFromModelParent(failures);
				return Status.OK_STATUS;
			} catch (EOModelException e) {
				throw new ExecutionException("Failed to remove object.", e);
			}
		}
	}
}