import mxgraph from '@mxgraph/core';

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

export default {
  title: 'Styles/Stylesheet',
  argTypes: {
    ...globalTypes
  }
};

const Template = ({ label, ...args }) => {
  const {
    mxGraph,
    mxPerimeter,
    mxConstants,
    mxEdgeStyle
  } = 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';

  // Creates the graph inside the DOM node.
  const graph = new mxGraph(container);

  // Disables basic selection and cell handling
  graph.setEnabled(false);

  // Returns a special label for edges. Note: This does
  // a supercall to use the default implementation.
  graph.getLabel = function(cell) {
    const label = mxGraph.prototype.getLabel.apply(this, arguments);

    if (cell.isEdge()) {
      return `Transfer ${label}`;
    }
    return label;
  };

  // Installs a custom global tooltip
  graph.setTooltips(true);
  graph.getTooltip = function(state) {
    const { cell } = state;
    const model = this.getModel();

    if (modcellel.isEdge()) {
      const source = this.getLabel(cell.getTerminal(true));
      const target = this.getLabel(cell.getTerminal(false));

      return `${source} -> ${target}`;
    }
    return this.getLabel(cell);
  };

  // Creates the default style for vertices
  let style = [];
  style.shape = mxConstants.SHAPE_RECTANGLE;
  style.perimiter = mxPerimeter.RectanglePerimeter;
  style.strokeColor = 'gray';
  style.rounded = true;
  style.fillColor = '#EEEEEE';
  style.gradientColor = 'white';
  style.fontColor = '#774400';
  style.align = mxConstants.ALIGN_CENTER;
  style.verticalAlign = mxConstants.ALIGN_MIDDLE;
  style.fontSize = '12';
  style.fontStyle = 1;
  graph.getStylesheet().putDefaultVertexStyle(style);

  // Creates the default style for edges
  style = [];
  style.shape = mxConstants.SHAPE_CONNECTOR;
  style.strokeColor = '#6482B9';
  style.align = mxConstants.ALIGN_CENTER;
  style.verticalAlign = mxConstants.ALIGN_MIDDLE;
  style.edge = mxEdgeStyle.ElbowConnector;
  style.endArrow = mxConstants.ARROW_CLASSIC;
  style.fontSize = '10';
  graph.getStylesheet().putDefaultEdgeStyle(style);

  // 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 v1 = graph.insertVertex(
      parent,
      null,
      'Interval 1',
      20,
      20,
      180,
      30
    );
    const v2 = graph.insertVertex(
      parent,
      null,
      'Interval 2',
      140,
      80,
      280,
      30
    );
    const v3 = graph.insertVertex(
      parent,
      null,
      'Interval 3',
      200,
      140,
      360,
      30
    );
    const v4 = graph.insertVertex(
      parent,
      null,
      'Interval 4',
      480,
      200,
      120,
      30
    );
    const v5 = graph.insertVertex(
      parent,
      null,
      'Interval 5',
      60,
      260,
      400,
      30
    );
    const e1 = graph.insertEdge(parent, null, '1', v1, v2);
    e1.getGeometry().points = [{ x: 160, y: 60 }];
    const e2 = graph.insertEdge(parent, null, '2', v1, v5);
    e2.getGeometry().points = [{ x: 80, y: 60 }];
    const e3 = graph.insertEdge(parent, null, '3', v2, v3);
    e3.getGeometry().points = [{ x: 280, y: 120 }];
    const e4 = graph.insertEdge(parent, null, '4', v3, v4);
    e4.getGeometry().points = [{ x: 500, y: 180 }];
    const e5 = graph.insertEdge(parent, null, '5', v3, v5);
    e5.getGeometry().points = [{ x: 380, y: 180 }];
  } finally {
    // Updates the display
    graph.getModel().endUpdate();
  }

  return container;
}

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