🎉 initial commit

src/lib/reservations.ts

@@ -1,15 +1,18 @@
 import { createServerFn } from '@tanstack/react-start'
-import { and, eq } from 'drizzle-orm'
+import { and, eq, ne } from 'drizzle-orm'
 import { z } from 'zod'
 
 import { db } from '#/db'
 import { reservations, restaurantTables } from '#/db/schema'
+import { requireStaffSession } from '#/lib/auth'
 
 export const getTables = createServerFn({ method: 'GET' }).handler(async () => {
   return db.select().from(restaurantTables).orderBy(restaurantTables.name)
 })
 
 export const getReservations = createServerFn({ method: 'GET' }).handler(async () => {
+  await requireStaffSession()
+
   return db
     .select({
       id: reservations.id,
@@ -84,3 +87,79 @@ export const createReservation = createServerFn({ method: 'POST' })
 
     return reservation
   })
+
+const updateReservationSchema = createReservationSchema.extend({
+  id: z.number().int().positive(),
+})
+
+export const updateReservation = createServerFn({ method: 'POST' })
+  .validator(updateReservationSchema)
+  .handler(async ({ data }) => {
+    await requireStaffSession()
+
+    const reservationTime = new Date(data.reservationTime)
+
+    const [table] = await db
+      .select()
+      .from(restaurantTables)
+      .where(eq(restaurantTables.id, data.tableId))
+
+    if (!table) {
+      throw new Error('Selected table does not exist')
+    }
+
+    if (data.partySize > table.seats) {
+      throw new Error(`${table.name} only seats up to ${table.seats} guests`)
+    }
+
+    const [conflict] = await db
+      .select()
+      .from(reservations)
+      .where(
+        and(
+          eq(reservations.tableId, data.tableId),
+          eq(reservations.reservationTime, reservationTime),
+          ne(reservations.id, data.id),
+        ),
+      )
+
+    if (conflict) {
+      throw new Error('This table is already booked for that time')
+    }
+
+    const [updated] = await db
+      .update(reservations)
+      .set({
+        tableId: data.tableId,
+        customerName: data.customerName,
+        customerEmail: data.customerEmail,
+        partySize: data.partySize,
+        reservationTime,
+      })
+      .where(eq(reservations.id, data.id))
+      .returning()
+
+    if (!updated) {
+      throw new Error('Reservation not found')
+    }
+
+    return updated
+  })
+
+const deleteReservationSchema = z.object({
+  id: z.number().int().positive(),
+})
+
+export const deleteReservation = createServerFn({ method: 'POST' })
+  .validator(deleteReservationSchema)
+  .handler(async ({ data }) => {
+    await requireStaffSession()
+
+    const [deleted] = await db.delete(reservations).where(eq(reservations.id, data.id)).returning()
+
+    if (!deleted) {
+      throw new Error('Reservation not found')
+    }
+
+    return deleted
+  })