import 'package:comwell_key_app/overview/models/bookings.dart';
import 'package:comwell_key_app/services/mappers/booking_mapper.dart';
import 'package:comwell_key_app/services/models/bookings_dto.dart';

extension BookingsMapper on BookingsDTO {
  Bookings toBookings() {
    return Bookings(
      current:
          current.map((booking) => booking.toBooking()).toList(),
      past: past.map((booking) => booking.toBooking()).toList(),
      cancelled:
          cancelled.map((booking) => booking.toBooking()).toList(),
    );
  }
}