From d17e9092617de474a18c530d010a6a6fa3f99f57 Mon Sep 17 00:00:00 2001
From: Radoslaw Szwajkowski <rszwajko@redhat.com>
Date: Wed, 17 Apr 2024 20:24:17 +0200
Subject: [PATCH] Re-enable fetching applications after application import

After successful import re-enable useFetchApplications() hook for the
next 15 seconds.

Signed-off-by: Radoslaw Szwajkowski <rszwajko@redhat.com>
---
 .../applications-table/applications-table.tsx | 32 ++++++-------------
 client/src/app/queries/applications.ts        |  6 ++--
 2 files changed, 13 insertions(+), 25 deletions(-)

diff --git a/client/src/app/pages/applications/applications-table/applications-table.tsx b/client/src/app/pages/applications/applications-table/applications-table.tsx
index 0e2afe6cd..61942abc7 100644
--- a/client/src/app/pages/applications/applications-table/applications-table.tsx
+++ b/client/src/app/pages/applications/applications-table/applications-table.tsx
@@ -66,7 +66,6 @@ import { checkAccess } from "@app/utils/rbac-utils";
 import WarningTriangleIcon from "@patternfly/react-icons/dist/esm/icons/warning-triangle-icon";
 
 // Hooks
-import { useQueryClient } from "@tanstack/react-query";
 import {
   deserializeFilterUrlParams,
   useLocalTableControls,
@@ -108,6 +107,7 @@ import { KebabDropdown } from "@app/components/KebabDropdown";
 import { useFetchArchetypes } from "@app/queries/archetypes";
 import { ApplicationFormModal } from "../components/application-form";
 import { ManageColumnsToolbar } from "./components/manage-columns-toolbar";
+import dayjs from "dayjs";
 
 export const ApplicationsTable: React.FC = () => {
   const { t } = useTranslation();
@@ -207,12 +207,17 @@ export const ApplicationsTable: React.FC = () => {
   const [reviewToDiscard, setReviewToDiscard] =
     React.useState<Application | null>(null);
 
+  const [endOfAppImportPeriod, setEndOfAppImportPeriod] = useState<dayjs.Dayjs>(
+    dayjs()
+  );
+
   const {
     data: applications,
     isFetching: isFetchingApplications,
     error: applicationsFetchError,
-    refetch: fetchApplications,
-  } = useFetchApplications(!hasActiveTasks);
+  } = useFetchApplications(() =>
+    hasActiveTasks || dayjs().isBefore(endOfAppImportPeriod) ? 5000 : false
+  );
 
   const { assessments, isFetching: isFetchingAssessments } =
     useFetchAssessments();
@@ -498,8 +503,6 @@ export const ApplicationsTable: React.FC = () => {
     isSelectionEnabled: true,
   });
 
-  const queryClient = useQueryClient();
-
   const {
     currentPageItems,
     numRenderedColumns,
@@ -559,10 +562,6 @@ export const ApplicationsTable: React.FC = () => {
     tasksWriteAccess = checkAccess(userScopes, tasksWriteScopes),
     reviewsWriteAccess = checkAccess(userScopes, reviewsWriteScopes);
 
-  const areAppsInWaves = selectedRows.some(
-    (application) => application.migrationWave !== null
-  );
-
   const importDropdownItems = importWriteAccess
     ? [
         <DropdownItem
@@ -1131,19 +1130,6 @@ export const ApplicationsTable: React.FC = () => {
           documentId={taskToView?.task}
           onClose={() => setTaskToView(undefined)}
         />
-        <Modal
-          isOpen={isApplicationImportModalOpen}
-          variant="medium"
-          title={t("dialog.title.importApplicationFile")}
-          onClose={() => setIsApplicationImportModalOpen((current) => !current)}
-        >
-          <ImportApplicationsForm
-            onSaved={() => {
-              setIsApplicationImportModalOpen(false);
-              fetchApplications();
-            }}
-          />
-        </Modal>
         <Modal
           isOpen={isDependenciesModalOpen}
           variant="medium"
@@ -1168,7 +1154,7 @@ export const ApplicationsTable: React.FC = () => {
           <ImportApplicationsForm
             onSaved={() => {
               setIsApplicationImportModalOpen(false);
-              fetchApplications();
+              setEndOfAppImportPeriod(dayjs().add(15, "s"));
             }}
           />
         </Modal>
diff --git a/client/src/app/queries/applications.ts b/client/src/app/queries/applications.ts
index 12639b102..153a13d01 100644
--- a/client/src/app/queries/applications.ts
+++ b/client/src/app/queries/applications.ts
@@ -32,13 +32,15 @@ interface DownloadOptions {
   mimeType: MimeType;
 }
 
-export const useFetchApplications = (refetchDisabled: boolean = false) => {
+export const useFetchApplications = (
+  refetchInterval: number | false | (() => number | false) = 5000
+) => {
   const queryClient = useQueryClient();
   const { isLoading, error, refetch, data } = useQuery({
     initialData: [],
     queryKey: [ApplicationsQueryKey],
     queryFn: getApplications,
-    refetchInterval: !refetchDisabled ? 5000 : false,
+    refetchInterval: refetchInterval,
     onSuccess: () => {
       queryClient.invalidateQueries([assessmentsByItemIdQueryKey]);
     },