2025-04-18 13:59:12 +08:00
|
|
|
import { useCallback } from 'react'
|
|
|
|
import produce from 'immer'
|
|
|
|
import { useStoreApi } from 'reactflow'
|
2025-06-24 09:10:30 +08:00
|
|
|
import { NodeRunningStatus } from '../types'
|
2025-04-18 13:59:12 +08:00
|
|
|
|
|
|
|
export const useNodesInteractionsWithoutSync = () => {
|
|
|
|
const store = useStoreApi()
|
|
|
|
|
|
|
|
const handleNodeCancelRunningStatus = useCallback(() => {
|
|
|
|
const {
|
|
|
|
getNodes,
|
|
|
|
setNodes,
|
|
|
|
} = store.getState()
|
|
|
|
|
|
|
|
const nodes = getNodes()
|
|
|
|
const newNodes = produce(nodes, (draft) => {
|
|
|
|
draft.forEach((node) => {
|
|
|
|
node.data._runningStatus = undefined
|
|
|
|
node.data._waitingRun = false
|
|
|
|
})
|
|
|
|
})
|
|
|
|
setNodes(newNodes)
|
|
|
|
}, [store])
|
|
|
|
|
2025-06-24 09:10:30 +08:00
|
|
|
const handleCancelAllNodeSuccessStatus = useCallback(() => {
|
|
|
|
const {
|
|
|
|
getNodes,
|
|
|
|
setNodes,
|
|
|
|
} = store.getState()
|
|
|
|
|
|
|
|
const nodes = getNodes()
|
|
|
|
const newNodes = produce(nodes, (draft) => {
|
|
|
|
draft.forEach((node) => {
|
|
|
|
if(node.data._runningStatus === NodeRunningStatus.Succeeded)
|
|
|
|
node.data._runningStatus = undefined
|
|
|
|
})
|
|
|
|
})
|
|
|
|
setNodes(newNodes)
|
|
|
|
}, [store])
|
|
|
|
|
|
|
|
const handleCancelNodeSuccessStatus = useCallback((nodeId: string) => {
|
|
|
|
const {
|
|
|
|
getNodes,
|
|
|
|
setNodes,
|
|
|
|
} = store.getState()
|
|
|
|
|
|
|
|
const newNodes = produce(getNodes(), (draft) => {
|
|
|
|
const node = draft.find(n => n.id === nodeId)
|
|
|
|
if (node && node.data._runningStatus === NodeRunningStatus.Succeeded) {
|
|
|
|
node.data._runningStatus = undefined
|
|
|
|
node.data._waitingRun = false
|
|
|
|
}
|
|
|
|
})
|
|
|
|
setNodes(newNodes)
|
|
|
|
}, [store])
|
|
|
|
|
2025-04-18 13:59:12 +08:00
|
|
|
return {
|
|
|
|
handleNodeCancelRunningStatus,
|
2025-06-24 09:10:30 +08:00
|
|
|
handleCancelAllNodeSuccessStatus,
|
|
|
|
handleCancelNodeSuccessStatus,
|
2025-04-18 13:59:12 +08:00
|
|
|
}
|
|
|
|
}
|