| 1 |
import { recordAgentActivity } from '@agent/api'; |
| 2 |
import { useWorkflowStore } from '@agent/state/workflows'; |
| 3 |
import { useCallback, useMemo } from '@wordpress/element'; |
| 4 |
|
| 5 |
export const useWhenFinishedToolProps = () => { |
| 6 |
const { whenFinishedToolProps, setWhenFinishedToolProps, getWorkflow } = |
| 7 |
useWorkflowStore(); |
| 8 |
|
| 9 |
const onConfirm = useCallback( |
| 10 |
(props = {}) => { |
| 11 |
if (!whenFinishedToolProps) return; |
| 12 |
const workflow = getWorkflow(); |
| 13 |
recordAgentActivity({ |
| 14 |
sessionId: workflow?.sessionId, |
| 15 |
action: 'workflow_tool_event', |
| 16 |
value: { trigger: 'confirm', workflow: workflow?.id }, |
| 17 |
}); |
| 18 |
window.dispatchEvent( |
| 19 |
new CustomEvent('extendify-agent:workflow-confirm', { |
| 20 |
detail: { ...props, whenFinishedToolProps }, |
| 21 |
}), |
| 22 |
); |
| 23 |
}, |
| 24 |
[whenFinishedToolProps, getWorkflow], |
| 25 |
); |
| 26 |
|
| 27 |
const onCancel = useCallback(() => { |
| 28 |
if (!whenFinishedToolProps) return; |
| 29 |
const workflow = getWorkflow(); |
| 30 |
recordAgentActivity({ |
| 31 |
sessionId: workflow?.sessionId, |
| 32 |
action: 'workflow_tool_event', |
| 33 |
value: { trigger: 'cancel', workflow: workflow?.id }, |
| 34 |
}); |
| 35 |
window.dispatchEvent( |
| 36 |
new CustomEvent('extendify-agent:workflow-cancel', { |
| 37 |
detail: { whenFinishedToolProps }, |
| 38 |
}), |
| 39 |
); |
| 40 |
}, [whenFinishedToolProps, getWorkflow]); |
| 41 |
|
| 42 |
const onRetry = useCallback(() => { |
| 43 |
if (!whenFinishedToolProps) return; |
| 44 |
setWhenFinishedToolProps(null); |
| 45 |
const workflow = getWorkflow(); |
| 46 |
recordAgentActivity({ |
| 47 |
sessionId: workflow?.sessionId, |
| 48 |
action: 'workflow_tool_event', |
| 49 |
value: { trigger: 'retry', workflow: workflow?.id }, |
| 50 |
}); |
| 51 |
window.dispatchEvent( |
| 52 |
new CustomEvent('extendify-agent:workflow-retry', { |
| 53 |
detail: { whenFinishedToolProps }, |
| 54 |
}), |
| 55 |
); |
| 56 |
}, [whenFinishedToolProps, setWhenFinishedToolProps, getWorkflow]); |
| 57 |
|
| 58 |
return useMemo(() => { |
| 59 |
if (!whenFinishedToolProps) return null; |
| 60 |
return { ...whenFinishedToolProps, onConfirm, onCancel, onRetry }; |
| 61 |
}, [whenFinishedToolProps, onConfirm, onCancel, onRetry]); |
| 62 |
}; |
| 63 |
|
