From 80815bd8a57034b4f4a223fc02365405ce7558b7 Mon Sep 17 00:00:00 2001
From: Lauris Skraucis <lauris.skraucis@gmail.com>
Date: Tue, 17 Dec 2024 08:28:18 +0100
Subject: [PATCH] feat: Booker atom slots callbacks (#18203)

---
 .../atoms/booker/BookerPlatformWrapper.tsx    |  7 ++++-
 packages/platform/atoms/hooks/useSlots.ts     | 28 +++++++++++++++++--
 2 files changed, 32 insertions(+), 3 deletions(-)

diff --git a/packages/platform/atoms/booker/BookerPlatformWrapper.tsx b/packages/platform/atoms/booker/BookerPlatformWrapper.tsx
index a922e9c101..1195005eb7 100644
--- a/packages/platform/atoms/booker/BookerPlatformWrapper.tsx
+++ b/packages/platform/atoms/booker/BookerPlatformWrapper.tsx
@@ -322,7 +322,12 @@ export const BookerPlatformWrapper = (
     onError: props.onCreateInstantBookingError,
   });
 
-  const slots = useSlots(event);
+  const slots = useSlots(event, {
+    onReserveSlotSuccess: props.onReserveSlotSuccess,
+    onReserveSlotError: props.onReserveSlotError,
+    onDeleteSlotSuccess: props.onDeleteSlotSuccess,
+    onDeleteSlotError: props.onDeleteSlotError,
+  });
 
   const { data: connectedCalendars, isPending: fetchingConnectedCalendars } = useConnectedCalendars({
     enabled: hasSession,
diff --git a/packages/platform/atoms/hooks/useSlots.ts b/packages/platform/atoms/hooks/useSlots.ts
index 53e70a703e..0ea05530db 100644
--- a/packages/platform/atoms/hooks/useSlots.ts
+++ b/packages/platform/atoms/hooks/useSlots.ts
@@ -6,31 +6,55 @@ import { useBookerStore } from "@calcom/features/bookings/Booker/store";
 import { useSlotReservationId } from "@calcom/features/bookings/Booker/useSlotReservationId";
 import type { BookerEvent } from "@calcom/features/bookings/types";
 import { MINUTES_TO_BOOK } from "@calcom/lib/constants";
+import type {
+  ApiErrorResponse,
+  ApiSuccessResponse,
+  ApiSuccessResponseWithoutData,
+} from "@calcom/platform-types";
 
 import { useDeleteSelectedSlot } from "./useDeleteSelectedSlot";
 import { useReserveSlot } from "./useReserveSlot";
 
+type UseSlotsCallbacks = {
+  onReserveSlotSuccess?: (data: ApiSuccessResponse<string>) => void;
+  onReserveSlotError?: (err: ApiErrorResponse) => void;
+  onDeleteSlotSuccess?: (data: ApiSuccessResponseWithoutData) => void;
+  onDeleteSlotError?: (err: ApiErrorResponse) => void;
+};
+
 export type UseSlotsReturnType = ReturnType<typeof useSlots>;
 
-export const useSlots = (event: { data?: Pick<BookerEvent, "id" | "length"> | null }) => {
+export const useSlots = (
+  event: { data?: Pick<BookerEvent, "id" | "length"> | null },
+  { onReserveSlotSuccess, onReserveSlotError, onDeleteSlotSuccess, onDeleteSlotError }: UseSlotsCallbacks = {}
+) => {
   const selectedDuration = useBookerStore((state) => state.selectedDuration);
   const [selectedTimeslot, setSelectedTimeslot] = useBookerStore(
     (state) => [state.selectedTimeslot, state.setSelectedTimeslot],
     shallow
   );
+
   const [slotReservationId, setSlotReservationId] = useSlotReservationId();
+
   const reserveSlotMutation = useReserveSlot({
     onSuccess: (res) => {
       setSlotReservationId(res.data);
+      onReserveSlotSuccess?.(res);
     },
+    onError: onReserveSlotError,
+  });
+
+  const removeSelectedSlot = useDeleteSelectedSlot({
+    onSuccess: onDeleteSlotSuccess,
+    onError: onDeleteSlotError,
   });
 
-  const removeSelectedSlot = useDeleteSelectedSlot();
   const handleRemoveSlot = () => {
     if (event?.data) {
       removeSelectedSlot.mutate({ uid: slotReservationId ?? undefined });
     }
   };
+
   const handleReserveSlot = () => {
     if (event?.data?.id && selectedTimeslot && (selectedDuration || event?.data?.length)) {
       reserveSlotMutation.mutate({