mirror of
https://github.com/langgenius/dify.git
synced 2026-02-12 14:04:47 +08:00
Signed-off-by: -LAN- <laipz8200@outlook.com> Co-authored-by: twwu <twwu@dify.ai> Co-authored-by: crazywoola <100913391+crazywoola@users.noreply.github.com> Co-authored-by: jyong <718720800@qq.com> Co-authored-by: Wu Tianwei <30284043+WTW0313@users.noreply.github.com> Co-authored-by: QuantumGhost <obelisk.reg+git@gmail.com> Co-authored-by: lyzno1 <yuanyouhuilyz@gmail.com> Co-authored-by: quicksand <quicksandzn@gmail.com> Co-authored-by: Jyong <76649700+JohnJyong@users.noreply.github.com> Co-authored-by: lyzno1 <92089059+lyzno1@users.noreply.github.com> Co-authored-by: zxhlyh <jasonapring2015@outlook.com> Co-authored-by: Yongtao Huang <yongtaoh2022@gmail.com> Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com> Co-authored-by: Joel <iamjoel007@gmail.com> Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com> Co-authored-by: nite-knite <nkCoding@gmail.com> Co-authored-by: Hanqing Zhao <sherry9277@gmail.com> Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com> Co-authored-by: Harry <xh001x@hotmail.com>
71 lines
1.8 KiB
TypeScript
71 lines
1.8 KiB
TypeScript
import {
|
|
memo,
|
|
} from 'react'
|
|
import {
|
|
Handle,
|
|
Position,
|
|
} from 'reactflow'
|
|
import {
|
|
BlockEnum,
|
|
} from '@/app/components/workflow/types'
|
|
import type { Node } from '@/app/components/workflow/types'
|
|
import cn from '@/utils/classnames'
|
|
|
|
type NodeHandleProps = {
|
|
handleId: string
|
|
handleClassName?: string
|
|
} & Pick<Node, 'id' | 'data'>
|
|
|
|
export const NodeTargetHandle = memo(({
|
|
data,
|
|
handleId,
|
|
handleClassName,
|
|
}: NodeHandleProps) => {
|
|
const connected = data._connectedTargetHandleIds?.includes(handleId)
|
|
|
|
return (
|
|
<>
|
|
<Handle
|
|
id={handleId}
|
|
type='target'
|
|
position={Position.Left}
|
|
className={cn(
|
|
'z-[1] !h-4 !w-4 !rounded-none !border-none !bg-transparent !outline-none',
|
|
'after:absolute after:left-1.5 after:top-1 after:h-2 after:w-0.5 after:bg-workflow-link-line-handle',
|
|
'transition-all hover:scale-125',
|
|
!connected && 'after:opacity-0',
|
|
data.type === BlockEnum.Start && 'opacity-0',
|
|
handleClassName,
|
|
)}
|
|
>
|
|
</Handle>
|
|
</>
|
|
)
|
|
})
|
|
NodeTargetHandle.displayName = 'NodeTargetHandle'
|
|
|
|
export const NodeSourceHandle = memo(({
|
|
data,
|
|
handleId,
|
|
handleClassName,
|
|
}: NodeHandleProps) => {
|
|
const connected = data._connectedSourceHandleIds?.includes(handleId)
|
|
|
|
return (
|
|
<Handle
|
|
id={handleId}
|
|
type='source'
|
|
position={Position.Right}
|
|
className={cn(
|
|
'group/handle z-[1] !h-4 !w-4 !rounded-none !border-none !bg-transparent !outline-none',
|
|
'after:absolute after:right-1.5 after:top-1 after:h-2 after:w-0.5 after:bg-workflow-link-line-handle',
|
|
'transition-all hover:scale-125',
|
|
!connected && 'after:opacity-0',
|
|
handleClassName,
|
|
)}
|
|
>
|
|
</Handle>
|
|
)
|
|
})
|
|
NodeSourceHandle.displayName = 'NodeSourceHandle'
|