jpa_diagram_editor/plugins/org.eclipse.jpt.jpadiagrameditor.ui/src/org/eclipse/jpt/jpadiagrameditor/ui/internal/feature/DiscardAndRemoveAllEntitiesFeature.java

/*******************************************************************************
 * <copyright>
 *
 * Copyright (c) 2005, 2010 SAP AG.
 * 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:
 *    Stefan Dimov - initial API, implementation and documentation
 *
 * </copyright>
 *
 *******************************************************************************/
package org.eclipse.jpt.jpadiagrameditor.ui.internal.feature;

import java.util.Iterator;

import org.eclipse.graphiti.features.IFeatureProvider;
import org.eclipse.graphiti.features.context.ICustomContext;
import org.eclipse.graphiti.features.context.impl.CustomContext;
import org.eclipse.graphiti.mm.pictograms.PictogramElement;
import org.eclipse.graphiti.mm.pictograms.Shape;
import org.eclipse.jface.dialogs.Dialog;
import org.eclipse.jface.dialogs.MessageDialog;
import org.eclipse.jpt.jpadiagrameditor.ui.internal.i18n.JPAEditorMessages;
import org.eclipse.swt.widgets.Display;


public class DiscardAndRemoveAllEntitiesFeature extends RemoveAllEntitiesFeature {

	public DiscardAndRemoveAllEntitiesFeature(IFeatureProvider fp) {
		super(fp);
	}
	
	public String getConfirmationText() {
		return JPAEditorMessages.JPAEditorToolBehaviorProvider_removeAndDiscardAllEntitiesConfirmation;
	}
	
	public void execute(ICustomContext context) {
		MessageDialog dlg = new MessageDialog(Display.getCurrent().getShells()[0], 
												JPAEditorMessages.JPAEditorToolBehaviorProvider_removeAllEntitiesMenu, 
												null, getConfirmationText(), 0, 
												new String[] {JPAEditorMessages.BTN_OK, JPAEditorMessages.BTN_CANCEL}, Dialog.CANCEL);
		if (dlg.open() != Dialog.OK)
			return;		
		Iterator<Shape> it = this.getDiagram().getChildren().iterator();
		while (it.hasNext()) {
			Shape sh = it.next();
			RestoreEntityFeature ft = new RestoreEntityFeature(getFeatureProvider());
			ICustomContext ctx = new CustomContext(new PictogramElement[] { sh });
			ft.execute(ctx);
			allShapes.add(sh);
		}
		super.execute(context);
	}		

}