import mxgraph from '@mxgraph/core';

import { globalTypes } from '../.storybook/preview';

export default {
  title: 'Layouts/Folding',
  argTypes: {
    ...globalTypes
  }
};

const Template = ({ label, ...args }) => {
  const {
    mxGraph,
    mxConstants,
    mxEdgeStyle,
    mxStackLayout,
    mxLayoutManager
  } = mxgraph;

  const container = document.createElement('div');
  container.style.position = 'relative';
  container.style.overflow = 'hidden';
  container.style.width = `${args.width}px`;
  container.style.height = `${args.height}px`;
  container.style.background = 'url(/images/grid.gif)';
  container.style.cursor = 'default';

  // Enables crisp rendering of rectangles in SVG
  mxConstants.ENTITY_SEGMENT = 20;

  // Creates the graph inside the given container
  const graph = new mxGraph(container);
  graph.setDropEnabled(true);

  // Disables global features
  graph.collapseToPreferredSize = false;
  graph.constrainChildren = false;
  graph.cellsSelectable = false;
  graph.extendParentsOnAdd = false;
  graph.extendParents = false;
  graph.border = 10;

  // Sets global styles
  let style = graph.getStylesheet().getDefaultEdgeStyle();
  style.edge = mxEdgeStyle.EntityRelation;
  style.rounded = true;

  style = graph.getStylesheet().getDefaultVertexStyle();
  style.fillColor = '#ffffff';
  style.shape = 'swimlane';
  style.startSize = 30;

  style = [];
  style.shape = mxConstants.SHAPE_RECTANGLE;
  style.strokeColor = 'none';
  style.fillColor = 'none';
  style.foldable = false;
  graph.getStylesheet().putCellStyle('column', style);

  // Installs auto layout for all levels
  const layout = new mxStackLayout(graph, true);
  layout.border = graph.border;
  const layoutMgr = new mxLayoutManager(graph);
  layoutMgr.getLayout = function(cell) {
    if (!cell.collapsed) {
      if (cell.parent !== graph.model.root) {
        layout.resizeParent = true;
        layout.horizontal = false;
        layout.spacing = 10;
      } else {
        layout.resizeParent = true;
        layout.horizontal = true;
        layout.spacing = 40;
      }

      return layout;
    }

    return null;
  };

  // Resizes the container
  graph.setResizeContainer(true);

  // Gets the default parent for inserting new cells. This
  // is normally the first child of the root (ie. layer 0).
  const parent = graph.getDefaultParent();

  // Adds cells to the model in a single step
  graph.getModel().beginUpdate();
  try {
    const col1 = graph.insertVertex(parent, null, '', 0, 0, 120, 0, 'column');

    const v1 = graph.insertVertex(col1, null, '1', 0, 0, 100, 30);
    v1.collapsed = true;

    const v11 = graph.insertVertex(v1, null, '1.1', 0, 0, 80, 30);
    v11.collapsed = true;

    const v111 = graph.insertVertex(v11, null, '1.1.1', 0, 0, 60, 30);
    const v112 = graph.insertVertex(v11, null, '1.1.2', 0, 0, 60, 30);

    const v12 = graph.insertVertex(v1, null, '1.2', 0, 0, 80, 30);

    const col2 = graph.insertVertex(parent, null, '', 0, 0, 120, 0, 'column');

    const v2 = graph.insertVertex(col2, null, '2', 0, 0, 100, 30);
    v2.collapsed = true;

    const v21 = graph.insertVertex(v2, null, '2.1', 0, 0, 80, 30);
    v21.collapsed = true;

    const v211 = graph.insertVertex(v21, null, '2.1.1', 0, 0, 60, 30);
    const v212 = graph.insertVertex(v21, null, '2.1.2', 0, 0, 60, 30);

    const v22 = graph.insertVertex(v2, null, '2.2', 0, 0, 80, 30);

    const v3 = graph.insertVertex(col2, null, '3', 0, 0, 100, 30);
    v3.collapsed = true;

    const v31 = graph.insertVertex(v3, null, '3.1', 0, 0, 80, 30);
    v31.collapsed = true;

    const v311 = graph.insertVertex(v31, null, '3.1.1', 0, 0, 60, 30);
    const v312 = graph.insertVertex(v31, null, '3.1.2', 0, 0, 60, 30);

    const v32 = graph.insertVertex(v3, null, '3.2', 0, 0, 80, 30);

    graph.insertEdge(parent, null, '', v111, v211);
    graph.insertEdge(parent, null, '', v112, v212);
    graph.insertEdge(parent, null, '', v112, v22);

    graph.insertEdge(parent, null, '', v12, v311);
    graph.insertEdge(parent, null, '', v12, v312);
    graph.insertEdge(parent, null, '', v12, v32);
  } finally {
    // Updates the display
    graph.getModel().endUpdate();
  }

  return container;
}

export const Default = Template.bind({});