feat #1117: Add EnableAll menu item when routes include multiple disa…

…bled steps
KaotoIO · Oct 11, 2024 · b399484 · b399484
1 parent c9f7a97
commit b399484
Showing 1 changed file with 51 additions and 13 deletions.
diff --git a/packages/ui/src/components/Visualization/Custom/ContextMenu/ItemDisableStep.tsx b/packages/ui/src/components/Visualization/Custom/ContextMenu/ItemDisableStep.tsx
@@ -1,6 +1,6 @@
-import { BanIcon, CheckIcon } from '@patternfly/react-icons';
-import { ContextMenuItem } from '@patternfly/react-topology';
-import { FunctionComponent, PropsWithChildren, useCallback, useContext } from 'react';
+import { BanIcon, CheckIcon, PowerOffIcon } from '@patternfly/react-icons';
+import { ContextMenuItem, useVisualizationController } from '@patternfly/react-topology';
+import { FunctionComponent, PropsWithChildren, useCallback, useContext, useMemo } from 'react';
 import { IDataTestID } from '../../../../models';
 import { IVisualizationNode } from '../../../../models/visualization/base-visual-entity';
 import { EntitiesContext } from '../../../../providers/entities.provider';
@@ -10,11 +10,32 @@ interface ItemDisableStepProps extends PropsWithChildren<IDataTestID> {
   vizNode: IVisualizationNode;
 }
 
+// eslint-disable-next-line @typescript-eslint/no-explicit-any
+function findAllDisabledNodes(node: any): IVisualizationNode[] | null {
+  const allDisabledNodes = [];
+  if (node?.data?.vizNode?.getComponentSchema()?.definition?.disabled) {
+    allDisabledNodes.push(node.data.vizNode);
+  }
+  if (node?.children) {
+    for (const child of node.children) {
+      const result = findAllDisabledNodes(child);
+      allDisabledNodes.push(...(result || []));
+    }
+  }
+  return allDisabledNodes;
+}
+
 export const ItemDisableStep: FunctionComponent<ItemDisableStepProps> = (props) => {
   const entitiesContext = useContext(EntitiesContext);
+  const controller = useVisualizationController();
 
   const isDisabled = !!props.vizNode.getComponentSchema()?.definition?.disabled;
 
+  const allDisabledNodes = useMemo(() => {
+    return findAllDisabledNodes(controller.getGraph()) || [];
+  }, [controller]);
+  const isMultiDisabled = allDisabledNodes.length > 1;
+
   const onToggleDisableNode = useCallback(() => {
     const newModel = props.vizNode.getComponentSchema()?.definition || {};
     setValue(newModel, 'disabled', !isDisabled);
@@ -23,17 +44,34 @@ export const ItemDisableStep: FunctionComponent<ItemDisableStepProps> = (props)
     entitiesContext?.updateEntitiesFromCamelResource();
   }, [entitiesContext, isDisabled, props.vizNode]);
 
+  const onEnableAllNodes = useCallback(() => {
+    allDisabledNodes.forEach((node) => {
+      const newModel = node.getComponentSchema()?.definition || {};
+      setValue(newModel, 'disabled', false);
+      node.updateModel(newModel);
+    });
+
+    entitiesContext?.updateEntitiesFromCamelResource();
+  }, [allDisabledNodes, entitiesContext]);
+
   return (
-    <ContextMenuItem onClick={onToggleDisableNode} data-testid={props['data-testid']}>
-      {isDisabled ? (
-        <>
-          <CheckIcon /> Enable
-        </>
-      ) : (
-        <>
-          <BanIcon /> Disable
-        </>
+    <>
+      <ContextMenuItem onClick={onToggleDisableNode} data-testid={props['data-testid']}>
+        {isDisabled ? (
+          <>
+            <CheckIcon /> Enable
+          </>
+        ) : (
+          <>
+            <BanIcon /> Disable
+          </>
+        )}
+      </ContextMenuItem>
+      {isDisabled && isMultiDisabled && (
+        <ContextMenuItem onClick={onEnableAllNodes}>
+          <PowerOffIcon /> Enable All
+        </ContextMenuItem>
       )}
-    </ContextMenuItem>
+    </>
   );
 };