diff --git a/src/components/EdgeToolbar.tsx b/src/components/EdgeToolbar.tsx index e19bf359..ca00436d 100644 --- a/src/components/EdgeToolbar.tsx +++ b/src/components/EdgeToolbar.tsx @@ -5,7 +5,7 @@ import { WorkflowEdgeData } from "@/types"; import { useMemo, useEffect, useState, useRef } from "react"; export function EdgeToolbar() { - const { edges, toggleEdgePause, removeEdge } = useWorkflowStore(); + const { edges, toggleEdgePause, removeEdge, setLoopCount } = useWorkflowStore(); const [clickPosition, setClickPosition] = useState<{ x: number; y: number } | null>(null); const previousSelectedEdgeId = useRef(null); @@ -79,6 +79,12 @@ export function EdgeToolbar() { } }; + const handleLoopCountChange = (delta: number) => { + if (selectedEdge) { + setLoopCount(selectedEdge.id, loopCount + delta); + } + }; + const handleDelete = () => { if (selectedEdge) { removeEdge(selectedEdge.id); @@ -88,6 +94,8 @@ export function EdgeToolbar() { if (!toolbarPosition || !selectedEdge) return null; const hasPause = selectedEdge.data?.hasPause; + const isLoop = selectedEdge.data?.isLoop; + const loopCount = selectedEdge.data?.loopCount ?? 3; return (
)} - + {loopCount} + +
+ + )} + {!isLoop && ( + + )}