@@ -6,18 +6,20 @@ import ReactFlow, {
66 Edge ,
77 useNodesState ,
88 useEdgesState ,
9+ Node ,
910} from 'react-flow-renderer' ;
1011import './index.less' ;
1112
1213export interface IPipelineInitNode {
13- label : string | Element ;
14+ label : React . ReactNode ;
15+ enable : boolean ;
1416 selected ?: boolean ;
1517 selectable ?: boolean ;
1618 [ key : string ] : any ;
1719}
1820export interface IPipelineInitNodeProps {
1921 nodes : IPipelineInitNode [ ] ;
20- onClick ?: ( node : IPipelineInitNode ) => void ;
22+ onClick ?: ( node : Node < any > ) => void ;
2123}
2224
2325const getData = ( nodes : IPipelineInitNode [ ] ) => {
@@ -44,6 +46,7 @@ const getData = (nodes: IPipelineInitNode[]) => {
4446 position : { x : gap , y : node . className === 'circle' ? - 10 : 0 } ,
4547 draggable : false ,
4648 connectable : false ,
49+ className : `${ node . className } ${ node . enable ? 'enable' : '' } ` ,
4750 ...node ,
4851 } ;
4952 if ( index === '0' ) {
@@ -70,8 +73,6 @@ const getData = (nodes: IPipelineInitNode[]) => {
7073 newEdges . push ( edgeObj ) ;
7174 lastNode = nodeObj ;
7275 }
73- console . log ( 'newNodes' , newNodes ) ;
74- console . log ( 'newEdges' , newEdges ) ;
7576 return { newNodes, newEdges } ;
7677} ;
7778
@@ -80,7 +81,6 @@ export const PipelineInitNode: FC<IPipelineInitNodeProps> = (props) => {
8081 const [ nodes , setNodes ] = useNodesState ( [ ] ) ;
8182 const [ edges , setEdges ] = useEdgesState ( [ ] ) ;
8283 useEffect ( ( ) => {
83- console . log ( 'originNodes' , originNodes ) ;
8484 const { newNodes, newEdges } = getData ( originNodes ) ;
8585 setNodes ( newNodes ) ;
8686 setEdges ( newEdges ) ;
0 commit comments