package org.activiti.designer.features;

import org.activiti.bpmn.model.Pool;
import org.activiti.designer.PluginImage;
import org.activiti.designer.util.platform.OSEnum;
import org.activiti.designer.util.platform.OSUtil;
import org.activiti.designer.util.style.StyleUtil;
import org.eclipse.graphiti.features.IFeatureProvider;
import org.eclipse.graphiti.features.context.IAddContext;
import org.eclipse.graphiti.features.impl.AbstractAddShapeFeature;
import org.eclipse.graphiti.mm.algorithms.Rectangle;
import org.eclipse.graphiti.mm.algorithms.Text;
import org.eclipse.graphiti.mm.algorithms.styles.Font;
import org.eclipse.graphiti.mm.algorithms.styles.Orientation;
import org.eclipse.graphiti.mm.pictograms.ContainerShape;
import org.eclipse.graphiti.mm.pictograms.Diagram;
import org.eclipse.graphiti.mm.pictograms.PictogramElement;
import org.eclipse.graphiti.mm.pictograms.Shape;
import org.eclipse.graphiti.services.Graphiti;
import org.eclipse.graphiti.services.IGaService;
import org.eclipse.graphiti.services.IPeCreateService;

public class AddPoolFeature extends AbstractAddShapeFeature {

  public AddPoolFeature(IFeatureProvider fp) {
    super(fp);
  }

  @Override
  public PictogramElement add(IAddContext context) {

    final Pool addedPool = (Pool) context.getNewObject();
    final ContainerShape parent = context.getTargetContainer();

    // CONTAINER SHAPE WITH ROUNDED RECTANGLE
    final IPeCreateService peCreateService = Graphiti.getPeCreateService();
    final ContainerShape containerShape = peCreateService.createContainerShape(parent, true);

    // check whether the context has a size (e.g. from a create feature)
    // otherwise define a default size for the shape
    final int width = context.getWidth() <= 0 ? 500 : context.getWidth();
    final int height = context.getHeight() <= 0 ? 150 : context.getHeight();

    final IGaService gaService = Graphiti.getGaService();
    {
      // create invisible outer rectangle expanded by
      // the width needed for the anchor
      final Rectangle invisibleRectangle = gaService.createInvisibleRectangle(containerShape);
      gaService.setLocationAndSize(invisibleRectangle, context.getX(), context.getY(), width, height);

      // create and set visible rectangle inside invisible rectangle
      Rectangle rectangle = gaService.createRectangle(invisibleRectangle);
      rectangle.setParentGraphicsAlgorithm(invisibleRectangle);
      rectangle.setStyle(StyleUtil.getStyleForPool(getDiagram()));
      gaService.setLocationAndSize(rectangle, 0, 0, width, height);

      // create link and wire it
      link(containerShape, addedPool);
    }

    // SHAPE WITH TEXT
    {
      // create shape for text
      final Shape shape = peCreateService.createShape(containerShape, false);

      // create and set text graphics algorithm
      final Text text = gaService.createDefaultText(getDiagram(), shape, addedPool.getName());
      text.setStyle(StyleUtil.getStyleForEvent(getDiagram()));
      text.setHorizontalAlignment(Orientation.ALIGNMENT_CENTER);
      text.setVerticalAlignment(Orientation.ALIGNMENT_MIDDLE);
      gaService.setLocationAndSize(text, 0, 0, 20, height);
      text.setAngle(-90);
      Font font = null;
      if (OSUtil.getOperatingSystem() == OSEnum.Mac) {
        font = gaService.manageFont(getDiagram(), text.getFont().getName(), 11, false, true);
      } else {
        font = gaService.manageDefaultFont(getDiagram(), false, true);
      }
      text.setFont(font);

      // create link and wire it
      link(shape, addedPool);
    }

    layoutPictogramElement(containerShape);
    return containerShape;
  }

  @Override
  public boolean canAdd(IAddContext context) {
    if (context.getNewObject() instanceof Pool) {

      if (context.getTargetContainer() instanceof Diagram) {
        return true;
      }
    }
    return false;
  }

  protected String getIcon() {
    return PluginImage.IMG_POOL.getImageKey();
  }

}