- optimization

Signed-off-by: Dave Richer <dave@imexsystems.ca>

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
+    if (location.pathname.includes("/manage/jobs") && jobId) {
-    taskObject.refetchQueries.push({
+      taskObject.refetchQueries.push({
-      query: GET_JOB_BY_PK,
+        query: GET_JOB_BY_PK,
-      variables: {
+        variables: {
-        id: jobId
+          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
+    handleRefetchQueries(taskObject, values.jobid);
-    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);
 
     const newTaskData = await insertTask(taskObject);
     const newTask = newTaskData?.data?.insert_tasks?.returning[0];