import React, { PureComponent } from 'react';

import Graph, { isEdge } from '../src';
// import Graph from '../dist/ReactGraph';

const SpecialNode = ({ data, onChange, styles }) => (
  <div
    style={{ background: '#FFCC00', padding: 10, borderRadius: 30, ...styles }}
  >
    <div>I am <strong>special</strong>!<br />{data.label}</div>
    <select onChange={(e) => onChange(e.target.value, data)}>
      <option value="1">1</option>
      <option value="2">2</option>
      <option value="3">3</option>
    </select>
  </div>
);

class App extends PureComponent {
  constructor() {
    super();

    const onChange = (id, d) => {
      this.setState(prevState => (
        {elements: prevState.elements.map(e => {
          if (isEdge(e)) {
            return e;
          }

          return {
            ...e,
            data: {
              ...e.data,
              label: '6' === e.id ? `option ${id} selected` : e.data.label
            }
          };
        })}
      ));
    }

    this.state = {
      elements: [
        { id: '1', type: 'input', data: { label: 'Tests' }, position: { x: 50, y: 50 } },
        { id: '2', data: { label: 'This is a node This is a node This is a node This is a node' }, position: { x: 100, y: 100 } },
        { id: '3', data: { label: 'This is a node' }, position: { x: 100, y: 200 }, style: { background: '#222', color: '#fff' } },
        { id: '4', type: 'output', data: { label: 'nody nodes' }, position: { x: 50, y: 300 } },
        { id: '5', type: 'default', data: { label: 'Another node'}, position: { x: 400, y: 300 } },
        { id: '6', type: 'special', onChange, data: { label: 'no option selected' }, position: { x: 400, y: 400 } },
        { source: '1', target: '2', type: 'bezier', animated: true, style: { stroke: 'orange' } },
        { source: '2', target: '3' },
        { source: '3', target: '4' },
        { source: '3', target: '5' },
        { source: '5', target: '6' }
      ]
    };
  }

  onLoad(graphInstance) {
    this.graphInstance = graphInstance;
    console.log('graph loaded:', this.graphInstance);

    this.graphInstance.fitView();
  }

  onChange(elements) {
    if (!this.graphInstance) {
      return false;
    }
    // console.log('graph changed', elements);
  }

  onFitView() {
    if (!this.graphInstance) {
      return false;
    }
    this.graphInstance.fitView();
  }

  onAdd() {
    this.setState(prevState => ({
      ...prevState,
      elements: prevState.elements.concat({
        data: { id: prevState.elements.length + 1, label: 'Added node' },
        position: { x: Math.random() * window.innerWidth, y: Math.random() * window.innerHeight }
      })
    }));
  }

  render() {
    return (
      <Graph
        elements={this.state.elements}
        onElementClick={node => console.log('clicked', node)}
        onElementsRemove={elements => console.log('remove', elements)}
        style={{ width: '100%', height: '100%' }}
        onLoad={graphInstance => this.onLoad(graphInstance)}
        onChange={(elements) => this.onChange(elements)}
        nodeTypes={{
          special: SpecialNode
        }}
      >
        <button
          type="button"
          style={{ position: 'absolute', right: '10px', bottom: '10px', zIndex: 4 }}
          onClick={() => this.onFitView()}
        >
          fit
        </button>
        <button
          type="button"
          style={{ position: 'absolute', bottom: '10px', left: '10px', zIndex: 4 }}
          onClick={() => this.onAdd()}
        >
          add
        </button>
      </Graph>
    );
  }
}

export default App;