dify/web/app/components/workflow/custom-connection-line.tsx

40 lines
758 B
TypeScript
Raw Normal View History

import { memo } from 'react'
import type { ConnectionLineComponentProps } from 'reactflow'
import {
Position,
getSimpleBezierPath,
} from 'reactflow'
const CustomConnectionLine = ({ fromX, fromY, toX, toY }: ConnectionLineComponentProps) => {
const [
edgePath,
] = getSimpleBezierPath({
sourceX: fromX,
sourceY: fromY,
sourcePosition: Position.Right,
targetX: toX,
targetY: toY,
targetPosition: Position.Left,
})
return (
<g>
<path
fill="none"
stroke='#D0D5DD'
strokeWidth={2}
d={edgePath}
/>
<rect
x={toX}
y={toY - 4}
width={2}
height={8}
fill='#2970FF'
/>
</g>
)
}
export default memo(CustomConnectionLine)