@uipath/apollo-react 4.12.1 → 4.12.2

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);
                 })

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

@@ -38,8 +38,8 @@ const useStageTaskDragHandler_cjs_namespaceObject = require("./hooks/useStageTas
 const external_StageNode_styles_cjs_namespaceObject = require("./StageNode.styles.cjs");
 const external_StageNodeTaskUtilities_cjs_namespaceObject = require("./StageNodeTaskUtilities.cjs");
 const external_StageTaskDragOverlay_cjs_namespaceObject = require("./StageTaskDragOverlay.cjs");
-const StageNodeSequentialTaskGroups = ({ props, sequentialTaskGroups, sequentialTasks, isReadOnly, selectedTaskId, taskWidthStyle, taskStateReference, hasContextMenu, handleTaskClick, setIsReplacingTask, handleReorderSequentialTasks })=>{
-    const { execution, onTaskClick, onTaskGroupModification, onTaskReorder, onReplaceTaskFromToolbox, hideParallelOptions, loadingTaskIds } = props;
+const StageNodeSequentialTaskGroups = ({ props, sequentialTaskGroups, sequentialTasks, allTasks, isReadOnly, selectedTaskId, taskWidthStyle, hasContextMenu, handleTaskClick, handleReorderSequentialTasks, generateReplaceTaskMenuItemForTask })=>{
+    const { execution, onTaskGroupModification, onTaskReorder, hideParallelOptions, loadingTaskIds } = props;
     const sequentialTaskIds = (0, external_react_namespaceObject.useMemo)(()=>sequentialTasks.map(({ task })=>task.id), [
         sequentialTasks
     ]);
@@ -62,25 +62,34 @@ const StageNodeSequentialTaskGroups = ({ props, sequentialTaskGroups, sequential
         onTaskGroupModification,
         sequentialTaskGroups
     ]);
-    const buildContextMenuItems = (0, external_react_namespaceObject.useCallback)((groupIndex, taskIndex)=>{
+    const buildContextMenuItems = (0, external_react_namespaceObject.useCallback)((groupIndex, _, taskId)=>{
         const taskGroup = sequentialTaskGroups[groupIndex] ?? [];
         const isParallel = taskGroup.length > 1;
         const items = [];
-        if (onReplaceTaskFromToolbox) {
-            items.push((0, external_StageNodeTaskUtilities_cjs_namespaceObject.getMenuItem)('replace-task', 'Replace task', ()=>{
-                taskStateReference.current = {
-                    isParallel,
-                    groupIndex,
-                    taskIndex
-                };
-                const taskId = taskGroup[taskIndex]?.id;
-                if (taskId) onTaskClick?.(taskId);
-                setIsReplacingTask(true);
-            }));
+        const replaceTaskMenuItem = generateReplaceTaskMenuItemForTask(taskId, isParallel);
+        if (replaceTaskMenuItem) {
+            items.push(replaceTaskMenuItem);
             items.push((0, external_StageNodeTaskUtilities_cjs_namespaceObject.getDivider)());
         }
-        if (onTaskGroupModification) {
-            const reGroupOptions = (0, external_StageNodeTaskUtilities_cjs_namespaceObject.getContextMenuItems)(isParallel, groupIndex, sequentialTaskGroups.length, taskIndex, taskGroup.length, (sequentialTaskGroups[groupIndex - 1]?.length ?? 0) > 1, (sequentialTaskGroups[groupIndex + 1]?.length ?? 0) > 1, handleTaskRegroup, hideParallelOptions);
+        let groupIndexInAllTasks;
+        let taskIndexInAllTasks;
+        for (const [allTasksGroupIndex, group] of allTasks.entries())for (const [allTasksTaskIndex, task] of group.entries())if (task.id === taskId) {
+            groupIndexInAllTasks = allTasksGroupIndex;
+            taskIndexInAllTasks = allTasksTaskIndex;
+            break;
+        }
+        if (onTaskGroupModification && void 0 !== groupIndexInAllTasks && void 0 !== taskIndexInAllTasks) {
+            const reGroupOptions = (0, external_StageNodeTaskUtilities_cjs_namespaceObject.getContextMenuItems)({
+                isParallelGroup: isParallel,
+                groupIndex,
+                tasksLength: sequentialTaskGroups.length,
+                groupIndexInAllTasks,
+                taskIndexInAllTasks,
+                isAboveParallel: (sequentialTaskGroups[groupIndex - 1]?.length ?? 0) > 1,
+                isBelowParallel: (sequentialTaskGroups[groupIndex + 1]?.length ?? 0) > 1,
+                reGroupTaskFunction: handleTaskRegroup,
+                hideParallelOptions
+            });
             return [
                 ...items,
                 ...reGroupOptions
@@ -88,14 +97,12 @@ const StageNodeSequentialTaskGroups = ({ props, sequentialTaskGroups, sequential
         }
         return items;
     }, [
-        onReplaceTaskFromToolbox,
-        onTaskClick,
         onTaskGroupModification,
         sequentialTaskGroups,
         hideParallelOptions,
         handleTaskRegroup,
-        setIsReplacingTask,
-        taskStateReference
+        generateReplaceTaskMenuItemForTask,
+        allTasks
     ]);
     const sensors = (0, core_namespaceObject.useSensors)((0, core_namespaceObject.useSensor)(core_namespaceObject.PointerSensor, {
         activationConstraint: {

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

@@ -1,16 +1,17 @@
-import { CSSProperties, RefObject } from 'react';
-import type { StageNodeProps, StageTaskGroup, StageTaskItem, TaskStateReference } from './StageNode.types';
-export declare const StageNodeSequentialTaskGroups: ({ props, sequentialTaskGroups, sequentialTasks, isReadOnly, selectedTaskId, taskWidthStyle, taskStateReference, hasContextMenu, handleTaskClick, setIsReplacingTask, handleReorderSequentialTasks, }: {
+import { CSSProperties } from 'react';
+import type { NodeMenuItem } from '../NodeContextMenu';
+import type { StageNodeProps, StageTaskGroup, StageTaskItem } from './StageNode.types';
+export declare const StageNodeSequentialTaskGroups: ({ props, sequentialTaskGroups, sequentialTasks, allTasks, isReadOnly, selectedTaskId, taskWidthStyle, hasContextMenu, handleTaskClick, handleReorderSequentialTasks, generateReplaceTaskMenuItemForTask, }: {
     props: StageNodeProps;
     sequentialTaskGroups: StageTaskItem[][];
     sequentialTasks: StageTaskGroup[];
+    allTasks: StageTaskItem[][];
     isReadOnly: boolean;
     selectedTaskId?: string;
     taskWidthStyle?: CSSProperties;
-    taskStateReference: RefObject<TaskStateReference>;
     hasContextMenu: boolean;
     handleTaskClick: (e: React.MouseEvent, taskElementId: string) => void;
-    setIsReplacingTask: (isReplacingTask: boolean) => void;
     handleReorderSequentialTasks: (newTasks: StageTaskItem[][]) => void;
+    generateReplaceTaskMenuItemForTask: (taskId: string, isParallel: boolean) => NodeMenuItem | undefined;
 }) => import("react/jsx-runtime").JSX.Element | null;
 //# sourceMappingURL=StageNodeSequentialTaskGroups.d.ts.map

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

@@ -1 +1 @@
-{"version":3,"file":"StageNodeSequentialTaskGroups.d.ts","sourceRoot":"","sources":["../../../../src/canvas/components/StageNode/StageNodeSequentialTaskGroups.tsx"],"names":[],"mappings":"AAeA,OAAO,EAAE,aAAa,EAAE,SAAS,EAAwB,MAAM,OAAO,CAAC;AAevE,OAAO,KAAK,EACV,cAAc,EACd,cAAc,EACd,aAAa,EACb,kBAAkB,EACnB,MAAM,mBAAmB,CAAC;AAI3B,eAAO,MAAM,6BAA6B,GAAI,sMAY3C;IACD,KAAK,EAAE,cAAc,CAAC;IACtB,oBAAoB,EAAE,aAAa,EAAE,EAAE,CAAC;IACxC,eAAe,EAAE,cAAc,EAAE,CAAC;IAClC,UAAU,EAAE,OAAO,CAAC;IACpB,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,cAAc,CAAC,EAAE,aAAa,CAAC;IAC/B,kBAAkB,EAAE,SAAS,CAAC,kBAAkB,CAAC,CAAC;IAClD,cAAc,EAAE,OAAO,CAAC;IACxB,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;IACvD,4BAA4B,EAAE,CAAC,QAAQ,EAAE,aAAa,EAAE,EAAE,KAAK,IAAI,CAAC;CACrE,mDAqLA,CAAC"}
+{"version":3,"file":"StageNodeSequentialTaskGroups.d.ts","sourceRoot":"","sources":["../../../../src/canvas/components/StageNode/StageNodeSequentialTaskGroups.tsx"],"names":[],"mappings":"AAeA,OAAO,EAAE,aAAa,EAAwB,MAAM,OAAO,CAAC;AAM5D,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AASvD,OAAO,KAAK,EAAE,cAAc,EAAE,cAAc,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AAIvF,eAAO,MAAM,6BAA6B,GAAI,4MAY3C;IACD,KAAK,EAAE,cAAc,CAAC;IACtB,oBAAoB,EAAE,aAAa,EAAE,EAAE,CAAC;IACxC,eAAe,EAAE,cAAc,EAAE,CAAC;IAClC,QAAQ,EAAE,aAAa,EAAE,EAAE,CAAC;IAC5B,UAAU,EAAE,OAAO,CAAC;IACpB,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,cAAc,CAAC,EAAE,aAAa,CAAC;IAC/B,cAAc,EAAE,OAAO,CAAC;IACxB,eAAe,EAAE,CAAC,CAAC,EAAE,KAAK,CAAC,UAAU,EAAE,aAAa,EAAE,MAAM,KAAK,IAAI,CAAC;IACtE,4BAA4B,EAAE,CAAC,QAAQ,EAAE,aAAa,EAAE,EAAE,KAAK,IAAI,CAAC;IACpE,kCAAkC,EAAE,CAClC,MAAM,EAAE,MAAM,EACd,UAAU,EAAE,OAAO,KAChB,YAAY,GAAG,SAAS,CAAC;CAC/B,mDAiLA,CAAC"}

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

@@ -8,10 +8,10 @@ import { GroupModificationType, moveGroupDown, moveGroupUp } from "../../utils/G
 import { DraggableTask } from "./DraggableTask.js";
 import { useStageTaskDragHandler } from "./hooks/useStageTaskDragHandler.js";
 import { StageParallelBracket, StageParallelLabel, StageTaskGroupContainer, StageTaskList } from "./StageNode.styles.js";
-import { getContextMenuItems, getDivider, getMenuItem } from "./StageNodeTaskUtilities.js";
+import { getContextMenuItems, getDivider } from "./StageNodeTaskUtilities.js";
 import { StageTaskDragOverlay } from "./StageTaskDragOverlay.js";
-const StageNodeSequentialTaskGroups = ({ props, sequentialTaskGroups, sequentialTasks, isReadOnly, selectedTaskId, taskWidthStyle, taskStateReference, hasContextMenu, handleTaskClick, setIsReplacingTask, handleReorderSequentialTasks })=>{
-    const { execution, onTaskClick, onTaskGroupModification, onTaskReorder, onReplaceTaskFromToolbox, hideParallelOptions, loadingTaskIds } = props;
+const StageNodeSequentialTaskGroups = ({ props, sequentialTaskGroups, sequentialTasks, allTasks, isReadOnly, selectedTaskId, taskWidthStyle, hasContextMenu, handleTaskClick, handleReorderSequentialTasks, generateReplaceTaskMenuItemForTask })=>{
+    const { execution, onTaskGroupModification, onTaskReorder, hideParallelOptions, loadingTaskIds } = props;
     const sequentialTaskIds = useMemo(()=>sequentialTasks.map(({ task })=>task.id), [
         sequentialTasks
     ]);
@@ -34,25 +34,34 @@ const StageNodeSequentialTaskGroups = ({ props, sequentialTaskGroups, sequential
         onTaskGroupModification,
         sequentialTaskGroups
     ]);
-    const buildContextMenuItems = useCallback((groupIndex, taskIndex)=>{
+    const buildContextMenuItems = useCallback((groupIndex, _, taskId)=>{
         const taskGroup = sequentialTaskGroups[groupIndex] ?? [];
         const isParallel = taskGroup.length > 1;
         const items = [];
-        if (onReplaceTaskFromToolbox) {
-            items.push(getMenuItem('replace-task', 'Replace task', ()=>{
-                taskStateReference.current = {
-                    isParallel,
-                    groupIndex,
-                    taskIndex
-                };
-                const taskId = taskGroup[taskIndex]?.id;
-                if (taskId) onTaskClick?.(taskId);
-                setIsReplacingTask(true);
-            }));
+        const replaceTaskMenuItem = generateReplaceTaskMenuItemForTask(taskId, isParallel);
+        if (replaceTaskMenuItem) {
+            items.push(replaceTaskMenuItem);
             items.push(getDivider());
         }
-        if (onTaskGroupModification) {
-            const reGroupOptions = getContextMenuItems(isParallel, groupIndex, sequentialTaskGroups.length, taskIndex, taskGroup.length, (sequentialTaskGroups[groupIndex - 1]?.length ?? 0) > 1, (sequentialTaskGroups[groupIndex + 1]?.length ?? 0) > 1, handleTaskRegroup, hideParallelOptions);
+        let groupIndexInAllTasks;
+        let taskIndexInAllTasks;
+        for (const [allTasksGroupIndex, group] of allTasks.entries())for (const [allTasksTaskIndex, task] of group.entries())if (task.id === taskId) {
+            groupIndexInAllTasks = allTasksGroupIndex;
+            taskIndexInAllTasks = allTasksTaskIndex;
+            break;
+        }
+        if (onTaskGroupModification && void 0 !== groupIndexInAllTasks && void 0 !== taskIndexInAllTasks) {
+            const reGroupOptions = getContextMenuItems({
+                isParallelGroup: isParallel,
+                groupIndex,
+                tasksLength: sequentialTaskGroups.length,
+                groupIndexInAllTasks,
+                taskIndexInAllTasks,
+                isAboveParallel: (sequentialTaskGroups[groupIndex - 1]?.length ?? 0) > 1,
+                isBelowParallel: (sequentialTaskGroups[groupIndex + 1]?.length ?? 0) > 1,
+                reGroupTaskFunction: handleTaskRegroup,
+                hideParallelOptions
+            });
             return [
                 ...items,
                 ...reGroupOptions
@@ -60,14 +69,12 @@ const StageNodeSequentialTaskGroups = ({ props, sequentialTaskGroups, sequential
         }
         return items;
     }, [
-        onReplaceTaskFromToolbox,
-        onTaskClick,
         onTaskGroupModification,
         sequentialTaskGroups,
         hideParallelOptions,
         handleTaskRegroup,
-        setIsReplacingTask,
-        taskStateReference
+        generateReplaceTaskMenuItemForTask,
+        allTasks
     ]);
     const sensors = useSensors(useSensor(PointerSensor, {
         activationConstraint: {

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

@@ -30,18 +30,18 @@ __webpack_require__.d(__webpack_exports__, {
     getContextMenuItems: ()=>getContextMenuItems
 });
 const GroupModificationUtils_cjs_namespaceObject = require("../../utils/GroupModificationUtils.cjs");
-const getContextMenuItems = (isParallelGroup, groupIndex, tasksLength, taskIndex, _taskGroupLength, isAboveParallel, isBelowParallel, reGroupTaskFunction, hideParallelOptions = false)=>{
+const getContextMenuItems = ({ isParallelGroup, groupIndex, tasksLength, groupIndexInAllTasks, taskIndexInAllTasks, isAboveParallel, isBelowParallel, reGroupTaskFunction, hideParallelOptions = false })=>{
     const CONTEXT_MENU_ITEMS = {
-        MOVE_UP: getMenuItem('move-up', 'Move up', ()=>reGroupTaskFunction(GroupModificationUtils_cjs_namespaceObject.GroupModificationType.TASK_GROUP_UP, groupIndex, taskIndex)),
-        MOVE_DOWN: getMenuItem('move-down', 'Move down', ()=>reGroupTaskFunction(GroupModificationUtils_cjs_namespaceObject.GroupModificationType.TASK_GROUP_DOWN, groupIndex, taskIndex)),
-        UNGROUP_ALL: getMenuItem('ungroup', 'Ungroup parallel tasks', ()=>reGroupTaskFunction(GroupModificationUtils_cjs_namespaceObject.GroupModificationType.UNGROUP_ALL_TASKS, groupIndex, taskIndex)),
-        SPLIT_TASK: getMenuItem('split', 'Remove from parallel group', ()=>reGroupTaskFunction(GroupModificationUtils_cjs_namespaceObject.GroupModificationType.SPLIT_GROUP, groupIndex, taskIndex)),
-        REMOVE_GROUP: getMenuItem('remove-group', 'Remove group from stage', ()=>reGroupTaskFunction(GroupModificationUtils_cjs_namespaceObject.GroupModificationType.REMOVE_GROUP, groupIndex, taskIndex)),
-        REMOVE_TASK: getMenuItem('remove-task', 'Delete task', ()=>reGroupTaskFunction(GroupModificationUtils_cjs_namespaceObject.GroupModificationType.REMOVE_TASK, groupIndex, taskIndex)),
-        CREATE_PARALLEL_GROUP_ABOVE: getMenuItem('group-with-up', 'Create parallel group with task above', ()=>reGroupTaskFunction(GroupModificationUtils_cjs_namespaceObject.GroupModificationType.MERGE_GROUP_UP, groupIndex, taskIndex)),
-        CREATE_PARALLEL_GROUP_BELOW: getMenuItem('group-with-down', 'Create parallel group with task below', ()=>reGroupTaskFunction(GroupModificationUtils_cjs_namespaceObject.GroupModificationType.MERGE_GROUP_DOWN, groupIndex, taskIndex)),
-        ADD_TO_PARALLEL_GROUP_ABOVE: getMenuItem('add-to-group-with-up', 'Add task to parallel group above', ()=>reGroupTaskFunction(GroupModificationUtils_cjs_namespaceObject.GroupModificationType.MERGE_GROUP_UP, groupIndex, taskIndex)),
-        ADD_TO_PARALLEL_GROUP_BELOW: getMenuItem('add-to-group-with-down', 'Add task to parallel group below', ()=>reGroupTaskFunction(GroupModificationUtils_cjs_namespaceObject.GroupModificationType.MERGE_GROUP_DOWN, groupIndex, taskIndex)),
+        MOVE_UP: getMenuItem('move-up', 'Move up', ()=>reGroupTaskFunction(GroupModificationUtils_cjs_namespaceObject.GroupModificationType.TASK_GROUP_UP, groupIndexInAllTasks, taskIndexInAllTasks)),
+        MOVE_DOWN: getMenuItem('move-down', 'Move down', ()=>reGroupTaskFunction(GroupModificationUtils_cjs_namespaceObject.GroupModificationType.TASK_GROUP_DOWN, groupIndexInAllTasks, taskIndexInAllTasks)),
+        UNGROUP_ALL: getMenuItem('ungroup', 'Ungroup parallel tasks', ()=>reGroupTaskFunction(GroupModificationUtils_cjs_namespaceObject.GroupModificationType.UNGROUP_ALL_TASKS, groupIndexInAllTasks, taskIndexInAllTasks)),
+        SPLIT_TASK: getMenuItem('split', 'Remove from parallel group', ()=>reGroupTaskFunction(GroupModificationUtils_cjs_namespaceObject.GroupModificationType.SPLIT_GROUP, groupIndexInAllTasks, taskIndexInAllTasks)),
+        REMOVE_GROUP: getMenuItem('remove-group', 'Remove group from stage', ()=>reGroupTaskFunction(GroupModificationUtils_cjs_namespaceObject.GroupModificationType.REMOVE_GROUP, groupIndexInAllTasks, taskIndexInAllTasks)),
+        REMOVE_TASK: getMenuItem('remove-task', 'Delete task', ()=>reGroupTaskFunction(GroupModificationUtils_cjs_namespaceObject.GroupModificationType.REMOVE_TASK, groupIndexInAllTasks, taskIndexInAllTasks)),
+        CREATE_PARALLEL_GROUP_ABOVE: getMenuItem('group-with-up', 'Create parallel group with task above', ()=>reGroupTaskFunction(GroupModificationUtils_cjs_namespaceObject.GroupModificationType.MERGE_GROUP_UP, groupIndexInAllTasks, taskIndexInAllTasks)),
+        CREATE_PARALLEL_GROUP_BELOW: getMenuItem('group-with-down', 'Create parallel group with task below', ()=>reGroupTaskFunction(GroupModificationUtils_cjs_namespaceObject.GroupModificationType.MERGE_GROUP_DOWN, groupIndexInAllTasks, taskIndexInAllTasks)),
+        ADD_TO_PARALLEL_GROUP_ABOVE: getMenuItem('add-to-group-with-up', 'Add task to parallel group above', ()=>reGroupTaskFunction(GroupModificationUtils_cjs_namespaceObject.GroupModificationType.MERGE_GROUP_UP, groupIndexInAllTasks, taskIndexInAllTasks)),
+        ADD_TO_PARALLEL_GROUP_BELOW: getMenuItem('add-to-group-with-down', 'Add task to parallel group below', ()=>reGroupTaskFunction(GroupModificationUtils_cjs_namespaceObject.GroupModificationType.MERGE_GROUP_DOWN, groupIndexInAllTasks, taskIndexInAllTasks)),
         DIVIDER: getDivider()
     };
     const items = [];

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

@@ -1,7 +1,17 @@
 import type { ReactNode } from 'react';
 import { GroupModificationType } from '../../utils/GroupModificationUtils';
 import type { NodeMenuAction, NodeMenuItem } from '../NodeContextMenu';
-export declare const getContextMenuItems: (isParallelGroup: boolean, groupIndex: number, tasksLength: number, taskIndex: number, _taskGroupLength: number, isAboveParallel: boolean, isBelowParallel: boolean, reGroupTaskFunction: (groupModificationType: GroupModificationType, groupIndex: number, taskIndex: number) => void, hideParallelOptions?: boolean) => NodeMenuItem[];
+export declare const getContextMenuItems: ({ isParallelGroup, groupIndex, tasksLength, groupIndexInAllTasks, taskIndexInAllTasks, isAboveParallel, isBelowParallel, reGroupTaskFunction, hideParallelOptions, }: {
+    isParallelGroup: boolean;
+    groupIndex: number;
+    tasksLength: number;
+    groupIndexInAllTasks: number;
+    taskIndexInAllTasks: number;
+    isAboveParallel: boolean;
+    isBelowParallel: boolean;
+    reGroupTaskFunction: (groupModificationType: GroupModificationType, groupIndex: number, taskIndex: number) => void;
+    hideParallelOptions?: boolean;
+}) => NodeMenuItem[];
 export declare function getMenuItem(id: string | undefined, label: string | undefined, onClick: () => void, isDisabled?: boolean): NodeMenuItem;
 export declare const getDivider: () => NodeMenuItem;
 export interface TransformedMenuItem {

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

@@ -1 +1 @@
-{"version":3,"file":"StageNodeTaskUtilities.d.ts","sourceRoot":"","sources":["../../../../src/canvas/components/StageNode/StageNodeTaskUtilities.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,OAAO,CAAC;AACvC,OAAO,EAAE,qBAAqB,EAAE,MAAM,oCAAoC,CAAC;AAC3E,OAAO,KAAK,EAAE,cAAc,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAIvE,eAAO,MAAM,mBAAmB,GAC9B,iBAAiB,OAAO,EACxB,YAAY,MAAM,EAClB,aAAa,MAAM,EACnB,WAAW,MAAM,EACjB,kBAAkB,MAAM,EACxB,iBAAiB,OAAO,EACxB,iBAAiB,OAAO,EACxB,qBAAqB,CACnB,qBAAqB,EAAE,qBAAqB,EAC5C,UAAU,EAAE,MAAM,EAClB,SAAS,EAAE,MAAM,KACd,IAAI,EACT,6BAA2B,KAC1B,YAAY,EAmFd,CAAC;AAEF,wBAAgB,WAAW,CACzB,EAAE,EAAE,MAAM,YAAO,EACjB,KAAK,EAAE,MAAM,YAAU,EACvB,OAAO,EAAE,MAAM,IAAI,EACnB,UAAU,UAAQ,GACjB,YAAY,CAEd;AAED,eAAO,MAAM,UAAU,QAAO,YAI7B,CAAC;AAEF,MAAM,WAAW,mBAAmB;IAClC,GAAG,EAAE,MAAM,CAAC;IACZ,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,SAAS,CAAC,EAAE,SAAS,CAAC;IACtB,QAAQ,CAAC,EAAE,OAAO,CAAC;IACnB,OAAO,CAAC,EAAE,MAAM,IAAI,CAAC;IACrB,OAAO,EAAE,MAAM,GAAG,WAAW,CAAC;IAC9B,OAAO,CAAC,EAAE,OAAO,CAAC;CACnB;AAMD,eAAO,MAAM,kBAAkB,GAC7B,WAAW,YAAY,EAAE,GAAG,SAAS,EACrC,aAAa,CAAC,IAAI,EAAE,cAAc,KAAK,IAAI,KAC1C,mBAAmB,EAwBrB,CAAC"}
+{"version":3,"file":"StageNodeTaskUtilities.d.ts","sourceRoot":"","sources":["../../../../src/canvas/components/StageNode/StageNodeTaskUtilities.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,OAAO,CAAC;AACvC,OAAO,EAAE,qBAAqB,EAAE,MAAM,oCAAoC,CAAC;AAC3E,OAAO,KAAK,EAAE,cAAc,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAIvE,eAAO,MAAM,mBAAmB,GAAI,sKAUjC;IACD,eAAe,EAAE,OAAO,CAAC;IACzB,UAAU,EAAE,MAAM,CAAC;IACnB,WAAW,EAAE,MAAM,CAAC;IACpB,oBAAoB,EAAE,MAAM,CAAC;IAC7B,mBAAmB,EAAE,MAAM,CAAC;IAC5B,eAAe,EAAE,OAAO,CAAC;IACzB,eAAe,EAAE,OAAO,CAAC;IACzB,mBAAmB,EAAE,CACnB,qBAAqB,EAAE,qBAAqB,EAC5C,UAAU,EAAE,MAAM,EAClB,SAAS,EAAE,MAAM,KACd,IAAI,CAAC;IACV,mBAAmB,CAAC,EAAE,OAAO,CAAC;CAC/B,KAAG,YAAY,EA+Hf,CAAC;AAEF,wBAAgB,WAAW,CACzB,EAAE,EAAE,MAAM,YAAO,EACjB,KAAK,EAAE,MAAM,YAAU,EACvB,OAAO,EAAE,MAAM,IAAI,EACnB,UAAU,UAAQ,GACjB,YAAY,CAEd;AAED,eAAO,MAAM,UAAU,QAAO,YAI7B,CAAC;AAEF,MAAM,WAAW,mBAAmB;IAClC,GAAG,EAAE,MAAM,CAAC;IACZ,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,SAAS,CAAC,EAAE,SAAS,CAAC;IACtB,QAAQ,CAAC,EAAE,OAAO,CAAC;IACnB,OAAO,CAAC,EAAE,MAAM,IAAI,CAAC;IACrB,OAAO,EAAE,MAAM,GAAG,WAAW,CAAC;IAC9B,OAAO,CAAC,EAAE,OAAO,CAAC;CACnB;AAMD,eAAO,MAAM,kBAAkB,GAC7B,WAAW,YAAY,EAAE,GAAG,SAAS,EACrC,aAAa,CAAC,IAAI,EAAE,cAAc,KAAK,IAAI,KAC1C,mBAAmB,EAwBrB,CAAC"}

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

@@ -1,16 +1,16 @@
 import { GroupModificationType } from "../../utils/GroupModificationUtils.js";
-const getContextMenuItems = (isParallelGroup, groupIndex, tasksLength, taskIndex, _taskGroupLength, isAboveParallel, isBelowParallel, reGroupTaskFunction, hideParallelOptions = false)=>{
+const getContextMenuItems = ({ isParallelGroup, groupIndex, tasksLength, groupIndexInAllTasks, taskIndexInAllTasks, isAboveParallel, isBelowParallel, reGroupTaskFunction, hideParallelOptions = false })=>{
     const CONTEXT_MENU_ITEMS = {
-        MOVE_UP: getMenuItem('move-up', 'Move up', ()=>reGroupTaskFunction(GroupModificationType.TASK_GROUP_UP, groupIndex, taskIndex)),
-        MOVE_DOWN: getMenuItem('move-down', 'Move down', ()=>reGroupTaskFunction(GroupModificationType.TASK_GROUP_DOWN, groupIndex, taskIndex)),
-        UNGROUP_ALL: getMenuItem('ungroup', 'Ungroup parallel tasks', ()=>reGroupTaskFunction(GroupModificationType.UNGROUP_ALL_TASKS, groupIndex, taskIndex)),
-        SPLIT_TASK: getMenuItem('split', 'Remove from parallel group', ()=>reGroupTaskFunction(GroupModificationType.SPLIT_GROUP, groupIndex, taskIndex)),
-        REMOVE_GROUP: getMenuItem('remove-group', 'Remove group from stage', ()=>reGroupTaskFunction(GroupModificationType.REMOVE_GROUP, groupIndex, taskIndex)),
-        REMOVE_TASK: getMenuItem('remove-task', 'Delete task', ()=>reGroupTaskFunction(GroupModificationType.REMOVE_TASK, groupIndex, taskIndex)),
-        CREATE_PARALLEL_GROUP_ABOVE: getMenuItem('group-with-up', 'Create parallel group with task above', ()=>reGroupTaskFunction(GroupModificationType.MERGE_GROUP_UP, groupIndex, taskIndex)),
-        CREATE_PARALLEL_GROUP_BELOW: getMenuItem('group-with-down', 'Create parallel group with task below', ()=>reGroupTaskFunction(GroupModificationType.MERGE_GROUP_DOWN, groupIndex, taskIndex)),
-        ADD_TO_PARALLEL_GROUP_ABOVE: getMenuItem('add-to-group-with-up', 'Add task to parallel group above', ()=>reGroupTaskFunction(GroupModificationType.MERGE_GROUP_UP, groupIndex, taskIndex)),
-        ADD_TO_PARALLEL_GROUP_BELOW: getMenuItem('add-to-group-with-down', 'Add task to parallel group below', ()=>reGroupTaskFunction(GroupModificationType.MERGE_GROUP_DOWN, groupIndex, taskIndex)),
+        MOVE_UP: getMenuItem('move-up', 'Move up', ()=>reGroupTaskFunction(GroupModificationType.TASK_GROUP_UP, groupIndexInAllTasks, taskIndexInAllTasks)),
+        MOVE_DOWN: getMenuItem('move-down', 'Move down', ()=>reGroupTaskFunction(GroupModificationType.TASK_GROUP_DOWN, groupIndexInAllTasks, taskIndexInAllTasks)),
+        UNGROUP_ALL: getMenuItem('ungroup', 'Ungroup parallel tasks', ()=>reGroupTaskFunction(GroupModificationType.UNGROUP_ALL_TASKS, groupIndexInAllTasks, taskIndexInAllTasks)),
+        SPLIT_TASK: getMenuItem('split', 'Remove from parallel group', ()=>reGroupTaskFunction(GroupModificationType.SPLIT_GROUP, groupIndexInAllTasks, taskIndexInAllTasks)),
+        REMOVE_GROUP: getMenuItem('remove-group', 'Remove group from stage', ()=>reGroupTaskFunction(GroupModificationType.REMOVE_GROUP, groupIndexInAllTasks, taskIndexInAllTasks)),
+        REMOVE_TASK: getMenuItem('remove-task', 'Delete task', ()=>reGroupTaskFunction(GroupModificationType.REMOVE_TASK, groupIndexInAllTasks, taskIndexInAllTasks)),
+        CREATE_PARALLEL_GROUP_ABOVE: getMenuItem('group-with-up', 'Create parallel group with task above', ()=>reGroupTaskFunction(GroupModificationType.MERGE_GROUP_UP, groupIndexInAllTasks, taskIndexInAllTasks)),
+        CREATE_PARALLEL_GROUP_BELOW: getMenuItem('group-with-down', 'Create parallel group with task below', ()=>reGroupTaskFunction(GroupModificationType.MERGE_GROUP_DOWN, groupIndexInAllTasks, taskIndexInAllTasks)),
+        ADD_TO_PARALLEL_GROUP_ABOVE: getMenuItem('add-to-group-with-up', 'Add task to parallel group above', ()=>reGroupTaskFunction(GroupModificationType.MERGE_GROUP_UP, groupIndexInAllTasks, taskIndexInAllTasks)),
+        ADD_TO_PARALLEL_GROUP_BELOW: getMenuItem('add-to-group-with-down', 'Add task to parallel group below', ()=>reGroupTaskFunction(GroupModificationType.MERGE_GROUP_DOWN, groupIndexInAllTasks, taskIndexInAllTasks)),
         DIVIDER: getDivider()
     };
     const items = [];

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@uipath/apollo-react",
-  "version": "4.12.1",
+  "version": "4.12.2",
   "description": "Apollo Design System - React component library with Material UI theming",
   "repository": {
     "type": "git",