import maxgraph from '@maxgraph/core';

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

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

const Template = ({ label, ...args }) => {
  const { Graph, mxKeyHandler, utils, mxDomUtils, CloneUtils, EventUtils } = maxgraph;

  const div = document.createElement('div');
  div.innerHTML = `
    Double-click the upper/lower half of the cell to edit different fields
    of the user object.
  `;

  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';
  div.appendChild(container);

  class MyCustomGraph extends Graph {
    getLabel(cell) {
      // Returns a HTML representation of the cell where the
      // upper half is the first value, lower half is second
      // value

      const table = document.createElement('table');
      table.style.height = '100%';
      table.style.width = '100%';

      const body = document.createElement('tbody');
      const tr1 = document.createElement('tr');
      const td1 = document.createElement('td');
      td1.style.textAlign = 'center';
      td1.style.fontSize = '12px';
      td1.style.color = '#774400';
      mxDomUtils.write(td1, cell.value.first);

      const tr2 = document.createElement('tr');
      const td2 = document.createElement('td');
      td2.style.textAlign = 'center';
      td2.style.fontSize = '12px';
      td2.style.color = '#774400';
      mxDomUtils.write(td2, cell.value.second);

      tr1.appendChild(td1);
      tr2.appendChild(td2);
      body.appendChild(tr1);
      body.appendChild(tr2);
      table.appendChild(body);

      return table;
    }

    getEditingValue(cell, evt) {
      // Returns the editing value for the given cell and event
      evt.fieldname = this.__getFieldnameForEvent(cell, evt);
      return cell.value[evt.fieldname] || '';
    }

    __getFieldnameForEvent(cell, evt) {
      // Helper method that returns the fieldname to be used for
      // a mouse event
      if (evt != null) {
        // Finds the relative coordinates inside the cell
        const point = utils.convertPoint(
          this.container,
          EventUtils.getClientX(evt),
          EventUtils.getClientY(evt)
        );
        const state = this.getView().getState(cell);

        if (state != null) {
          point.x -= state.x;
          point.y -= state.y;

          // Returns second if mouse in second half of cell
          if (point.y > state.height / 2) {
            return 'second';
          }
        }
      }
      return 'first';
    }

    labelChanged(cell, newValue, trigger) {
      // Sets the new value for the given cell and trigger
      const name = trigger != null ? trigger.fieldname : null;

      if (name != null) {
        // Clones the user object for correct undo and puts
        // the new value in the correct field.
        const value = CloneUtils.clone(cell.value);
        value[name] = newValue;
        newValue = value;

        super.labelChanged(cell, newValue, trigger);
      }
    }
  }

  // Creates the graph inside the given container
  const graph = new MyCustomGraph(container);
  graph.setHtmlLabels(true);

  // Adds handling of return and escape keystrokes for editing
  const keyHandler = new mxKeyHandler(graph);

  // Sample user objects with 2 fields
  const value = {};
  value.first = 'First value';
  value.second = 'Second value';

  // 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.batchUpdate(() => {
    const v1 = graph.insertVertex({
      parent,
      value,
      position: [100, 60],
      size: [120, 80],
      style: 'overflow=fill;',
    });
  });

  return div;
};

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