Bookings UI improvements (#826)

server/routers/viewer.tsx

@@ -1,4 +1,4 @@
-import { Prisma } from "@prisma/client";
+import { Prisma, BookingStatus } from "@prisma/client";
 import { TRPCError } from "@trpc/server";
 import { z } from "zod";
 
@@ -20,8 +20,25 @@ export const viewerRouter = createProtectedRouter()
     },
   })
   .query("bookings", {
-    async resolve({ ctx }) {
+    input: z.object({
+      status: z.enum(["upcoming", "past", "cancelled"]).optional(),
+    }),
+    async resolve({ ctx, input }) {
       const { prisma, user } = ctx;
+      const bookingListingByStatus = input.status || "upcoming";
+      const bookingListingFilters: Record<typeof bookingListingByStatus, Prisma.BookingWhereInput[]> = {
+        upcoming: [{ endTime: { gte: new Date() } }],
+        past: [{ endTime: { lte: new Date() } }],
+        cancelled: [{ status: { equals: BookingStatus.CANCELLED } }],
+      };
+      const bookingListingOrderby: Record<typeof bookingListingByStatus, Prisma.BookingOrderByInput> = {
+        upcoming: { startTime: "desc" },
+        past: { startTime: "asc" },
+        cancelled: { startTime: "asc" },
+      };
+      const passedBookingsFilter = bookingListingFilters[bookingListingByStatus];
+      const orderBy = bookingListingOrderby[bookingListingByStatus];
+
       const bookingsQuery = await prisma.booking.findMany({
         where: {
           OR: [
@@ -36,6 +53,7 @@ export const viewerRouter = createProtectedRouter()
               },
             },
           ],
+          AND: passedBookingsFilter,
         },
         select: {
           uid: true,
@@ -58,9 +76,7 @@ export const viewerRouter = createProtectedRouter()
           },
           status: true,
         },
-        orderBy: {
-          startTime: "asc",
-        },
+        orderBy,
       });
 
       const bookings = bookingsQuery.reverse().map((booking) => {