pfadi-bussle/db/booking.ts

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<Booking>(
  {
    // 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<string[]> {
  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: Booking) => booking.days)
    .flat()
    .sort()
})

const Model: mongoose.Model<Booking> =
  mongoose.models.Booking || mongoose.model('Booking', BookingSchema)

export default Model