supersonic/webapp/packages/supersonic-fe/src/pages/SemanticModel/SemanticGraph/components/ToolBar.tsx

import G6 from '@antv/g6';
import { createDom } from '@antv/dom-util';

const searchIconSvgPath = `<path d="M909.6 854.5L649.9 594.8C690.2 542.7 712 479 712 412c0-80.2-31.3-155.4-87.9-212.1-56.6-56.7-132-87.9-212.1-87.9s-155.5 31.3-212.1 87.9C143.2 256.5 112 331.8 112 412c0 80.1 31.3 155.5 87.9 212.1C256.5 680.8 331.8 712 412 712c67 0 130.6-21.8 182.7-62l259.7 259.6a8.2 8.2 0 0011.6 0l43.6-43.5a8.2 8.2 0 000-11.6zM570.4 570.4C528 612.7 471.8 636 412 636s-116-23.3-158.4-65.6C211.3 528 188 471.8 188 412s23.3-116.1 65.6-158.4C296 211.3 352.2 188 412 188s116.1 23.2 158.4 65.6S636 352.2 636 412s-23.3 116.1-65.6 158.4z" />`;

const searchNode = (graph) => {
  const toolBarSearchInput = document.getElementById('toolBarSearchInput') as HTMLInputElement;
  const searchText = toolBarSearchInput.value.trim();
  let lastFoundNode = null;
  graph.getNodes().forEach((node) => {
    const model = node.getModel();
    const isFound = searchText && model.label.includes(searchText);
    if (isFound) {
      graph.setItemState(node, 'active', true);
      lastFoundNode = node;
    } else {
      graph.setItemState(node, 'active', false);
    }
  });

  if (lastFoundNode) {
    // 将视图移动到找到的节点位置
    graph.focusItem(lastFoundNode, true, {
      duration: 300,
      easing: 'easeCubic',
    });
  }
};

const generatorSearchInputDom = (graph) => {
  const domString =
    '<input placeholder="请输入指标/维度名称" class="ant-input" id="toolBarSearchInput" type="text" value="" />';
  const searchInputDom = createDom(domString);
  searchInputDom.addEventListener('keydown', (event) => {
    if (event.key === 'Enter') {
      searchNode(graph);
    }
  });
  return searchInputDom;
};

const generatorSearchBtnDom = (graph) => {
  const domString = `<button
  id="toolBarSearchBtn"
  type="button"
  class="ant-btn ant-btn-default ant-btn-icon-only ant-input-search-button"
>
  <span role="img" aria-label="search" class="anticon anticon-search">
    <svg
      viewBox="64 64 896 896"
      focusable="false"
      data-icon="search"
      width="1em"
      height="1em"
      fill="currentColor"
      aria-hidden="true"
    >
      ${searchIconSvgPath}
    </svg>
  </span>
</button>`;
  const searchBtnDom = createDom(domString);
  searchBtnDom.addEventListener('click', () => {
    searchNode(graph);
  });
  return searchBtnDom;
};

const searchInputDOM = (graph) => {
  const searchInputDom = generatorSearchInputDom(graph);
  const searchBtnDom = generatorSearchBtnDom(graph);
  const searchInput = `
  <div id="searchInputContent" class="g6-component-toolbar-search-input" style="position: absolute;top: 38px;width: 190px;left: 0;display:none">
    <span class="ant-input-group-wrapper ant-input-search" >
      <span class="ant-input-wrapper ant-input-group" id="toolBarSearchWrapper">
        <span class="ant-input-group-addon"></span>
      </span>
    </span>
  </div>`;
  const searchDom = createDom(searchInput);
  const searchWrapperDom = searchDom.querySelector('#toolBarSearchWrapper');
  searchWrapperDom.insertBefore(searchInputDom, searchWrapperDom.firstChild);
  searchWrapperDom.querySelector('.ant-input-group-addon').appendChild(searchBtnDom);
  return searchDom;
};

const initToolBar = () => {
  const toolBarInstance = new G6.ToolBar();
  const config = toolBarInstance._cfgs;
  const defaultContentDomString = config.getContent();
  const defaultContentDom = createDom(defaultContentDomString);
  // @ts-ignore
  const elements = defaultContentDom.querySelectorAll('li[code="redo"], li[code="undo"]');
  elements.forEach((element) => {
    element.remove();
  });
  const searchBtnDom = `<li code="search">
  <svg
    viewBox="64 64 896 896"
    class="icon"
    data-icon="search"
    width="24"
    height="24"
    fill="currentColor"
    aria-hidden="true"
  >
    ${searchIconSvgPath}
  </svg>
</li>`;
  defaultContentDom.insertAdjacentHTML('afterbegin', searchBtnDom);
  let searchInputContentVisible = false;
  const toolbar = new G6.ToolBar({
    position: { x: 10, y: 10 },
    className: 'semantic-graph-toolbar',
    getContent: (graph) => {
      const searchInput = searchInputDOM(graph);
      const content = `<div class="g6-component-toolbar-content">${defaultContentDom.outerHTML}</div>`;
      const contentDom = createDom(content);
      contentDom.appendChild(searchInput);
      return contentDom;
    },
    handleClick: (code, graph) => {
      if (code === 'search') {
        const searchText = document.getElementById('searchInputContent');
        if (searchText) {
          const visible = searchInputContentVisible ? 'none' : 'block';
          searchText.style.display = visible;
          searchInputContentVisible = !searchInputContentVisible;
        }
      } else {
        // handleDefaultOperator public方法缺失graph作为参数传入，将graph挂载在cfgs上，源码通过get会获取到graph,完成默认code的执行逻辑
        toolBarInstance._cfgs.graph = graph;
        toolBarInstance.handleDefaultOperator(code);
      }
    },
  });

  return toolbar;
};
export default initToolBar;