web/packages/booking-flow/lib/bookingFlowConfig/bookingFlowConfigContext.tsx

"use client"

import { createContext, useContext } from "react"

import type { BookingFlowConfig } from "./bookingFlowConfig"
import type { BookingFlowVariant } from "./bookingFlowVariants"

type BookingFlowConfigContextData = BookingFlowConfig

const BookingFlowConfigContext = createContext<
  BookingFlowConfigContextData | undefined
>(undefined)

export const useIsPartner = (variant: BookingFlowVariant) => {
  const context = useContext(BookingFlowConfigContext)

  if (!context) {
    throw new Error(
      "useBookingFlowConfig must be used within a BookingFlowConfigContextProvider. Did you forget to use BookingFlowConfig in the consuming app?"
    )
  }

  return context.variant === variant
}

export const useBookingFlowConfig = (): BookingFlowConfigContextData => {
  const context = useContext(BookingFlowConfigContext)

  if (!context) {
    throw new Error(
      "useBookingFlowConfig must be used within a BookingFlowConfigContextProvider. Did you forget to use BookingFlowConfig in the consuming app?"
    )
  }

  return context
}

export function BookingFlowConfigContextProvider({
  children,
  config,
}: {
  children: React.ReactNode
  config: BookingFlowConfig
}) {
  return (
    <BookingFlowConfigContext.Provider value={config}>
      {children}
    </BookingFlowConfigContext.Provider>
  )
}