From a65e48b98adef0550d046683b882ff92cb45f46f Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?R=C4=B1dvan=20Akca?= <akca.ridvann@gmail.com>
Date: Fri, 5 Apr 2024 16:10:32 +0200
Subject: [PATCH] fix: refetch app flows after delete and duplicate

---
 packages/web/src/components/AppFlows/index.jsx  |  2 +-
 .../src/components/FlowContextMenu/index.jsx    | 17 ++++++++++++-----
 packages/web/src/components/FlowRow/index.jsx   |  6 ++++--
 3 files changed, 17 insertions(+), 8 deletions(-)

diff --git a/packages/web/src/components/AppFlows/index.jsx b/packages/web/src/components/AppFlows/index.jsx
index 23ea2f14..ed984d2d 100644
--- a/packages/web/src/components/AppFlows/index.jsx
+++ b/packages/web/src/components/AppFlows/index.jsx
@@ -47,7 +47,7 @@ function AppFlows(props) {
   return (
     <>
       {flows?.map((appFlow) => (
-        <AppFlowRow key={appFlow.id} flow={appFlow} />
+        <AppFlowRow key={appFlow.id} flow={appFlow} appKey={appKey} />
       ))}
 
       {pageInfo && pageInfo.totalPages > 1 && (
diff --git a/packages/web/src/components/FlowContextMenu/index.jsx b/packages/web/src/components/FlowContextMenu/index.jsx
index 1d124ceb..9be5ef9f 100644
--- a/packages/web/src/components/FlowContextMenu/index.jsx
+++ b/packages/web/src/components/FlowContextMenu/index.jsx
@@ -2,6 +2,8 @@ import PropTypes from 'prop-types';
 import { useMutation } from '@apollo/client';
 import Menu from '@mui/material/Menu';
 import MenuItem from '@mui/material/MenuItem';
+import { useQueryClient } from '@tanstack/react-query';
+
 import useEnqueueSnackbar from 'hooks/useEnqueueSnackbar';
 import * as React from 'react';
 import { Link } from 'react-router-dom';
@@ -13,17 +15,20 @@ import { DUPLICATE_FLOW } from 'graphql/mutations/duplicate-flow';
 import useFormatMessage from 'hooks/useFormatMessage';
 
 function ContextMenu(props) {
-  const { flowId, onClose, anchorEl, onDuplicateFlow, onDeleteFlow } = props;
+  const { flowId, onClose, anchorEl, onDuplicateFlow, onDeleteFlow, appKey } =
+    props;
   const enqueueSnackbar = useEnqueueSnackbar();
-  const [deleteFlow] = useMutation(DELETE_FLOW);
-  const [duplicateFlow] = useMutation(DUPLICATE_FLOW);
   const formatMessage = useFormatMessage();
+  const queryClient = useQueryClient();
+  const [duplicateFlow] = useMutation(DUPLICATE_FLOW);
+  const [deleteFlow] = useMutation(DELETE_FLOW);
 
   const onFlowDuplicate = React.useCallback(async () => {
     await duplicateFlow({
       variables: { input: { id: flowId } },
     });
 
+    await queryClient.invalidateQueries({ queryKey: ['appFlows', appKey] });
     enqueueSnackbar(formatMessage('flow.successfullyDuplicated'), {
       variant: 'success',
       SnackbarProps: {
@@ -33,7 +38,7 @@ function ContextMenu(props) {
 
     onDuplicateFlow?.();
     onClose();
-  }, [flowId, onClose, duplicateFlow, onDuplicateFlow]);
+  }, [flowId, onClose, duplicateFlow, queryClient, onDuplicateFlow]);
 
   const onFlowDelete = React.useCallback(async () => {
     await deleteFlow({
@@ -49,13 +54,14 @@ function ContextMenu(props) {
       },
     });
 
+    await queryClient.invalidateQueries({ queryKey: ['appFlows', appKey] });
     enqueueSnackbar(formatMessage('flow.successfullyDeleted'), {
       variant: 'success',
     });
 
     onDeleteFlow?.();
     onClose();
-  }, [flowId, onClose, deleteFlow, onDeleteFlow]);
+  }, [flowId, onClose, deleteFlow, queryClient, onDeleteFlow]);
 
   return (
     <Menu
@@ -100,6 +106,7 @@ ContextMenu.propTypes = {
   ]).isRequired,
   onDeleteFlow: PropTypes.func,
   onDuplicateFlow: PropTypes.func,
+  appKey: PropTypes.string.isRequired,
 };
 
 export default ContextMenu;
diff --git a/packages/web/src/components/FlowRow/index.jsx b/packages/web/src/components/FlowRow/index.jsx
index 75af128a..e93b3a9b 100644
--- a/packages/web/src/components/FlowRow/index.jsx
+++ b/packages/web/src/components/FlowRow/index.jsx
@@ -1,4 +1,5 @@
 import * as React from 'react';
+import PropTypes from 'prop-types';
 import { Link } from 'react-router-dom';
 import Card from '@mui/material/Card';
 import IconButton from '@mui/material/IconButton';
@@ -6,7 +7,6 @@ import CardActionArea from '@mui/material/CardActionArea';
 import Chip from '@mui/material/Chip';
 import MoreHorizIcon from '@mui/icons-material/MoreHoriz';
 import { DateTime } from 'luxon';
-import PropTypes from 'prop-types';
 
 import FlowAppIcons from 'components/FlowAppIcons';
 import FlowContextMenu from 'components/FlowContextMenu';
@@ -37,7 +37,7 @@ function FlowRow(props) {
   const formatMessage = useFormatMessage();
   const contextButtonRef = React.useRef(null);
   const [anchorEl, setAnchorEl] = React.useState(null);
-  const { flow, onDuplicateFlow, onDeleteFlow } = props;
+  const { flow, onDuplicateFlow, onDeleteFlow, appKey } = props;
   const handleClose = () => {
     setAnchorEl(null);
   };
@@ -116,6 +116,7 @@ function FlowRow(props) {
           anchorEl={anchorEl}
           onDeleteFlow={onDeleteFlow}
           onDuplicateFlow={onDuplicateFlow}
+          appKey={appKey}
         />
       )}
     </>
@@ -126,6 +127,7 @@ FlowRow.propTypes = {
   flow: FlowPropType.isRequired,
   onDeleteFlow: PropTypes.func,
   onDuplicateFlow: PropTypes.func,
+  appKey: PropTypes.string.isRequired,
 };
 
 export default FlowRow;