package editor.features.create;

import org.eclipse.graphiti.features.IFeatureProvider;
import org.eclipse.graphiti.features.context.ICreateContext;
import org.eclipse.graphiti.features.impl.AbstractCreateFeature;
import org.eclipse.graphiti.mm.pictograms.ContainerShape;

public class CreateElementFeature extends AbstractCreateFeature {
  private ElementType type;

  public CreateElementFeature(IFeatureProvider fp, ElementType type) {
    super(fp, null, null);

    this.type = type;
  }

  @Override
  public Object[] create(ICreateContext context) {
    Element element = doCreate(context);

    addGraphicalRepresentation(context, element);

    return new Object[] { element };
  }

  protected Element doCreate(ICreateContext context) {
    ContainerShape cpe = context.getTargetContainer();
    ElementContainer cbo = (ElementContainer) getBusinessObjectForPictogramElement(cpe);

    Element bo = new Element();
    bo.setType(type);

    cbo.getElements().add(bo);

    return bo;
  }

  @Override
  public String getCreateName() {
    switch (type) {
    case TYPE1:
      return "Type 1";
    case TYPE2:
      return "Type 2";
    default:
      return super.getCreateImageId();
    }
  }

  @Override
  public String getCreateDescription() {
    switch (type) {
    case TYPE1:
      return "Create Type 1";
    case TYPE2:
      return "Create Type 2";
    default:
      return super.getCreateImageId();
    }
  }

  @Override
  public String getCreateImageId() {
    switch (type) {
    case TYPE1:
      return "type1";
    case TYPE2:
      return "type2";
    default:
      return super.getCreateImageId();
    }
  }

  @Override
  public boolean canCreate(ICreateContext context) {
    ContainerShape pe = context.getTargetContainer();
    Object bo = getBusinessObjectForPictogramElement(pe);

    return bo instanceof ElementContainer;
  }
}