import * as mongoose from 'mongoose' import { v4 as uuidv4 } from 'uuid' import { dateFormatBackend, getDays } from '../helpers/date' import { BookerDocument } from './booker' import { BOOKING_STATUS } from './bookingStatus' export interface BookingDocument extends mongoose.Document, mongoose.SchemaTimestampsConfig { uuid: string booker: BookerDocument startDate: Date endDate: Date status: BOOKING_STATUS purpose: string org: string destination: string days?: string[] } export interface BookingModel extends mongoose.Model { findBookedDays(): Promise } const BookingSchema = new mongoose.Schema( { // need a seperate uuid to be able to target a booking anonimously uuid: { type: String, default: uuidv4, index: true, }, booker: { type: mongoose.Schema.Types.ObjectId, ref: 'Booker', required: true, }, startDate: { type: Date, required: true, get: dateFormatBackend, min: new Date(), }, endDate: { type: Date, required: false, get: dateFormatBackend, min: new Date(), }, status: { type: String, enum: Object.values(BOOKING_STATUS), required: true, default: 'requested', }, purpose: { type: String, required: false }, org: { type: String, required: false }, destination: { type: String, required: false }, }, { timestamps: true, toJSON: { virtuals: true, getters: true }, toObject: { virtuals: true, getters: true }, } ) BookingSchema.virtual('days').get(function () { return getDays({ startDate: this.startDate, endDate: this.endDate }) }) BookingSchema.static('findBookedDays', async function (): Promise { console.log('in findBookedDays this is', this) const bookings = await this.find( { status: { $in: [BOOKING_STATUS.REQUESTED, BOOKING_STATUS.CONFIRMED] }, $or: [ { endDate: { $gt: new Date() } }, { startDate: { $gt: new Date() } }, ], }, 'startDate endDate' ).exec() return bookings .map((booking: BookingDocument) => booking.days) .flat() .sort() }) const Model: BookingModel = mongoose.model( 'Booking', BookingSchema ) export default Model