import * as mongoose from 'mongoose' import { v4 as uuidv4 } from 'uuid' import { dateFormatBackend, getDays } from '../helpers/date' import { Booker } from './booker' import { BOOKING_STATUS } from './bookingStatus' export interface Booking extends mongoose.Document, mongoose.SchemaTimestampsConfig { uuid: string booker: Booker startDate: Date endDate: Date status: BOOKING_STATUS purpose: string org: string destination: string days?: string[] } 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 }) }) const Model: mongoose.Model = mongoose.models.Booking || mongoose.model('Booking', BookingSchema) export default Model