autobaan/src/runner.ts

116 lines
3.3 KiB
TypeScript

import puppeteer, {
Browser,
BrowserConnectOptions,
BrowserLaunchArgumentOptions,
ElementHandle,
LaunchOptions,
Page,
} from 'puppeteer'
import { DateTime, Opponent, Reservation, timeToString } from './reservation'
export class Runner {
private readonly username: string
private readonly password: string
private readonly reservations: Reservation[]
private browser: Browser | undefined
private page: Page | undefined
public constructor(
username: string,
password: string,
reservations: Reservation[]
) {
this.username = username
this.password = password
this.reservations = reservations
}
public async run(
options?: LaunchOptions &
BrowserLaunchArgumentOptions &
BrowserConnectOptions
): Promise<Reservation[]> {
this.browser = await puppeteer.launch(options)
this.page = await this.browser.newPage()
await this.login()
return await this.makeReservations()
}
private async login() {
await this.page?.goto('https://squashcity.baanreserveren.nl/')
await this.page
?.waitForSelector('input[name=username]')
.then((i) => i?.type(this.username))
await this.page
?.$('input[name=password]')
.then((i) => i?.type(this.password))
await this.page?.$('button').then((b) => b?.click())
}
private async makeReservations(): Promise<Reservation[]> {
for (let i = 0; i < this.reservations.length; i++) {
await this.makeReservation(this.reservations[i])
}
return this.reservations
}
private async makeReservation(reservation: Reservation): Promise<void> {
try {
await this.navigateToDay(reservation.dateTime)
await this.selectAvailableTime(reservation)
await this.selectOpponent(reservation.opponent)
await this.confirmReservation()
reservation.booked = true
} catch (err) {
console.error(err)
}
}
private async navigateToDay(dt: DateTime): Promise<void> {
await this.page
?.waitForSelector(`td#cal_${dt.year}_${dt.month}_${dt.day}`)
.then((d) => d?.click())
await this.page?.waitForSelector(
`td#cal_${dt.year}_${dt.month}_${dt.day}.selected`
)
}
private async selectAvailableTime(res: Reservation): Promise<void> {
let freeCourt: ElementHandle | null | undefined
let i = 0
while (i < res.possibleTimes.length && !freeCourt) {
const possibleTime = res.possibleTimes[i]
const timeString = timeToString(possibleTime)
const selector =
`tr[data-time='${timeString}']` + `> td.free[rowspan='3'][type='free']`
freeCourt = await this.page?.$(selector)
i++
}
if (!freeCourt) {
throw new Error('No free court available')
}
await freeCourt.click()
}
private async selectOpponent(opponent: Opponent): Promise<void> {
const player2Search = await this.page?.waitForSelector(
'tr.res-make-player-2 > td > input'
)
await player2Search?.type(opponent.name)
await this.page?.waitForNetworkIdle()
await this.page
?.$('select.br-user-select[name="players[2]"]')
.then((d) => d?.select(opponent.id))
}
private async confirmReservation(): Promise<void> {
await this.page?.$('input#__make_submit').then((b) => b?.click())
await this.page
?.waitForSelector('input#__make_submit2')
.then((b) => b?.click())
}
}