From e911a7b5b02ab45a2fcdd29916eb39baf62facc9 Mon Sep 17 00:00:00 2001
From: Lorenzo <lorenzo.cavazzi.tech@gmail.com>
Date: Thu, 15 Jan 2026 14:31:24 +0100
Subject: [PATCH 1/3] chore: update admin page visualization

---
 .../features/admin/UpdateResourceClassButton.tsx    | 13 ++++++++++---
 .../admin/UpdateResourcePoolQuotaButton.tsx         |  7 ++++---
 .../admin/UpdateResourcePoolRemoteButton.tsx        |  7 ++++---
 3 files changed, 18 insertions(+), 9 deletions(-)

diff --git a/client/src/features/admin/UpdateResourceClassButton.tsx b/client/src/features/admin/UpdateResourceClassButton.tsx
index dce6d1306f..6a4ffe7caa 100644
--- a/client/src/features/admin/UpdateResourceClassButton.tsx
+++ b/client/src/features/admin/UpdateResourceClassButton.tsx
@@ -18,7 +18,13 @@
 
 import cx from "classnames";
 import { useCallback, useEffect, useState } from "react";
-import { CheckLg, PlusLg, TrashFill, XLg } from "react-bootstrap-icons";
+import {
+  CheckLg,
+  PencilSquare,
+  PlusLg,
+  TrashFill,
+  XLg,
+} from "react-bootstrap-icons";
 import { Controller, useFieldArray, useForm } from "react-hook-form";
 import {
   Button,
@@ -60,7 +66,8 @@ export default function UpdateResourceClassButton({
   return (
     <>
       <Button size="sm" color="outline-primary" onClick={toggle}>
-        Update
+        <PencilSquare className={cx("bi", "me-1")} />
+        Edit
       </Button>
       <UpdateResourceClassModal
         isOpen={isOpen}
@@ -453,7 +460,7 @@ function UpdateResourceClassModal({
       <ModalFooter>
         <Button color="outline-primary" onClick={toggle}>
           <XLg className={cx("bi", "me-1")} />
-          Close
+          Cancel
         </Button>
         <Button
           color="primary"
diff --git a/client/src/features/admin/UpdateResourcePoolQuotaButton.tsx b/client/src/features/admin/UpdateResourcePoolQuotaButton.tsx
index 5bc5f997b0..3ee8fca7d1 100644
--- a/client/src/features/admin/UpdateResourcePoolQuotaButton.tsx
+++ b/client/src/features/admin/UpdateResourcePoolQuotaButton.tsx
@@ -18,7 +18,7 @@
 
 import cx from "classnames";
 import { useCallback, useEffect, useState } from "react";
-import { CheckLg, XLg } from "react-bootstrap-icons";
+import { CheckLg, PencilSquare, XLg } from "react-bootstrap-icons";
 import { Controller, useForm } from "react-hook-form";
 import {
   Button,
@@ -62,7 +62,8 @@ export default function UpdateResourcePoolQuotaButton({
         disabled={resourcePool.quota == null}
         onClick={toggle}
       >
-        Update
+        <PencilSquare className={cx("bi", "me-1")} />
+        Edit
       </Button>
       <UpdateResourcePoolQuotaModal
         isOpen={isOpen}
@@ -200,7 +201,7 @@ function UpdateResourcePoolQuotaModal({
       <ModalFooter>
         <Button color="outline-primary" onClick={toggle}>
           <XLg className={cx("bi", "me-1")} />
-          Close
+          Cancel
         </Button>
         <Button
           color="primary"
diff --git a/client/src/features/admin/UpdateResourcePoolRemoteButton.tsx b/client/src/features/admin/UpdateResourcePoolRemoteButton.tsx
index 249d9bbc88..4c45351ec6 100644
--- a/client/src/features/admin/UpdateResourcePoolRemoteButton.tsx
+++ b/client/src/features/admin/UpdateResourcePoolRemoteButton.tsx
@@ -18,7 +18,7 @@
 
 import cx from "classnames";
 import { useCallback, useEffect, useState } from "react";
-import { CheckLg, XLg } from "react-bootstrap-icons";
+import { CheckLg, PencilSquare, XLg } from "react-bootstrap-icons";
 import { useForm } from "react-hook-form";
 import {
   Button,
@@ -53,7 +53,8 @@ export default function UpdateResourcePoolRemoteButton({
   return (
     <>
       <Button size="sm" color="outline-primary" onClick={toggle}>
-        Update
+        <PencilSquare className={cx("bi", "me-1")} />
+        Edit
       </Button>
       <UpdateResourcePoolRemoteModal
         isOpen={isOpen}
@@ -163,7 +164,7 @@ function UpdateResourcePoolRemoteModal({
       <ModalFooter>
         <Button color="outline-primary" onClick={toggle}>
           <XLg className={cx("bi", "me-1")} />
-          Close
+          Cancel
         </Button>
         <Button
           color="primary"

From 4ab3506580f4663031a81ff5e47b4505dee9624e Mon Sep 17 00:00:00 2001
From: Lorenzo <lorenzo.cavazzi.tech@gmail.com>
Date: Thu, 15 Jan 2026 14:32:00 +0100
Subject: [PATCH 2/3] feat: allow admins to set/update pause warnings

---
 .../features/admin/AddResourcePoolButton.tsx  | 47 ++++++++-
 client/src/features/admin/AdminPage.tsx       | 17 +++-
 .../UpdateResourcePoolThresholdsButton.tsx    | 99 +++++++++++++++----
 .../admin/adminComputeResources.types.ts      |  2 +
 4 files changed, 141 insertions(+), 24 deletions(-)

diff --git a/client/src/features/admin/AddResourcePoolButton.tsx b/client/src/features/admin/AddResourcePoolButton.tsx
index 8f24446b5c..07fc517cb5 100644
--- a/client/src/features/admin/AddResourcePoolButton.tsx
+++ b/client/src/features/admin/AddResourcePoolButton.tsx
@@ -39,6 +39,7 @@ import {
   usePostResourcePoolsMutation,
   type RemoteConfiguration,
 } from "../sessionsV2/api/computeResources.api";
+import { PAUSE_SESSION_WARNING_GRACE_PERIOD_SECONDS } from "../sessionsV2/session.constants";
 import { useGetNotebooksVersionQuery } from "../versions/versions.api";
 import type { ResourcePoolForm } from "./adminComputeResources.types";
 import ResourcePoolClusterIdInput from "./forms/ResourcePoolClusterIdInput";
@@ -105,6 +106,7 @@ function AddResourcePoolModal({ isOpen, toggle }: AddResourcePoolModalProps) {
         gpu: defaultQuota.gpu,
       },
       idleThresholdMinutes: undefined,
+      pauseWarningMinutes: undefined,
       hibernationThresholdMinutes: undefined,
       clusterId: "",
       remote: {
@@ -159,6 +161,9 @@ function AddResourcePoolModal({ isOpen, toggle }: AddResourcePoolModalProps) {
           idle_threshold: data.idleThresholdMinutes
             ? data.idleThresholdMinutes * 60
             : undefined,
+          hibernation_warning_period: data.pauseWarningMinutes
+            ? data.pauseWarningMinutes * 60
+            : undefined,
           name: data.name,
           public: data.public,
           quota: data.quota,
@@ -180,6 +185,7 @@ function AddResourcePoolModal({ isOpen, toggle }: AddResourcePoolModalProps) {
         gpu: defaultQuota.gpu,
       },
       idleThresholdMinutes: undefined,
+      pauseWarningMinutes: undefined,
       hibernationThresholdMinutes: undefined,
       clusterId: "",
       remote: {
@@ -248,9 +254,10 @@ function AddResourcePoolModal({ isOpen, toggle }: AddResourcePoolModalProps) {
             />
             <div className="invalid-feedback">Please provide a name</div>
           </div>
+
           <div>
             <Label className="form-label" for="addResourcePoolIdleThreshold">
-              Maximum idle time before hibernating (minutes)
+              Maximum idle time before pausing (minutes)
             </Label>
             <Controller
               control={control}
@@ -282,6 +289,44 @@ function AddResourcePoolModal({ isOpen, toggle }: AddResourcePoolModalProps) {
               isLoading={notebookVersion.isLoading}
             />
           </div>
+
+          <div>
+            <Label
+              className="form-label"
+              for="addResourcePoolPauseWarningMinutes"
+            >
+              How long in advance should users be warned about pausing (minutes)
+            </Label>
+            <Controller
+              control={control}
+              name="pauseWarningMinutes"
+              render={({ field }) => (
+                <Input
+                  className={cx(
+                    "form-control",
+                    errors.pauseWarningMinutes && "is-invalid"
+                  )}
+                  id="addResourcePoolPauseWarningMinutes"
+                  min="0"
+                  placeholder="pause warning"
+                  step="1"
+                  type="number"
+                  {...field}
+                />
+              )}
+              rules={{ min: 0 }}
+            />
+            <div className="invalid-feedback">
+              Please enter 0 (or leave it blank) for default or anything greater
+              than 0 to specify a custom value.
+            </div>
+            <Label className="form-text">
+              Default:{" "}
+              {toFullHumanDuration(PAUSE_SESSION_WARNING_GRACE_PERIOD_SECONDS)}.
+              The value cannot be higher than Max idle time.
+            </Label>
+          </div>
+
           <div>
             <Label
               className="form-label"
diff --git a/client/src/features/admin/AdminPage.tsx b/client/src/features/admin/AdminPage.tsx
index 8de7dec9e2..28634560fd 100644
--- a/client/src/features/admin/AdminPage.tsx
+++ b/client/src/features/admin/AdminPage.tsx
@@ -314,6 +314,7 @@ function ResourcePoolItem({ resourcePool }: ResourcePoolItemProps) {
 function ResourcePoolThresholds({ resourcePool }: ResourcePoolItemProps) {
   const {
     idle_threshold: idleThreshold,
+    hibernation_warning_period: hibernationWarningPeriod,
     hibernation_threshold: hibernationThreshold,
   } = resourcePool;
 
@@ -325,8 +326,8 @@ function ResourcePoolThresholds({ resourcePool }: ResourcePoolItemProps) {
         "align-items-center",
         "row",
         "row-cols-1",
-        "row-cols-sm-3",
-        "row-cols-lg-4",
+        "row-cols-sm-4",
+        "row-cols-md-5",
         "text-end"
       )}
     >
@@ -347,6 +348,18 @@ function ResourcePoolThresholds({ resourcePool }: ResourcePoolItemProps) {
             : "unknown"}
         </span>
       </div>
+      <div className="col">
+        Warn in advance{" "}
+        <span className="text-nowrap">
+          {hibernationWarningPeriod
+            ? toFullHumanDuration(hibernationWarningPeriod)
+            : isLoading
+            ? "(Loading...)"
+            : isError
+            ? "unavailable"
+            : "(unknown default)"}
+        </span>
+      </div>
       <div className="col">
         Delete after{" "}
         <span className="text-nowrap">
diff --git a/client/src/features/admin/UpdateResourcePoolThresholdsButton.tsx b/client/src/features/admin/UpdateResourcePoolThresholdsButton.tsx
index c631592ed3..dde231c03d 100644
--- a/client/src/features/admin/UpdateResourcePoolThresholdsButton.tsx
+++ b/client/src/features/admin/UpdateResourcePoolThresholdsButton.tsx
@@ -17,8 +17,8 @@
  */
 
 import cx from "classnames";
-import { useCallback, useEffect, useState } from "react";
-import { CheckLg, XLg } from "react-bootstrap-icons";
+import { useCallback, useEffect, useMemo, useState } from "react";
+import { CheckLg, PencilSquare, XLg } from "react-bootstrap-icons";
 import { Controller, useForm } from "react-hook-form";
 import {
   Button,
@@ -33,10 +33,12 @@ import {
 
 import { RtkOrNotebooksError } from "~/components/errors/RtkErrorAlert";
 import { Loader } from "~/components/Loader";
+import { toFullHumanDuration } from "~/utils/helpers/DurationUtils";
 import {
   usePatchResourcePoolsByResourcePoolIdMutation,
   type ResourcePoolWithId,
 } from "../sessionsV2/api/computeResources.api";
+import { PAUSE_SESSION_WARNING_GRACE_PERIOD_SECONDS } from "../sessionsV2/session.constants";
 import { useGetNotebooksVersionQuery } from "../versions/versions.api";
 import { ResourcePoolDefaultThreshold } from "./AddResourcePoolButton";
 import type { UpdateResourcePoolThresholdsForm } from "./adminComputeResources.types";
@@ -62,7 +64,8 @@ export default function UpdateResourcePoolThresholdsButton({
   return (
     <div key={localKey}>
       <Button color="outline-primary" onClick={toggle} size="sm">
-        Update
+        <PencilSquare className={cx("bi", "me-1")} />
+        Edit
       </Button>
       <UpdateResourcePoolThresholdsModal
         isOpen={isOpen}
@@ -90,20 +93,32 @@ function UpdateResourcePoolThresholdsModal({
   const notebookVersion = useGetNotebooksVersionQuery();
 
   // Form state
-  const {
-    control,
-    formState: { errors },
-    handleSubmit,
-    reset,
-  } = useForm<UpdateResourcePoolThresholdsForm>({
-    defaultValues: {
+  const defaultValues = useMemo(
+    () => ({
       idleThresholdMinutes: resourcePool.idle_threshold
         ? resourcePool.idle_threshold / 60
         : undefined,
+      pauseWarningMinutes: resourcePool.hibernation_warning_period
+        ? resourcePool.hibernation_warning_period / 60
+        : undefined,
       hibernationThresholdMinutes: resourcePool.hibernation_threshold
         ? resourcePool.hibernation_threshold / 60
         : undefined,
-    },
+    }),
+    [
+      resourcePool.idle_threshold,
+      resourcePool.hibernation_warning_period,
+      resourcePool.hibernation_threshold,
+    ]
+  );
+
+  const {
+    control,
+    formState: { errors },
+    handleSubmit,
+    reset,
+  } = useForm<UpdateResourcePoolThresholdsForm>({
+    defaultValues,
   });
 
   // Handle invoking API to update resource pools
@@ -117,6 +132,9 @@ function UpdateResourcePoolThresholdsModal({
           idle_threshold: data.idleThresholdMinutes
             ? data.idleThresholdMinutes * 60
             : undefined,
+          hibernation_warning_period: data.pauseWarningMinutes
+            ? data.pauseWarningMinutes * 60
+            : undefined,
           hibernation_threshold: data.hibernationThresholdMinutes
             ? data.hibernationThresholdMinutes * 60
             : undefined,
@@ -126,17 +144,21 @@ function UpdateResourcePoolThresholdsModal({
     [id, updateResourcePool]
   );
 
-  // Reset form and close modal on successful submissions
+  // Reset form to show up-to-date values
   useEffect(() => {
     if (!result.isSuccess) {
       return;
     }
+    reset(defaultValues);
     toggle();
-  }, [result.isSuccess, toggle]);
+  }, [result.isSuccess, defaultValues, reset, toggle]);
+
+  useEffect(() => {
+    if (isOpen) reset(defaultValues);
+  }, [isOpen, reset, defaultValues]);
 
   useEffect(() => {
     if (!isOpen) {
-      reset();
       result.reset();
     }
   }, [isOpen, reset, result]);
@@ -158,16 +180,12 @@ function UpdateResourcePoolThresholdsModal({
           Please note that changes only affect new sessions, not already running
           ones.
         </p>
-        <Form
-          className="form-rk-green"
-          noValidate
-          onSubmit={handleSubmit(onSubmit)}
-        >
+        <Form noValidate onSubmit={handleSubmit(onSubmit)}>
           {result.error && <RtkOrNotebooksError error={result.error} />}
 
           <div className="mb-3">
             <Label className="form-label" for="updateResourcePoolIdleThreshold">
-              Maximum idle time before hibernating (minutes)
+              Maximum idle time before pausing (minutes)
             </Label>
             <Controller
               control={control}
@@ -205,7 +223,46 @@ function UpdateResourcePoolThresholdsModal({
               </Label>
             )}
           </div>
+
           <div className="mb-3">
+            <Label
+              className="form-label"
+              for="updateResourcePoolPauseWarningMinutes"
+            >
+              How long in advance should users be warned about pausing (minutes)
+            </Label>
+            <Controller
+              control={control}
+              name="pauseWarningMinutes"
+              render={({ field }) => (
+                <Input
+                  className={cx(
+                    "form-control",
+                    errors.pauseWarningMinutes && "is-invalid"
+                  )}
+                  id="updateResourcePoolPauseWarningMinutes"
+                  placeholder="pause warning"
+                  type="number"
+                  min="0"
+                  step="1"
+                  required={false}
+                  {...field}
+                />
+              )}
+              rules={{ min: 0 }}
+            />
+            <div className="invalid-feedback">
+              Please enter 0 (or leave it blank) for default or anything greater
+              than 0 to specify a custom value.
+            </div>
+            <Label className="form-text">
+              Default:{" "}
+              {toFullHumanDuration(PAUSE_SESSION_WARNING_GRACE_PERIOD_SECONDS)}.
+              The value cannot be higher than Max idle time.
+            </Label>
+          </div>
+
+          <div>
             <Label
               className="form-label"
               for="updateResourcePoolHibernationThreshold"
@@ -252,7 +309,7 @@ function UpdateResourcePoolThresholdsModal({
       <ModalFooter>
         <Button color="outline-primary" onClick={toggle}>
           <XLg className={cx("bi", "me-1")} />
-          Close
+          Cancel
         </Button>
         <Button
           color="primary"
diff --git a/client/src/features/admin/adminComputeResources.types.ts b/client/src/features/admin/adminComputeResources.types.ts
index 9c43cd5342..4c5481e13f 100644
--- a/client/src/features/admin/adminComputeResources.types.ts
+++ b/client/src/features/admin/adminComputeResources.types.ts
@@ -21,6 +21,7 @@ export interface ResourcePoolForm {
   public: boolean;
   quota: ResourcePoolFormQuota;
   hibernationThresholdMinutes?: number;
+  pauseWarningMinutes?: number;
   idleThresholdMinutes?: number;
   clusterId: string;
   remote: RemoteConfiguration;
@@ -65,5 +66,6 @@ export interface ResourceClassFormNodeAffinity {
 
 export interface UpdateResourcePoolThresholdsForm {
   idleThresholdMinutes?: number;
+  pauseWarningMinutes?: number;
   hibernationThresholdMinutes?: number;
 }

From bc40f9fd9abe8360ae6adcfdcf0db1e10e973e3d Mon Sep 17 00:00:00 2001
From: Lorenzo <lorenzo.cavazzi.tech@gmail.com>
Date: Thu, 15 Jan 2026 14:32:16 +0100
Subject: [PATCH 3/3] feat: get pause warnings -- TEMP

---
 .../SessionShowPage/ShowSessionPage.tsx       | 22 +++--
 .../sessionsV2/api/computeResources.api.ts    |  5 ++
 .../features/sessionsV2/session.constants.tsx |  2 +-
 .../sessionsV2/useSessionPauseWarning.hook.ts | 81 +++++++++++++++++++
 4 files changed, 102 insertions(+), 8 deletions(-)
 create mode 100644 client/src/features/sessionsV2/useSessionPauseWarning.hook.ts

diff --git a/client/src/features/sessionsV2/SessionShowPage/ShowSessionPage.tsx b/client/src/features/sessionsV2/SessionShowPage/ShowSessionPage.tsx
index 23f400d648..5fa1e09ee5 100644
--- a/client/src/features/sessionsV2/SessionShowPage/ShowSessionPage.tsx
+++ b/client/src/features/sessionsV2/SessionShowPage/ShowSessionPage.tsx
@@ -67,13 +67,11 @@ import {
   usePatchSessionsBySessionIdMutation,
 } from "../api/sessionsV2.api";
 import PauseOrDeleteSessionModal from "../PauseOrDeleteSessionModal";
-import {
-  PAUSE_SESSION_WARNING_DEBOUNCE_SECONDS,
-  PAUSE_SESSION_WARNING_GRACE_PERIOD_SECONDS,
-} from "../session.constants";
+import { PAUSE_SESSION_WARNING_DEBOUNCE_SECONDS } from "../session.constants";
 import { getSessionFavicon } from "../session.utils";
 import { SessionV2 } from "../sessionsV2.types";
 import SessionLaunchLinkModal from "../SessionView/SessionLaunchLinkModal";
+import useSessionPauseWarning from "../useSessionPauseWarning.hook";
 import PauseWarningModal from "./PauseWarningModal";
 import SessionAlerts from "./SessionAlerts";
 import SessionIframe from "./SessionIframe";
@@ -140,8 +138,15 @@ export default function ShowSessionPage() {
     }
   }, [postponePauseError]);
 
+  const sessionPauseWarning = useSessionPauseWarning({
+    classId: thisSession?.resource_class_id,
+  });
+
   // Set next pause warning notification based on the current session data
   useEffect(() => {
+    if (!sessionPauseWarning.isDone || !sessionPauseWarning.pauseWarningSeconds)
+      return;
+
     const willHibernateAt = thisSession?.status.will_hibernate_at
       ? new Date(thisSession.status.will_hibernate_at)
       : null;
@@ -150,11 +155,14 @@ export default function ShowSessionPage() {
       return;
     }
     const notificationTime = new Date(
-      willHibernateAt.getTime() -
-        PAUSE_SESSION_WARNING_GRACE_PERIOD_SECONDS * 1000
+      willHibernateAt.getTime() - sessionPauseWarning.pauseWarningSeconds * 1000
     );
     setNextPauseWarning(notificationTime);
-  }, [thisSession?.status.will_hibernate_at]);
+  }, [
+    thisSession?.status.will_hibernate_at,
+    sessionPauseWarning.pauseWarningSeconds,
+    sessionPauseWarning.isDone,
+  ]);
 
   // Handle showing the pause warning
   useEffect(() => {
diff --git a/client/src/features/sessionsV2/api/computeResources.api.ts b/client/src/features/sessionsV2/api/computeResources.api.ts
index 6da27dd356..737db7700a 100644
--- a/client/src/features/sessionsV2/api/computeResources.api.ts
+++ b/client/src/features/sessionsV2/api/computeResources.api.ts
@@ -53,6 +53,10 @@ export const computeResourcesApi = withFixedEndpoints.enhanceEndpoints({
             ]
           : ["ResourcePool"],
     },
+    getResourcePoolsByResourcePoolId: {
+      providesTags: (result, _error, { resourcePoolId }) =>
+        result ? [{ id: resourcePoolId, type: "ResourcePool" }] : [],
+    },
     postResourcePools: {
       invalidatesTags: ["ResourcePool", "ResourceClass"],
     },
@@ -114,6 +118,7 @@ export const computeResourcesApi = withFixedEndpoints.enhanceEndpoints({
 export const {
   // "resource pools" hooks
   useGetResourcePoolsQuery,
+  useGetResourcePoolsByResourcePoolIdQuery,
   usePostResourcePoolsMutation,
   usePatchResourcePoolsByResourcePoolIdMutation,
   useDeleteResourcePoolsByResourcePoolIdMutation,
diff --git a/client/src/features/sessionsV2/session.constants.tsx b/client/src/features/sessionsV2/session.constants.tsx
index a745202f22..44fe571d2c 100644
--- a/client/src/features/sessionsV2/session.constants.tsx
+++ b/client/src/features/sessionsV2/session.constants.tsx
@@ -163,4 +163,4 @@ export const LAUNCHER_CONTAINER_IMAGE_QUERY_DEBOUNCE = 1_000;
 
 export const PAUSE_SESSION_WARNING_DEBOUNCE_SECONDS = 30;
 
-export const PAUSE_SESSION_WARNING_GRACE_PERIOD_SECONDS = 10 * 60;
+export const PAUSE_SESSION_WARNING_GRACE_PERIOD_SECONDS = 15 * 60;
diff --git a/client/src/features/sessionsV2/useSessionPauseWarning.hook.ts b/client/src/features/sessionsV2/useSessionPauseWarning.hook.ts
new file mode 100644
index 0000000000..9f5f7de4d9
--- /dev/null
+++ b/client/src/features/sessionsV2/useSessionPauseWarning.hook.ts
@@ -0,0 +1,81 @@
+/*!
+ * Copyright 2026 - Swiss Data Science Center (SDSC)
+ * A partnership between École Polytechnique Fédérale de Lausanne (EPFL) and
+ * Eidgenössische Technische Hochschule Zürich (ETHZ).
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+ * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License
+ */
+
+import { skipToken } from "@reduxjs/toolkit/query/react";
+import { useEffect, useState } from "react";
+
+import {
+  useGetClassesByClassIdQuery,
+  useGetResourcePoolsByResourcePoolIdQuery,
+} from "./api/computeResources.api";
+import { PAUSE_SESSION_WARNING_GRACE_PERIOD_SECONDS } from "./session.constants";
+
+interface StartPauseWarningProps {
+  classId?: number;
+}
+
+export default function useSessionPauseWarning({
+  classId,
+}: StartPauseWarningProps) {
+  const [isDone, setIsDone] = useState(false);
+  const [pauseWarningSeconds, setPauseWarningSeconds] = useState<
+    number | undefined
+  >(undefined);
+  const resourceClasses = useGetClassesByClassIdQuery(
+    classId ? { classId: classId.toString() } : skipToken
+  );
+
+  const resourcePool = useGetResourcePoolsByResourcePoolIdQuery(
+    resourceClasses.data?.id ? { resourcePoolId: 2 } : skipToken
+  );
+
+  useEffect(() => {
+    if (!classId || resourceClasses.isLoading || resourcePool.isLoading) {
+      return;
+    }
+    if (resourceClasses.isError || resourcePool.isError) {
+      setPauseWarningSeconds(PAUSE_SESSION_WARNING_GRACE_PERIOD_SECONDS);
+      setIsDone(true);
+      return;
+    }
+    if (
+      resourcePool.isSuccess &&
+      resourcePool?.data.hibernation_warning_period
+    ) {
+      setPauseWarningSeconds(resourcePool.data.hibernation_warning_period);
+    } else {
+      setPauseWarningSeconds(PAUSE_SESSION_WARNING_GRACE_PERIOD_SECONDS);
+    }
+    setIsDone(true);
+  }, [
+    classId,
+    resourceClasses.isError,
+    resourcePool.isError,
+    resourceClasses.isLoading,
+    resourcePool.isLoading,
+    resourcePool.isSuccess,
+    resourcePool.data,
+  ]);
+
+  return {
+    isError: resourceClasses.isError || resourcePool.isError,
+    isLoading: resourceClasses.isLoading || resourcePool.isLoading,
+    isDone,
+    pauseWarningSeconds,
+  };
+}