qinglong/src/pages/scenario/flowgram/components/node-adder/index.tsx

import { useCallback, useMemo, useState } from 'react';

import { useClientContext } from '@flowgram.ai/fixed-layout-editor';
import { type FlowNodeEntity } from '@flowgram.ai/fixed-layout-editor';
import { Popover, message, Typography } from 'antd';

import { NodeList } from '../node-list';
import { readData } from '../../shortcuts/utils';
import { generateNodeId } from './utils';
import { PasteIcon, Wrap } from './styles';
import { CopyOutlined, PlusOutlined } from '@ant-design/icons';

const generateNewIdForChildren = (n: FlowNodeEntity): FlowNodeEntity => {
  if (n.blocks) {
    return {
      ...n,
      id: generateNodeId(n),
      blocks: n.blocks.map((b) => generateNewIdForChildren(b)),
    } as FlowNodeEntity;
  } else {
    return {
      ...n,
      id: generateNodeId(n),
    } as FlowNodeEntity;
  }
};

export default function Adder(props: {
  from: FlowNodeEntity;
  to?: FlowNodeEntity;
  hoverActivated: boolean;
}) {
  const { from } = props;
  const isVertical = from.isVertical;
  const [visible, setVisible] = useState(false);
  const { playground, operation, clipboard } = useClientContext();

  const [pasteIconVisible, setPasteIconVisible] = useState(false);

  const activated = useMemo(
    () => props.hoverActivated && !playground.config.readonly,
    [props.hoverActivated, playground.config.readonly]
  );

  const add = (addProps: any) => {
    const blocks = addProps.blocks ? addProps.blocks : undefined;
    const block = operation.addFromNode(from, {
      ...addProps,
      blocks,
    });
    setTimeout(() => {
      playground.scrollToView({
        bounds: block.bounds,
        scrollToCenter: true,
      });
    }, 10);
    setVisible(false);
  };

  const handlePaste = useCallback(async (e: any) => {
    try {
      e.stopPropagation();
      const nodes = await readData(clipboard);

      if (!nodes) {
        message.error({
          content: 'The clipboard content has been updated, please copy the node again.',
        });
        return;
      }

      nodes.reverse().forEach((n: FlowNodeEntity) => {
        const newNodeData = generateNewIdForChildren(n);
        operation.addFromNode(from, newNodeData);
      });

      message.success({
        content: 'Paste successfully!',
      });
    } catch (error) {
      console.error(error);
        message.error({
        content: (
          <Typography.Text>
            Paste failed, please check if you have permission to read the clipboard,
          </Typography.Text>
        ),
      });
    }
  }, []);
  if (playground.config.readonly) return null;

  return (
    <Popover
      visible={visible}
      onVisibleChange={setVisible}
      content={<NodeList onSelect={add} from={from} />}
      placement="right"
      trigger="click"
      align={{ offset: [30, 0] }}
      overlayStyle={{
        padding: 0,
      }}
    >
      <Wrap
        style={
          props.hoverActivated
            ? {
                width: 15,
                height: 15,
              }
            : {}
        }
        onMouseDown={(e) => e.stopPropagation()}
      >
        {props.hoverActivated ? (
          <PlusOutlined
            onClick={() => {
              setVisible(true);
            }}
            onMouseEnter={() => {
              const data = clipboard.readText();
              setPasteIconVisible(!!data);
            }}
            style={{
              backgroundColor: '#fff',
              color: '#3370ff',
              borderRadius: 15,
            }}
          />
        ) : (
          ''
        )}
        {activated && pasteIconVisible && (
          <Popover placement="top" showArrow content="Paste">
            <PasteIcon
              onClick={handlePaste}
              style={
                isVertical
                  ? {
                      right: -25,
                      top: 0,
                    }
                  : {
                      right: 0,
                      top: -20,
                    }
              }
            >
              <CopyOutlined
                style={{
                  backgroundColor: 'var(--semi-color-bg-0)',
                  borderRadius: 15,
                }}
              />
            </PasteIcon>
          </Popover>
        )}
      </Wrap>
    </Popover>
  );
}