gg-editor/src/components/ContextMenu/index.js

A visual graph editor based on G6 and React

import React from 'react';
import { pick } from '@utils';
import Editor from '@components/Base/Editor';
import { CONTEXT_MENU_CONTAINER } from '@common/constants';
import withGGEditorContext from '@common/context/GGEditorContext/withGGEditorContext';
import Menu from './Menu';

class ContextMenu extends React.Component {
  contextMenu = null;

  get containerId() {
    const { editor } = this.props;

    return `${CONTEXT_MENU_CONTAINER}_${editor.id}`;
  }

  componentDidMount() {
    const { editor } = this.props;

    this.contextMenu = new Editor.Contextmenu({
      container: this.containerId,
    });

    editor.add(this.contextMenu);
  }

  render() {
    const { children } = this.props;

    return (
      <div id={this.containerId} {...pick(this.props, ['style', 'className'])}>
        {children}
      </div>
    );
  }
}

export const NodeMenu = Menu.create('node');
export const EdgeMenu = Menu.create('edge');
export const GroupMenu = Menu.create('group');
export const MultiMenu = Menu.create('multi');
export const CanvasMenu = Menu.create('canvas');

export default withGGEditorContext(ContextMenu);