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

package/dist/index.d.cts

@@ -272,7 +272,15 @@ type AutoSchedulerModalProps = {
         url: string;
         headers?: Record<string, string>;
     };
+    getTokenURLAndHeaders: {
+        url: string;
+        headers?: Record<string, string>;
+    };
     onSolution: (solution: ScheduleSolution) => Promise<void>;
+    /** When provided, fetches stored config for a location set to prefill the form */
+    getStoredConfigForLocations?: (locationIds: string[]) => AutoSchedulerModalConfig | null;
+    /** When provided, persists config after successful schedule creation */
+    persistConfigForLocations?: (locationIds: string[], config: AutoSchedulerModalConfig) => void;
 };
 type Screen = "configure" | "creating" | "failed";
 type StepStatus = "pending" | "active" | "done";
@@ -393,7 +401,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, onSolution, }: 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, }: AutoSchedulerModalProps): react_jsx_runtime.JSX.Element;
 
 /**
  * Self-contained AutoSchedulerModal that wraps itself in ChakraProvider.

package/dist/index.d.ts

@@ -272,7 +272,15 @@ type AutoSchedulerModalProps = {
         url: string;
         headers?: Record<string, string>;
     };
+    getTokenURLAndHeaders: {
+        url: string;
+        headers?: Record<string, string>;
+    };
     onSolution: (solution: ScheduleSolution) => Promise<void>;
+    /** When provided, fetches stored config for a location set to prefill the form */
+    getStoredConfigForLocations?: (locationIds: string[]) => AutoSchedulerModalConfig | null;
+    /** When provided, persists config after successful schedule creation */
+    persistConfigForLocations?: (locationIds: string[], config: AutoSchedulerModalConfig) => void;
 };
 type Screen = "configure" | "creating" | "failed";
 type StepStatus = "pending" | "active" | "done";
@@ -393,7 +401,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, onSolution, }: 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, }: AutoSchedulerModalProps): react_jsx_runtime.JSX.Element;
 
 /**
  * Self-contained AutoSchedulerModal that wraps itself in ChakraProvider.

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: [
@@ -1421,14 +1422,65 @@ function buildSchedulePayload(input) {
   };
 }
 
-// src/api/solveSchedule.ts
+// src/api/api.ts
 import axios from "axios";
 var axiosInstance = axios.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;
 }
 
@@ -1791,7 +1843,10 @@ function AutoSchedulerModal({
   theme,
   cssVarsRoot,
   generateRecommendationsURLAndHeaders,
-  onSolution
+  getTokenURLAndHeaders,
+  onSolution,
+  getStoredConfigForLocations,
+  persistConfigForLocations
 }) {
   const baseTheme = useTheme();
   const { chakraOverride, styleOverrides } = useMemo2(
@@ -1813,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) => ({
@@ -1878,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);
@@ -1909,6 +1969,23 @@ function AutoSchedulerModal({
       setScreen("configure");
     }
   }, [isOpen, initialConfig]);
+  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]);
   useEffect(() => () => clearTimers(), []);
   const stepStatusForIndex = (index) => {
     if (index <= completedStepIndex) {
@@ -1985,52 +2062,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;
@@ -2041,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");