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

package/dist/index.cjs

@@ -1437,14 +1437,29 @@ 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 solveSchedule(baseURL, payload, 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.post(
+    `${cleanBaseUrl}/api/schedule/solve`,
+    payload,
+    { headers: Object.keys(headers).length > 0 ? headers : void 0 }
+  );
   return response.data;
 }
 
@@ -1799,7 +1814,10 @@ function AutoSchedulerModal({
   theme,
   cssVarsRoot,
   generateRecommendationsURLAndHeaders,
-  onSolution
+  getTokenURLAndHeaders,
+  onSolution,
+  getStoredConfigForLocations,
+  persistConfigForLocations
 }) {
   const baseTheme = (0, import_react4.useTheme)();
   const { chakraOverride, styleOverrides } = (0, import_react3.useMemo)(
@@ -1917,6 +1935,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,14 +2028,30 @@ 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 response = await solveSchedule(
+          baseURL,
+          payload,
+          {
+            authorization: `Bearer ${token}`
+          }
+        );
         if (runIdRef.current !== runId) {
           return;
         }