@uipath/apollo-react 3.62.1 → 3.63.0-pr431.329c320

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

@@ -1,3 +1,3 @@
 import type { StageNodeProps } from './StageNode.types';
-export declare const StageNode: import("react").MemoExoticComponent<(props: StageNodeProps) => import("react/jsx-runtime").JSX.Element>;
+export declare const StageNode: import("react").MemoExoticComponent<({ dragging, selected, id, width, execution, stageDetails, addTaskLabel, addTaskLoading, replaceTaskLabel, taskOptions, menuItems, pendingReplaceTask, onStageClick, onTaskAdd, onAddTaskFromToolbox, onTaskToolboxSearch, onTaskClick, onTaskGroupModification, onStageTitleChange, onTaskReorder, onReplaceTaskFromToolbox, onTaskPlay, }: StageNodeProps) => import("react/jsx-runtime").JSX.Element>;
 //# sourceMappingURL=StageNode.d.ts.map

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

@@ -1 +1 @@
-{"version":3,"file":"StageNode.d.ts","sourceRoot":"","sources":["../../../../src/canvas/components/StageNode/StageNode.tsx"],"names":[],"mappings":"AAqDA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,mBAAmB,CAAC;AA2pBxD,eAAO,MAAM,SAAS,8CAjpBa,cAAc,6CAipBA,CAAC"}
+{"version":3,"file":"StageNode.d.ts","sourceRoot":"","sources":["../../../../src/canvas/components/StageNode/StageNode.tsx"],"names":[],"mappings":"AA0DA,OAAO,KAAK,EAEV,cAAc,EAEf,MAAM,mBAAmB,CAAC;AAotB3B,eAAO,MAAM,SAAS,mXA7BnB,cAAc,6CA6BgC,CAAC"}

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

@@ -3,7 +3,7 @@ import { DndContext, DragOverlay, KeyboardSensor, PointerSensor, closestCenter,
 import { SortableContext, sortableKeyboardCoordinates, verticalListSortingStrategy } from "@dnd-kit/sortable";
 import { FontVariantToken, Padding, Spacing } from "@uipath/apollo-core";
 import { Column, Row } from "../../layouts/index.js";
-import { Position, useStore, useViewport } from "../../xyflow/react.js";
+import { Position, useStore, useStoreApi, useViewport } from "../../xyflow/react.js";
 import { ApIcon, ApIconButton, ApLink, ApTooltip, ApTypography } from "../../../material/index.js";
 import { memo, useCallback, useEffect, useMemo, useRef, useState } from "react";
 import { createPortal } from "react-dom";
@@ -12,13 +12,39 @@ import { useButtonHandles } from "../ButtonHandle/useButtonHandles.js";
 import { ExecutionStatusIcon } from "../ExecutionStatusIcon/index.js";
 import { FloatingCanvasPanel } from "../FloatingCanvasPanel/index.js";
 import { NodeContextMenu } from "../NodeContextMenu/index.js";
-import { useNodeSelection } from "../NodePropertiesPanel/hooks/index.js";
+import { useSetNodeSelection } from "../NodePropertiesPanel/hooks/index.js";
 import { Toolbox } from "../Toolbox/index.js";
 import { DraggableTask, TaskContent } from "./DraggableTask.js";
 import { INDENTATION_WIDTH, STAGE_CONTENT_INSET, StageContainer, StageContent, StageHeader, StageParallelBracket, StageParallelLabel, StageTask, StageTaskGroup, StageTaskList, StageTitleContainer, StageTitleInput } from "./StageNode.styles.js";
 import { flattenTasks, getProjection, reorderTasks } from "./StageNode.utils.js";
 import { getContextMenuItems, getDivider, getMenuItem } from "./StageNodeTaskUtilities.js";
-const StageNodeComponent = (props)=>{
+const StageNode_StageTaskDragOverlay = /*#__PURE__*/ memo(function({ activeTask, isActiveTaskParallel, taskWidthStyle }) {
+    const { zoom } = useViewport();
+    const dragOverlayStyle = useMemo(()=>({
+            transform: `scale(${zoom})`,
+            transformOrigin: 'top left'
+        }), [
+        zoom
+    ]);
+    return /*#__PURE__*/ createPortal(/*#__PURE__*/ jsx(DragOverlay, {
+        children: activeTask ? /*#__PURE__*/ jsx("div", {
+            style: dragOverlayStyle,
+            children: /*#__PURE__*/ jsx(StageTask, {
+                selected: true,
+                isParallel: isActiveTaskParallel,
+                style: {
+                    cursor: 'grabbing',
+                    ...taskWidthStyle
+                },
+                children: /*#__PURE__*/ jsx(TaskContent, {
+                    task: activeTask,
+                    isDragging: true
+                })
+            })
+        }) : null
+    }), document.body);
+});
+const StageNodeInner = (props)=>{
     const { dragging, selected, id, width, execution, stageDetails, addTaskLabel = 'Add task', addTaskLoading = false, replaceTaskLabel = 'Replace task', taskOptions = [], menuItems, pendingReplaceTask, onStageClick, onTaskAdd, onAddTaskFromToolbox, onTaskToolboxSearch, onTaskClick, onTaskGroupModification, onStageTitleChange, onTaskReorder, onReplaceTaskFromToolbox, onTaskPlay } = props;
     const taskWidth = width ? width - STAGE_CONTENT_INSET : void 0;
     const tasks = useMemo(()=>stageDetails?.tasks || [], [
@@ -38,9 +64,6 @@ const StageNodeComponent = (props)=>{
     const status = execution?.stageStatus?.status;
     const statusLabel = execution?.stageStatus?.label;
     const stageDuration = execution?.stageStatus?.duration;
-    const reGroupTaskFunction = useMemo(()=>onTaskGroupModification || (()=>{}), [
-        onTaskGroupModification
-    ]);
     const isStageTitleEditable = !!onStageTitleChange && !isReadOnly;
     const [isHovered, setIsHovered] = useState(false);
     const [label, setLabel] = useState(props.stageDetails.label);
@@ -57,6 +80,7 @@ const StageNodeComponent = (props)=>{
         taskIndex: -1
     });
     const isConnecting = useStore((state)=>!!state.connectionClickStartHandle);
+    const storeApi = useStoreApi();
     const connectedHandleIds = useConnectedHandles(id);
     const [isAddingTask, setIsAddingTask] = useState(false);
     const [isReplacingTask, setIsReplacingTask] = useState(false);
@@ -96,7 +120,6 @@ const StageNodeComponent = (props)=>{
         tasks,
         activeDragId
     ]);
-    const { zoom } = useViewport();
     const projected = useMemo(()=>{
         if (!activeDragId || !overId) return null;
         return getProjection(tasks, activeDragId, overId, offsetLeft);
@@ -179,18 +202,21 @@ const StageNodeComponent = (props)=>{
         handleStageTitleClickToSave,
         isStageTitleEditing
     ]);
-    const contextMenuItems = useCallback((isParallel, groupIndex, taskIndex, tasksLength, taskGroupLength, isAboveParallel, isBelowParallel)=>{
+    const hasContextMenu = !!(onReplaceTaskFromToolbox || onTaskGroupModification);
+    const buildContextMenuItems = useCallback((groupIndex, taskIndex)=>{
+        const taskGroup = tasks[groupIndex] ?? [];
+        const isParallel = taskGroup.length > 1;
         const items = [];
         if (onReplaceTaskFromToolbox) {
             items.push(getMenuItem('replace-task', 'Replace task', ()=>{
-                const taskId = tasks[groupIndex]?.[taskIndex]?.id;
+                const taskId = taskGroup[taskIndex]?.id;
                 if (taskId) onTaskClick?.(taskId);
                 setIsReplacingTask(true);
             }));
             items.push(getDivider());
         }
         if (onTaskGroupModification) {
-            const reGroupOptions = getContextMenuItems(isParallel, groupIndex, tasksLength, taskIndex, taskGroupLength, isAboveParallel, isBelowParallel, reGroupTaskFunction);
+            const reGroupOptions = getContextMenuItems(isParallel, groupIndex, tasks.length, taskIndex, taskGroup.length, (tasks[groupIndex - 1]?.length ?? 0) > 1, (tasks[groupIndex + 1]?.length ?? 0) > 1, onTaskGroupModification);
             return [
                 ...items,
                 ...reGroupOptions
@@ -201,10 +227,9 @@ const StageNodeComponent = (props)=>{
         onReplaceTaskFromToolbox,
         onTaskClick,
         onTaskGroupModification,
-        reGroupTaskFunction,
         tasks
     ]);
-    const { setSelectedNodeId } = useNodeSelection();
+    const { setSelectedNodeId } = useSetNodeSelection();
     const handleStageClick = useCallback(()=>{
         onStageClick?.();
     }, [
@@ -327,9 +352,9 @@ const StageNodeComponent = (props)=>{
         setActiveDragId(event.active.id);
     }, []);
     const handleDragMove = useCallback((event)=>{
-        setOffsetLeft(event.delta.x / zoom);
+        setOffsetLeft(event.delta.x / storeApi.getState().transform[2]);
     }, [
-        zoom
+        storeApi
     ]);
     const handleDragOver = useCallback((event)=>{
         setOverId(event.over?.id ?? null);
@@ -365,12 +390,6 @@ const StageNodeComponent = (props)=>{
         } : void 0, [
         taskWidth
     ]);
-    const dragOverlayStyle = useMemo(()=>({
-            transform: `scale(${zoom})`,
-            transformOrigin: 'top left'
-        }), [
-        zoom
-    ]);
     return /*#__PURE__*/ jsxs("div", {
         "data-testid": `stage-${id}`,
         style: {
@@ -510,13 +529,12 @@ const StageNodeComponent = (props)=>{
                                                                     taskExecution: taskExecution,
                                                                     isSelected: selectedTaskId === task.id,
                                                                     isParallel: isParallel,
-                                                                    contextMenuItems: contextMenuItems(isParallel, groupIndex, taskIndex, tasks.length, taskGroup.length, (tasks[groupIndex - 1]?.length ?? 0) > 1, (tasks[groupIndex + 1]?.length ?? 0) > 1),
                                                                     onTaskClick: handleTaskClick,
                                                                     projectedDepth: task.id === activeDragId && projected ? projected.depth : void 0,
                                                                     onTaskPlay: onTaskPlay,
                                                                     isDragDisabled: !onTaskReorder,
-                                                                    zoom: zoom,
-                                                                    ...(onTaskGroupModification || onReplaceTaskFromToolbox) && {
+                                                                    ...hasContextMenu && {
+                                                                        getContextMenuItems: ()=>buildContextMenuItems(groupIndex, taskIndex),
                                                                         onMenuOpen: ()=>{
                                                                             taskStateReference.current = {
                                                                                 isParallel,
@@ -534,23 +552,11 @@ const StageNodeComponent = (props)=>{
                                         })
                                     })
                                 }),
-                                /*#__PURE__*/ createPortal(/*#__PURE__*/ jsx(DragOverlay, {
-                                    children: activeTask ? /*#__PURE__*/ jsx("div", {
-                                        style: dragOverlayStyle,
-                                        children: /*#__PURE__*/ jsx(StageTask, {
-                                            selected: true,
-                                            isParallel: isActiveTaskParallel,
-                                            style: {
-                                                cursor: 'grabbing',
-                                                ...taskWidthStyle
-                                            },
-                                            children: /*#__PURE__*/ jsx(TaskContent, {
-                                                task: activeTask,
-                                                isDragging: true
-                                            })
-                                        })
-                                    }) : null
-                                }), document.body)
+                                /*#__PURE__*/ jsx(StageNode_StageTaskDragOverlay, {
+                                    activeTask: activeTask,
+                                    isActiveTaskParallel: isActiveTaskParallel,
+                                    taskWidthStyle: taskWidthStyle
+                                })
                             ]
                         }) : /*#__PURE__*/ jsx(Column, {
                             py: 2,
@@ -604,5 +610,30 @@ const StageNodeComponent = (props)=>{
         ]
     });
 };
+const StageNodeInnerMemo = /*#__PURE__*/ memo(StageNodeInner);
+const StageNodeComponent = ({ dragging, selected, id, width, execution, stageDetails, addTaskLabel, addTaskLoading, replaceTaskLabel, taskOptions, menuItems, pendingReplaceTask, onStageClick, onTaskAdd, onAddTaskFromToolbox, onTaskToolboxSearch, onTaskClick, onTaskGroupModification, onStageTitleChange, onTaskReorder, onReplaceTaskFromToolbox, onTaskPlay })=>/*#__PURE__*/ jsx(StageNodeInnerMemo, {
+        dragging: dragging,
+        selected: selected,
+        id: id,
+        width: width,
+        execution: execution,
+        stageDetails: stageDetails,
+        addTaskLabel: addTaskLabel,
+        addTaskLoading: addTaskLoading,
+        replaceTaskLabel: replaceTaskLabel,
+        taskOptions: taskOptions,
+        menuItems: menuItems,
+        pendingReplaceTask: pendingReplaceTask,
+        onStageClick: onStageClick,
+        onTaskAdd: onTaskAdd,
+        onAddTaskFromToolbox: onAddTaskFromToolbox,
+        onTaskToolboxSearch: onTaskToolboxSearch,
+        onTaskClick: onTaskClick,
+        onTaskGroupModification: onTaskGroupModification,
+        onStageTitleChange: onStageTitleChange,
+        onTaskReorder: onTaskReorder,
+        onReplaceTaskFromToolbox: onReplaceTaskFromToolbox,
+        onTaskPlay: onTaskPlay
+    });
 const StageNode = /*#__PURE__*/ memo(StageNodeComponent);
 export { StageNode };

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

@@ -48,6 +48,53 @@ const external_TaskIcon_index_cjs_namespaceObject = require("../TaskIcon/index.c
 const external_StageConnectionEdge_cjs_namespaceObject = require("./StageConnectionEdge.cjs");
 const external_StageEdge_cjs_namespaceObject = require("./StageEdge.cjs");
 const external_StageNode_cjs_namespaceObject = require("./StageNode.cjs");
+const external_StageNode_stories_utils_cjs_namespaceObject = require("./StageNode.stories.utils.cjs");
+const DefaultCanvasDecorator = ({ initialNodes, initialEdges = [] })=>{
+    const [nodes, _setNodes, onNodesChange] = (0, react_cjs_namespaceObject.useNodesState)(initialNodes);
+    const [edges, setEdges, onEdgesChange] = (0, react_cjs_namespaceObject.useEdgesState)(initialEdges);
+    const onConnect = (0, external_react_namespaceObject.useCallback)((connection)=>{
+        setEdges((eds)=>(0, react_cjs_namespaceObject.addEdge)(connection, eds));
+    }, [
+        setEdges
+    ]);
+    const nodeTypes = (0, external_react_namespaceObject.useMemo)(()=>({
+            stage: external_StageNode_stories_utils_cjs_namespaceObject.StageNodeWrapper
+        }), []);
+    const edgeTypes = (0, external_react_namespaceObject.useMemo)(()=>({
+            stage: external_StageEdge_cjs_namespaceObject.StageEdge
+        }), []);
+    const defaultEdgeOptions = (0, external_react_namespaceObject.useMemo)(()=>({
+            type: 'stage'
+        }), []);
+    return /*#__PURE__*/ (0, jsx_runtime_namespaceObject.jsx)("div", {
+        style: {
+            width: '100vw',
+            height: '100vh'
+        },
+        children: /*#__PURE__*/ (0, jsx_runtime_namespaceObject.jsx)(react_cjs_namespaceObject.ReactFlowProvider, {
+            children: /*#__PURE__*/ (0, jsx_runtime_namespaceObject.jsx)(external_BaseCanvas_index_cjs_namespaceObject.BaseCanvas, {
+                nodes: nodes,
+                edges: edges,
+                onNodesChange: onNodesChange,
+                onEdgesChange: onEdgesChange,
+                onConnect: onConnect,
+                nodeTypes: nodeTypes,
+                edgeTypes: edgeTypes,
+                mode: "design",
+                connectionMode: react_cjs_namespaceObject.ConnectionMode.Strict,
+                defaultEdgeOptions: defaultEdgeOptions,
+                connectionLineComponent: external_StageConnectionEdge_cjs_namespaceObject.StageConnectionEdge,
+                elevateEdgesOnSelect: true,
+                children: /*#__PURE__*/ (0, jsx_runtime_namespaceObject.jsx)(react_cjs_namespaceObject.Panel, {
+                    position: "bottom-right",
+                    children: /*#__PURE__*/ (0, jsx_runtime_namespaceObject.jsx)(external_CanvasPositionControls_cjs_namespaceObject.CanvasPositionControls, {
+                        translations: external_types_cjs_namespaceObject.DefaultCanvasTranslations
+                    })
+                })
+            })
+        })
+    });
+};
 const meta = {
     title: 'Canvas/StageNode',
     component: external_StageNode_cjs_namespaceObject.StageNode,
@@ -57,10 +104,6 @@ const meta = {
     decorators: [
         (Story, context)=>{
             if (context.parameters?.useCustomRender) return /*#__PURE__*/ (0, jsx_runtime_namespaceObject.jsx)(Story, {});
-            const StageNodeWrapper = (props)=>/*#__PURE__*/ (0, jsx_runtime_namespaceObject.jsx)(external_StageNode_cjs_namespaceObject.StageNode, {
-                    ...props,
-                    ...props.data
-                });
             const initialNodes = context.parameters?.nodes || [
                 {
                     id: '1',
@@ -80,51 +123,9 @@ const meta = {
                 }
             ];
             const initialEdges = context.parameters?.edges || [];
-            const [nodes, _setNodes, onNodesChange] = (0, react_cjs_namespaceObject.useNodesState)(initialNodes);
-            const [edges, setEdges, onEdgesChange] = (0, react_cjs_namespaceObject.useEdgesState)(initialEdges);
-            const onConnect = (0, external_react_namespaceObject.useCallback)((connection)=>{
-                setEdges((eds)=>(0, react_cjs_namespaceObject.addEdge)(connection, eds));
-            }, [
-                setEdges
-            ]);
-            const nodeTypes = (0, external_react_namespaceObject.useMemo)(()=>({
-                    stage: StageNodeWrapper
-                }), [
-                StageNodeWrapper
-            ]);
-            const edgeTypes = (0, external_react_namespaceObject.useMemo)(()=>({
-                    stage: external_StageEdge_cjs_namespaceObject.StageEdge
-                }), []);
-            const defaultEdgeOptions = (0, external_react_namespaceObject.useMemo)(()=>({
-                    type: 'stage'
-                }), []);
-            return /*#__PURE__*/ (0, jsx_runtime_namespaceObject.jsx)("div", {
-                style: {
-                    width: '100vw',
-                    height: '100vh'
-                },
-                children: /*#__PURE__*/ (0, jsx_runtime_namespaceObject.jsx)(react_cjs_namespaceObject.ReactFlowProvider, {
-                    children: /*#__PURE__*/ (0, jsx_runtime_namespaceObject.jsx)(external_BaseCanvas_index_cjs_namespaceObject.BaseCanvas, {
-                        nodes: nodes,
-                        edges: edges,
-                        onNodesChange: onNodesChange,
-                        onEdgesChange: onEdgesChange,
-                        onConnect: onConnect,
-                        nodeTypes: nodeTypes,
-                        edgeTypes: edgeTypes,
-                        mode: "design",
-                        connectionMode: react_cjs_namespaceObject.ConnectionMode.Strict,
-                        defaultEdgeOptions: defaultEdgeOptions,
-                        connectionLineComponent: external_StageConnectionEdge_cjs_namespaceObject.StageConnectionEdge,
-                        elevateEdgesOnSelect: true,
-                        children: /*#__PURE__*/ (0, jsx_runtime_namespaceObject.jsx)(react_cjs_namespaceObject.Panel, {
-                            position: "bottom-right",
-                            children: /*#__PURE__*/ (0, jsx_runtime_namespaceObject.jsx)(external_CanvasPositionControls_cjs_namespaceObject.CanvasPositionControls, {
-                                translations: external_types_cjs_namespaceObject.DefaultCanvasTranslations
-                            })
-                        })
-                    })
-                })
+            return /*#__PURE__*/ (0, jsx_runtime_namespaceObject.jsx)(DefaultCanvasDecorator, {
+                initialNodes: initialNodes,
+                initialEdges: initialEdges
             });
         }
     ],
@@ -200,7 +201,7 @@ const DocumentIcon = ()=>/*#__PURE__*/ (0, jsx_runtime_namespaceObject.jsxs)("sv
         strokeWidth: "1.5",
         children: [
             /*#__PURE__*/ (0, jsx_runtime_namespaceObject.jsx)("path", {
-                d: "M14 2H6C4.9 2 4 2.9 4 4V20C20 21.1 4.9 22 6 22H18C19.1 22 20 21.1 20 20V8L14 2Z"
+                d: "M14 2H6C4.9 2 4 2.9 4 4V20C4 21.1 4.9 22 6 22H18C19.1 22 20 21.1 20 20V8L14 2Z"
             }),
             /*#__PURE__*/ (0, jsx_runtime_namespaceObject.jsx)("path", {
                 d: "M14 2V8H20"
@@ -1311,17 +1312,9 @@ const initialTasks = [
     ]
 ];
 const DraggableTaskReorderingStory = ()=>{
-    const StageNodeWrapper = (0, external_react_namespaceObject.useMemo)(()=>function(props) {
-            return /*#__PURE__*/ (0, jsx_runtime_namespaceObject.jsx)(external_StageNode_cjs_namespaceObject.StageNode, {
-                ...props,
-                ...props.data
-            });
-        }, []);
     const nodeTypes = (0, external_react_namespaceObject.useMemo)(()=>({
-            stage: StageNodeWrapper
-        }), [
-        StageNodeWrapper
-    ]);
+            stage: external_StageNode_stories_utils_cjs_namespaceObject.StageNodeWrapper
+        }), []);
     const edgeTypes = (0, external_react_namespaceObject.useMemo)(()=>({
             stage: external_StageEdge_cjs_namespaceObject.StageEdge
         }), []);
@@ -1490,17 +1483,9 @@ const availableTaskOptions = [
     }
 ];
 const AddAndReplaceTasksStory = ()=>{
-    const StageNodeWrapper = (0, external_react_namespaceObject.useMemo)(()=>function(props) {
-            return /*#__PURE__*/ (0, jsx_runtime_namespaceObject.jsx)(external_StageNode_cjs_namespaceObject.StageNode, {
-                ...props,
-                ...props.data
-            });
-        }, []);
     const nodeTypes = (0, external_react_namespaceObject.useMemo)(()=>({
-            stage: StageNodeWrapper
-        }), [
-        StageNodeWrapper
-    ]);
+            stage: external_StageNode_stories_utils_cjs_namespaceObject.StageNodeWrapper
+        }), []);
     const edgeTypes = (0, external_react_namespaceObject.useMemo)(()=>({
             stage: external_StageEdge_cjs_namespaceObject.StageEdge
         }), []);
@@ -1741,20 +1726,25 @@ const AddAndReplaceTasks = {
     args: {}
 };
 const InlineTitleEditStory = ()=>{
-    const StageNodeWrapper = (0, external_react_namespaceObject.useMemo)(()=>function(props) {
-            return /*#__PURE__*/ (0, jsx_runtime_namespaceObject.jsx)(external_StageNode_cjs_namespaceObject.StageNode, {
-                ...props,
-                ...props.data
-            });
-        }, []);
     const nodeTypes = (0, external_react_namespaceObject.useMemo)(()=>({
-            stage: StageNodeWrapper
-        }), [
-        StageNodeWrapper
-    ]);
+            stage: external_StageNode_stories_utils_cjs_namespaceObject.StageNodeWrapper
+        }), []);
     const edgeTypes = (0, external_react_namespaceObject.useMemo)(()=>({
             stage: external_StageEdge_cjs_namespaceObject.StageEdge
         }), []);
+    const setNodesRef = (0, external_react_namespaceObject.useRef)(null);
+    const createTitleChangeHandler = (0, external_react_namespaceObject.useCallback)((nodeId)=>(newTitle)=>{
+            setNodesRef.current((nds)=>nds.map((node)=>node.id === nodeId ? {
+                        ...node,
+                        data: {
+                            ...node.data,
+                            stageDetails: {
+                                ...node.data.stageDetails,
+                                label: newTitle
+                            }
+                        }
+                    } : node));
+        }, []);
     const initialNodes = (0, external_react_namespaceObject.useMemo)(()=>[
             {
                 id: 'editable-stage',
@@ -1786,7 +1776,8 @@ const InlineTitleEditStory = ()=>{
                     },
                     onTaskAdd: ()=>{
                         window.alert('Add task functionality - this would open a dialog to add a new task');
-                    }
+                    },
+                    onStageTitleChange: createTitleChangeHandler('editable-stage')
                 }
             },
             {
@@ -1812,36 +1803,16 @@ const InlineTitleEditStory = ()=>{
                     },
                     onTaskAdd: ()=>{
                         window.alert('Add task functionality - this would open a dialog to add a new task');
-                    }
+                    },
+                    onStageTitleChange: createTitleChangeHandler('long-title-stage')
                 }
             }
-        ], []);
-    const [nodes, setNodes, onNodesChange] = (0, react_cjs_namespaceObject.useNodesState)(initialNodes);
-    const [edges, setEdges, onEdgesChange] = (0, react_cjs_namespaceObject.useEdgesState)([]);
-    const createTitleChangeHandler = (0, external_react_namespaceObject.useCallback)((nodeId)=>(newTitle)=>{
-            setNodes((nds)=>nds.map((node)=>node.id === nodeId ? {
-                        ...node,
-                        data: {
-                            ...node.data,
-                            stageDetails: {
-                                ...node.data.stageDetails,
-                                label: newTitle
-                            }
-                        }
-                    } : node));
-        }, [
-        setNodes
-    ]);
-    const nodesWithHandlers = (0, external_react_namespaceObject.useMemo)(()=>nodes.map((node)=>({
-                ...node,
-                data: {
-                    ...node.data,
-                    onStageTitleChange: createTitleChangeHandler(node.id)
-                }
-            })), [
-        nodes,
+        ], [
         createTitleChangeHandler
     ]);
+    const [nodes, setNodes, onNodesChange] = (0, react_cjs_namespaceObject.useNodesState)(initialNodes);
+    setNodesRef.current = setNodes;
+    const [edges, setEdges, onEdgesChange] = (0, react_cjs_namespaceObject.useEdgesState)([]);
     const onConnect = (0, external_react_namespaceObject.useCallback)((connection)=>setEdges((eds)=>(0, react_cjs_namespaceObject.addEdge)(connection, eds)), [
         setEdges
     ]);
@@ -1852,7 +1823,7 @@ const InlineTitleEditStory = ()=>{
         },
         children: /*#__PURE__*/ (0, jsx_runtime_namespaceObject.jsx)(react_cjs_namespaceObject.ReactFlowProvider, {
             children: /*#__PURE__*/ (0, jsx_runtime_namespaceObject.jsx)(external_BaseCanvas_index_cjs_namespaceObject.BaseCanvas, {
-                nodes: nodesWithHandlers,
+                nodes: nodes,
                 edges: edges,
                 onNodesChange: onNodesChange,
                 onEdgesChange: onEdgesChange,
@@ -1943,20 +1914,33 @@ const loadedTaskOptionsWithChildren = [
     }
 ];
 const AddTaskLoadingStory = ()=>{
-    const StageNodeWrapper = (0, external_react_namespaceObject.useMemo)(()=>function(props) {
-            return /*#__PURE__*/ (0, jsx_runtime_namespaceObject.jsx)(external_StageNode_cjs_namespaceObject.StageNode, {
-                ...props,
-                ...props.data
-            });
-        }, []);
     const nodeTypes = (0, external_react_namespaceObject.useMemo)(()=>({
-            stage: StageNodeWrapper
-        }), [
-        StageNodeWrapper
-    ]);
+            stage: external_StageNode_stories_utils_cjs_namespaceObject.StageNodeWrapper
+        }), []);
     const edgeTypes = (0, external_react_namespaceObject.useMemo)(()=>({
             stage: external_StageEdge_cjs_namespaceObject.StageEdge
         }), []);
+    const setNodesRef = (0, external_react_namespaceObject.useRef)(null);
+    const timeoutRef = (0, external_react_namespaceObject.useRef)(void 0);
+    const handleAddTaskFromToolbox = (0, external_react_namespaceObject.useCallback)((nodeId, _taskItem)=>{
+        clearTimeout(timeoutRef.current);
+        setNodesRef.current((nds)=>nds.map((n)=>n.id === nodeId ? {
+                    ...n,
+                    data: {
+                        ...n.data,
+                        addTaskLoading: true
+                    }
+                } : n));
+        timeoutRef.current = setTimeout(()=>{
+            setNodesRef.current((nds)=>nds.map((n)=>n.id === nodeId ? {
+                        ...n,
+                        data: {
+                            ...n.data,
+                            addTaskLoading: false
+                        }
+                    } : n));
+        }, 2000);
+    }, []);
     const initialNodes = (0, external_react_namespaceObject.useMemo)(()=>[
             {
                 id: 'loading-stage-empty',
@@ -1972,7 +1956,10 @@ const AddTaskLoadingStory = ()=>{
                         tasks: []
                     },
                     addTaskLoading: true,
-                    taskOptions: []
+                    taskOptions: [],
+                    onAddTaskFromToolbox: (taskItem)=>{
+                        handleAddTaskFromToolbox('loading-stage-empty', taskItem);
+                    }
                 }
             },
             {
@@ -1997,11 +1984,17 @@ const AddTaskLoadingStory = ()=>{
                         ]
                     },
                     addTaskLoading: false,
-                    taskOptions: loadedTaskOptionsWithChildren
+                    taskOptions: loadedTaskOptionsWithChildren,
+                    onAddTaskFromToolbox: (taskItem)=>{
+                        handleAddTaskFromToolbox('loading-stage-children', taskItem);
+                    }
                 }
             }
-        ], []);
-    const [nodesState, setNodes, onNodesChange] = (0, react_cjs_namespaceObject.useNodesState)(initialNodes);
+        ], [
+        handleAddTaskFromToolbox
+    ]);
+    const [nodes, setNodes, onNodesChange] = (0, react_cjs_namespaceObject.useNodesState)(initialNodes);
+    setNodesRef.current = setNodes;
     const [edges, setEdges, onEdgesChange] = (0, react_cjs_namespaceObject.useEdgesState)([]);
     (0, external_react_namespaceObject.useEffect)(()=>{
         const timeout = setTimeout(()=>{
@@ -2018,41 +2011,7 @@ const AddTaskLoadingStory = ()=>{
     }, [
         setNodes
     ]);
-    const timeoutRef = (0, external_react_namespaceObject.useRef)(void 0);
     (0, external_react_namespaceObject.useEffect)(()=>()=>clearTimeout(timeoutRef.current), []);
-    const handleAddTaskFromToolbox = (0, external_react_namespaceObject.useCallback)((nodeId, _taskItem)=>{
-        clearTimeout(timeoutRef.current);
-        setNodes((nds)=>nds.map((n)=>n.id === nodeId ? {
-                    ...n,
-                    data: {
-                        ...n.data,
-                        addTaskLoading: true
-                    }
-                } : n));
-        timeoutRef.current = setTimeout(()=>{
-            setNodes((nds)=>nds.map((n)=>n.id === nodeId ? {
-                        ...n,
-                        data: {
-                            ...n.data,
-                            addTaskLoading: false
-                        }
-                    } : n));
-        }, 2000);
-    }, [
-        setNodes
-    ]);
-    const nodesWithHandler = (0, external_react_namespaceObject.useMemo)(()=>nodesState.map((node)=>({
-                ...node,
-                data: {
-                    ...node.data,
-                    onAddTaskFromToolbox: (taskItem)=>{
-                        handleAddTaskFromToolbox(node.id, taskItem);
-                    }
-                }
-            })), [
-        nodesState,
-        handleAddTaskFromToolbox
-    ]);
     const onConnect = (0, external_react_namespaceObject.useCallback)((connection)=>setEdges((eds)=>(0, react_cjs_namespaceObject.addEdge)(connection, eds)), [
         setEdges
     ]);
@@ -2063,7 +2022,7 @@ const AddTaskLoadingStory = ()=>{
         },
         children: /*#__PURE__*/ (0, jsx_runtime_namespaceObject.jsx)(react_cjs_namespaceObject.ReactFlowProvider, {
             children: /*#__PURE__*/ (0, jsx_runtime_namespaceObject.jsx)(external_BaseCanvas_index_cjs_namespaceObject.BaseCanvas, {
-                nodes: nodesWithHandler,
+                nodes: nodes,
                 edges: edges,
                 onNodesChange: onNodesChange,
                 onEdgesChange: onEdgesChange,

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

@@ -1 +1 @@
-{"version":3,"file":"StageNode.stories.d.ts","sourceRoot":"","sources":["../../../../src/canvas/components/StageNode/StageNode.stories.tsx"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,kBAAkB,CAAC;AAyBvD,OAAO,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AAGxC,QAAA,MAAM,IAAI,EAAE,IAAI,CAAC,OAAO,SAAS,CA0FD,CAAC;AAEjC,eAAe,IAAI,CAAC;AACpB,KAAK,KAAK,GAAG,QAAQ,CAAC,OAAO,IAAI,CAAC,CAAC;AA4CnC,eAAO,MAAM,OAAO,EAAE,KAoHrB,CAAC;AAEF,eAAO,MAAM,aAAa,EAAE,KAsE3B,CAAC;AAEF,eAAO,MAAM,eAAe,EAAE,KA8M7B,CAAC;AAEF,eAAO,MAAM,yBAAyB,EAAE,KA+HvC,CAAC;AAEF,eAAO,MAAM,sBAAsB,EAAE,KAoKpC,CAAC;AA8GF,eAAO,MAAM,uBAAuB,EAAE,KAOrC,CAAC;AAkVF,eAAO,MAAM,kBAAkB,EAAE,KAOhC,CAAC;AA2HF,eAAO,MAAM,kBAAkB,EAAE,KAOhC,CAAC;AA0KF,eAAO,MAAM,cAAc,EAAE,KAO5B,CAAC;AAEF,eAAO,MAAM,UAAU,EAAE,KAiHxB,CAAC"}
+{"version":3,"file":"StageNode.stories.d.ts","sourceRoot":"","sources":["../../../../src/canvas/components/StageNode/StageNode.stories.tsx"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,kBAAkB,CAAC;AAyBvD,OAAO,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AAmDxC,QAAA,MAAM,IAAI,EAAE,IAAI,CAAC,OAAO,SAAS,CA8CD,CAAC;AAEjC,eAAe,IAAI,CAAC;AACpB,KAAK,KAAK,GAAG,QAAQ,CAAC,OAAO,IAAI,CAAC,CAAC;AA4CnC,eAAO,MAAM,OAAO,EAAE,KAoHrB,CAAC;AAEF,eAAO,MAAM,aAAa,EAAE,KAsE3B,CAAC;AAEF,eAAO,MAAM,eAAe,EAAE,KA8M7B,CAAC;AAEF,eAAO,MAAM,yBAAyB,EAAE,KA+HvC,CAAC;AAEF,eAAO,MAAM,sBAAsB,EAAE,KAoKpC,CAAC;AAsGF,eAAO,MAAM,uBAAuB,EAAE,KAOrC,CAAC;AA0UF,eAAO,MAAM,kBAAkB,EAAE,KAOhC,CAAC;AA4GF,eAAO,MAAM,kBAAkB,EAAE,KAOhC,CAAC;AAqJF,eAAO,MAAM,cAAc,EAAE,KAO5B,CAAC;AAEF,eAAO,MAAM,UAAU,EAAE,KAiHxB,CAAC"}