diff --git a/packages/twenty-front/src/modules/workflow/workflow-diagram/components/WorkflowDiagramDefaultEdgeEditable.tsx b/packages/twenty-front/src/modules/workflow/workflow-diagram/components/WorkflowDiagramDefaultEdgeEditable.tsx index 2129f7cce0d..efbf80695b1 100644 --- a/packages/twenty-front/src/modules/workflow/workflow-diagram/components/WorkflowDiagramDefaultEdgeEditable.tsx +++ b/packages/twenty-front/src/modules/workflow/workflow-diagram/components/WorkflowDiagramDefaultEdgeEditable.tsx @@ -14,6 +14,7 @@ import { useStartNodeCreation } from '@/workflow/workflow-diagram/hooks/useStart import { type WorkflowDiagramEdge } from '@/workflow/workflow-diagram/types/WorkflowDiagram'; import { useCreateStep } from '@/workflow/workflow-steps/hooks/useCreateStep'; import { useDeleteEdge } from '@/workflow/workflow-steps/hooks/useDeleteEdge'; +import { useIsFeatureEnabled } from '@/workspace/hooks/useIsFeatureEnabled'; import { EdgeLabelRenderer, type EdgeProps, @@ -23,6 +24,7 @@ import { type MouseEvent, useContext } from 'react'; import { useSetRecoilState } from 'recoil'; import { isDefined } from 'twenty-shared/utils'; import { IconFilter, IconPlus, IconTrash } from 'twenty-ui/display'; +import { FeatureFlagKey } from '~/generated/graphql'; type WorkflowDiagramDefaultEdgeEditableProps = EdgeProps; @@ -37,6 +39,10 @@ export const WorkflowDiagramDefaultEdgeEditable = ({ markerStart, markerEnd, }: WorkflowDiagramDefaultEdgeEditableProps) => { + const isWorkflowBranchEnabled = useIsFeatureEnabled( + FeatureFlagKey.IS_WORKFLOW_BRANCH_ENABLED, + ); + const { isInRightDrawer } = useContext(ActionMenuContext); const { isEdgeHovered } = useEdgeHovered(); @@ -135,10 +141,14 @@ export const WorkflowDiagramDefaultEdgeEditable = ({ Icon: IconPlus, onClick: handleNodeButtonClick, }, - { - Icon: IconTrash, - onClick: handleDeleteBranch, - }, + ...(isWorkflowBranchEnabled + ? [ + { + Icon: IconTrash, + onClick: handleDeleteBranch, + }, + ] + : []), ]} selected={nodeCreationStarted} /> diff --git a/packages/twenty-front/src/modules/workflow/workflow-diagram/components/WorkflowDiagramFilterEdgeEditable.tsx b/packages/twenty-front/src/modules/workflow/workflow-diagram/components/WorkflowDiagramFilterEdgeEditable.tsx index 9332547de23..3ee717c3cad 100644 --- a/packages/twenty-front/src/modules/workflow/workflow-diagram/components/WorkflowDiagramFilterEdgeEditable.tsx +++ b/packages/twenty-front/src/modules/workflow/workflow-diagram/components/WorkflowDiagramFilterEdgeEditable.tsx @@ -28,6 +28,7 @@ import { import { getWorkflowDiagramNodeSelectedColors } from '@/workflow/workflow-diagram/utils/getWorkflowDiagramNodeSelectedColors'; import { useDeleteEdge } from '@/workflow/workflow-steps/hooks/useDeleteEdge'; import { useDeleteStep } from '@/workflow/workflow-steps/hooks/useDeleteStep'; +import { useIsFeatureEnabled } from '@/workspace/hooks/useIsFeatureEnabled'; import { css } from '@emotion/react'; import styled from '@emotion/styled'; import { useLingui } from '@lingui/react/macro'; @@ -48,6 +49,7 @@ import { } from 'twenty-ui/display'; import { IconButtonGroup } from 'twenty-ui/input'; import { MenuItem } from 'twenty-ui/navigation'; +import { FeatureFlagKey } from '~/generated/graphql'; type WorkflowDiagramFilterEdgeEditableProps = EdgeProps; @@ -94,6 +96,9 @@ export const WorkflowDiagramFilterEdgeEditable = ({ assertFilterEdgeDataOrThrow(data); const { t } = useLingui(); + const isWorkflowBranchEnabled = useIsFeatureEnabled( + FeatureFlagKey.IS_WORKFLOW_BRANCH_ENABLED, + ); const [edgePath, labelX, labelY] = getBezierPath({ sourceX, @@ -271,11 +276,13 @@ export const WorkflowDiagramFilterEdgeEditable = ({ LeftIcon={IconPlus} onClick={handleAddNodeButtonClick} /> - + {isWorkflowBranchEnabled && ( + + )} }