@@ -6,12 +6,13 @@ import {
6
6
useReactFlow ,
7
7
} from '@xyflow/react' ;
8
8
import produce from 'immer' ;
9
- import React , { memo , useState } from 'react' ;
9
+ import React , { memo , useContext , useState } from 'react' ;
10
10
import { shallow } from 'zustand/shallow' ;
11
11
import { useStore } from '../../hooks/useStore' ;
12
12
import { uuid } from '../../utils' ;
13
13
import NodeSelectPopover from '../NodesPopover' ;
14
14
import './index.less' ;
15
+ import { ConfigContext } from '../../models/context' ;
15
16
16
17
export default memo ( ( edge : any ) => {
17
18
const { id, selected, sourceX, sourceY, targetX, targetY, source, target } =
@@ -26,6 +27,9 @@ export default memo((edge: any) => {
26
27
targetY,
27
28
} ) ;
28
29
30
+ const { hideLineInsertBtn } =
31
+ useContext ( ConfigContext ) ;
32
+
29
33
const {
30
34
nodes,
31
35
edges,
@@ -45,7 +49,7 @@ export default memo((edge: any) => {
45
49
onEdgesChange : state . onEdgesChange ,
46
50
} ) ,
47
51
shallow
48
- ) ;
52
+ ) ;
49
53
50
54
const handleAddNode = ( data : any ) => {
51
55
const { screenToFlowPosition } = reactflow ;
@@ -122,11 +126,13 @@ export default memo((edge: any) => {
122
126
>
123
127
< CloseOutlined style = { { color : '#fff' , fontSize : 10 } } />
124
128
</ div >
125
- < NodeSelectPopover placement = "right" addNode = { handleAddNode } >
126
- < div className = "line-icon-box" >
127
- < PlusOutlined style = { { color : '#fff' , fontSize : 10 } } />
128
- </ div >
129
- </ NodeSelectPopover >
129
+ {
130
+ ! hideLineInsertBtn && < NodeSelectPopover placement = "right" addNode = { handleAddNode } >
131
+ < div className = "line-icon-box" >
132
+ < PlusOutlined style = { { color : '#fff' , fontSize : 10 } } />
133
+ </ div >
134
+ </ NodeSelectPopover >
135
+ }
130
136
</ div >
131
137
</ div >
132
138
</ EdgeLabelRenderer >
0 commit comments