/*
 * Copyright 2013-2016 consulo.io
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package consulo.diagram.builder.impl;

import com.intellij.util.containers.HashMap;
import com.mxgraph.model.mxCell;
import com.mxgraph.swing.mxGraphComponent;
import com.mxgraph.util.mxRectangle;
import com.mxgraph.util.mxUtils;
import com.mxgraph.view.mxGraph;
import consulo.diagram.builder.GraphBuilder;
import consulo.diagram.builder.GraphNode;
import consulo.diagram.builder.GraphPositionStrategy;
import org.jetbrains.annotations.NotNull;

import javax.swing.*;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.Map;

/**
 * @author VISTALL
 * @since 22:38/15.10.13
 */
public class GraphBuilderImpl implements GraphBuilder {
  private List<GraphNode<?>> myGraphNodes = new ArrayList<GraphNode<?>>();

  public GraphBuilderImpl() {

  }

  @NotNull
  @Override
  public JComponent getComponent() {
    mxGraph graph = new mxGraph();
    graph.setCellsResizable(false);
    graph.setEdgeLabelsMovable(false);
    graph.setConnectableEdges(false);
    graph.setCellsEditable(false);
    graph.setAllowDanglingEdges(false);

    Object parent = graph.getDefaultParent();

    graph.getModel().beginUpdate();
    try
    {
      Map<GraphNode<?>, Object> map = new HashMap<GraphNode<?>, Object>();

      int i = 1;
      for (GraphNode<?> graphNode : myGraphNodes) {
        mxRectangle labelSize = mxUtils.getLabelSize(graphNode.getValue().toString(), Collections.<String, Object> emptyMap(), false, 1.);
        mxCell vertex = graph.insertVertex(parent, null, graphNode.getValue(), 0, 0, labelSize.getWidth() + 5, labelSize.getHeight() + 6);

        map.put(graphNode, vertex);
        i++;
      }

      for (Map.Entry<GraphNode<?>, Object> entry : map.entrySet()) {
        List<GraphNode<?>> arrowNodes = entry.getKey().getArrowNodes();

        for (GraphNode<?> arrowNode : arrowNodes) {
          graph.insertEdge(parent, null, null, entry.getValue(), map.get(arrowNode));
        }
      }
    }
    finally
    {
      graph.getModel().endUpdate();
    }

    mxGraphComponent mxGraphComponent = new mxGraphComponent(graph){

    };

    return mxGraphComponent;
  }

  @NotNull
  @Override
  public <E> GraphNode<E> createNode(E value, GraphPositionStrategy strategy) {
    GraphNodeImpl<E> graphNode = new GraphNodeImpl<E>(value, strategy);
    myGraphNodes.add(graphNode);
    return graphNode;
  }
}