feat: ✨ add filtering by restriction code to WebSoc endpoint

apps/api/src/routes/v1/graphql/schema/enum.graphql

@@ -43,6 +43,29 @@ enum SectionType {
   Tap
   Tut
 }
+
+"The set of valid restriction codes."
+enum RestrictionCode {
+  A
+  B
+  C
+  D
+  E
+  F
+  G
+  H
+  I
+  J
+  K
+  L
+  M
+  N
+  O
+  R
+  S
+  X
+}
+
 "The set of valid options for filtering full courses."
 enum FullCourses {
   ANY

apps/api/src/routes/v1/graphql/schema/websoc.graphql

@@ -176,6 +176,7 @@ extend type Query {
     maxCapacity: String
     fullCourses: FullCourses
     cancelledCourses: CancelledCourses
+    excludeRestrictionCodes: [RestrictionCode!]
   ): WebsocAPIResponse!
   "Get data on all available departments."
   depts: [Department!]!

apps/api/src/routes/v1/rest/websoc/lib.ts

@@ -1,3 +1,4 @@
+import { $Enums } from "@libs/db";
 import type { Prisma } from "@libs/db";
 import type { WebsocAPIOptions } from "@libs/uc-irvine-lib/websoc";
 import { notNull } from "@libs/utils";
@@ -200,6 +201,17 @@ export function constructPrismaQuery(parsedQuery: Query): Prisma.WebsocSectionWh
     );
   }
 
+  if (parsedQuery.excludeRestrictionCodes) {
+    console.log(parsedQuery.excludeRestrictionCodes as $Enums.RestrictionCode[]);
+    AND.push({
+      NOT: {
+        restrictionCodes: {
+          hasSome: parsedQuery.excludeRestrictionCodes as $Enums.RestrictionCode[],
+        },
+      },
+    });
+  }
+
   return {
     AND,
 

apps/api/src/routes/v1/rest/websoc/schema.ts

@@ -51,6 +51,12 @@ export const QuerySchema = z
     units: z.string().array().or(z.string()).optional().transform(flattenStringsAndSplit),
     startTime: z.optional(z.literal("").or(z.string().regex(/([1-9]|1[0-2]):[0-5][0-9][ap]m/))),
     endTime: z.optional(z.literal("").or(z.string().regex(/([1-9]|1[0-2]):[0-5][0-9][ap]m/))),
+    excludeRestrictionCodes: z
+      .string()
+      .array()
+      .or(z.string())
+      .optional()
+      .transform(flattenStringsAndSplit),
   })
   .refine((x) => x.cache || !x.cacheOnly, {
     message: "cacheOnly cannot be true if cache is false",

libs/db/prisma/schema.prisma

@@ -44,6 +44,27 @@ enum WebsocSectionType {
   Tut
 }
 
+enum RestrictionCode {
+  A
+  B
+  C
+  D
+  E
+  F
+  G
+  H
+  I
+  J
+  K
+  L
+  M
+  N
+  O
+  R
+  S
+  X
+}
+
 // Models
 
 model CalendarTerm {
@@ -223,25 +244,26 @@ model WebsocSectionMeeting {
 }
 
 model WebsocSection {
-  year          String
-  quarter       Quarter
-  sectionCode   Int
-  timestamp     DateTime
-  geCategories  Json
-  department    String
-  courseNumber  String
-  courseNumeric Int
-  instructors   WebsocSectionInstructor[]
-  courseTitle   String
-  sectionType   WebsocSectionType
-  units         String
-  meetings      WebsocSectionMeeting[]
-  maxCapacity   Int
-  sectionFull   Boolean
-  waitlistFull  Boolean
-  overEnrolled  Boolean
-  cancelled     Boolean
-  data          Json
+  year             String
+  quarter          Quarter
+  sectionCode      Int
+  timestamp        DateTime
+  geCategories     Json
+  department       String
+  courseNumber     String
+  courseNumeric    Int
+  instructors      WebsocSectionInstructor[]
+  courseTitle      String
+  sectionType      WebsocSectionType
+  units            String
+  meetings         WebsocSectionMeeting[]
+  maxCapacity      Int
+  sectionFull      Boolean
+  waitlistFull     Boolean
+  overEnrolled     Boolean
+  cancelled        Boolean
+  restrictionCodes RestrictionCode[]
+  data             Json
 
   @@id([year, quarter, sectionCode, timestamp])
   @@unique([year, quarter, sectionCode, timestamp], name: "idx")

services/websoc-scraper-v2/index.ts

@@ -1,3 +1,4 @@
+import { $Enums } from "@libs/db";
 import { PrismaClient } from "@libs/db";
 import { getTermDateData } from "@libs/uc-irvine-lib/registrar";
 import type {
@@ -116,6 +117,7 @@ type ProcessedSection = {
     waitlistFull: boolean;
     overEnrolled: boolean;
     cancelled: boolean;
+    restrictionCodes: $Enums.RestrictionCode[];
     data: object;
   };
 };
@@ -391,6 +393,9 @@ async function scrape(name: string, term: Term) {
                     parseInt(section.numCurrentlyEnrolled.totalEnrolled, 10) >
                     parseInt(section.maxCapacity, 10),
                   cancelled: section.sectionComment.includes("***  CANCELLED  ***"),
+                  restrictionCodes: section.restrictions
+                    ? (section.restrictions.split(/ and | or /) as $Enums.RestrictionCode[])
+                    : [],
                   data: isolateSection({ school, department, course, section }),
                 },
               };