@uipath/apollo-react 4.12.1-pr556.9394355 → 4.12.2

package/dist/canvas/components/HierarchicalCanvas/HierarchicalCanvasWithControls.cjs

@@ -34,7 +34,7 @@ const index_cjs_namespaceObject = require("../../core/index.cjs");
 const canvasStore_cjs_namespaceObject = require("../../stores/canvasStore.cjs");
 const CanvasEventBus_cjs_namespaceObject = require("../../utils/CanvasEventBus.cjs");
 const icon_registry_cjs_namespaceObject = require("../../utils/icon-registry.cjs");
-const LoopNodePreview_cjs_namespaceObject = require("../LoopNode/LoopNodePreview.cjs");
+const createAddNodePreview_cjs_namespaceObject = require("../AddNodePanel/createAddNodePreview.cjs");
 const external_HierarchicalCanvas_cjs_namespaceObject = require("./HierarchicalCanvas.cjs");
 const createDemoCanvases = ()=>{
     const rootCanvas = {
@@ -332,10 +332,13 @@ const CanvasWithControlsContent = ({ initialCanvases, initialPath, onCanvasesCha
     const removeEdge = (0, canvasStore_cjs_namespaceObject.useCanvasStore)(canvasStore_cjs_namespaceObject.selectRemoveEdge);
     const updateNodes = (0, canvasStore_cjs_namespaceObject.useCanvasStore)(canvasStore_cjs_namespaceObject.selectUpdateNodes);
     (0, external_react_namespaceObject.useEffect)(()=>{
-        const unsubscribe = CanvasEventBus_cjs_namespaceObject.canvasEventBus.on('handle:action', (event)=>{
-            (0, LoopNodePreview_cjs_namespaceObject.applyHandleActionAddPreview)(event, reactFlowInstance);
-        });
-        return unsubscribe;
+        const handleAction = (event)=>{
+            if (reactFlowInstance) (0, createAddNodePreview_cjs_namespaceObject.createAddNodePreview)(event.nodeId, event.handleId, reactFlowInstance);
+        };
+        CanvasEventBus_cjs_namespaceObject.canvasEventBus.on('handle:action', handleAction);
+        return ()=>{
+            CanvasEventBus_cjs_namespaceObject.canvasEventBus.off('handle:action', handleAction);
+        };
     }, [
         reactFlowInstance
     ]);

package/dist/canvas/components/HierarchicalCanvas/HierarchicalCanvasWithControls.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"HierarchicalCanvasWithControls.d.ts","sourceRoot":"","sources":["../../../../src/canvas/components/HierarchicalCanvas/HierarchicalCanvasWithControls.tsx"],"names":[],"mappings":"AAOA,OAAO,KAAK,KAAK,MAAM,OAAO,CAAC;AAwd/B,eAAO,MAAM,8BAA8B,EAAE,KAAK,CAAC,EA8BlD,CAAC"}
+{"version":3,"file":"HierarchicalCanvasWithControls.d.ts","sourceRoot":"","sources":["../../../../src/canvas/components/HierarchicalCanvas/HierarchicalCanvasWithControls.tsx"],"names":[],"mappings":"AAOA,OAAO,KAAK,KAAK,MAAM,OAAO,CAAC;AA6d/B,eAAO,MAAM,8BAA8B,EAAE,KAAK,CAAC,EA8BlD,CAAC"}

package/dist/canvas/components/HierarchicalCanvas/HierarchicalCanvasWithControls.js

@@ -6,7 +6,7 @@ import { NodeRegistryProvider } from "../../core/index.js";
 import { selectAddNode, selectCurrentCanvas, selectCurrentPathLength, selectRemoveEdge, selectRemoveNode, selectUpdateNodes, useCanvasStore } from "../../stores/canvasStore.js";
 import { canvasEventBus } from "../../utils/CanvasEventBus.js";
 import { CanvasIcon } from "../../utils/icon-registry.js";
-import { applyHandleActionAddPreview } from "../LoopNode/LoopNodePreview.js";
+import { createAddNodePreview } from "../AddNodePanel/createAddNodePreview.js";
 import { HierarchicalCanvas } from "./HierarchicalCanvas.js";
 const createDemoCanvases = ()=>{
     const rootCanvas = {
@@ -304,10 +304,13 @@ const CanvasWithControlsContent = ({ initialCanvases, initialPath, onCanvasesCha
     const removeEdge = useCanvasStore(selectRemoveEdge);
     const updateNodes = useCanvasStore(selectUpdateNodes);
     useEffect(()=>{
-        const unsubscribe = canvasEventBus.on('handle:action', (event)=>{
-            applyHandleActionAddPreview(event, reactFlowInstance);
-        });
-        return unsubscribe;
+        const handleAction = (event)=>{
+            if (reactFlowInstance) createAddNodePreview(event.nodeId, event.handleId, reactFlowInstance);
+        };
+        canvasEventBus.on('handle:action', handleAction);
+        return ()=>{
+            canvasEventBus.off('handle:action', handleAction);
+        };
     }, [
         reactFlowInstance
     ]);

package/dist/canvas/components/StageNode/DraggableTask.cjs

@@ -50,11 +50,12 @@ const DraggableTaskComponent = ({ task, taskExecution, isSelected, isParallel, g
     }, []);
     const handleGetContextMenuItems = (0, external_react_namespaceObject.useCallback)(()=>{
         if (!getContextMenuItems) return [];
-        return getContextMenuItems(groupIndex, taskIndex);
+        return getContextMenuItems(groupIndex, taskIndex, task.id);
     }, [
         getContextMenuItems,
         groupIndex,
-        taskIndex
+        taskIndex,
+        task.id
     ]);
     const { attributes, listeners, setNodeRef, transition, transform, isDragging } = (0, sortable_namespaceObject.useSortable)({
         id: task.id,

package/dist/canvas/components/StageNode/DraggableTask.js

@@ -22,11 +22,12 @@ const DraggableTaskComponent = ({ task, taskExecution, isSelected, isParallel, g
     }, []);
     const handleGetContextMenuItems = useCallback(()=>{
         if (!getContextMenuItems) return [];
-        return getContextMenuItems(groupIndex, taskIndex);
+        return getContextMenuItems(groupIndex, taskIndex, task.id);
     }, [
         getContextMenuItems,
         groupIndex,
-        taskIndex
+        taskIndex,
+        task.id
     ]);
     const { attributes, listeners, setNodeRef, transition, transform, isDragging } = useSortable({
         id: task.id,

package/dist/canvas/components/StageNode/DraggableTask.types.d.ts

@@ -7,7 +7,7 @@ export interface DraggableTaskProps {
     isParallel: boolean;
     groupIndex: number;
     taskIndex: number;
-    getContextMenuItems?: (groupIndex: number, taskIndex: number) => NodeMenuItem[];
+    getContextMenuItems?: (groupIndex: number, taskIndex: number, taskId: string) => NodeMenuItem[];
     onTaskClick: (e: React.MouseEvent, taskId: string) => void;
     onTaskPlay?: (taskId: string) => Promise<void>;
     isDragDisabled?: boolean;

package/dist/canvas/components/StageNode/DraggableTask.types.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"DraggableTask.types.d.ts","sourceRoot":"","sources":["../../../../src/canvas/components/StageNode/DraggableTask.types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AACvD,OAAO,KAAK,EAAE,kBAAkB,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AAE3E,MAAM,WAAW,kBAAkB;IACjC,IAAI,EAAE,aAAa,CAAC;IACpB,aAAa,CAAC,EAAE,kBAAkB,CAAC;IACnC,UAAU,EAAE,OAAO,CAAC;IACpB,UAAU,EAAE,OAAO,CAAC;IACpB,UAAU,EAAE,MAAM,CAAC;IACnB,SAAS,EAAE,MAAM,CAAC;IAClB,mBAAmB,CAAC,EAAE,CAAC,UAAU,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,KAAK,YAAY,EAAE,CAAC;IAChF,WAAW,EAAE,CAAC,CAAC,EAAE,KAAK,CAAC,UAAU,EAAE,MAAM,EAAE,MAAM,KAAK,IAAI,CAAC;IAC3D,UAAU,CAAC,EAAE,CAAC,MAAM,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IAC/C,cAAc,CAAC,EAAE,OAAO,CAAC;IACzB,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,aAAa,CAAC,EAAE,OAAO,CAAC;CACzB"}
+{"version":3,"file":"DraggableTask.types.d.ts","sourceRoot":"","sources":["../../../../src/canvas/components/StageNode/DraggableTask.types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AACvD,OAAO,KAAK,EAAE,kBAAkB,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AAE3E,MAAM,WAAW,kBAAkB;IACjC,IAAI,EAAE,aAAa,CAAC;IACpB,aAAa,CAAC,EAAE,kBAAkB,CAAC;IACnC,UAAU,EAAE,OAAO,CAAC;IACpB,UAAU,EAAE,OAAO,CAAC;IACpB,UAAU,EAAE,MAAM,CAAC;IACnB,SAAS,EAAE,MAAM,CAAC;IAClB,mBAAmB,CAAC,EAAE,CAAC,UAAU,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,KAAK,YAAY,EAAE,CAAC;IAChG,WAAW,EAAE,CAAC,CAAC,EAAE,KAAK,CAAC,UAAU,EAAE,MAAM,EAAE,MAAM,KAAK,IAAI,CAAC;IAC3D,UAAU,CAAC,EAAE,CAAC,MAAM,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IAC/C,cAAc,CAAC,EAAE,OAAO,CAAC;IACzB,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,aAAa,CAAC,EAAE,OAAO,CAAC;CACzB"}

package/dist/canvas/components/StageNode/StageNodeAdhocTaskGroups.cjs

@@ -28,34 +28,20 @@ __webpack_require__.d(__webpack_exports__, {
 });
 const jsx_runtime_namespaceObject = require("react/jsx-runtime");
 const external_react_namespaceObject = require("react");
-const GroupModificationUtils_cjs_namespaceObject = require("../../utils/GroupModificationUtils.cjs");
 const external_AdhocTask_cjs_namespaceObject = require("./AdhocTask.cjs");
 const external_StageNode_styles_cjs_namespaceObject = require("./StageNode.styles.cjs");
-const external_StageNodeTaskUtilities_cjs_namespaceObject = require("./StageNodeTaskUtilities.cjs");
-const StageNodeAdhocTaskGroupsInner = ({ props, adhocTasks, isReadOnly, selectedTaskId, taskStateReference, marginTop, handleTaskClick, setIsReplacingTask })=>{
-    const { execution, onTaskClick, onTaskGroupModification, onReplaceTaskFromToolbox, onTaskPlay, loadingTaskIds } = props;
-    const getAdhocContextMenuItems = (0, external_react_namespaceObject.useCallback)((groupIndex, taskIndex, taskId)=>{
+const StageNodeAdhocTaskGroupsInner = ({ props, adhocTasks, isReadOnly, selectedTaskId, marginTop, handleTaskClick, generateReplaceTaskMenuItemForTask, generateDeleteTaskMenuItemForTask })=>{
+    const { execution, onTaskGroupModification, onReplaceTaskFromToolbox, onTaskPlay, loadingTaskIds } = props;
+    const getAdhocContextMenuItems = (0, external_react_namespaceObject.useCallback)((taskId)=>{
         const items = [];
-        if (onReplaceTaskFromToolbox) items.push((0, external_StageNodeTaskUtilities_cjs_namespaceObject.getMenuItem)('replace-task', 'Replace task', ()=>{
-            taskStateReference.current = {
-                isParallel: false,
-                groupIndex,
-                taskIndex
-            };
-            onTaskClick?.(taskId);
-            setIsReplacingTask(true);
-        }));
-        if (onTaskGroupModification) {
-            if (items.length > 0) items.push((0, external_StageNodeTaskUtilities_cjs_namespaceObject.getDivider)());
-            items.push((0, external_StageNodeTaskUtilities_cjs_namespaceObject.getMenuItem)('remove-task', 'Delete task', ()=>onTaskGroupModification(GroupModificationUtils_cjs_namespaceObject.GroupModificationType.REMOVE_TASK, groupIndex, taskIndex)));
-        }
+        const replaceTaskMenuItem = generateReplaceTaskMenuItemForTask(taskId, false);
+        if (replaceTaskMenuItem) items.push(replaceTaskMenuItem);
+        const deleteTaskMenuItem = generateDeleteTaskMenuItemForTask(taskId);
+        if (deleteTaskMenuItem) items.push(deleteTaskMenuItem);
         return items;
     }, [
-        onReplaceTaskFromToolbox,
-        onTaskClick,
-        onTaskGroupModification,
-        setIsReplacingTask,
-        taskStateReference
+        generateReplaceTaskMenuItemForTask,
+        generateDeleteTaskMenuItemForTask
     ]);
     if (0 === adhocTasks.length) return null;
     return /*#__PURE__*/ (0, jsx_runtime_namespaceObject.jsxs)(external_StageNode_styles_cjs_namespaceObject.StageAdditionalTasksSection, {
@@ -70,7 +56,7 @@ const StageNodeAdhocTaskGroupsInner = ({ props, adhocTasks, isReadOnly, selected
                 })
             }),
             /*#__PURE__*/ (0, jsx_runtime_namespaceObject.jsx)(external_StageNode_styles_cjs_namespaceObject.StageTaskList, {
-                children: adhocTasks.map(({ task, groupIndex, taskIndex })=>{
+                children: adhocTasks.map(({ task })=>{
                     const taskExecution = execution?.taskStatus?.[task.id];
                     return /*#__PURE__*/ (0, jsx_runtime_namespaceObject.jsx)(external_AdhocTask_cjs_namespaceObject.AdhocTaskItem, {
                         task: task,
@@ -80,7 +66,7 @@ const StageNodeAdhocTaskGroupsInner = ({ props, adhocTasks, isReadOnly, selected
                         onTaskPlay: onTaskPlay,
                         isTaskLoading: loadingTaskIds?.has(task.id),
                         ...(onTaskGroupModification || onReplaceTaskFromToolbox) && !isReadOnly && {
-                            getContextMenuItems: ()=>getAdhocContextMenuItems(groupIndex, taskIndex, task.id)
+                            getContextMenuItems: ()=>getAdhocContextMenuItems(task.id)
                         }
                     }, task.id);
                 })

package/dist/canvas/components/StageNode/StageNodeAdhocTaskGroups.d.ts

@@ -1,13 +1,13 @@
-import { RefObject } from 'react';
-import type { StageNodeProps, StageTaskGroup, TaskStateReference } from './StageNode.types';
-export declare const StageNodeAdhocTaskGroups: import("react").MemoExoticComponent<({ props, adhocTasks, isReadOnly, selectedTaskId, taskStateReference, marginTop, handleTaskClick, setIsReplacingTask, }: {
+import type { NodeMenuItem } from '../NodeContextMenu';
+import type { StageNodeProps, StageTaskGroup } from './StageNode.types';
+export declare const StageNodeAdhocTaskGroups: import("react").MemoExoticComponent<({ props, adhocTasks, isReadOnly, selectedTaskId, marginTop, handleTaskClick, generateReplaceTaskMenuItemForTask, generateDeleteTaskMenuItemForTask, }: {
     props: StageNodeProps;
     adhocTasks: StageTaskGroup[];
     isReadOnly: boolean;
     selectedTaskId?: string;
-    taskStateReference: RefObject<TaskStateReference>;
     marginTop: string;
     handleTaskClick: (e: React.MouseEvent, taskElementId: string) => void;
-    setIsReplacingTask: (isReplacingTask: boolean) => void;
+    generateReplaceTaskMenuItemForTask: (taskId: string, isParallel: boolean) => NodeMenuItem | undefined;
+    generateDeleteTaskMenuItemForTask: (taskId: string) => NodeMenuItem | undefined;
 }) => import("react/jsx-runtime").JSX.Element | null>;
 //# sourceMappingURL=StageNodeAdhocTaskGroups.d.ts.map

package/dist/canvas/components/StageNode/StageNodeAdhocTaskGroups.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"StageNodeAdhocTaskGroups.d.ts","sourceRoot":"","sources":["../../../../src/canvas/components/StageNode/StageNodeAdhocTaskGroups.tsx"],"names":[],"mappings":"AAAA,OAAO,EAAQ,SAAS,EAAe,MAAM,OAAO,CAAC;AASrD,OAAO,KAAK,EAAE,cAAc,EAAE,cAAc,EAAE,kBAAkB,EAAE,MAAM,mBAAmB,CAAC;AAwG5F,eAAO,MAAM,wBAAwB,+JA5FlC;IACD,KAAK,EAAE,cAAc,CAAC;IACtB,UAAU,EAAE,cAAc,EAAE,CAAC;IAC7B,UAAU,EAAE,OAAO,CAAC;IACpB,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,kBAAkB,EAAE,SAAS,CAAC,kBAAkB,CAAC,CAAC;IAClD,SAAS,EAAE,MAAM,CAAC;IAClB,eAAe,EAAE,CAAC,CAAC,EAAE,KAAK,CAAC,UAAU,EAAE,aAAa,EAAE,MAAM,KAAK,IAAI,CAAC;IACtE,kBAAkB,EAAE,CAAC,eAAe,EAAE,OAAO,KAAK,IAAI,CAAC;CACxD,oDAmF0E,CAAC"}
+{"version":3,"file":"StageNodeAdhocTaskGroups.d.ts","sourceRoot":"","sources":["../../../../src/canvas/components/StageNode/StageNodeAdhocTaskGroups.tsx"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAOvD,OAAO,KAAK,EAAE,cAAc,EAAE,cAAc,EAAE,MAAM,mBAAmB,CAAC;AAqFxE,eAAO,MAAM,wBAAwB,8LA1ElC;IACD,KAAK,EAAE,cAAc,CAAC;IACtB,UAAU,EAAE,cAAc,EAAE,CAAC;IAC7B,UAAU,EAAE,OAAO,CAAC;IACpB,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,SAAS,EAAE,MAAM,CAAC;IAClB,eAAe,EAAE,CAAC,CAAC,EAAE,KAAK,CAAC,UAAU,EAAE,aAAa,EAAE,MAAM,KAAK,IAAI,CAAC;IACtE,kCAAkC,EAAE,CAClC,MAAM,EAAE,MAAM,EACd,UAAU,EAAE,OAAO,KAChB,YAAY,GAAG,SAAS,CAAC;IAC9B,iCAAiC,EAAE,CAAC,MAAM,EAAE,MAAM,KAAK,YAAY,GAAG,SAAS,CAAC;CACjF,oDA8D0E,CAAC"}

package/dist/canvas/components/StageNode/StageNodeAdhocTaskGroups.js

@@ -1,33 +1,19 @@
 import { jsx, jsxs } from "react/jsx-runtime";
 import { memo, useCallback } from "react";
-import { GroupModificationType } from "../../utils/GroupModificationUtils.js";
 import { AdhocTaskItem } from "./AdhocTask.js";
 import { StageAdditionalTasksHeaderSection, StageAdditionalTasksSection, StageTaskList } from "./StageNode.styles.js";
-import { getDivider, getMenuItem } from "./StageNodeTaskUtilities.js";
-const StageNodeAdhocTaskGroupsInner = ({ props, adhocTasks, isReadOnly, selectedTaskId, taskStateReference, marginTop, handleTaskClick, setIsReplacingTask })=>{
-    const { execution, onTaskClick, onTaskGroupModification, onReplaceTaskFromToolbox, onTaskPlay, loadingTaskIds } = props;
-    const getAdhocContextMenuItems = useCallback((groupIndex, taskIndex, taskId)=>{
+const StageNodeAdhocTaskGroupsInner = ({ props, adhocTasks, isReadOnly, selectedTaskId, marginTop, handleTaskClick, generateReplaceTaskMenuItemForTask, generateDeleteTaskMenuItemForTask })=>{
+    const { execution, onTaskGroupModification, onReplaceTaskFromToolbox, onTaskPlay, loadingTaskIds } = props;
+    const getAdhocContextMenuItems = useCallback((taskId)=>{
         const items = [];
-        if (onReplaceTaskFromToolbox) items.push(getMenuItem('replace-task', 'Replace task', ()=>{
-            taskStateReference.current = {
-                isParallel: false,
-                groupIndex,
-                taskIndex
-            };
-            onTaskClick?.(taskId);
-            setIsReplacingTask(true);
-        }));
-        if (onTaskGroupModification) {
-            if (items.length > 0) items.push(getDivider());
-            items.push(getMenuItem('remove-task', 'Delete task', ()=>onTaskGroupModification(GroupModificationType.REMOVE_TASK, groupIndex, taskIndex)));
-        }
+        const replaceTaskMenuItem = generateReplaceTaskMenuItemForTask(taskId, false);
+        if (replaceTaskMenuItem) items.push(replaceTaskMenuItem);
+        const deleteTaskMenuItem = generateDeleteTaskMenuItemForTask(taskId);
+        if (deleteTaskMenuItem) items.push(deleteTaskMenuItem);
         return items;
     }, [
-        onReplaceTaskFromToolbox,
-        onTaskClick,
-        onTaskGroupModification,
-        setIsReplacingTask,
-        taskStateReference
+        generateReplaceTaskMenuItemForTask,
+        generateDeleteTaskMenuItemForTask
     ]);
     if (0 === adhocTasks.length) return null;
     return /*#__PURE__*/ jsxs(StageAdditionalTasksSection, {
@@ -42,7 +28,7 @@ const StageNodeAdhocTaskGroupsInner = ({ props, adhocTasks, isReadOnly, selected
                 })
             }),
             /*#__PURE__*/ jsx(StageTaskList, {
-                children: adhocTasks.map(({ task, groupIndex, taskIndex })=>{
+                children: adhocTasks.map(({ task })=>{
                     const taskExecution = execution?.taskStatus?.[task.id];
                     return /*#__PURE__*/ jsx(AdhocTaskItem, {
                         task: task,
@@ -52,7 +38,7 @@ const StageNodeAdhocTaskGroupsInner = ({ props, adhocTasks, isReadOnly, selected
                         onTaskPlay: onTaskPlay,
                         isTaskLoading: loadingTaskIds?.has(task.id),
                         ...(onTaskGroupModification || onReplaceTaskFromToolbox) && !isReadOnly && {
-                            getContextMenuItems: ()=>getAdhocContextMenuItems(groupIndex, taskIndex, task.id)
+                            getContextMenuItems: ()=>getAdhocContextMenuItems(task.id)
                         }
                     }, task.id);
                 })

package/dist/canvas/components/StageNode/StageNodeAllTaskGroups.cjs

@@ -31,11 +31,13 @@ const apollo_core_namespaceObject = require("@uipath/apollo-core");
 const index_cjs_namespaceObject = require("../../layouts/index.cjs");
 const apollo_wind_namespaceObject = require("@uipath/apollo-wind");
 const external_react_namespaceObject = require("react");
+const external_utils_index_cjs_namespaceObject = require("../../utils/index.cjs");
 const useStageTasksByGroups_cjs_namespaceObject = require("./hooks/useStageTasksByGroups.cjs");
 const external_StageNode_styles_cjs_namespaceObject = require("./StageNode.styles.cjs");
 const external_StageNodeAdhocTaskGroups_cjs_namespaceObject = require("./StageNodeAdhocTaskGroups.cjs");
 const external_StageNodeEventDrivenTaskGroups_cjs_namespaceObject = require("./StageNodeEventDrivenTaskGroups.cjs");
 const external_StageNodeSequentialTaskGroups_cjs_namespaceObject = require("./StageNodeSequentialTaskGroups.cjs");
+const external_StageNodeTaskUtilities_cjs_namespaceObject = require("./StageNodeTaskUtilities.cjs");
 const StageNodeAllTaskGroupsInner = ({ props, isReadOnly, taskWidthStyle, taskStateReference, setSelectedNodeId, handleTaskAddClick, setIsReplacingTask })=>{
     const { id, stageDetails, onTaskAdd, onAddTaskFromToolbox, onTaskClick, onTaskGroupModification, onTaskReorder, onReplaceTaskFromToolbox } = props;
     const allTasks = (0, external_react_namespaceObject.useMemo)(()=>stageDetails?.tasks || [], [
@@ -66,6 +68,47 @@ const StageNodeAllTaskGroupsInner = ({ props, isReadOnly, taskWidthStyle, taskSt
         setSelectedNodeId,
         id
     ]);
+    const generateReplaceTaskMenuItemForTask = (0, external_react_namespaceObject.useCallback)((taskId, isParallel)=>{
+        if (!onReplaceTaskFromToolbox) return;
+        let groupIndex;
+        let taskIndex;
+        for (const [allTasksGroupIndex, group] of allTasks.entries())for (const [allTasksTaskIndex, task] of group.entries())if (task.id === taskId) {
+            groupIndex = allTasksGroupIndex;
+            taskIndex = allTasksTaskIndex;
+            break;
+        }
+        if (void 0 === groupIndex || void 0 === taskIndex) return;
+        return (0, external_StageNodeTaskUtilities_cjs_namespaceObject.getMenuItem)('replace-task', 'Replace task', ()=>{
+            taskStateReference.current = {
+                isParallel,
+                groupIndex,
+                taskIndex
+            };
+            onTaskClick?.(taskId);
+            setIsReplacingTask(true);
+        });
+    }, [
+        onReplaceTaskFromToolbox,
+        allTasks,
+        onTaskClick,
+        setIsReplacingTask,
+        taskStateReference
+    ]);
+    const generateDeleteTaskMenuItemForTask = (0, external_react_namespaceObject.useCallback)((taskId)=>{
+        if (!onTaskGroupModification) return;
+        let groupIndex;
+        let taskIndex;
+        for (const [allTasksGroupIndex, group] of allTasks.entries())for (const [allTasksTaskIndex, task] of group.entries())if (task.id === taskId) {
+            groupIndex = allTasksGroupIndex;
+            taskIndex = allTasksTaskIndex;
+            break;
+        }
+        if (void 0 === groupIndex || void 0 === taskIndex) return;
+        return (0, external_StageNodeTaskUtilities_cjs_namespaceObject.getMenuItem)('remove-task', 'Delete task', ()=>onTaskGroupModification(external_utils_index_cjs_namespaceObject.GroupModificationType.REMOVE_TASK, groupIndex, taskIndex));
+    }, [
+        allTasks,
+        onTaskGroupModification
+    ]);
     return /*#__PURE__*/ (0, jsx_runtime_namespaceObject.jsx)(external_StageNode_styles_cjs_namespaceObject.StageContent, {
         children: 0 === sequentialTaskGroups.length && 0 === adhocTaskGroups.length && 0 === eventDrivenTaskGroups.length ? /*#__PURE__*/ (0, jsx_runtime_namespaceObject.jsx)(index_cjs_namespaceObject.Column, {
             py: 2,
@@ -89,31 +132,31 @@ const StageNodeAllTaskGroupsInner = ({ props, isReadOnly, taskWidthStyle, taskSt
                     isReadOnly: isReadOnly,
                     selectedTaskId: selectedTaskId,
                     taskWidthStyle: taskWidthStyle,
-                    taskStateReference: taskStateReference,
                     hasContextMenu: hasContextMenu,
                     handleTaskClick: handleTaskClick,
-                    setIsReplacingTask: setIsReplacingTask,
-                    handleReorderSequentialTasks: handleReorderSequentialTasks
+                    handleReorderSequentialTasks: handleReorderSequentialTasks,
+                    allTasks: allTasks,
+                    generateReplaceTaskMenuItemForTask: generateReplaceTaskMenuItemForTask
                 }),
                 /*#__PURE__*/ (0, jsx_runtime_namespaceObject.jsx)(external_StageNodeEventDrivenTaskGroups_cjs_namespaceObject.StageNodeEventDrivenTaskGroups, {
                     props: props,
                     eventDrivenTasks: eventDrivenTasks,
                     isReadOnly: isReadOnly,
                     selectedTaskId: selectedTaskId,
-                    taskStateReference: taskStateReference,
                     marginTop: sequentialTaskGroups.length > 0 ? apollo_core_namespaceObject.Spacing.SpacingS : '0px',
                     handleTaskClick: handleTaskClick,
-                    setIsReplacingTask: setIsReplacingTask
+                    generateReplaceTaskMenuItemForTask: generateReplaceTaskMenuItemForTask,
+                    generateDeleteTaskMenuItemForTask: generateDeleteTaskMenuItemForTask
                 }),
                 /*#__PURE__*/ (0, jsx_runtime_namespaceObject.jsx)(external_StageNodeAdhocTaskGroups_cjs_namespaceObject.StageNodeAdhocTaskGroups, {
                     props: props,
                     adhocTasks: adhocTasks,
                     isReadOnly: isReadOnly,
                     selectedTaskId: selectedTaskId,
-                    taskStateReference: taskStateReference,
                     marginTop: sequentialTaskGroups.length > 0 || eventDrivenTaskGroups.length > 0 ? apollo_core_namespaceObject.Spacing.SpacingS : '0px',
                     handleTaskClick: handleTaskClick,
-                    setIsReplacingTask: setIsReplacingTask
+                    generateReplaceTaskMenuItemForTask: generateReplaceTaskMenuItemForTask,
+                    generateDeleteTaskMenuItemForTask: generateDeleteTaskMenuItemForTask
                 })
             ]
         })

package/dist/canvas/components/StageNode/StageNodeAllTaskGroups.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"StageNodeAllTaskGroups.d.ts","sourceRoot":"","sources":["../../../../src/canvas/components/StageNode/StageNodeAllTaskGroups.tsx"],"names":[],"mappings":"AAGA,OAAO,EAAE,aAAa,EAAQ,SAAS,EAAwB,MAAM,OAAO,CAAC;AAG7E,OAAO,KAAK,EAAE,cAAc,EAAiB,kBAAkB,EAAE,MAAM,mBAAmB,CAAC;AAiI3F,eAAO,MAAM,sBAAsB,8JApHhC;IACD,KAAK,EAAE,cAAc,CAAC;IACtB,UAAU,EAAE,OAAO,CAAC;IACpB,cAAc,CAAC,EAAE,aAAa,CAAC;IAC/B,kBAAkB,EAAE,SAAS,CAAC,kBAAkB,CAAC,CAAC;IAClD,iBAAiB,EAAE,CAAC,MAAM,EAAE,MAAM,KAAK,IAAI,CAAC;IAC5C,kBAAkB,EAAE,CAAC,KAAK,EAAE,KAAK,CAAC,UAAU,KAAK,IAAI,CAAC;IACtD,kBAAkB,EAAE,CAAC,eAAe,EAAE,OAAO,KAAK,IAAI,CAAC;CACxD,6CA4GsE,CAAC"}
+{"version":3,"file":"StageNodeAllTaskGroups.d.ts","sourceRoot":"","sources":["../../../../src/canvas/components/StageNode/StageNodeAllTaskGroups.tsx"],"names":[],"mappings":"AAGA,OAAO,EAAE,aAAa,EAAQ,SAAS,EAAwB,MAAM,OAAO,CAAC;AAI7E,OAAO,KAAK,EAAE,cAAc,EAAiB,kBAAkB,EAAE,MAAM,mBAAmB,CAAC;AAgM3F,eAAO,MAAM,sBAAsB,8JAlLhC;IACD,KAAK,EAAE,cAAc,CAAC;IACtB,UAAU,EAAE,OAAO,CAAC;IACpB,cAAc,CAAC,EAAE,aAAa,CAAC;IAC/B,kBAAkB,EAAE,SAAS,CAAC,kBAAkB,CAAC,CAAC;IAClD,iBAAiB,EAAE,CAAC,MAAM,EAAE,MAAM,KAAK,IAAI,CAAC;IAC5C,kBAAkB,EAAE,CAAC,KAAK,EAAE,KAAK,CAAC,UAAU,KAAK,IAAI,CAAC;IACtD,kBAAkB,EAAE,CAAC,eAAe,EAAE,OAAO,KAAK,IAAI,CAAC;CACxD,6CA0KsE,CAAC"}

package/dist/canvas/components/StageNode/StageNodeAllTaskGroups.js

@@ -3,11 +3,13 @@ import { Spacing } from "@uipath/apollo-core";
 import { Column } from "../../layouts/index.js";
 import { Button } from "@uipath/apollo-wind";
 import { memo, useCallback, useMemo } from "react";
+import { GroupModificationType } from "../../utils/index.js";
 import { useStageTasksByGroups } from "./hooks/useStageTasksByGroups.js";
 import { StageContent } from "./StageNode.styles.js";
 import { StageNodeAdhocTaskGroups } from "./StageNodeAdhocTaskGroups.js";
 import { StageNodeEventDrivenTaskGroups } from "./StageNodeEventDrivenTaskGroups.js";
 import { StageNodeSequentialTaskGroups } from "./StageNodeSequentialTaskGroups.js";
+import { getMenuItem } from "./StageNodeTaskUtilities.js";
 const StageNodeAllTaskGroupsInner = ({ props, isReadOnly, taskWidthStyle, taskStateReference, setSelectedNodeId, handleTaskAddClick, setIsReplacingTask })=>{
     const { id, stageDetails, onTaskAdd, onAddTaskFromToolbox, onTaskClick, onTaskGroupModification, onTaskReorder, onReplaceTaskFromToolbox } = props;
     const allTasks = useMemo(()=>stageDetails?.tasks || [], [
@@ -38,6 +40,47 @@ const StageNodeAllTaskGroupsInner = ({ props, isReadOnly, taskWidthStyle, taskSt
         setSelectedNodeId,
         id
     ]);
+    const generateReplaceTaskMenuItemForTask = useCallback((taskId, isParallel)=>{
+        if (!onReplaceTaskFromToolbox) return;
+        let groupIndex;
+        let taskIndex;
+        for (const [allTasksGroupIndex, group] of allTasks.entries())for (const [allTasksTaskIndex, task] of group.entries())if (task.id === taskId) {
+            groupIndex = allTasksGroupIndex;
+            taskIndex = allTasksTaskIndex;
+            break;
+        }
+        if (void 0 === groupIndex || void 0 === taskIndex) return;
+        return getMenuItem('replace-task', 'Replace task', ()=>{
+            taskStateReference.current = {
+                isParallel,
+                groupIndex,
+                taskIndex
+            };
+            onTaskClick?.(taskId);
+            setIsReplacingTask(true);
+        });
+    }, [
+        onReplaceTaskFromToolbox,
+        allTasks,
+        onTaskClick,
+        setIsReplacingTask,
+        taskStateReference
+    ]);
+    const generateDeleteTaskMenuItemForTask = useCallback((taskId)=>{
+        if (!onTaskGroupModification) return;
+        let groupIndex;
+        let taskIndex;
+        for (const [allTasksGroupIndex, group] of allTasks.entries())for (const [allTasksTaskIndex, task] of group.entries())if (task.id === taskId) {
+            groupIndex = allTasksGroupIndex;
+            taskIndex = allTasksTaskIndex;
+            break;
+        }
+        if (void 0 === groupIndex || void 0 === taskIndex) return;
+        return getMenuItem('remove-task', 'Delete task', ()=>onTaskGroupModification(GroupModificationType.REMOVE_TASK, groupIndex, taskIndex));
+    }, [
+        allTasks,
+        onTaskGroupModification
+    ]);
     return /*#__PURE__*/ jsx(StageContent, {
         children: 0 === sequentialTaskGroups.length && 0 === adhocTaskGroups.length && 0 === eventDrivenTaskGroups.length ? /*#__PURE__*/ jsx(Column, {
             py: 2,
@@ -61,31 +104,31 @@ const StageNodeAllTaskGroupsInner = ({ props, isReadOnly, taskWidthStyle, taskSt
                     isReadOnly: isReadOnly,
                     selectedTaskId: selectedTaskId,
                     taskWidthStyle: taskWidthStyle,
-                    taskStateReference: taskStateReference,
                     hasContextMenu: hasContextMenu,
                     handleTaskClick: handleTaskClick,
-                    setIsReplacingTask: setIsReplacingTask,
-                    handleReorderSequentialTasks: handleReorderSequentialTasks
+                    handleReorderSequentialTasks: handleReorderSequentialTasks,
+                    allTasks: allTasks,
+                    generateReplaceTaskMenuItemForTask: generateReplaceTaskMenuItemForTask
                 }),
                 /*#__PURE__*/ jsx(StageNodeEventDrivenTaskGroups, {
                     props: props,
                     eventDrivenTasks: eventDrivenTasks,
                     isReadOnly: isReadOnly,
                     selectedTaskId: selectedTaskId,
-                    taskStateReference: taskStateReference,
                     marginTop: sequentialTaskGroups.length > 0 ? Spacing.SpacingS : '0px',
                     handleTaskClick: handleTaskClick,
-                    setIsReplacingTask: setIsReplacingTask
+                    generateReplaceTaskMenuItemForTask: generateReplaceTaskMenuItemForTask,
+                    generateDeleteTaskMenuItemForTask: generateDeleteTaskMenuItemForTask
                 }),
                 /*#__PURE__*/ jsx(StageNodeAdhocTaskGroups, {
                     props: props,
                     adhocTasks: adhocTasks,
                     isReadOnly: isReadOnly,
                     selectedTaskId: selectedTaskId,
-                    taskStateReference: taskStateReference,
                     marginTop: sequentialTaskGroups.length > 0 || eventDrivenTaskGroups.length > 0 ? Spacing.SpacingS : '0px',
                     handleTaskClick: handleTaskClick,
-                    setIsReplacingTask: setIsReplacingTask
+                    generateReplaceTaskMenuItemForTask: generateReplaceTaskMenuItemForTask,
+                    generateDeleteTaskMenuItemForTask: generateDeleteTaskMenuItemForTask
                 })
             ]
         })

package/dist/canvas/components/StageNode/StageNodeEventDrivenTaskGroups.cjs

@@ -28,34 +28,20 @@ __webpack_require__.d(__webpack_exports__, {
 });
 const jsx_runtime_namespaceObject = require("react/jsx-runtime");
 const external_react_namespaceObject = require("react");
-const GroupModificationUtils_cjs_namespaceObject = require("../../utils/GroupModificationUtils.cjs");
 const external_EventDrivenTask_cjs_namespaceObject = require("./EventDrivenTask.cjs");
 const external_StageNode_styles_cjs_namespaceObject = require("./StageNode.styles.cjs");
-const external_StageNodeTaskUtilities_cjs_namespaceObject = require("./StageNodeTaskUtilities.cjs");
-const StageNodeEventDrivenTaskGroupsInner = ({ props, eventDrivenTasks, isReadOnly, selectedTaskId, taskStateReference, marginTop, handleTaskClick, setIsReplacingTask })=>{
-    const { execution, onTaskClick, onTaskGroupModification, onReplaceTaskFromToolbox, loadingTaskIds } = props;
-    const getEventDrivenContextMenuItems = (0, external_react_namespaceObject.useCallback)((groupIndex, taskIndex, taskId)=>{
+const StageNodeEventDrivenTaskGroupsInner = ({ props, eventDrivenTasks, isReadOnly, selectedTaskId, marginTop, handleTaskClick, generateReplaceTaskMenuItemForTask, generateDeleteTaskMenuItemForTask })=>{
+    const { execution, onTaskGroupModification, onReplaceTaskFromToolbox, loadingTaskIds } = props;
+    const getEventDrivenContextMenuItems = (0, external_react_namespaceObject.useCallback)((taskId)=>{
         const items = [];
-        if (onReplaceTaskFromToolbox) items.push((0, external_StageNodeTaskUtilities_cjs_namespaceObject.getMenuItem)('replace-task', 'Replace task', ()=>{
-            taskStateReference.current = {
-                isParallel: false,
-                groupIndex,
-                taskIndex
-            };
-            onTaskClick?.(taskId);
-            setIsReplacingTask(true);
-        }));
-        if (onTaskGroupModification) {
-            if (items.length > 0) items.push((0, external_StageNodeTaskUtilities_cjs_namespaceObject.getDivider)());
-            items.push((0, external_StageNodeTaskUtilities_cjs_namespaceObject.getMenuItem)('remove-task', 'Delete task', ()=>onTaskGroupModification(GroupModificationUtils_cjs_namespaceObject.GroupModificationType.REMOVE_TASK, groupIndex, taskIndex)));
-        }
+        const replaceTaskMenuItem = generateReplaceTaskMenuItemForTask(taskId, false);
+        if (replaceTaskMenuItem) items.push(replaceTaskMenuItem);
+        const deleteTaskMenuItem = generateDeleteTaskMenuItemForTask(taskId);
+        if (deleteTaskMenuItem) items.push(deleteTaskMenuItem);
         return items;
     }, [
-        onReplaceTaskFromToolbox,
-        onTaskClick,
-        onTaskGroupModification,
-        setIsReplacingTask,
-        taskStateReference
+        generateReplaceTaskMenuItemForTask,
+        generateDeleteTaskMenuItemForTask
     ]);
     if (0 === eventDrivenTasks.length) return null;
     return /*#__PURE__*/ (0, jsx_runtime_namespaceObject.jsxs)(external_StageNode_styles_cjs_namespaceObject.StageAdditionalTasksSection, {
@@ -70,7 +56,7 @@ const StageNodeEventDrivenTaskGroupsInner = ({ props, eventDrivenTasks, isReadOn
                 })
             }),
             /*#__PURE__*/ (0, jsx_runtime_namespaceObject.jsx)(external_StageNode_styles_cjs_namespaceObject.StageTaskList, {
-                children: eventDrivenTasks.map(({ task, groupIndex, taskIndex })=>{
+                children: eventDrivenTasks.map(({ task })=>{
                     const taskExecution = execution?.taskStatus?.[task.id];
                     return /*#__PURE__*/ (0, jsx_runtime_namespaceObject.jsx)(external_EventDrivenTask_cjs_namespaceObject.EventDrivenTaskItem, {
                         task: task,
@@ -79,7 +65,7 @@ const StageNodeEventDrivenTaskGroupsInner = ({ props, eventDrivenTasks, isReadOn
                         onTaskClick: handleTaskClick,
                         isTaskLoading: loadingTaskIds?.has(task.id),
                         ...(onTaskGroupModification || onReplaceTaskFromToolbox) && !isReadOnly && {
-                            getContextMenuItems: ()=>getEventDrivenContextMenuItems(groupIndex, taskIndex, task.id)
+                            getContextMenuItems: ()=>getEventDrivenContextMenuItems(task.id)
                         }
                     }, task.id);
                 })

package/dist/canvas/components/StageNode/StageNodeEventDrivenTaskGroups.d.ts

@@ -1,13 +1,13 @@
-import { RefObject } from 'react';
-import type { StageNodeProps, StageTaskGroup, TaskStateReference } from './StageNode.types';
-export declare const StageNodeEventDrivenTaskGroups: import("react").MemoExoticComponent<({ props, eventDrivenTasks, isReadOnly, selectedTaskId, taskStateReference, marginTop, handleTaskClick, setIsReplacingTask, }: {
+import type { NodeMenuItem } from '../NodeContextMenu';
+import type { StageNodeProps, StageTaskGroup } from './StageNode.types';
+export declare const StageNodeEventDrivenTaskGroups: import("react").MemoExoticComponent<({ props, eventDrivenTasks, isReadOnly, selectedTaskId, marginTop, handleTaskClick, generateReplaceTaskMenuItemForTask, generateDeleteTaskMenuItemForTask, }: {
     props: StageNodeProps;
     eventDrivenTasks: StageTaskGroup[];
     isReadOnly: boolean;
     selectedTaskId?: string;
-    taskStateReference: RefObject<TaskStateReference>;
     marginTop: string;
     handleTaskClick: (e: React.MouseEvent, taskElementId: string) => void;
-    setIsReplacingTask: (isReplacingTask: boolean) => void;
+    generateReplaceTaskMenuItemForTask: (taskId: string, isParallel: boolean) => NodeMenuItem | undefined;
+    generateDeleteTaskMenuItemForTask: (taskId: string) => NodeMenuItem | undefined;
 }) => import("react/jsx-runtime").JSX.Element | null>;
 //# sourceMappingURL=StageNodeEventDrivenTaskGroups.d.ts.map

package/dist/canvas/components/StageNode/StageNodeEventDrivenTaskGroups.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"StageNodeEventDrivenTaskGroups.d.ts","sourceRoot":"","sources":["../../../../src/canvas/components/StageNode/StageNodeEventDrivenTaskGroups.tsx"],"names":[],"mappings":"AAAA,OAAO,EAAQ,SAAS,EAAe,MAAM,OAAO,CAAC;AASrD,OAAO,KAAK,EAAE,cAAc,EAAE,cAAc,EAAE,kBAAkB,EAAE,MAAM,mBAAmB,CAAC;AAsG5F,eAAO,MAAM,8BAA8B,qKA1FxC;IACD,KAAK,EAAE,cAAc,CAAC;IACtB,gBAAgB,EAAE,cAAc,EAAE,CAAC;IACnC,UAAU,EAAE,OAAO,CAAC;IACpB,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,kBAAkB,EAAE,SAAS,CAAC,kBAAkB,CAAC,CAAC;IAClD,SAAS,EAAE,MAAM,CAAC;IAClB,eAAe,EAAE,CAAC,CAAC,EAAE,KAAK,CAAC,UAAU,EAAE,aAAa,EAAE,MAAM,KAAK,IAAI,CAAC;IACtE,kBAAkB,EAAE,CAAC,eAAe,EAAE,OAAO,KAAK,IAAI,CAAC;CACxD,oDAiFsF,CAAC"}
+{"version":3,"file":"StageNodeEventDrivenTaskGroups.d.ts","sourceRoot":"","sources":["../../../../src/canvas/components/StageNode/StageNodeEventDrivenTaskGroups.tsx"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAOvD,OAAO,KAAK,EAAE,cAAc,EAAE,cAAc,EAAE,MAAM,mBAAmB,CAAC;AA8ExE,eAAO,MAAM,8BAA8B,oMAnExC;IACD,KAAK,EAAE,cAAc,CAAC;IACtB,gBAAgB,EAAE,cAAc,EAAE,CAAC;IACnC,UAAU,EAAE,OAAO,CAAC;IACpB,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,SAAS,EAAE,MAAM,CAAC;IAClB,eAAe,EAAE,CAAC,CAAC,EAAE,KAAK,CAAC,UAAU,EAAE,aAAa,EAAE,MAAM,KAAK,IAAI,CAAC;IACtE,kCAAkC,EAAE,CAClC,MAAM,EAAE,MAAM,EACd,UAAU,EAAE,OAAO,KAChB,YAAY,GAAG,SAAS,CAAC;IAC9B,iCAAiC,EAAE,CAAC,MAAM,EAAE,MAAM,KAAK,YAAY,GAAG,SAAS,CAAC;CACjF,oDAuDsF,CAAC"}

package/dist/canvas/components/StageNode/StageNodeEventDrivenTaskGroups.js

@@ -1,33 +1,19 @@
 import { jsx, jsxs } from "react/jsx-runtime";
 import { memo, useCallback } from "react";
-import { GroupModificationType } from "../../utils/GroupModificationUtils.js";
 import { EventDrivenTaskItem } from "./EventDrivenTask.js";
 import { StageAdditionalTasksHeaderSection, StageAdditionalTasksSection, StageTaskList } from "./StageNode.styles.js";
-import { getDivider, getMenuItem } from "./StageNodeTaskUtilities.js";
-const StageNodeEventDrivenTaskGroupsInner = ({ props, eventDrivenTasks, isReadOnly, selectedTaskId, taskStateReference, marginTop, handleTaskClick, setIsReplacingTask })=>{
-    const { execution, onTaskClick, onTaskGroupModification, onReplaceTaskFromToolbox, loadingTaskIds } = props;
-    const getEventDrivenContextMenuItems = useCallback((groupIndex, taskIndex, taskId)=>{
+const StageNodeEventDrivenTaskGroupsInner = ({ props, eventDrivenTasks, isReadOnly, selectedTaskId, marginTop, handleTaskClick, generateReplaceTaskMenuItemForTask, generateDeleteTaskMenuItemForTask })=>{
+    const { execution, onTaskGroupModification, onReplaceTaskFromToolbox, loadingTaskIds } = props;
+    const getEventDrivenContextMenuItems = useCallback((taskId)=>{
         const items = [];
-        if (onReplaceTaskFromToolbox) items.push(getMenuItem('replace-task', 'Replace task', ()=>{
-            taskStateReference.current = {
-                isParallel: false,
-                groupIndex,
-                taskIndex
-            };
-            onTaskClick?.(taskId);
-            setIsReplacingTask(true);
-        }));
-        if (onTaskGroupModification) {
-            if (items.length > 0) items.push(getDivider());
-            items.push(getMenuItem('remove-task', 'Delete task', ()=>onTaskGroupModification(GroupModificationType.REMOVE_TASK, groupIndex, taskIndex)));
-        }
+        const replaceTaskMenuItem = generateReplaceTaskMenuItemForTask(taskId, false);
+        if (replaceTaskMenuItem) items.push(replaceTaskMenuItem);
+        const deleteTaskMenuItem = generateDeleteTaskMenuItemForTask(taskId);
+        if (deleteTaskMenuItem) items.push(deleteTaskMenuItem);
         return items;
     }, [
-        onReplaceTaskFromToolbox,
-        onTaskClick,
-        onTaskGroupModification,
-        setIsReplacingTask,
-        taskStateReference
+        generateReplaceTaskMenuItemForTask,
+        generateDeleteTaskMenuItemForTask
     ]);
     if (0 === eventDrivenTasks.length) return null;
     return /*#__PURE__*/ jsxs(StageAdditionalTasksSection, {
@@ -42,7 +28,7 @@ const StageNodeEventDrivenTaskGroupsInner = ({ props, eventDrivenTasks, isReadOn
                 })
             }),
             /*#__PURE__*/ jsx(StageTaskList, {
-                children: eventDrivenTasks.map(({ task, groupIndex, taskIndex })=>{
+                children: eventDrivenTasks.map(({ task })=>{
                     const taskExecution = execution?.taskStatus?.[task.id];
                     return /*#__PURE__*/ jsx(EventDrivenTaskItem, {
                         task: task,
@@ -51,7 +37,7 @@ const StageNodeEventDrivenTaskGroupsInner = ({ props, eventDrivenTasks, isReadOn
                         onTaskClick: handleTaskClick,
                         isTaskLoading: loadingTaskIds?.has(task.id),
                         ...(onTaskGroupModification || onReplaceTaskFromToolbox) && !isReadOnly && {
-                            getContextMenuItems: ()=>getEventDrivenContextMenuItems(groupIndex, taskIndex, task.id)
+                            getContextMenuItems: ()=>getEventDrivenContextMenuItems(task.id)
                         }
                     }, task.id);
                 })