✨ feat: 更新proflow配置的类型

ant-design · Oct 8, 2023 · 3b06dde · 3b06dde
1 parent b914cd4
commit 3b06dde
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 26 deletions.
diff --git a/src/ProFlow/constants.tsx b/src/ProFlow/constants.tsx
@@ -1,3 +1,4 @@
+import { ProFlowNode, ProFlowNodeData } from '@/constants';
 import { Node } from 'reactflow';
 
 export enum NodeSelect {
@@ -6,24 +7,6 @@ export enum NodeSelect {
   WARNING = 'WARNING',
   DEFAULT = 'DEFAULT',
 }
-export interface ProFlowNode {
-  id: string;
-  group?: boolean;
-  select?: NodeSelect;
-  data: ProFlowNodeData | ProFlowNode[];
-}
-
-export interface ProFlowNodeData {
-  title: string;
-  describe: string;
-  logo: string;
-}
-
-export interface ProFLowEdge {
-  id: string;
-  source: string;
-  target: string;
-}
 
 export interface InitialNode extends Node {
   width?: number;

diff --git a/src/ProFlow/index.tsx b/src/ProFlow/index.tsx
@@ -1,7 +1,6 @@
 import React, { useMemo, type CSSProperties, type MouseEvent as ReactMouseEvent } from 'react';
-import ReactFlow, { Background, BackgroundVariant, Edge, Node, useEdgesState } from 'reactflow';
-import { ProFlowController, RadiusEdge } from '../index';
-import { ProFLowEdge, ProFlowNode } from './constants';
+import ReactFlow, { Background, BackgroundVariant, Edge, Node } from 'reactflow';
+import { ProFlowController, ProFlowEdge, ProFlowNode, RadiusEdge } from '../index';
 import { convertMappingFrom, getRenderData } from './helper';
 import { useStyles } from './styles';
 
@@ -12,7 +11,7 @@ interface ProFlowProps {
   onPaneClick: (event: ReactMouseEvent) => void;
   onNodeClick: (event: ReactMouseEvent, node: Node) => void;
   nodes: ProFlowNode[];
-  edges: ProFLowEdge[];
+  edges: ProFlowEdge[];
   className?: string;
   style?: CSSProperties;
   miniMap?: boolean;
@@ -40,10 +39,7 @@ const ProFlow: React.FC<Partial<ProFlowProps>> = (props) => {
       };
     }
   }, [mapping]);
-  const [_edges] = useEdgesState(renderData.edges);
-
-  console.log(renderData.edges);
-  console.log(_edges);
+  // const [_edges] = useEdgesState(renderData.edges);
 
   return (
     <ReactFlow

diff --git a/src/constants.tsx b/src/constants.tsx
@@ -4,13 +4,27 @@ export enum NodeSelect {
   WARNING = 'WARNING',
   DEFAULT = 'DEFAULT',
 }
+
+export enum EdgeType {
+  default = 'default',
+  radius = 'radius',
+}
+
+// 暴露给用户使用的Node配置
 export interface ProFlowNode {
   id: string;
   group?: boolean;
   select?: NodeSelect;
   data: ProFlowNodeData | ProFlowNode[];
 }
 
+// 暴露给用户使用的Edge配置
+export interface ProFlowEdge {
+  id: string;
+  source: string;
+  target: string;
+  type?: EdgeType;
+}
 export interface ProFlowNodeData {
   title: string;
   describe: string;