From 3cac4ef927219814903cf3c885a528b62c6ad927 Mon Sep 17 00:00:00 2001
From: x032205 <x032205@gmail.com>
Date: Thu, 12 Jun 2025 11:43:32 -0400
Subject: [PATCH] Reviews

---
 .../hooks/api/secretScanningV2/mutations.tsx  |  1 +
 .../SecretScanningFindingsTable.tsx           |  1 +
 .../SecretScanningUpdateFindingModal.tsx      | 38 ++++++++++++-------
 3 files changed, 27 insertions(+), 13 deletions(-)

diff --git a/frontend/src/hooks/api/secretScanningV2/mutations.tsx b/frontend/src/hooks/api/secretScanningV2/mutations.tsx
index a936c2a4f..beb1cffa0 100644
--- a/frontend/src/hooks/api/secretScanningV2/mutations.tsx
+++ b/frontend/src/hooks/api/secretScanningV2/mutations.tsx
@@ -115,6 +115,7 @@ export const useTriggerSecretScanningDataSource = () => {
   });
 };
 
+// If possible, use useUpdateMultipleSecretScanningFinding instead.
 export const useUpdateSecretScanningFinding = () => {
   const queryClient = useQueryClient();
   return useMutation({
diff --git a/frontend/src/pages/secret-scanning/SecretScanningFindingsPage/components/SecretScanningFindingsTable.tsx b/frontend/src/pages/secret-scanning/SecretScanningFindingsPage/components/SecretScanningFindingsTable.tsx
index 2061c27c6..57bc57a4e 100644
--- a/frontend/src/pages/secret-scanning/SecretScanningFindingsPage/components/SecretScanningFindingsTable.tsx
+++ b/frontend/src/pages/secret-scanning/SecretScanningFindingsPage/components/SecretScanningFindingsTable.tsx
@@ -418,6 +418,7 @@ export const SecretScanningFindingsTable = ({ findings }: Props) => {
       <SecretScanningUpdateFindingModal
         isOpen={popUp.updateFinding.isOpen}
         onOpenChange={(isOpen) => handlePopUpToggle("updateFinding", isOpen)}
+        onComplete={() => setSelectedRows([])}
         findings={popUp.updateFinding.data}
       />
       {selectedRows.length > 0 && (
diff --git a/frontend/src/pages/secret-scanning/SecretScanningFindingsPage/components/SecretScanningUpdateFindingModal.tsx b/frontend/src/pages/secret-scanning/SecretScanningFindingsPage/components/SecretScanningUpdateFindingModal.tsx
index d7a17572a..25e5bf77e 100644
--- a/frontend/src/pages/secret-scanning/SecretScanningFindingsPage/components/SecretScanningUpdateFindingModal.tsx
+++ b/frontend/src/pages/secret-scanning/SecretScanningFindingsPage/components/SecretScanningUpdateFindingModal.tsx
@@ -17,14 +17,14 @@ import { SECRET_SCANNING_FINDING_STATUS_ICON_MAP } from "@app/helpers/secretScan
 import {
   SecretScanningFindingStatus,
   TSecretScanningFinding,
-  useUpdateMultipleSecretScanningFinding,
-  useUpdateSecretScanningFinding
+  useUpdateMultipleSecretScanningFinding
 } from "@app/hooks/api/secretScanningV2";
 
 type Props = {
   findings?: TSecretScanningFinding[];
   isOpen: boolean;
   onOpenChange: (isOpen: boolean) => void;
+  onComplete?: () => void;
 };
 
 const FormSchema = z.object({
@@ -40,7 +40,6 @@ type ContentProps = {
 };
 
 const Content = ({ findings, onComplete }: ContentProps) => {
-  const updateFinding = useUpdateSecretScanningFinding();
   const updateMultipleFindings = useUpdateMultipleSecretScanningFinding();
 
   const single = findings.length === 1;
@@ -67,12 +66,14 @@ const Content = ({ findings, onComplete }: ContentProps) => {
           }))
         );
       } else {
-        await updateFinding.mutateAsync({
-          ...data,
-          status: data.status,
-          findingId: findings[0].id,
-          projectId: findings[0].projectId
-        });
+        await updateMultipleFindings.mutateAsync([
+          {
+            ...data,
+            status: data.status,
+            findingId: findings[0].id,
+            projectId: findings[0].projectId
+          }
+        ]);
       }
 
       createNotification({
@@ -135,8 +136,8 @@ const Content = ({ findings, onComplete }: ContentProps) => {
       <div className="flex w-full flex-row-reverse justify-between gap-4 pt-4">
         <Button
           type="submit"
-          isLoading={updateFinding.isPending}
-          isDisabled={updateFinding.isPending}
+          isLoading={updateMultipleFindings.isPending}
+          isDisabled={updateMultipleFindings.isPending}
           colorSchema="secondary"
         >
           Update Status
@@ -149,7 +150,12 @@ const Content = ({ findings, onComplete }: ContentProps) => {
   );
 };
 
-export const SecretScanningUpdateFindingModal = ({ findings, isOpen, onOpenChange }: Props) => {
+export const SecretScanningUpdateFindingModal = ({
+  findings,
+  isOpen,
+  onOpenChange,
+  onComplete
+}: Props) => {
   if (!findings?.length) return null;
 
   return (
@@ -158,7 +164,13 @@ export const SecretScanningUpdateFindingModal = ({ findings, isOpen, onOpenChang
         title={`Update Finding${findings.length === 1 ? "" : "s"}`}
         subTitle="Update the status or leave remarks"
       >
-        <Content findings={findings} onComplete={() => onOpenChange(false)} />
+        <Content
+          findings={findings}
+          onComplete={() => {
+            onOpenChange(false);
+            if (onComplete) onComplete();
+          }}
+        />
       </ModalContent>
     </Modal>
   );