From 1f52ed7290bc191c566824369065a6b012bec81a Mon Sep 17 00:00:00 2001
From: Akarsh Jain <72064462+akarsh-jain-790@users.noreply.github.com>
Date: Fri, 30 May 2025 16:09:02 +0530
Subject: [PATCH] feat: Add loadingStateChildren prop to
 AvailabilitySettingsPlatformWrapper (#21602)

Co-authored-by: Morgan <33722304+ThyMinimalDev@users.noreply.github.com>
---
 .../wrappers/AvailabilitySettingsPlatformWrapper.tsx   | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/packages/platform/atoms/availability/wrappers/AvailabilitySettingsPlatformWrapper.tsx b/packages/platform/atoms/availability/wrappers/AvailabilitySettingsPlatformWrapper.tsx
index 2ea3443816..dfdcc83953 100644
--- a/packages/platform/atoms/availability/wrappers/AvailabilitySettingsPlatformWrapper.tsx
+++ b/packages/platform/atoms/availability/wrappers/AvailabilitySettingsPlatformWrapper.tsx
@@ -41,6 +41,7 @@ type AvailabilitySettingsPlatformWrapperProps = {
   disableToasts?: boolean;
   isDryRun?: boolean;
   noScheduleChildren?: ReactNode;
+  loadingStateChildren?: ReactNode;
 };
 
 export const AvailabilitySettingsPlatformWrapper = ({
@@ -59,6 +60,7 @@ export const AvailabilitySettingsPlatformWrapper = ({
   disableToasts,
   isDryRun = false,
   noScheduleChildren,
+  loadingStateChildren,
 }: AvailabilitySettingsPlatformWrapperProps) => {
   const { isLoading, data: schedule } = useSchedule(id);
   const { data: schedules } = useSchedules();
@@ -123,7 +125,13 @@ export const AvailabilitySettingsPlatformWrapper = ({
     }
   };
 
-  if (isLoading) return <div className="px-10 py-4 text-xl">Loading...</div>;
+  if (isLoading) {
+    return (
+      <>
+        {loadingStateChildren ? loadingStateChildren : <div className="px-10 py-4 text-xl">Loading...</div>}
+      </>
+    );
+  }
 
   if (!atomSchedule) {
     return noScheduleChildren ? (