From 9c699a634be590895b533b89aa53f7a0c3b5375d Mon Sep 17 00:00:00 2001
From: Dave Richer <dave@imexsystems.ca>
Date: Fri, 12 Apr 2024 11:35:53 -0400
Subject: [PATCH] - optimization

Signed-off-by: Dave Richer <dave@imexsystems.ca>
---
 .../task-upsert-modal.container.jsx           | 40 ++++++++++---------
 1 file changed, 21 insertions(+), 19 deletions(-)

diff --git a/client/src/components/task-upsert-modal/task-upsert-modal.container.jsx b/client/src/components/task-upsert-modal/task-upsert-modal.container.jsx
index 5da4ae224..e39ef5de0 100644
--- a/client/src/components/task-upsert-modal/task-upsert-modal.container.jsx
+++ b/client/src/components/task-upsert-modal/task-upsert-modal.container.jsx
@@ -109,8 +109,18 @@ export function TaskUpsertModalContainer({ bodyshop, currentUser, taskUpsert, to
   /**
    * Handle refetch queries
    * @param taskObject
+   * @param jobId
    */
-  const handleRefetchQueries = (taskObject) => {
+  const handleRefetchQueries = (taskObject, jobId) => {
+    if (location.pathname.includes("/manage/jobs") && jobId) {
+      taskObject.refetchQueries.push({
+        query: GET_JOB_BY_PK,
+        variables: {
+          id: jobId
+        }
+      });
+    }
+
     if (query && Object.keys(query).length) {
       taskObject.refetchQueries.push(Object.keys(query)[0]);
     } else {
@@ -138,15 +148,16 @@ export function TaskUpsertModalContainer({ bodyshop, currentUser, taskUpsert, to
       refetchQueries: []
     };
 
-    // We need to make sure this is updated everywhere
-    taskObject.refetchQueries.push({
-      query: GET_JOB_BY_PK,
-      variables: {
-        id: jobId
-      }
-    });
+    if (location.pathname.includes("/manage/jobs") && jobId) {
+      taskObject.refetchQueries.push({
+        query: GET_JOB_BY_PK,
+        variables: {
+          id: jobId
+        }
+      });
+    }
 
-    handleRefetchQueries(taskObject);
+    handleRefetchQueries(taskObject, jobId);
 
     const taskData = await updateTask(taskObject);
 
@@ -210,16 +221,7 @@ export function TaskUpsertModalContainer({ bodyshop, currentUser, taskUpsert, to
       refetchQueries: []
     };
 
-    // We need to make sure this is updated everywhere
-    taskObject.refetchQueries.push({
-      query: GET_JOB_BY_PK,
-      variables: {
-        id: values.jobid
-      }
-    });
-
-    // We have a relationship query, so we need to refetch the query
-    handleRefetchQueries(taskObject);
+    handleRefetchQueries(taskObject, values.jobid);
 
     const newTaskData = await insertTask(taskObject);
     const newTask = newTaskData?.data?.insert_tasks?.returning[0];