org.eclipse.ogee.designer/src/org/eclipse/ogee/designer/features/ODataCreateEntityTypeFeature.java

/*******************************************************************************
 *  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 org.eclipse.graphiti.features.IFeatureProvider;
import org.eclipse.graphiti.features.context.IContext;
import org.eclipse.graphiti.features.context.ICreateContext;
import org.eclipse.graphiti.features.impl.AbstractCreateFeature;
import org.eclipse.graphiti.mm.pictograms.Diagram;
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.utils.ODataShapeUtil;

public class ODataCreateEntityTypeFeature extends AbstractCreateFeature {
	/**
	 * ArifactFactory instance used to create all the Business objects.
	 */
	private ArtifactFactory artifactFactory;

	public ODataCreateEntityTypeFeature(IFeatureProvider fp,
			ShapesFactory shapesFactory, ArtifactFactory artifactFactory) {
		super(fp, Messages.ODATAEDITOR_ENTITYTYPE_NAME,
				Messages.ODATAEDITOR_ENTITYTYPE_DESC);
		this.artifactFactory = artifactFactory;
	}

	/**
	 * Returns true if the target is Diagram only as the Entity can only be
	 * created on a Diagram
	 * 
	 * @param context
	 *            ICreateContext instance
	 * @return boolean true if Entity can be created
	 */
	@Override
	public boolean canCreate(ICreateContext context) {

		return context.getTargetContainer() instanceof Diagram;
	}

	@Override
	public Object[] create(ICreateContext context) {

		// Entity Type creation
		Object entityType = this.artifactFactory.createDefaultEntityType();

		// Default EntitySet creation
		this.artifactFactory.createDefaultEntitySet(entityType);

		// Default Property ID creation
		this.artifactFactory.createIDProperty(entityType);

		ODataShapeUtil.addGraphicalRepresentation(context, entityType,
				getFeatureProvider());
		return new Object[] { entityType };
	}

	/**
	 * Returns Entity Type Display Name.
	 * 
	 * @return String the Name of the Entity
	 */
	@Override
	public String getCreateName() {
		return Messages.ODATAEDITOR_ENTITYTYPE_NAME;
	}

	/**
	 * Returns Entity Type Description.
	 * 
	 * @return String Entity Type Description
	 */
	@Override
	public String getCreateDescription() {
		return Messages.ODATAEDITOR_ENTITYTYPE_DESC;
	}

	/**
	 * Returns Image ID.
	 * 
	 * @return String Image ID
	 */
	@Override
	public String getCreateImageId() {
		return ODataImageProvider.IMG_ENTITY;
	}

	@Override
	public boolean canUndo(IContext context) {
		// TODO Auto-generated method stub
		return false;
	}

	@Override
	public boolean hasDoneChanges() {
		// TODO Auto-generated method stub
		return false;
	}
}