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

package/dist/index.cjs

@@ -905,6 +905,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: [
@@ -1437,14 +1438,65 @@ function buildSchedulePayload(input) {
   };
 }
 
-// src/api/solveSchedule.ts
+// src/api/api.ts
 var import_axios = __toESM(require("axios"), 1);
 var axiosInstance = import_axios.default.create({
   headers: { "Content-Type": "application/json" }
 });
-async function solveSchedule(baseURL, payload) {
+async function getToken(fullUrl, headers) {
+  var _a;
+  const tokenRes = await axiosInstance.get(fullUrl, {
+    ...headers ? { headers } : {}
+  });
+  return (_a = tokenRes.data) == null ? void 0 : _a.token;
+}
+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-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 response = await axiosInstance.post(`${cleanBaseUrl}/api/schedule/solve`, payload);
+  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;
 }
 
@@ -1799,7 +1851,10 @@ function AutoSchedulerModal({
   theme,
   cssVarsRoot,
   generateRecommendationsURLAndHeaders,
-  onSolution
+  getTokenURLAndHeaders,
+  onSolution,
+  getStoredConfigForLocations,
+  persistConfigForLocations
 }) {
   const baseTheme = (0, import_react4.useTheme)();
   const { chakraOverride, styleOverrides } = (0, import_react3.useMemo)(
@@ -1821,6 +1876,7 @@ function AutoSchedulerModal({
   const [lastFailure, setLastFailure] = (0, import_react3.useState)(null);
   const [selectedLocationIds, setSelectedLocationIds] = (0, import_react3.useState)([]);
   const timersRef = (0, import_react3.useRef)([]);
+  const pollIntervalRef = (0, import_react3.useRef)(null);
   const runIdRef = (0, import_react3.useRef)(0);
   const selectedByJurisdiction = (0, import_react3.useMemo)(
     () => jurisdictions.map((jurisdiction) => ({
@@ -1886,6 +1942,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);
@@ -1917,6 +1977,23 @@ function AutoSchedulerModal({
       setScreen("configure");
     }
   }, [isOpen, initialConfig]);
+  (0, import_react3.useEffect)(() => {
+    if (selectedLocationIds.length === 0 || initialConfig || !getStoredConfigForLocations) {
+      return;
+    }
+    const stored = getStoredConfigForLocations(selectedLocationIds);
+    if (stored) {
+      setSelectedConstraintIds(stored.selectedConstraintIds);
+      setConstraintValues(stored.constraintValues);
+      setSelectedOptimizationIds(stored.selectedOptimizationIds);
+      setOptimizationValues(stored.optimizationValues);
+    } else {
+      setSelectedConstraintIds([]);
+      setConstraintValues({});
+      setSelectedOptimizationIds([]);
+      setOptimizationValues({});
+    }
+  }, [selectedLocationIds, initialConfig, getStoredConfigForLocations]);
   (0, import_react3.useEffect)(() => () => clearTimers(), []);
   const stepStatusForIndex = (index) => {
     if (index <= completedStepIndex) {
@@ -1993,52 +2070,63 @@ function AutoSchedulerModal({
       selectedOptimizationIds,
       optimizationValues
     });
-    const solvePromise = solveSchedule(baseURL, payload);
+    persistConfigForLocations == null ? void 0 : persistConfigForLocations(selectedLocationIds, {
+      selectedLocationIds,
+      selectedConstraintIds,
+      constraintValues,
+      selectedOptimizationIds,
+      optimizationValues
+    });
     const runId = resetBeforeSubmit();
     schedule(STEP_INTERVAL_MS, async () => {
-      var _a, _b;
       setCompletedStepIndex(0);
       setActiveStepIndex(1);
       try {
-        const response = await solvePromise;
+        const token = await getToken(getTokenURLAndHeaders.url, getTokenURLAndHeaders.headers);
+        if (!token) {
+          throw new Error("Failed to get auto-scheduler 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;
@@ -2049,6 +2137,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");