web/apps/scandic-web/components/HotelReservation/MyStay/stores/myStayTotalPrice.ts

import { create } from "zustand"

interface RoomPrice {
  id: string
  totalPrice: number
  currencyCode: string
  isMainBooking?: boolean
}

interface MyStayTotalPriceState {
  rooms: RoomPrice[]
  totalPrice: number
  currencyCode: string
  actions: {
    // Add a single room price
    setRoomPrice: (room: RoomPrice) => void

    // Get the calculated total
    getTotalPrice: () => number
  }
}

export const useMyStayTotalPriceStore = create<MyStayTotalPriceState>(
  (set, get) => ({
    rooms: [],
    totalPrice: 0,
    currencyCode: "",
    actions: {
      setRoomPrice: (room) => {
        set((state) => {
          // Check if room with this ID already exists
          const existingIndex = state.rooms.findIndex((r) => r.id === room.id)
          let newRooms = [...state.rooms]

          if (existingIndex >= 0) {
            // Update existing room
            newRooms[existingIndex] = room
          } else {
            // Add new room
            newRooms.push(room)
          }

          // Get currency from main booking or first room
          const mainRoom = newRooms.find((r) => r.isMainBooking) || newRooms[0]
          const currencyCode = mainRoom?.currencyCode || ""

          // Calculate total (only same currency for now)
          const total = newRooms.reduce((sum, r) => {
            if (r.currencyCode === currencyCode) {
              return sum + r.totalPrice
            }
            return sum
          }, 0)

          return {
            rooms: newRooms,
            totalPrice: total,
            currencyCode,
          }
        })
      },

      getTotalPrice: () => {
        return get().totalPrice
      },
    },
  })
)