Add Status Update Logic

services/alert/backend/src/core/db/schema/course/index.ts

@@ -1,3 +1,4 @@
 export * from './auth.user.model'
 export * from './course.model'
 export * from './section.model'
+export * from "./statusupdate.model"

services/alert/backend/src/core/db/schema/course/statusupdate.model.ts

@@ -0,0 +1,33 @@
+import {
+	bigint,
+	bigserial,
+	boolean,
+	doublePrecision,
+	pgTable,
+	text,
+	timestamp,
+	varchar,
+} from 'drizzle-orm/pg-core'
+
+import { $section } from '@/core/db/schema/course/section.model'
+
+/**
+ * This model was autogenerated / edited using drizzle-kit.
+ * The source of truth is the model defined in the Django backend,
+ * so be sure to sync changes with the backend.
+ * DO NOT EXPORT/SYNC THIS MODEL.
+ */
+export const $statusUpdate = pgTable("courses_statusupdate", {
+	id: bigserial({ mode: "bigint" }).primaryKey().notNull(),
+	oldStatus: varchar("old_status", { length: 16 }).notNull(),
+	newStatus: varchar("new_status", { length: 16 }).notNull(),
+	createdAt: timestamp("created_at", { withTimezone: true, mode: 'string' }).notNull(),
+	alertSent: boolean("alert_sent").notNull(),
+	requestBody: text("request_body").notNull(),
+	// You can use { mode: "bigint" } if numbers are exceeding js number limitations
+	sectionId: bigint("section_id", { mode: "number" })
+        .notNull()
+        .references(() => $section.id),
+	inAddDropPeriod: boolean("in_add_drop_period").notNull(),
+	percentThroughAddDropPeriod: doublePrecision("percent_through_add_drop_period"),
+})

services/alert/webhook/src/core/save.ts

@@ -0,0 +1,40 @@
+import db, { eq, and } from "@pennlabs/pca-backend/db"
+import { $section, $statusUpdate, $course } from "@pennlabs/pca-backend/db/schema/course"
+import { Status, type WebhookPayload } from "@/types/alert"
+import { ENV } from '@/core/env'
+
+export const updateCourseStatus = async (payload: WebhookPayload, requestBody: string) => {
+    const sectionQuery = await db
+        .select({
+            id: $section.id,
+            status: $section.status
+        })
+        .from($section)
+        .where(and(eq($section.fullCode, payload.section_id_normalized), eq($course.semester, ENV.CURRENT_SEMESTER)))
+        .innerJoin($course, eq($section.courseId, $course.id))
+    const previousStatus = sectionQuery[0]?.status;
+    const sectionId = sectionQuery[0]?.id;
+    const typedPreviousStatus = Status[previousStatus as keyof typeof Status];
+    if (typedPreviousStatus !== payload.previous_status) {
+        return false;
+    }
+
+    await db
+        .update($section)
+        .set({
+            status: payload.status
+        })
+        .where(eq($section.id, sectionId))
+    await db
+        .insert($statusUpdate)
+        .values({
+            oldStatus: payload.previous_status,
+            newStatus: payload.status,
+            createdAt: new Date().toISOString(),
+            alertSent: true,
+            requestBody: requestBody,
+            sectionId: Number(sectionId),
+            inAddDropPeriod: false,
+        });
+    return true;
+}

services/alert/webhook/src/main.ts

@@ -3,6 +3,7 @@ import { z, ZodError } from "zod"
 import { Status } from "@/types/alert"
 import { sendCourseAlertEmail } from "@/core/ses"
 import { getRegisteredUsers } from "@/core/query"
+import { updateCourseStatus } from "./core/save"
 
 const statusSchema = z.nativeEnum(Status)
 
@@ -26,7 +27,9 @@ export const handler: APIGatewayProxyHandlerV2 = async (event) => {
 		if (!event.body) throw new BodyMissingError()
 		const payload = webhookPayloadSchema.parse(JSON.parse(event.body))
 		console.log("[org.pennlabs.courses.alert.webhook]: Incoming alert", payload)
+		const isValidStatusUpdate = await updateCourseStatus(payload, event.body);
 		if (
+			isValidStatusUpdate &&
 			payload.previous_status !== Status.OPEN &&
 			payload.status == Status.OPEN
 		) {

services/alert/webhook/src/types/alert.ts

@@ -4,3 +4,11 @@ export enum Status {
 	CANCELLED = "X",
 	UNLISTED = "",
 }
+
+export type WebhookPayload = {
+	previous_status: Status,
+	section_id: string;
+	section_id_normalized: string;
+	status: Status;
+	term: string;
+};