web/packages/booking-flow/lib/components/BookingWidget/GuestsRoomsPicker/ChildSelector/index.tsx

"use client"

import { useFormContext } from "react-hook-form"
import { useIntl } from "react-intl"

import Caption from "@scandic-hotels/design-system/Caption"

import Counter from "../Counter"
import ChildInfoSelector from "./ChildInfoSelector"

import styles from "./child-selector.module.css"

import type { Child } from "@scandic-hotels/trpc/types/child"

type ChildSelectorProps = {
  roomIndex: number
  currentAdults: number
  currentChildren: Child[]
  childrenInAdultsBed: number
}
export default function ChildSelector({
  roomIndex = 0,
  currentAdults,
  childrenInAdultsBed,
  currentChildren,
}: ChildSelectorProps) {
  const intl = useIntl()
  const childrenLabel = intl.formatMessage({
    defaultMessage: "Children",
  })
  const { setValue } = useFormContext()

  function increaseChildrenCount(roomIndex: number) {
    if (currentChildren.length < 5) {
      setValue(
        `rooms.${roomIndex}.childrenInRoom.${currentChildren.length}`,
        {
          age: undefined,
          bed: undefined,
        },
        { shouldDirty: true }
      )
    }
  }
  function decreaseChildrenCount(roomIndex: number) {
    if (currentChildren.length > 0) {
      currentChildren.pop()
      setValue(`rooms.${roomIndex}.childrenInRoom`, currentChildren, {
        shouldDirty: true,
      })
    }
  }

  return (
    <>
      <section className={styles.container}>
        <Caption color="uiTextHighContrast" type="bold">
          {childrenLabel}
        </Caption>
        <Counter
          count={currentChildren.length}
          handleOnDecrease={() => {
            decreaseChildrenCount(roomIndex)
          }}
          handleOnIncrease={() => {
            increaseChildrenCount(roomIndex)
          }}
          disableDecrease={currentChildren.length == 0}
          disableIncrease={currentChildren.length == 5}
        />
      </section>
      {currentChildren.map((child, index) => (
        <ChildInfoSelector
          roomIndex={roomIndex}
          index={index}
          child={child}
          adults={currentAdults}
          key={"child_" + index}
          childrenInAdultsBed={childrenInAdultsBed}
        />
      ))}
    </>
  )
}