@easyteam/auto-scheduler-modal-ui 0.1.1 → 0.1.2

package/dist/index.js

@@ -889,6 +889,7 @@ var SCHEDULE_STEPS = [
   { title: "Finalizing schedule" }
 ];
 var STEP_INTERVAL_MS = 1200;
+var POLL_INTERVAL_MS = 2500;
 var OPTIMIZING_STEP_INDEX = 2;
 var NETWORK_ERROR_FAILURE = {
   violatedConstraints: [
@@ -1433,19 +1434,55 @@ async function getToken(fullUrl, headers) {
   });
   return (_a = tokenRes.data) == null ? void 0 : _a.token;
 }
-async function solveSchedule(baseURL, payload, options) {
+async function submitSolveJob(baseURL, payload, options) {
   const cleanBaseUrl = baseURL.replace(/\/$/, "");
   const headers = {};
   if (options == null ? void 0 : options.authorization) {
     headers.Authorization = options.authorization;
   }
   const response = await axiosInstance.post(
-    `${cleanBaseUrl}/api/schedule/solve`,
+    `${cleanBaseUrl}/api/schedule/solve-async`,
     payload,
     { headers: Object.keys(headers).length > 0 ? headers : void 0 }
   );
   return response.data;
 }
+async function getJobStatus(baseURL, jobId, options) {
+  const cleanBaseUrl = baseURL.replace(/\/$/, "");
+  const headers = {};
+  if (options == null ? void 0 : options.authorization) {
+    headers.Authorization = options.authorization;
+  }
+  const response = await axiosInstance.get(
+    `${cleanBaseUrl}/api/schedule/status/${jobId}`,
+    { headers: Object.keys(headers).length > 0 ? headers : void 0 }
+  );
+  return response.data;
+}
+async function getJobResult(baseURL, jobId, options) {
+  const cleanBaseUrl = baseURL.replace(/\/$/, "");
+  const headers = {};
+  if (options == null ? void 0 : options.authorization) {
+    headers.Authorization = options.authorization;
+  }
+  const response = await axiosInstance.get(
+    `${cleanBaseUrl}/api/schedule/result/${jobId}`,
+    { headers: Object.keys(headers).length > 0 ? headers : void 0 }
+  );
+  return response.data;
+}
+async function getJobSolution(baseURL, jobId, options) {
+  const cleanBaseUrl = baseURL.replace(/\/$/, "");
+  const headers = {};
+  if (options == null ? void 0 : options.authorization) {
+    headers.Authorization = options.authorization;
+  }
+  const response = await axiosInstance.get(
+    `${cleanBaseUrl}/api/schedule/solution/${jobId}`,
+    { headers: Object.keys(headers).length > 0 ? headers : void 0 }
+  );
+  return response.data;
+}
 
 // src/utils/violationMessages.ts
 var SYSTEM_CONSTRAINT_MESSAGES = {
@@ -1831,6 +1868,7 @@ function AutoSchedulerModal({
   const [lastFailure, setLastFailure] = useState2(null);
   const [selectedLocationIds, setSelectedLocationIds] = useState2([]);
   const timersRef = useRef([]);
+  const pollIntervalRef = useRef(null);
   const runIdRef = useRef(0);
   const selectedByJurisdiction = useMemo2(
     () => jurisdictions.map((jurisdiction) => ({
@@ -1896,6 +1934,10 @@ function AutoSchedulerModal({
   const clearTimers = () => {
     timersRef.current.forEach((timerId) => clearTimeout(timerId));
     timersRef.current = [];
+    if (pollIntervalRef.current) {
+      clearInterval(pollIntervalRef.current);
+      pollIntervalRef.current = null;
+    }
   };
   const resetProgress = () => {
     setActiveStepIndex(0);
@@ -2029,7 +2071,6 @@ function AutoSchedulerModal({
     });
     const runId = resetBeforeSubmit();
     schedule(STEP_INTERVAL_MS, async () => {
-      var _a, _b;
       setCompletedStepIndex(0);
       setActiveStepIndex(1);
       try {
@@ -2037,51 +2078,47 @@ function AutoSchedulerModal({
         if (!token) {
           throw new Error("Failed to get auto-scheduler token");
         }
-        const response = await solveSchedule(
-          baseURL,
-          payload,
-          {
-            authorization: `Bearer ${token}`
-          }
-        );
+        const auth = { authorization: `Bearer ${token}` };
+        const { jobId } = await submitSolveJob(baseURL, payload, auth);
         if (runIdRef.current !== runId) {
           return;
         }
-        if (!response.explanation.isFeasible) {
-          const result = mapViolationsToResult(
-            (_a = response.explanation.violations) != null ? _a : [],
-            response.explanation.summary
-          );
-          const selectedLocations = jurisdictions.flatMap((j) => j.locations).filter(
-            (l) => selectedLocationIds.includes(l.id)
-          );
-          const recommendationPayload = generateRecommendationsURLAndHeaders ? {
-            context: {
-              selectedLocationIds,
-              selectedConstraintIds,
-              constraintValues,
-              selectedOptimizationIds,
-              optimizationValues,
-              timezone: (_b = selectedLocations[0]) == null ? void 0 : _b.timezone
-            },
-            locations: selectedLocations,
-            employees,
-            openShifts: openShifts.filter((s) => !s.employeeId),
-            assignedShifts: openShifts.filter((s) => Boolean(s.employeeId)),
-            timeOffs: timeOffs != null ? timeOffs : [],
-            explanation: response.explanation
-          } : void 0;
-          setLastFailure({
-            ...result,
-            recommendedFixes: [],
-            recommendationPayload
-          });
-          setScreen("failed");
-        } else {
-          setCompletedStepIndex(1);
-          setActiveStepIndex(OPTIMIZING_STEP_INDEX);
-          afterSuccessSteps(runId, response);
-        }
+        setCompletedStepIndex(1);
+        setActiveStepIndex(OPTIMIZING_STEP_INDEX);
+        const pollForResult = async () => {
+          if (runIdRef.current !== runId) return;
+          try {
+            const statusRes = await getJobStatus(baseURL, jobId, auth);
+            if (runIdRef.current !== runId) return;
+            if (statusRes.status === "NOT_SOLVING" || statusRes.status === "TERMINATED_EARLY") {
+              if (pollIntervalRef.current) {
+                clearInterval(pollIntervalRef.current);
+                pollIntervalRef.current = null;
+              }
+              const response = await getJobResult(baseURL, jobId, auth);
+              if (runIdRef.current !== runId) return;
+              let responseForHandling = response;
+              if (response.explanation.isFeasible) {
+                const solution = await getJobSolution(baseURL, jobId, auth);
+                if (runIdRef.current !== runId) return;
+                responseForHandling = { ...response, solution };
+              }
+              handleSolveResult(runId, responseForHandling);
+              return;
+            }
+          } catch (err) {
+            if (runIdRef.current !== runId) return;
+            if (pollIntervalRef.current) {
+              clearInterval(pollIntervalRef.current);
+              pollIntervalRef.current = null;
+            }
+            console.error(err);
+            setLastFailure(NETWORK_ERROR_FAILURE);
+            setScreen("failed");
+          }
+        };
+        pollForResult();
+        pollIntervalRef.current = setInterval(pollForResult, POLL_INTERVAL_MS);
       } catch (err) {
         if (runIdRef.current !== runId) {
           return;
@@ -2092,6 +2129,42 @@ function AutoSchedulerModal({
       }
     }, runId);
   };
+  const handleSolveResult = (runId, response) => {
+    var _a, _b;
+    if (!response.explanation.isFeasible) {
+      const result = mapViolationsToResult(
+        (_a = response.explanation.violations) != null ? _a : [],
+        response.explanation.summary
+      );
+      const selectedLocations = jurisdictions.flatMap((j) => j.locations).filter(
+        (l) => selectedLocationIds.includes(l.id)
+      );
+      const recommendationPayload = generateRecommendationsURLAndHeaders ? {
+        context: {
+          selectedLocationIds,
+          selectedConstraintIds,
+          constraintValues,
+          selectedOptimizationIds,
+          optimizationValues,
+          timezone: (_b = selectedLocations[0]) == null ? void 0 : _b.timezone
+        },
+        locations: selectedLocations,
+        employees,
+        openShifts: openShifts.filter((s) => !s.employeeId),
+        assignedShifts: openShifts.filter((s) => Boolean(s.employeeId)),
+        timeOffs: timeOffs != null ? timeOffs : [],
+        explanation: response.explanation
+      } : void 0;
+      setLastFailure({
+        ...result,
+        recommendedFixes: [],
+        recommendationPayload
+      });
+      setScreen("failed");
+    } else {
+      afterSuccessSteps(runId, response);
+    }
+  };
   const handleAdjustConstraints = () => {
     clearTimers();
     setScreen("configure");