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

package/dist/index.d.cts

@@ -118,6 +118,8 @@ type OpenShift = {
     };
     requiredPosition: string;
     requiredCount: number;
+    isDraft?: boolean;
+    isPublished?: boolean;
     employeePreferences: Array<{
         employeeId: string;
         level: "PREFERRED" | "AVAILABLE" | "UNAVAILABLE" | string;
@@ -281,6 +283,7 @@ type AutoSchedulerModalProps = {
     getStoredConfigForLocations?: (locationIds: string[]) => AutoSchedulerModalConfig | null;
     /** When provided, persists config after successful schedule creation */
     persistConfigForLocations?: (locationIds: string[], config: AutoSchedulerModalConfig) => void;
+    errorToast?: (msg: string) => void;
 };
 type Screen = "configure" | "creating" | "failed";
 type StepStatus = "pending" | "active" | "done";
@@ -374,6 +377,7 @@ type ViolatedConstraintsPanelProps = {
     };
     styles?: Record<string, Record<string, unknown>>;
     styleOverrides?: ViolatedConstraintsPanelStyleOverrides;
+    errorToast?: (msg: string) => void;
 };
 type SolveApiResponse = {
     problemId: string;
@@ -401,7 +405,7 @@ type ScheduleRecommendation = {
     why_this_resolves_it: string;
 };
 
-declare function AutoSchedulerModal({ baseURL, isOpen, title, bannerText, jurisdictions, openShifts, employees, timeOffs, locationPlaceholder, onClose, cancelLabel, primaryActionLabel, onPrimaryAction: _onPrimaryAction, onFixManually, initialConfig, theme, cssVarsRoot, generateRecommendationsURLAndHeaders, getTokenURLAndHeaders, onSolution, getStoredConfigForLocations, persistConfigForLocations, }: AutoSchedulerModalProps): react_jsx_runtime.JSX.Element;
+declare function AutoSchedulerModal({ baseURL, isOpen, title, bannerText, jurisdictions, openShifts, employees, timeOffs, locationPlaceholder, onClose, cancelLabel, primaryActionLabel, onPrimaryAction: _onPrimaryAction, onFixManually, initialConfig, theme, cssVarsRoot, generateRecommendationsURLAndHeaders, getTokenURLAndHeaders, onSolution, getStoredConfigForLocations, persistConfigForLocations, errorToast, }: AutoSchedulerModalProps): react_jsx_runtime.JSX.Element;
 
 /**
  * Self-contained AutoSchedulerModal that wraps itself in ChakraProvider.
@@ -410,7 +414,7 @@ declare function AutoSchedulerModal({ baseURL, isOpen, title, bannerText, jurisd
  */
 declare function AutoSchedulerModalWithProvider(props: AutoSchedulerModalProps): react_jsx_runtime.JSX.Element;
 
-declare function ViolatedConstraintsPanel({ violatedConstraints, title, subtitle, showSecondaryButton, secondaryButtonTitle, onSecondaryButtonClick, generateRecommendationsURLAndHeaders, styles: stylesProp, styleOverrides, }: ViolatedConstraintsPanelProps): react_jsx_runtime.JSX.Element;
+declare function ViolatedConstraintsPanel({ violatedConstraints, title, subtitle, showSecondaryButton, secondaryButtonTitle, onSecondaryButtonClick, generateRecommendationsURLAndHeaders, styles: stylesProp, styleOverrides, errorToast, }: ViolatedConstraintsPanelProps): react_jsx_runtime.JSX.Element;
 
 declare function ViolatedConstraintsPanelWithProvider(props: ViolatedConstraintsPanelProps): react_jsx_runtime.JSX.Element;
 

package/dist/index.d.ts

@@ -118,6 +118,8 @@ type OpenShift = {
     };
     requiredPosition: string;
     requiredCount: number;
+    isDraft?: boolean;
+    isPublished?: boolean;
     employeePreferences: Array<{
         employeeId: string;
         level: "PREFERRED" | "AVAILABLE" | "UNAVAILABLE" | string;
@@ -281,6 +283,7 @@ type AutoSchedulerModalProps = {
     getStoredConfigForLocations?: (locationIds: string[]) => AutoSchedulerModalConfig | null;
     /** When provided, persists config after successful schedule creation */
     persistConfigForLocations?: (locationIds: string[], config: AutoSchedulerModalConfig) => void;
+    errorToast?: (msg: string) => void;
 };
 type Screen = "configure" | "creating" | "failed";
 type StepStatus = "pending" | "active" | "done";
@@ -374,6 +377,7 @@ type ViolatedConstraintsPanelProps = {
     };
     styles?: Record<string, Record<string, unknown>>;
     styleOverrides?: ViolatedConstraintsPanelStyleOverrides;
+    errorToast?: (msg: string) => void;
 };
 type SolveApiResponse = {
     problemId: string;
@@ -401,7 +405,7 @@ type ScheduleRecommendation = {
     why_this_resolves_it: string;
 };
 
-declare function AutoSchedulerModal({ baseURL, isOpen, title, bannerText, jurisdictions, openShifts, employees, timeOffs, locationPlaceholder, onClose, cancelLabel, primaryActionLabel, onPrimaryAction: _onPrimaryAction, onFixManually, initialConfig, theme, cssVarsRoot, generateRecommendationsURLAndHeaders, getTokenURLAndHeaders, onSolution, getStoredConfigForLocations, persistConfigForLocations, }: AutoSchedulerModalProps): react_jsx_runtime.JSX.Element;
+declare function AutoSchedulerModal({ baseURL, isOpen, title, bannerText, jurisdictions, openShifts, employees, timeOffs, locationPlaceholder, onClose, cancelLabel, primaryActionLabel, onPrimaryAction: _onPrimaryAction, onFixManually, initialConfig, theme, cssVarsRoot, generateRecommendationsURLAndHeaders, getTokenURLAndHeaders, onSolution, getStoredConfigForLocations, persistConfigForLocations, errorToast, }: AutoSchedulerModalProps): react_jsx_runtime.JSX.Element;
 
 /**
  * Self-contained AutoSchedulerModal that wraps itself in ChakraProvider.
@@ -410,7 +414,7 @@ declare function AutoSchedulerModal({ baseURL, isOpen, title, bannerText, jurisd
  */
 declare function AutoSchedulerModalWithProvider(props: AutoSchedulerModalProps): react_jsx_runtime.JSX.Element;
 
-declare function ViolatedConstraintsPanel({ violatedConstraints, title, subtitle, showSecondaryButton, secondaryButtonTitle, onSecondaryButtonClick, generateRecommendationsURLAndHeaders, styles: stylesProp, styleOverrides, }: ViolatedConstraintsPanelProps): react_jsx_runtime.JSX.Element;
+declare function ViolatedConstraintsPanel({ violatedConstraints, title, subtitle, showSecondaryButton, secondaryButtonTitle, onSecondaryButtonClick, generateRecommendationsURLAndHeaders, styles: stylesProp, styleOverrides, errorToast, }: ViolatedConstraintsPanelProps): react_jsx_runtime.JSX.Element;
 
 declare function ViolatedConstraintsPanelWithProvider(props: ViolatedConstraintsPanelProps): react_jsx_runtime.JSX.Element;
 

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: [
@@ -1344,7 +1345,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 {
@@ -1352,8 +1353,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,
@@ -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 = {
@@ -1655,12 +1692,15 @@ function ViolatedConstraintsPanel({
   onSecondaryButtonClick,
   generateRecommendationsURLAndHeaders,
   styles: stylesProp,
-  styleOverrides
+  styleOverrides,
+  errorToast
 }) {
   const [showRecommendations, setShowRecommendations] = useState(false);
   const [isExpanded, setIsExpanded] = useState(true);
   const [recommendations, setRecommendations] = useState(null);
   const [isLoadingRecommendations, setIsLoadingRecommendations] = useState(false);
+  const [isDisabledRecommendations, setIsDisabledRecommendations] = useState(false);
+  const [recommendationsButtonText, setRecommendationsButtonText] = useState("Show recommendations");
   const styles = useMemo(() => {
     if (stylesProp) return stylesProp;
     return buildStyles(defaultStyles, styleOverrides);
@@ -1671,12 +1711,10 @@ function ViolatedConstraintsPanel({
   const displayRecommendations = recommendations !== null ? recommendations : violatedConstraints.recommendedFixes;
   const hasRecommendations = displayRecommendations.length > 0;
   const handleToggleRecommendations = useCallback(async () => {
-    var _a, _b;
     if (showRecommendations) {
       setShowRecommendations(false);
       return;
     }
-    setShowRecommendations(true);
     if (recommendations !== null) {
       return;
     }
@@ -1696,9 +1734,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([]);
       }
@@ -1713,7 +1763,8 @@ function ViolatedConstraintsPanel({
     recommendations,
     canFetchRecommendations,
     violatedConstraints.recommendationPayload,
-    generateRecommendationsURLAndHeaders
+    generateRecommendationsURLAndHeaders,
+    errorToast
   ]);
   return /* @__PURE__ */ jsxs2(Box, { sx: styles.violationPanel, children: [
     /* @__PURE__ */ jsxs2(HStack, { sx: styles.violationHeaderRow, children: [
@@ -1765,9 +1816,9 @@ function ViolatedConstraintsPanel({
             variant: "outline",
             size: "xs",
             onClick: handleToggleRecommendations,
-            isDisabled: isLoadingRecommendations,
+            isDisabled: isLoadingRecommendations || isDisabledRecommendations,
             sx: { ...styles.recommendationsToggle, minWidth: "140px" },
-            children: isLoadingRecommendations ? /* @__PURE__ */ jsx2(Spinner, { size: "sm", thickness: "2px" }) : showRecommendations ? "Hide recommendations" : "Show recommendations"
+            children: isLoadingRecommendations ? /* @__PURE__ */ jsx2(Spinner, { size: "sm", thickness: "2px" }) : showRecommendations ? "Hide recommendations" : recommendationsButtonText
           }
         ) : null,
         showSecondaryButton && secondaryButtonTitle && onSecondaryButtonClick ? /* @__PURE__ */ jsx2(
@@ -1809,7 +1860,8 @@ function AutoSchedulerModal({
   getTokenURLAndHeaders,
   onSolution,
   getStoredConfigForLocations,
-  persistConfigForLocations
+  persistConfigForLocations,
+  errorToast
 }) {
   const baseTheme = useTheme();
   const { chakraOverride, styleOverrides } = useMemo2(
@@ -1831,6 +1883,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) => ({
@@ -1845,6 +1898,11 @@ function AutoSchedulerModal({
     (total, jurisdiction) => total + jurisdiction.selectedLocations.length,
     0
   );
+  const locationIdsWithOpenShifts = useMemo2(() => {
+    return new Set(
+      openShifts.filter((shift) => !shift.employeeId).map((shift) => shift.location.id)
+    );
+  }, [openShifts]);
   const selectedConstraints = useMemo2(
     () => HARD_CONSTRAINT_OPTIONS.filter((option) => selectedConstraintIds.includes(option.id)),
     [selectedConstraintIds]
@@ -1896,6 +1954,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);
@@ -1945,6 +2007,12 @@ function AutoSchedulerModal({
     }
   }, [selectedLocationIds, initialConfig, getStoredConfigForLocations]);
   useEffect(() => () => clearTimers(), []);
+  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";
@@ -2029,7 +2097,6 @@ function AutoSchedulerModal({
     });
     const runId = resetBeforeSubmit();
     schedule(STEP_INTERVAL_MS, async () => {
-      var _a, _b;
       setCompletedStepIndex(0);
       setActiveStepIndex(1);
       try {
@@ -2037,51 +2104,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,11 +2155,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;
@@ -2160,6 +2262,7 @@ function AutoSchedulerModal({
       violatedConstraints: lastFailure,
       title: violationTitle,
       subtitle: "",
+      errorToast,
       generateRecommendationsURLAndHeaders,
       styles
     }
@@ -2219,13 +2322,18 @@ function AutoSchedulerModal({
             HStack2,
             {
               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__ */ jsx3(
                   Checkbox,
                   {
                     id: `location-${location.id}`,
                     isChecked: selectedLocationIds.includes(location.id),
+                    isDisabled: !locationIdsWithOpenShifts.has(location.id),
                     onChange: () => handleToggleLocation(location.id)
                   }
                 ),