diff --git a/database/migrations/1739366336570-ReservationStatuses.ts b/database/migrations/1739366336570-ReservationStatuses.ts new file mode 100644 index 0000000..24572b2 --- /dev/null +++ b/database/migrations/1739366336570-ReservationStatuses.ts @@ -0,0 +1,31 @@ +import { MigrationInterface, QueryRunner } from 'typeorm' + +export class ReservationStatuses1739366336570 implements MigrationInterface { + name = 'ReservationStatuses1739366336570' + + public async up(queryRunner: QueryRunner): Promise { + await queryRunner.query( + `CREATE TABLE "temporary_reservations" ("id" varchar PRIMARY KEY NOT NULL, "dateRangeStart" datetime NOT NULL, "dateRangeEnd" datetime NOT NULL, "status" varchar(32) NOT NULL DEFAULT ('pending'), "waitingListId" integer, "ownerId" varchar(32) NOT NULL, "opponents" json NOT NULL)`, + ) + await queryRunner.query( + `INSERT INTO "temporary_reservations"("id", "dateRangeStart", "dateRangeEnd", "status", "waitingListId", "ownerId", "opponents") SELECT "id", "dateRangeStart", "dateRangeEnd", CASE "waitListed" WHEN true THEN "on_waiting_list" ELSE "pending" END, "waitingListId", "ownerId", "opponents" FROM "reservations"`, + ) + await queryRunner.query(`DROP TABLE "reservations"`) + await queryRunner.query( + `ALTER TABLE "temporary_reservations" RENAME TO "reservations"`, + ) + } + + public async down(queryRunner: QueryRunner): Promise { + await queryRunner.query( + `ALTER TABLE "reservations" RENAME TO "temporary_reservations"`, + ) + await queryRunner.query( + `CREATE TABLE "reservations" ("id" varchar PRIMARY KEY NOT NULL, "dateRangeStart" datetime NOT NULL, "dateRangeEnd" datetime NOT NULL, "waitListed" boolean NOT NULL DEFAULT (0), "waitingListId" integer, "ownerId" varchar(32) NOT NULL, "opponents" json NOT NULL)`, + ) + await queryRunner.query( + `INSERT INTO "reservations"("id", "dateRangeStart", "dateRangeEnd", "waitListed", "waitingListId", "ownerId", "opponents") SELECT "id", "dateRangeStart", "dateRangeEnd", CASE "status" WHEN "on_waiting_list" THEN true ELSE false, "waitingListId", "ownerId", "opponents" FROM "temporary_reservations"`, + ) + await queryRunner.query(`DROP TABLE "temporary_reservations"`) + } +} diff --git a/src/ntfy/provider.ts b/src/ntfy/provider.ts index b47be76..4d1a9e6 100644 --- a/src/ntfy/provider.ts +++ b/src/ntfy/provider.ts @@ -121,14 +121,14 @@ export class NtfyProvider implements OnApplicationBootstrap { ) } - async sendReservationWaitlistedNotification( + async sendReservationOnWaitingListNotification( reservationId: string, startTime: Dayjs, endTime: Dayjs, ) { await this.publishQueue.add( ...NtfyProvider.defaultJob({ - title: 'Reservation waitlisted', + title: 'Reservation added to waiting list', message: `${reservationId} - ${startTime.format()} to ${endTime.format()}`, tags: [MessageTags.badminton, MessageTags.hourglass], }), @@ -138,7 +138,7 @@ export class NtfyProvider implements OnApplicationBootstrap { async sendWaitListEmailReceivedNotification(subject: string) { await this.publishQueue.add( ...NtfyProvider.defaultJob({ - title: 'Wait listed reservation available', + title: 'Reservation on waiting list has become available', message: `${subject}`, tags: [MessageTags.badminton, MessageTags.hourglass], }), diff --git a/src/reservations/entity.ts b/src/reservations/entity.ts index 1948fe4..fff32fc 100644 --- a/src/reservations/entity.ts +++ b/src/reservations/entity.ts @@ -1,4 +1,5 @@ import { Exclude, Transform, Type } from 'class-transformer' +import { IsEnum } from 'class-validator' import { Dayjs } from 'dayjs' import { Column, Entity, PrimaryGeneratedColumn } from 'typeorm' @@ -12,6 +13,12 @@ export interface Opponent { name: string } +export enum ReservationStatus { + Pending = 'pending', + OnWaitingList = 'on_waiting_list', + Booked = 'booked', +} + @Entity({ name: 'reservations' }) export class Reservation { @PrimaryGeneratedColumn('uuid') @@ -39,8 +46,13 @@ export class Reservation { @Column('json', { nullable: false }) opponents: Opponent[] - @Column('boolean', { default: false }) - waitListed: boolean + @Column('varchar', { + length: 32, + nullable: false, + default: ReservationStatus.Pending, + }) + @IsEnum(ReservationStatus) + status: ReservationStatus @Column('int', { nullable: true }) waitingListId: number diff --git a/src/reservations/service.ts b/src/reservations/service.ts index 9788fa0..391dfd6 100644 --- a/src/reservations/service.ts +++ b/src/reservations/service.ts @@ -6,7 +6,7 @@ import { Repository } from 'typeorm' import dayjs from '../common/dayjs' import { LoggerService } from '../logger/service.logger' import { BaanReserverenService } from '../runner/baanreserveren/service' -import { Reservation } from './entity' +import { Reservation, ReservationStatus } from './entity' @Injectable() export class ReservationsService { @@ -50,7 +50,9 @@ export class ReservationsService { endDate: dayjs().add(7, 'days').toISOString(), }, ) - .andWhere(`waitListed = false`) + .andWhere('status <> :status', { + status: ReservationStatus.OnWaitingList, + }) .orderBy('dateRangeStart', 'ASC') return await query.getMany() @@ -65,7 +67,7 @@ export class ReservationsService { .andWhere(`DATE(dateRangeEnd) >= DATE(:date)`, { date: date.toISOString(), }) - .andWhere('waitListed = true') + .andWhere('status = :status', { status: ReservationStatus.OnWaitingList }) .getMany() } diff --git a/src/reservations/worker.ts b/src/reservations/worker.ts index 3552462..c7415aa 100644 --- a/src/reservations/worker.ts +++ b/src/reservations/worker.ts @@ -1,6 +1,5 @@ import { Process, Processor } from '@nestjs/bull' import { Inject } from '@nestjs/common' -import { Job } from 'bull' import { instanceToPlain, plainToInstance } from 'class-transformer' import { LoggerService } from '../logger/service.logger' @@ -11,7 +10,7 @@ import { } from '../runner/baanreserveren/service' import { RESERVATIONS_QUEUE_NAME, ReservationsJob } from './config' import { DAILY_RESERVATIONS_ATTEMPTS } from './cron' -import { Reservation } from './entity' +import { Reservation, ReservationStatus } from './entity' import { ReservationsService } from './service' @Processor(RESERVATIONS_QUEUE_NAME) @@ -63,10 +62,10 @@ export class ReservationsWorker { } if ( (shouldWaitlist || attemptsMade === DAILY_RESERVATIONS_ATTEMPTS) && - !reservation.waitListed + reservation.status !== ReservationStatus.OnWaitingList ) { this.loggerService.log('Adding reservation to waiting list') - await this.ntfyProvider.sendReservationWaitlistedNotification( + await this.ntfyProvider.sendReservationOnWaitingListNotification( reservation.id, reservation.dateRangeStart, reservation.dateRangeEnd, @@ -89,7 +88,9 @@ export class ReservationsWorker { } else { await this.brService.performReservation(reservation, timeSensitive) } - await this.reservationsService.deleteById(reservation.id) + await this.reservationsService.update(reservation.id, { + status: ReservationStatus.Booked, + }) } catch (error: unknown) { await this.handleReservationErrors( error as Error, @@ -105,12 +106,12 @@ export class ReservationsWorker { timeSensitive = true, ) { try { - const waitingListId = await this.brService.addReservationToWaitList( + const waitingListId = await this.brService.addReservationToWaitingList( reservation, timeSensitive, ) await this.reservationsService.update(reservation.id, { - waitListed: true, + status: ReservationStatus.OnWaitingList, waitingListId, }) } catch (error: unknown) { diff --git a/src/runner/baanreserveren/service.ts b/src/runner/baanreserveren/service.ts index 6bc9635..a7aa9ab 100644 --- a/src/runner/baanreserveren/service.ts +++ b/src/runner/baanreserveren/service.ts @@ -9,7 +9,11 @@ import dayjs from '../../common/dayjs' import { LoggerService } from '../../logger/service.logger' import { MONITORING_QUEUE_NAME, MonitoringQueue } from '../../monitoring/config' import { MonitorType } from '../../monitoring/entity' -import { Opponent, Reservation } from '../../reservations/entity' +import { + Opponent, + Reservation, + ReservationStatus, +} from '../../reservations/entity' import { EmptyPage } from '../pages/empty' export const BAAN_RESERVEREN_ROOT_URL = 'https://squashcity.baanreserveren.nl' @@ -741,7 +745,7 @@ export class BaanReserverenService { throw new NoCourtAvailableError('Could not reserve court') } - public async addReservationToWaitList( + public async addReservationToWaitingList( reservation: Reservation, timeSensitive = true, ) { @@ -774,7 +778,11 @@ export class BaanReserverenService { public async removeReservationFromWaitList(reservation: Reservation) { try { - if (!reservation.waitListed || !reservation.waitingListId) return + if ( + reservation.status !== ReservationStatus.OnWaitingList || + !reservation.waitingListId + ) + return await this.init() await this.navigateToWaitingList() await this.deleteWaitingListEntryRowById(reservation.waitingListId)