From 0722f52619784552a0b3202ba8badac4b0c6be70 Mon Sep 17 00:00:00 2001
From: Kartik Sharma <kartik.sharma@marcoaiot.com>
Date: Wed, 30 Jul 2025 16:56:27 +0530
Subject: [PATCH] Deleted Employee Still Visible Until Manual Refresh.

---
 src/hooks/useEmployees.js | 21 ++++++++++++++-------
 1 file changed, 14 insertions(+), 7 deletions(-)

diff --git a/src/hooks/useEmployees.js b/src/hooks/useEmployees.js
index b911d2dc..40db5105 100644
--- a/src/hooks/useEmployees.js
+++ b/src/hooks/useEmployees.js
@@ -241,9 +241,9 @@ export const useSuspendEmployee = ({
     (store) => store.localVariables.projectId
   );
   return useMutation({
-    mutationFn: (id) => {
+    mutationFn: async (employeeId) => {
       setemployeeLodaing(true);
-      return EmployeeRepository.deleteEmployee(id);
+      return await EmployeeRepository.deleteEmployee(employeeId);
     },
 
     onSuccess: () => {
@@ -254,14 +254,21 @@ export const useSuspendEmployee = ({
       });
       showToast("Employee deleted successfully.", "success");
       setIsDeleteModalOpen(false);
+
+      // Invalidate only the required employee-related queries
+      queryClient.invalidateQueries({ queryKey: ["employee", employeeId] });
+      queryClient.invalidateQueries({ queryKey: ["allEmployees"] });
+
+      if (selectedProjectId) {
+        queryClient.invalidateQueries({ queryKey: ["projectEmployees", selectedProjectId] });
+      }
     },
 
     onError: (error) => {
-      const message =
-        error.response?.data?.message ||
-        error.message ||
-        "An unexpected error occurred";
-      showToast(message, "error");
+      showToast(
+        error.response?.data?.message || error.message || "An unexpected error occurred",
+        "error"
+      );
       setIsDeleteModalOpen(false);
     },