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

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: [
@@ -1360,7 +1361,7 @@ function buildRulesPayload(selectedConstraintIds, constraintValues, selectedOpti
   return { ruleIds: allSelectedRuleIds, ruleOverrides };
 }
 function toApiShift(shift, includeEmployeePreferences) {
-  var _a, _b;
+  var _a;
   const startDateTime = "startDateTime" in shift ? shift.startDateTime : shift.startTime;
   const endDateTime = "endDateTime" in shift ? shift.endDateTime : shift.endTime;
   return {
@@ -1368,8 +1369,8 @@ function toApiShift(shift, includeEmployeePreferences) {
     startDateTime,
     endDateTime,
     ...shift.employeeId ? { employeeId: String(shift.employeeId) } : {},
-    requiredPosition: (_a = shift.requiredPosition) != null ? _a : void 0,
-    requiredCount: (_b = shift.requiredCount) != null ? _b : 1,
+    requiredPosition: shift.requiredPosition || void 0,
+    requiredCount: (_a = shift.requiredCount) != null ? _a : 1,
     location: shift.location ? {
       id: String(shift.location.id),
       timezone: "timezone" in shift.location ? shift.location.timezone : void 0,
@@ -1449,19 +1450,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 = {
@@ -1663,12 +1700,15 @@ function ViolatedConstraintsPanel({
   onSecondaryButtonClick,
   generateRecommendationsURLAndHeaders,
   styles: stylesProp,
-  styleOverrides
+  styleOverrides,
+  errorToast
 }) {
   const [showRecommendations, setShowRecommendations] = (0, import_react.useState)(false);
   const [isExpanded, setIsExpanded] = (0, import_react.useState)(true);
   const [recommendations, setRecommendations] = (0, import_react.useState)(null);
   const [isLoadingRecommendations, setIsLoadingRecommendations] = (0, import_react.useState)(false);
+  const [isDisabledRecommendations, setIsDisabledRecommendations] = (0, import_react.useState)(false);
+  const [recommendationsButtonText, setRecommendationsButtonText] = (0, import_react.useState)("Show recommendations");
   const styles = (0, import_react.useMemo)(() => {
     if (stylesProp) return stylesProp;
     return buildStyles(defaultStyles, styleOverrides);
@@ -1679,12 +1719,10 @@ function ViolatedConstraintsPanel({
   const displayRecommendations = recommendations !== null ? recommendations : violatedConstraints.recommendedFixes;
   const hasRecommendations = displayRecommendations.length > 0;
   const handleToggleRecommendations = (0, import_react.useCallback)(async () => {
-    var _a, _b;
     if (showRecommendations) {
       setShowRecommendations(false);
       return;
     }
-    setShowRecommendations(true);
     if (recommendations !== null) {
       return;
     }
@@ -1704,9 +1742,21 @@ function ViolatedConstraintsPanel({
           }
         }
       );
-      const data = (_b = (_a = recResponse.data) == null ? void 0 : _a.data) != null ? _b : recResponse.data;
-      if (data) {
-        setRecommendations(data.map((r) => r.recommendation));
+      const { success, data, msg } = recResponse.data;
+      const { isExpired, recommendations: recommendations2 } = data != null ? data : {};
+      if (!success) {
+        const errorMsg = msg || "Failed to fetch recommendations";
+        errorToast == null ? void 0 : errorToast(errorMsg);
+        if (isExpired) {
+          setIsDisabledRecommendations(true);
+          setRecommendationsButtonText("Daily limit reached");
+        }
+        return;
+      }
+      setShowRecommendations(true);
+      const fetchedRecommendations = recommendations2;
+      if (fetchedRecommendations) {
+        setRecommendations(fetchedRecommendations.map((r) => r.recommendation));
       } else {
         setRecommendations([]);
       }
@@ -1721,7 +1771,8 @@ function ViolatedConstraintsPanel({
     recommendations,
     canFetchRecommendations,
     violatedConstraints.recommendationPayload,
-    generateRecommendationsURLAndHeaders
+    generateRecommendationsURLAndHeaders,
+    errorToast
   ]);
   return /* @__PURE__ */ (0, import_jsx_runtime2.jsxs)(import_react2.Box, { sx: styles.violationPanel, children: [
     /* @__PURE__ */ (0, import_jsx_runtime2.jsxs)(import_react2.HStack, { sx: styles.violationHeaderRow, children: [
@@ -1773,9 +1824,9 @@ function ViolatedConstraintsPanel({
             variant: "outline",
             size: "xs",
             onClick: handleToggleRecommendations,
-            isDisabled: isLoadingRecommendations,
+            isDisabled: isLoadingRecommendations || isDisabledRecommendations,
             sx: { ...styles.recommendationsToggle, minWidth: "140px" },
-            children: isLoadingRecommendations ? /* @__PURE__ */ (0, import_jsx_runtime2.jsx)(import_react2.Spinner, { size: "sm", thickness: "2px" }) : showRecommendations ? "Hide recommendations" : "Show recommendations"
+            children: isLoadingRecommendations ? /* @__PURE__ */ (0, import_jsx_runtime2.jsx)(import_react2.Spinner, { size: "sm", thickness: "2px" }) : showRecommendations ? "Hide recommendations" : recommendationsButtonText
           }
         ) : null,
         showSecondaryButton && secondaryButtonTitle && onSecondaryButtonClick ? /* @__PURE__ */ (0, import_jsx_runtime2.jsx)(
@@ -1817,7 +1868,8 @@ function AutoSchedulerModal({
   getTokenURLAndHeaders,
   onSolution,
   getStoredConfigForLocations,
-  persistConfigForLocations
+  persistConfigForLocations,
+  errorToast
 }) {
   const baseTheme = (0, import_react4.useTheme)();
   const { chakraOverride, styleOverrides } = (0, import_react3.useMemo)(
@@ -1839,6 +1891,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) => ({
@@ -1853,6 +1906,11 @@ function AutoSchedulerModal({
     (total, jurisdiction) => total + jurisdiction.selectedLocations.length,
     0
   );
+  const locationIdsWithOpenShifts = (0, import_react3.useMemo)(() => {
+    return new Set(
+      openShifts.filter((shift) => !shift.employeeId).map((shift) => shift.location.id)
+    );
+  }, [openShifts]);
   const selectedConstraints = (0, import_react3.useMemo)(
     () => HARD_CONSTRAINT_OPTIONS.filter((option) => selectedConstraintIds.includes(option.id)),
     [selectedConstraintIds]
@@ -1904,6 +1962,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);
@@ -1953,6 +2015,12 @@ function AutoSchedulerModal({
     }
   }, [selectedLocationIds, initialConfig, getStoredConfigForLocations]);
   (0, import_react3.useEffect)(() => () => clearTimers(), []);
+  (0, import_react3.useEffect)(() => {
+    setSelectedLocationIds((previous) => {
+      const filtered = previous.filter((id) => locationIdsWithOpenShifts.has(id));
+      return filtered.length === previous.length ? previous : filtered;
+    });
+  }, [locationIdsWithOpenShifts]);
   const stepStatusForIndex = (index) => {
     if (index <= completedStepIndex) {
       return "done";
@@ -2037,7 +2105,6 @@ function AutoSchedulerModal({
     });
     const runId = resetBeforeSubmit();
     schedule(STEP_INTERVAL_MS, async () => {
-      var _a, _b;
       setCompletedStepIndex(0);
       setActiveStepIndex(1);
       try {
@@ -2045,51 +2112,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;
@@ -2100,11 +2163,50 @@ 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");
   };
   const handleToggleLocation = (locationId) => {
+    if (!locationIdsWithOpenShifts.has(locationId)) {
+      return;
+    }
     if (selectedLocationIds.includes(locationId)) {
       setSelectedLocationIds(selectedLocationIds.filter((id) => id !== locationId));
       return;
@@ -2168,6 +2270,7 @@ function AutoSchedulerModal({
       violatedConstraints: lastFailure,
       title: violationTitle,
       subtitle: "",
+      errorToast,
       generateRecommendationsURLAndHeaders,
       styles
     }
@@ -2227,13 +2330,18 @@ function AutoSchedulerModal({
             import_react4.HStack,
             {
               as: "label",
-              sx: { ...styles.optionRow, cursor: "pointer" },
+              sx: {
+                ...styles.optionRow,
+                cursor: locationIdsWithOpenShifts.has(location.id) ? "pointer" : "not-allowed",
+                opacity: locationIdsWithOpenShifts.has(location.id) ? 1 : 0.55
+              },
               children: [
                 /* @__PURE__ */ (0, import_jsx_runtime3.jsx)(
                   import_react4.Checkbox,
                   {
                     id: `location-${location.id}`,
                     isChecked: selectedLocationIds.includes(location.id),
+                    isDisabled: !locationIdsWithOpenShifts.has(location.id),
                     onChange: () => handleToggleLocation(location.id)
                   }
                 ),