import {
  BookingOptionGroup,
  BookingOptionPax,
  BookingOptionUnit,
  BookingPackage,
  BookingPackageOption,
  BookingPackageRoom,
  ChangedLine,
  PerBookingPackageOption,
  PerPaxPackageOption,
  PerUnitPackageOption
} from '@qite/tide-client/build/types';

export const updatePackageDetails = (packageDetails: BookingPackage, changedLines: ChangedLine[]): BookingPackage => {
  return {
    ...packageDetails,
    options: packageDetails.options.map((opt) => {
      return {
        ...opt,
        rooms: opt.isSelected ? updateRooms(opt.rooms, changedLines) : opt.rooms,
        groups: opt.isSelected ? updateBookingGroups(opt.groups, changedLines) : opt.groups,
        optionUnits: opt.isSelected ? updateOptionUnits(opt.optionUnits, changedLines) : opt.optionUnits,
        optionPax: opt.isSelected ? updateOptionPax(opt.optionPax, changedLines) : opt.optionPax
      } as BookingPackageOption;
    })
  };
};

// ROOMS
const updateRooms = (rooms: BookingPackageRoom[], changedLines: ChangedLine[]) => {
  return rooms.map((room) => ({
    ...room,
    options: room.options.map((option) => ({
      ...option,
      price: changedLines.find((x) => x.entryLineGuid == option.entryLineGuid)?.price ?? option.price
    }))
  }));
};

// PAX
const updateOptionPax = (pax: BookingOptionPax[], changedLines: ChangedLine[]): BookingOptionPax[] => {
  return pax.map((p) => ({
    ...p,
    groups: updatePaxGroups(p.groups, changedLines)
  }));
};

const updatePaxGroups = (groups: BookingOptionGroup<PerPaxPackageOption>[], changedLines: ChangedLine[]): BookingOptionGroup<PerPaxPackageOption>[] => {
  return groups.map((g) => ({
    ...g,
    options: g.options.map((o) => ({
      ...o,
      groups: updatePaxGroups(o.groups, changedLines),
      line: {
        ...o.line,
        price: changedLines.find((cl) => cl.entryLineGuid === o.line.entryLineGuid)?.price ?? o.line.price
      },
      alternatives: o.alternatives.map((a) => ({
        ...a,
        price: changedLines.find((cl) => cl.entryLineGuid === a.entryLineGuid)?.price ?? a.price
      }))
    }))
  }));
};

// UNITS
const updateOptionUnits = (units: BookingOptionUnit[], changedLines: ChangedLine[]): BookingOptionUnit[] => {
  return units.map((u) => ({
    ...u,
    groups: updateUnitGroups(u.groups, changedLines)
  }));
};

const updateUnitGroups = (groups: BookingOptionGroup<PerUnitPackageOption>[], changedLines: ChangedLine[]): BookingOptionGroup<PerUnitPackageOption>[] => {
  return groups.map((g) => ({
    ...g,
    options: g.options.map((o) => ({
      ...o,
      groups: updateUnitGroups(o.groups, changedLines),
      pax: updateOptionPax(o.pax, changedLines),
      line: {
        ...o.line,
        price: changedLines.find((cl) => cl.entryLineGuid === o.line.entryLineGuid)?.price ?? o.line.price
      },
      alternatives: o.alternatives.map((a) => ({
        ...a,
        price: changedLines.find((cl) => cl.entryLineGuid === a.entryLineGuid)?.price ?? a.price
      }))
    }))
  }));
};

// BOOKING
const updateBookingGroups = (
  groups: BookingOptionGroup<PerBookingPackageOption>[],
  changedLines: ChangedLine[]
): BookingOptionGroup<PerBookingPackageOption>[] => {
  return groups.map((g) => ({
    ...g,
    options: g.options.map((o) => ({
      ...o,
      groups: updateBookingGroups(o.groups, changedLines),
      units: updateOptionUnits(o.units, changedLines),
      pax: updateOptionPax(o.pax, changedLines),
      line: {
        ...o.line,
        price: changedLines.find((cl) => cl.entryLineGuid === o.line.entryLineGuid)?.price ?? o.line.price
      },
      alternatives: o.alternatives.map((a) => ({
        ...a,
        price: changedLines.find((cl) => cl.entryLineGuid === a.entryLineGuid)?.price ?? a.price
      }))
    }))
  }));
};