From 404ef900b2fdc7bb174fd5451f3815fcb8bf5665 Mon Sep 17 00:00:00 2001
From: jile1997 <1154416769@qq.com>
Date: Thu, 2 Jan 2025 14:11:53 +0800
Subject: [PATCH 1/2] =?UTF-8?q?feat:=E5=A2=9E=E5=8A=A0=E6=8E=A7=E5=88=B6?=
 =?UTF-8?q?=E9=85=8D=E7=BD=AE?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../src/components/CustomNode/index.tsx       | 45 ++++++++++------
 .../x-flow/src/operator/Control/index.tsx     | 53 +++++++++++++------
 packages/x-flow/src/types.ts                  | 15 +++++-
 3 files changed, 80 insertions(+), 33 deletions(-)

diff --git a/packages/x-flow/src/components/CustomNode/index.tsx b/packages/x-flow/src/components/CustomNode/index.tsx
index 15d156a75..2c560478e 100644
--- a/packages/x-flow/src/components/CustomNode/index.tsx
+++ b/packages/x-flow/src/components/CustomNode/index.tsx
@@ -20,6 +20,9 @@ export default memo((props: any) => {
     useContext(ConfigContext);
   const deletable = globalConfig?.edge?.deletable ?? true;
 
+  // const isConnectableStart = globalConfig?.handle?.isConnectableStart ?? true;
+  // const isConnectableEnd = globalConfig?.handle?.isConnectableEnd ?? true;
+
   const NodeWidget =
     widgets[`${capitalize(type)}Node`] || widgets['CommonNode'];
   const [isHovered, setIsHovered] = useState(false);
@@ -92,6 +95,24 @@ export default memo((props: any) => {
     deleteNode(id);
   }, [pasteNode]);
 
+  const defaultAction = (e, sourceHandle) => {
+    if (e.key === 'copy') {
+      handleCopyNode();
+    } else if (e.key === 'paste') {
+      handlePasteNode();
+    } else if (e.key === 'delete') {
+      handleDeleteNode();
+    } else if (e.key.startsWith('paste-')) {
+      if (sourceHandle) {
+        handlePasteNode({
+          sourceHandle,
+        });
+      } else {
+        handlePasteNode();
+      }
+    }
+  };
+
   const itemClick = e => {
     if (!e.key) {
       return;
@@ -105,23 +126,11 @@ export default memo((props: any) => {
       if (type === 'Switch' && e.key.startsWith('paste-') && sourceHandle) {
         data['sourceHandle'] = sourceHandle;
       }
-      onMenuItemClick(data);
+      onMenuItemClick(data, () => {
+        defaultAction(e, sourceHandle);
+      });
     } else {
-      if (e.key === 'copy') {
-        handleCopyNode();
-      } else if (e.key === 'paste') {
-        handlePasteNode();
-      } else if (e.key === 'delete') {
-        handleDeleteNode();
-      } else if (e.key.startsWith('paste-')) {
-        if (sourceHandle) {
-          handlePasteNode({
-            sourceHandle,
-          });
-        } else {
-          handlePasteNode();
-        }
-      }
+      defaultAction(e, sourceHandle);
     }
   };
 
@@ -187,6 +196,8 @@ export default memo((props: any) => {
           type="target"
           position={targetPosition}
           isConnectable={isConnectable}
+          // isConnectableStart={isConnectableStart}
+          // isConnectableEnd={isConnectableEnd}
         />
       )}
       {selected && (
@@ -234,6 +245,8 @@ export default memo((props: any) => {
             selected={selected}
             isHovered={isHovered}
             handleAddNode={handleAddNode}
+            // isConnectableStart={isConnectableStart}
+            // isConnectableEnd={isConnectableEnd}
           />
         </>
       )}
diff --git a/packages/x-flow/src/operator/Control/index.tsx b/packages/x-flow/src/operator/Control/index.tsx
index 2496d08d7..cd8f9238c 100644
--- a/packages/x-flow/src/operator/Control/index.tsx
+++ b/packages/x-flow/src/operator/Control/index.tsx
@@ -1,17 +1,22 @@
 import { Button, Tooltip } from 'antd';
 import type { MouseEvent } from 'react';
-import React, { memo } from 'react';
+import React, { memo, useContext } from 'react';
 import IconView from '../../components/IconView';
 import NodeSelectPopover from '../../components/NodesPopover';
 import { useStore, useStoreApi } from '../../hooks/useStore';
+import { ConfigContext } from '../../models/context';
 import { useEventEmitterContextContext } from '../../models/event-emitter';
 
-import './index.less';
 import { useFullscreen } from 'ahooks';
+import './index.less';
 
 const Control = (props: any) => {
   const { addNode, xflowRef } = props;
-  const [isFullscreen, {toggleFullscreen }] = useFullscreen(xflowRef);
+  const [isFullscreen, { toggleFullscreen }] = useFullscreen(xflowRef);
+  const { globalConfig } = useContext(ConfigContext);
+
+  const hideAddNode = globalConfig?.controls?.hideAddNode ?? false;
+  const hideAnnotate = globalConfig?.controls?.hideAnnotate ?? false;
 
   const addNote = (e: MouseEvent<HTMLDivElement>) => {
     e.stopPropagation();
@@ -27,22 +32,38 @@ const Control = (props: any) => {
 
   return (
     <div className='fai-reactflow-control'>
-      <NodeSelectPopover addNode={addNode}>
-        <Tooltip title='添加节点' getPopupContainer={() => document.getElementById('xflow-container') as HTMLElement}>
+      {!hideAddNode && (
+        <NodeSelectPopover addNode={addNode}>
+          <Tooltip
+            title="添加节点"
+            getPopupContainer={() =>
+              document.getElementById('xflow-container') as HTMLElement
+            }
+          >
+            <Button
+              type="text"
+              icon={<IconView type="icon-add-circle" className="icon" />}
+            />
+          </Tooltip>
+        </NodeSelectPopover>
+      )}
+      {!hideAnnotate && (
+        <Tooltip
+          title="添加注释"
+          getPopupContainer={() =>
+            document.getElementById('xflow-container') as HTMLElement
+          }
+        >
           <Button
-            type='text'
-            icon={<IconView type='icon-add-circle' className='icon' />}
+            type="text"
+            icon={
+              <IconView type="icon-sticky-note-add-line" className="icon" />
+            }
+            onClick={addNote}
           />
         </Tooltip>
-      </NodeSelectPopover>
-      <Tooltip title='添加注释' getPopupContainer={() => document.getElementById('xflow-container') as HTMLElement}>
-        <Button
-          type='text'
-          icon={<IconView type='icon-sticky-note-add-line' className='icon' />}
-          onClick={addNote}
-        />
-      </Tooltip>
-      <div className='separator'></div>
+      )}
+      {!(hideAddNode && hideAnnotate) && <div className='separator'></div>}
       <Tooltip title='指针模式' getPopupContainer={() => document.getElementById('xflow-container') as HTMLElement}>
         <Button
           type='text'
diff --git a/packages/x-flow/src/types.ts b/packages/x-flow/src/types.ts
index c7f9be232..958daa05a 100644
--- a/packages/x-flow/src/types.ts
+++ b/packages/x-flow/src/types.ts
@@ -77,6 +77,16 @@ export interface TEdge {
   deletable?: boolean; // 是否允许删除线条 初始化的edges不受此项影响
 }
 
+export interface TControl{
+  hideAddNode?:boolean
+  hideAnnotate?:boolean
+}
+
+// export interface THandle{
+//   isConnectableStart?:boolean
+//   isConnectableEnd?:boolean
+// }
+
 export interface FlowProps {
   initialValues?: {
     nodes: any[];
@@ -97,10 +107,13 @@ export interface FlowProps {
     nodePanel?: TNodePanel;
     nodeView?: TNodeView;
     edge?: TEdge;
+    controls?:TControl
+    //handle?:THandle
   };
   logPanel?: TLogPanel; // 日志面板配置
   onNodeClick?: NodeMouseHandler;
-  onMenuItemClick: (itemInfo: ItemInfo) => void;
+  onMenuItemClick: (itemInfo: ItemInfo,defaultAction:()=>void) => void;
+
 }
 interface ItemInfo {
   key: 'copy' | 'paste' | 'delete' | string;

From af7c1ec9efdcae27633423b1611aa3f73d972514 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=98=94=E6=A2=A6?= <hemengying.hmy@alibaba-inc.com>
Date: Thu, 2 Jan 2025 14:44:00 +0800
Subject: [PATCH 2/2] =?UTF-8?q?=20fix=EF=BC=9A=E5=A2=9E=E5=8A=A0"@monaco-e?=
 =?UTF-8?q?ditor/react=20dep?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 packages/x-flow/package.json | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/packages/x-flow/package.json b/packages/x-flow/package.json
index 4397063cb..936c87de9 100644
--- a/packages/x-flow/package.json
+++ b/packages/x-flow/package.json
@@ -56,7 +56,8 @@
     "tinycolor2": "^1.6.0",
     "use-context-selector": "^1.4.1",
     "zundo": "^2.1.0",
-    "zustand": "^4.5.4"
+    "zustand": "^4.5.4",
+    "@monaco-editor/react": "^4.4.6"
   },
   "devDependencies": {
     "deep-equal": "^2.0.3",