org.eclipse.ogee.designer/src/org/eclipse/ogee/designer/features/ODataAddEnumPropertyFeature.java - gerrit/ogee/org.eclipse.ogee - Git at Google

 /*******************************************************************************
  *  Copyright (c) 2012-2014 SAP SE.
  *  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:
  *  SAP SE - initial API and implementation and/or initial documentation
  *
  *******************************************************************************/
 package org.eclipse.ogee.designer.features;

 import java.util.ArrayList;
 import java.util.Iterator;
 import java.util.List;
 import java.util.StringTokenizer;

 import org.eclipse.emf.common.util.EList;
 import org.eclipse.graphiti.features.IFeatureProvider;
 import org.eclipse.graphiti.features.context.IContext;
 import org.eclipse.graphiti.features.context.ICustomContext;
 import org.eclipse.graphiti.features.context.impl.AddContext;
 import org.eclipse.graphiti.features.custom.AbstractCustomFeature;
 import org.eclipse.graphiti.mm.pictograms.ContainerShape;
 import org.eclipse.graphiti.mm.pictograms.PictogramElement;
 import org.eclipse.graphiti.mm.pictograms.Shape;
 import org.eclipse.ogee.designer.factories.ArtifactFactory;
 import org.eclipse.ogee.designer.factories.ShapesFactory;
 import org.eclipse.ogee.designer.messages.Messages;
 import org.eclipse.ogee.designer.providers.ODataImageProvider;
 import org.eclipse.ogee.designer.shapes.IdentifierShape;
 import org.eclipse.ogee.designer.utils.ODataLayoutUtil;
 import org.eclipse.ogee.designer.utils.ODataShapeUtil;
 import org.eclipse.ogee.designer.utils.PropertyUtil;
 import org.eclipse.ogee.model.odata.ComplexType;
 import org.eclipse.ogee.model.odata.EntityType;
 import org.eclipse.ogee.model.odata.Property;

 /**
  * Custom Feature Provider to add enum types to an Entity type or a complex
  * type.
  *
  */
 public class ODataAddEnumPropertyFeature extends AbstractCustomFeature {

 	/**
 	 * artifact factory
 	 */
 	private ArtifactFactory artifactFactory;
 	/**
 	 * shapes factory
 	 */
 	private ShapesFactory shapesFactory;

 	/**
 	 * Constructor
 	 *
 	 * @param fp
 	 *            Feature provider instance is passed
 	 * @param artifactFactory
 	 *            ArtifactFactory instance is passed
 	 * @param shapesFactory
 	 *            ShapesFactory instance is passed
 	 */
 	public ODataAddEnumPropertyFeature(IFeatureProvider fp,
 			ArtifactFactory artifactFactory, ShapesFactory shapesFactory) {
 		super(fp);
 		this.artifactFactory = artifactFactory;
 		this.shapesFactory = shapesFactory;
 	}

 	/**
 	 * Returns if an enum can be added(The feature can be executed). In this
 	 * case this has to be done only in case of Entity Types or Complex Types.
 	 *
 	 * @param context
 	 *            ICustomContext instance is passed
 	 */
 	@Override
 	public boolean canExecute(ICustomContext context) {
 		boolean ret = false;
 		PictogramElement[] pes = context.getPictogramElements();
 		if (pes != null && pes.length == 1) {
 			Object bo = getBusinessObjectForPictogramElement(pes[0]);
 			if (bo instanceof EntityType || bo instanceof ComplexType) {
 				ret = true;
 			}
 		}
 		return ret;
 	}

 	@Override
 	public String getName() {
 		return Messages.ODATAEDITOR_ENUMPROPERTY_NAME;
 	}

 	@Override
 	public String getDescription() {
 		return Messages.ODATAEDITOR_ENUMPROPERTY_DESC;
 	}

 	@Override
 	public String getImageId() {
 		return ODataImageProvider.IMG_ENUM;
 	}

 	@Override
 	public boolean isAvailable(IContext context) {
 		return true;
 	}

 	/**
 	 * Check for user input and create the new enum type
 	 *
 	 * @param context
 	 *            ICustomContext instance
 	 * @see org.eclipse.graphiti.features.custom.ICustomFeature#execute(org.eclipse.graphiti.features.context.ICustomContext)
 	 */
 	@Override
 	public void execute(ICustomContext context) {
 		String enumTypeName;
 		AddContext addContext;
 		Property enumType;
 		PictogramElement[] pes = context.getPictogramElements();
 		if (pes != null && pes.length == 1) {
 			Object bo = getBusinessObjectForPictogramElement(pes[0]);
 			if (bo instanceof EntityType) {
 				PictogramElement mainPictogramElement = pes[0];
 				EntityType etype = (EntityType) bo;
 				List<Property> enumList = new ArrayList<Property>();
 				enumList.addAll(etype.getProperties());
 				enumList.addAll(etype.getKeys());
 				enumTypeName = getNameForProperty(enumList);

 				if (enumTypeName != null && enumTypeName.trim().length() != 0) {
 					EList<Shape> children = ((ContainerShape) mainPictogramElement)
 							.getChildren();
 					IdentifierShape identifierShape = null;
 					ContainerShape sectionShape;
 					Shape titleImage;
 					for (Shape shape : children) {
 						/*
 						 * Create new property and its shape in the specified
 						 * area
 						 */
 						if (PropertyUtil.PROPERTY_RECTANGLE.equals(PropertyUtil
 								.getID(shape))) {
 							sectionShape = (ContainerShape) shape;
 							enumType = this.artifactFactory
 									.createEnumTypeProperty(etype, enumTypeName);
 							addContext = new AddContext();
 							addContext.setNewObject(enumType);

 							/* Visual shape */
 							identifierShape = this.shapesFactory
 									.createPropertyShape(sectionShape,
 											enumTypeName,
 											ODataImageProvider.IMG_ENUM,
 											addContext, enumType,
 											enumType.getType(), false,
 											PropertyUtil.ENTITY_PROPERTY_NAME);

 							titleImage = ODataLayoutUtil
 									.expandSection(sectionShape);
 							if (titleImage != null) {
 								ODataShapeUtil.updatePictogramElement(
 										titleImage, getFeatureProvider());

 							}
 							/* Update the layout */
 							ODataShapeUtil.layoutPictogramElement(
 									mainPictogramElement, getFeatureProvider());

 							identifierShape.updateSelection();
 							identifierShape.enableDirectEdit();
 							break;
 						}
 					}
 				}

 			} else if (bo instanceof ComplexType) {
 				PictogramElement mainPictogramElement = pes[0];
 				ComplexType ctype = (ComplexType) bo;

 				List<Property> propertyList = new ArrayList<Property>();
 				propertyList.addAll(ctype.getProperties());
 				enumTypeName = getNameForProperty(propertyList);

 				if (enumTypeName != null && enumTypeName.trim().length() != 0) {
 					EList<Shape> children = ((ContainerShape) mainPictogramElement)
 							.getChildren();
 					IdentifierShape identifierShape = null;
 					ContainerShape sectionShape;
 					Shape titleImage;
 					for (Shape shape : children) {
 						/*
 						 * Create new property and its shape in the specified
 						 * area
 						 */
 						if (PropertyUtil.PROPERTY_RECTANGLE.equals(PropertyUtil
 								.getID(shape))) {
 							sectionShape = (ContainerShape) shape;
 							enumType = this.artifactFactory
 									.createEnumTypeProperty(ctype, enumTypeName);
 							addContext = new AddContext();
 							addContext.setNewObject(enumType);

 							/* Visual shape */
 							identifierShape = this.shapesFactory
 									.createPropertyShape(sectionShape,
 											enumTypeName,
 											ODataImageProvider.IMG_ENUM,
 											addContext, enumType,
 											enumType.getType(), false,
 											PropertyUtil.ENTITY_PROPERTY_NAME);
 							titleImage = ODataLayoutUtil
 									.expandSection(sectionShape);
 							if (titleImage != null) {
 								ODataShapeUtil.updatePictogramElement(
 										titleImage, getFeatureProvider());

 							}
 							/* Update the layout */
 							ODataShapeUtil.layoutPictogramElement(
 									mainPictogramElement, getFeatureProvider());

 							identifierShape.updateSelection();
 							identifierShape.enableDirectEdit();
 							break;
 						}
 					}
 				}

 			}
 		}
 	}

 	/**
 	 * Creates a unique name for the new Entity
 	 *
 	 * @param properties
 	 * @return String new unique name
 	 */
 	private String getNameForProperty(List<Property> properties) {
 		int uniqueEnumTypeID = 1;
 		String propertyName;
 		StringBuffer newName = new StringBuffer(
 				Messages.ODATAEDITOR_PROPERTIES_DIALOG_PROPERTY);
 		List<Integer> numberList = new ArrayList<Integer>();
 		Property property;
 		StringTokenizer st;
 		String propertyNumber;
 		int index;
 		for (Iterator<Property> iterator = properties.iterator(); iterator
 				.hasNext();) {
 			property = iterator.next();
 			propertyName = property.getName();
 			if (propertyName
 					.startsWith(Messages.ODATAEDITOR_PROPERTIES_DIALOG_PROPERTY)) {
 				st = new StringTokenizer(propertyName,
 						Messages.ODATAEDITOR_PROPERTIES_DIALOG_PROPERTY);
 				if (propertyName
 						.equalsIgnoreCase(Messages.ODATAEDITOR_PROPERTIES_DIALOG_PROPERTY)) {
 					numberList.add(Integer.valueOf(1));
 				}
 				while (st.hasMoreTokens()) {
 					propertyNumber = st.nextToken();
 					if (propertyNumber != null
 							&& propertyNumber.trim().length() > 0) {
 						try {
 							index = Integer.parseInt(propertyNumber);
 							numberList.add(Integer.valueOf(index));
 						} catch (NumberFormatException e) {
 							// Digest...don't log
 						}
 					}
 				}
 			}
 		}
 		for (Iterator<Integer> iterator = numberList.iterator(); iterator
 				.hasNext();) {
 			if (!numberList.contains(Integer.valueOf(uniqueEnumTypeID))) {
 				if (uniqueEnumTypeID != 1) {
 					newName.append(uniqueEnumTypeID);
 				}
 				break;
 			}
 			uniqueEnumTypeID++;
 		}
 		return newName.toString();
 	}
 }
	/*******************************************************************************
	* Copyright (c) 2012-2014 SAP SE.
	* 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:
	* SAP SE - initial API and implementation and/or initial documentation
	*
	*******************************************************************************/
	package org.eclipse.ogee.designer.features;

	import java.util.ArrayList;
	import java.util.Iterator;
	import java.util.List;
	import java.util.StringTokenizer;

	import org.eclipse.emf.common.util.EList;
	import org.eclipse.graphiti.features.IFeatureProvider;
	import org.eclipse.graphiti.features.context.IContext;
	import org.eclipse.graphiti.features.context.ICustomContext;
	import org.eclipse.graphiti.features.context.impl.AddContext;
	import org.eclipse.graphiti.features.custom.AbstractCustomFeature;
	import org.eclipse.graphiti.mm.pictograms.ContainerShape;
	import org.eclipse.graphiti.mm.pictograms.PictogramElement;
	import org.eclipse.graphiti.mm.pictograms.Shape;
	import org.eclipse.ogee.designer.factories.ArtifactFactory;
	import org.eclipse.ogee.designer.factories.ShapesFactory;
	import org.eclipse.ogee.designer.messages.Messages;
	import org.eclipse.ogee.designer.providers.ODataImageProvider;
	import org.eclipse.ogee.designer.shapes.IdentifierShape;
	import org.eclipse.ogee.designer.utils.ODataLayoutUtil;
	import org.eclipse.ogee.designer.utils.ODataShapeUtil;
	import org.eclipse.ogee.designer.utils.PropertyUtil;
	import org.eclipse.ogee.model.odata.ComplexType;
	import org.eclipse.ogee.model.odata.EntityType;
	import org.eclipse.ogee.model.odata.Property;

	/**
	* Custom Feature Provider to add enum types to an Entity type or a complex
	* type.
	*
	*/
	public class ODataAddEnumPropertyFeature extends AbstractCustomFeature {

	/**
	* artifact factory
	*/
	private ArtifactFactory artifactFactory;
	/**
	* shapes factory
	*/
	private ShapesFactory shapesFactory;

	/**
	* Constructor
	*
	* @param fp
	* Feature provider instance is passed
	* @param artifactFactory
	* ArtifactFactory instance is passed
	* @param shapesFactory
	* ShapesFactory instance is passed
	*/
	public ODataAddEnumPropertyFeature(IFeatureProvider fp,
	ArtifactFactory artifactFactory, ShapesFactory shapesFactory) {
	super(fp);
	this.artifactFactory = artifactFactory;
	this.shapesFactory = shapesFactory;
	}

	/**
	* Returns if an enum can be added(The feature can be executed). In this
	* case this has to be done only in case of Entity Types or Complex Types.
	*
	* @param context
	* ICustomContext instance is passed
	*/
	@Override
	public boolean canExecute(ICustomContext context) {
	boolean ret = false;
	PictogramElement[] pes = context.getPictogramElements();
	if (pes != null && pes.length == 1) {
	Object bo = getBusinessObjectForPictogramElement(pes[0]);
	if (bo instanceof EntityType \|\| bo instanceof ComplexType) {
	ret = true;
	}
	}
	return ret;
	}

	@Override
	public String getName() {
	return Messages.ODATAEDITOR_ENUMPROPERTY_NAME;
	}

	@Override
	public String getDescription() {
	return Messages.ODATAEDITOR_ENUMPROPERTY_DESC;
	}

	@Override
	public String getImageId() {
	return ODataImageProvider.IMG_ENUM;
	}

	@Override
	public boolean isAvailable(IContext context) {
	return true;
	}

	/**
	* Check for user input and create the new enum type
	*
	* @param context
	* ICustomContext instance
	* @see org.eclipse.graphiti.features.custom.ICustomFeature#execute(org.eclipse.graphiti.features.context.ICustomContext)
	*/
	@Override
	public void execute(ICustomContext context) {
	String enumTypeName;
	AddContext addContext;
	Property enumType;
	PictogramElement[] pes = context.getPictogramElements();
	if (pes != null && pes.length == 1) {
	Object bo = getBusinessObjectForPictogramElement(pes[0]);
	if (bo instanceof EntityType) {
	PictogramElement mainPictogramElement = pes[0];
	EntityType etype = (EntityType) bo;
	List<Property> enumList = new ArrayList<Property>();
	enumList.addAll(etype.getProperties());
	enumList.addAll(etype.getKeys());
	enumTypeName = getNameForProperty(enumList);

	if (enumTypeName != null && enumTypeName.trim().length() != 0) {
	EList<Shape> children = ((ContainerShape) mainPictogramElement)
	.getChildren();
	IdentifierShape identifierShape = null;
	ContainerShape sectionShape;
	Shape titleImage;
	for (Shape shape : children) {
	/*
	* Create new property and its shape in the specified
	* area
	*/
	if (PropertyUtil.PROPERTY_RECTANGLE.equals(PropertyUtil
	.getID(shape))) {
	sectionShape = (ContainerShape) shape;
	enumType = this.artifactFactory
	.createEnumTypeProperty(etype, enumTypeName);
	addContext = new AddContext();
	addContext.setNewObject(enumType);

	/* Visual shape */
	identifierShape = this.shapesFactory
	.createPropertyShape(sectionShape,
	enumTypeName,
	ODataImageProvider.IMG_ENUM,
	addContext, enumType,
	enumType.getType(), false,
	PropertyUtil.ENTITY_PROPERTY_NAME);

	titleImage = ODataLayoutUtil
	.expandSection(sectionShape);
	if (titleImage != null) {
	ODataShapeUtil.updatePictogramElement(
	titleImage, getFeatureProvider());

	}
	/* Update the layout */
	ODataShapeUtil.layoutPictogramElement(
	mainPictogramElement, getFeatureProvider());

	identifierShape.updateSelection();
	identifierShape.enableDirectEdit();
	break;
	}
	}
	}

	} else if (bo instanceof ComplexType) {
	PictogramElement mainPictogramElement = pes[0];
	ComplexType ctype = (ComplexType) bo;

	List<Property> propertyList = new ArrayList<Property>();
	propertyList.addAll(ctype.getProperties());
	enumTypeName = getNameForProperty(propertyList);

	if (enumTypeName != null && enumTypeName.trim().length() != 0) {
	EList<Shape> children = ((ContainerShape) mainPictogramElement)
	.getChildren();
	IdentifierShape identifierShape = null;
	ContainerShape sectionShape;
	Shape titleImage;
	for (Shape shape : children) {
	/*
	* Create new property and its shape in the specified
	* area
	*/
	if (PropertyUtil.PROPERTY_RECTANGLE.equals(PropertyUtil
	.getID(shape))) {
	sectionShape = (ContainerShape) shape;
	enumType = this.artifactFactory
	.createEnumTypeProperty(ctype, enumTypeName);
	addContext = new AddContext();
	addContext.setNewObject(enumType);

	/* Visual shape */
	identifierShape = this.shapesFactory
	.createPropertyShape(sectionShape,
	enumTypeName,
	ODataImageProvider.IMG_ENUM,
	addContext, enumType,
	enumType.getType(), false,
	PropertyUtil.ENTITY_PROPERTY_NAME);
	titleImage = ODataLayoutUtil
	.expandSection(sectionShape);
	if (titleImage != null) {
	ODataShapeUtil.updatePictogramElement(
	titleImage, getFeatureProvider());

	}
	/* Update the layout */
	ODataShapeUtil.layoutPictogramElement(
	mainPictogramElement, getFeatureProvider());

	identifierShape.updateSelection();
	identifierShape.enableDirectEdit();
	break;
	}
	}
	}

	}
	}
	}

	/**
	* Creates a unique name for the new Entity
	*
	* @param properties
	* @return String new unique name
	*/
	private String getNameForProperty(List<Property> properties) {
	int uniqueEnumTypeID = 1;
	String propertyName;
	StringBuffer newName = new StringBuffer(
	Messages.ODATAEDITOR_PROPERTIES_DIALOG_PROPERTY);
	List<Integer> numberList = new ArrayList<Integer>();
	Property property;
	StringTokenizer st;
	String propertyNumber;
	int index;
	for (Iterator<Property> iterator = properties.iterator(); iterator
	.hasNext();) {
	property = iterator.next();
	propertyName = property.getName();
	if (propertyName
	.startsWith(Messages.ODATAEDITOR_PROPERTIES_DIALOG_PROPERTY)) {
	st = new StringTokenizer(propertyName,
	Messages.ODATAEDITOR_PROPERTIES_DIALOG_PROPERTY);
	if (propertyName
	.equalsIgnoreCase(Messages.ODATAEDITOR_PROPERTIES_DIALOG_PROPERTY)) {
	numberList.add(Integer.valueOf(1));
	}
	while (st.hasMoreTokens()) {
	propertyNumber = st.nextToken();
	if (propertyNumber != null
	&& propertyNumber.trim().length() > 0) {
	try {
	index = Integer.parseInt(propertyNumber);
	numberList.add(Integer.valueOf(index));
	} catch (NumberFormatException e) {
	// Digest...don't log
	}
	}
	}
	}
	}
	for (Iterator<Integer> iterator = numberList.iterator(); iterator
	.hasNext();) {
	if (!numberList.contains(Integer.valueOf(uniqueEnumTypeID))) {
	if (uniqueEnumTypeID != 1) {
	newName.append(uniqueEnumTypeID);
	}
	break;
	}
	uniqueEnumTypeID++;
	}
	return newName.toString();
	}
	}