feat: workflow agent node permissions tab (#16092)

packages/twenty-front/src/modules/workflow/workflow-steps/hooks/__tests__/useDeleteStep.test.tsx

@@ -1,5 +1,9 @@
+import React from 'react';
+
 import { useDeleteStep } from '@/workflow/workflow-steps/hooks/useDeleteStep';
 import { renderHook } from '@testing-library/react';
+import { RecoilRoot } from 'recoil';
+import { WorkflowVisualizerComponentInstanceContext } from '../../../workflow-diagram/states/contexts/WorkflowVisualizerComponentInstanceContext';
 
 const mockDeleteWorkflowVersionStep = jest.fn();
 const mockGetUpdatableWorkflowVersion = jest.fn();
@@ -33,6 +37,36 @@ jest.mock('@/command-menu/hooks/useCommandMenu', () => ({
   }),
 }));
 
+jest.mock('@/workflow/hooks/useWorkflowWithCurrentVersion', () => ({
+  useWorkflowWithCurrentVersion: () => undefined,
+}));
+
+jest.mock(
+  '@/workflow/workflow-steps/workflow-actions/ai-agent-action/hooks/useResetWorkflowAiAgentPermissionsStateOnCommandMenuClose',
+  () => ({
+    useResetWorkflowAiAgentPermissionsStateOnCommandMenuClose: () => ({
+      resetPermissionState: jest.fn(),
+    }),
+  }),
+);
+
+const wrapper = ({ children }: { children: React.ReactNode }) => {
+  const workflowVisualizerComponentInstanceId =
+    'workflow-visualizer-instance-id';
+
+  return (
+    <RecoilRoot>
+      <WorkflowVisualizerComponentInstanceContext.Provider
+        value={{
+          instanceId: workflowVisualizerComponentInstanceId,
+        }}
+      >
+        {children}
+      </WorkflowVisualizerComponentInstanceContext.Provider>
+    </RecoilRoot>
+  );
+};
+
 describe('useDeleteStep', () => {
   beforeEach(() => {
     jest.clearAllMocks();
@@ -55,7 +89,9 @@ describe('useDeleteStep', () => {
       },
     });
 
-    const { result } = renderHook(() => useDeleteStep());
+    const { result } = renderHook(() => useDeleteStep(), {
+      wrapper,
+    });
     await result.current.deleteStep(mockStepId);
 
     expect(mockGetUpdatableWorkflowVersion).toHaveBeenCalled();

packages/twenty-front/src/modules/workflow/workflow-steps/hooks/useDeleteStep.ts

@@ -1,19 +1,36 @@
 import { useCommandMenu } from '@/command-menu/hooks/useCommandMenu';
+import { useRecoilComponentValue } from '@/ui/utilities/state/component-state/hooks/useRecoilComponentValue';
 import { useGetUpdatableWorkflowVersionOrThrow } from '@/workflow/hooks/useGetUpdatableWorkflowVersionOrThrow';
 import { useStepsOutputSchema } from '@/workflow/hooks/useStepsOutputSchema';
+import { useWorkflowWithCurrentVersion } from '@/workflow/hooks/useWorkflowWithCurrentVersion';
+import { workflowVisualizerWorkflowIdComponentState } from '@/workflow/states/workflowVisualizerWorkflowIdComponentState';
 import { useDeleteWorkflowVersionStep } from '@/workflow/workflow-steps/hooks/useDeleteWorkflowVersionStep';
+import { useResetWorkflowAiAgentPermissionsStateOnCommandMenuClose } from '@/workflow/workflow-steps/workflow-actions/ai-agent-action/hooks/useResetWorkflowAiAgentPermissionsStateOnCommandMenuClose';
+import { isDefined } from 'twenty-shared/utils';
 
 export const useDeleteStep = () => {
+  const { resetPermissionState } =
+    useResetWorkflowAiAgentPermissionsStateOnCommandMenuClose();
   const { deleteWorkflowVersionStep } = useDeleteWorkflowVersionStep();
   const { deleteStepsOutputSchema } = useStepsOutputSchema();
 
   const { getUpdatableWorkflowVersion } =
     useGetUpdatableWorkflowVersionOrThrow();
   const { closeCommandMenu } = useCommandMenu();
+  const workflowVisualizerWorkflowId = useRecoilComponentValue(
+    workflowVisualizerWorkflowIdComponentState,
+  );
+  const workflow = useWorkflowWithCurrentVersion(workflowVisualizerWorkflowId);
 
   const deleteStep = async (stepId: string) => {
     const workflowVersionId = await getUpdatableWorkflowVersion();
 
+    const isAiAgentStep =
+      isDefined(workflow?.currentVersion?.steps) &&
+      workflow.currentVersion.steps.some(
+        (step) => step.id === stepId && step.type === 'AI_AGENT',
+      );
+
     await deleteWorkflowVersionStep({
       workflowVersionId,
       stepId,
@@ -25,6 +42,10 @@ export const useDeleteStep = () => {
       stepIds: [stepId],
       workflowVersionId,
     });
+
+    if (isAiAgentStep) {
+      resetPermissionState();
+    }
   };
 
   return {