behavior3editor/render-process/index.tsx

import * as ReactDOM from "react-dom";
import Editor from "./Editor";
import React, { Component } from "react";
import { Layout, Tabs } from 'antd';
import { ipcRenderer, remote } from 'electron';
import * as path from 'path';
import * as fs from 'fs';
import MainEventType from "../common/MainEventType";
import Properties from "./Properties";
import { BehaviorTreeModel } from "../common/BehaviorTreeModel";
import Settings from "../main-process/Settings";
import G6 from '@antv/g6';

import 'antd/dist/antd.dark.css';
import './index.css';
import { ModelConfig } from "@antv/g6/lib/types";

const { Header, Sider, Content, Footer } = Layout;
const { TabPane } = Tabs;

interface MainState {
  workspace: string;
  filepaths: string[];
  curPath?: string;
}

const ICON_MAP = {
  a: 'https://gw.alipayobjects.com/mdn/rms_8fd2eb/afts/img/A*0HC-SawWYUoAAAAAAAAAAABkARQnAQ',
  b: 'https://gw.alipayobjects.com/mdn/rms_8fd2eb/afts/img/A*sxK0RJ1UhNkAAAAAAAAAAABkARQnAQ',
};

export default class Main extends Component {
  state: MainState = {
    workspace: '',
    filepaths: [],
  }

  componentWillMount() {
    G6.registerNode(
      'TreeNode',
      {
        options: {
          type: 'rect',
          labelCfg: {
            style: {
              fill: 'blue',
              fontSize: 10
            }
          },
          style: {
            fill: "white",
            stroke: '#72CC4A',
            width: 150
          },
          stateStyles: {
            hover: {
              fill: '#d3adf7',
            },
            selected: {
              stroke: '#000',
              lineWidth: 3,
            },
            dragSrc: {
              fill: 'gray',
            }
          },
        },
        draw(cfg, group) {
          const color = cfg.error ? '#F4664A' : '#30BF78';
          var size = cfg.size ? cfg.size as number[] : [150, 40];
          const w = size[0];
          const h = size[1];
          const r = 2;
          const shape = group.addShape('rect', {
            attrs: {
              x: -w / 2,
              y: -h / 2,
              width: w,
              height: h,
              stroke: color,
              fill: 'white',
              radius: r,
            },
            name: 'main-box',
            draggable: true,
          });

          // title text
          group.addShape('text', {
            attrs: {
              textBaseline: 'top',
              x: -w / 2 + 8,
              y: -h / 2 + 2,
              lineHeight: 20,
              text: cfg.id,
              fill: 'black',
            },
            name: 'title',
          });

          group.addShape('rect', {
            name: 'drag-up',
            attrs: {
              x: -w / 2,
              y: -h / 2,
              width: w * 0.6,
              height: h / 2,
              fill: 'blue',
              fillOpacity: 0.5,
              radius: r,
            },
            visible: false,
          });

          group.addShape('rect', {
            name: 'drag-down',
            attrs: {
              x: -w / 2,
              y: 0,
              width: w * 0.6,
              height: h / 2,
              fill: 'green',
              fillOpacity: 0.5,
              radius: r,
            },
            visible: false,
          });

          group.addShape('rect', {
            name: 'drag-right',
            attrs: {
              x: w * 0.1,
              y: -h / 2,
              width: w * 0.4,
              height: h,
              fill: 'yellow',
              fillOpacity: 0.5,
              radius: r,
            },
            visible: false,
          });
          return shape;
        },

      },
      'single-node',
    );
  }

  componentDidMount() {
    ipcRenderer.on(MainEventType.OPEN_FILE, (event: any, path: any) => {
      this.openFile(path);
    });

    ipcRenderer.on(MainEventType.OPEN_WORKSPACE, (event: any, workspace: any) => {
      console.log("on open workspace", workspace);
      document.title = workspace;
      this.setState({ workspace });
    });

    this.setState({ workspace: this.getLastWorkspace() });
  }

  getLastWorkspace() {
    const settings: Settings = remote.getGlobal("settings");
    return settings.recentWorkspaces[0];
  }

  openFile(path: string) {
    console.log("on open file", path);
    if (this.state.filepaths.indexOf(path) < 0) {
      const filepaths = this.state.filepaths;
      filepaths.push(path);
      this.setState({ filepaths, curPath: path });
    } else {
      this.setState({ curPath: path });
    }
  }

  loadAllTrees() {
    const workspace = this.state.workspace
    if (workspace == '' || !fs.existsSync(workspace)) {
      console.log("workspace not exist", workspace);
      return;
    }

    const files = fs.readdirSync(workspace);
    const list = []
    files.forEach((filename) => {
      console.log("load path", filename);
      const stat = fs.statSync(path.join(workspace, filename))
      if (stat.isFile()) {
        let tree = this.loadTree(filename.slice(0, -5));
        list.push(tree);
        console.log("loaded", filename);
      }
    })
  }

  loadTree(name: string) {
    const filename = this.state.workspace + '/' + name + '.json';
    if (!fs.existsSync(filename)) {
      return;
    }
    const str = fs.readFileSync(filename, 'utf8');
    let tree: BehaviorTreeModel = JSON.parse(str);
    tree.name = name;
    return tree;
  }

  renderJson() {
    const path = this.state.curPath;
    if (path) {
      const str = fs.readFileSync(path, 'utf8');
      return (<div>{str}</div>)
    }
  }

  render() {
    const { workspace, filepaths } = this.state;
    document.title = `行为树编辑器 - ${workspace}`;
    return (
      <Layout className="body">
        <Sider className="sider" width={250}>
          <Properties
            workspace={this.state.workspace}
            onOpenTree={(path) => {
              this.openFile(path);
            }}
          />
        </Sider>
        <Content className="content">
          <Tabs
            hideAdd
            className="tabs"
            type="editable-card"
          >
            {
              filepaths.map(filepath => {
                return (
                  <TabPane tab={path.basename(filepath)} key={filepath}>
                    <Editor filepath={filepath} />
                  </TabPane>
                );
              })
            }
          </Tabs>
        </Content>
      </Layout>
    )
  }
};

ReactDOM.render(
  <Main />,
  document.getElementById('root') as HTMLElement
);