schedules working

src/background/background.ts

@@ -9,6 +9,7 @@ import { sessionStore } from '../shared/storage/sessionStore';
 import browserActionHandler from './handler/browserActionHandler';
 import hotReloadingHandler from './handler/hotReloadingHandler';
 import tabManagementHandler from './handler/tabManagementHandler';
+import userScheduleHandler from './handler/userScheduleHandler';
 
 onServiceWorkerAlive();
 
@@ -34,6 +35,7 @@ const messageListener = new MessageListener<BACKGROUND_MESSAGES>({
     ...browserActionHandler,
     ...hotReloadingHandler,
     ...tabManagementHandler,
+    ...userScheduleHandler,
 });
 
 messageListener.listen();

src/background/handler/userScheduleHandler.ts

@@ -0,0 +1,16 @@
+import { MessageHandler } from 'chrome-extension-toolkit';
+import { UserScheduleMessages } from 'src/shared/messages/UserScheduleMessages';
+import { Course } from 'src/shared/types/Course';
+import addCourse from '../lib/addCourse';
+import removeCourse from '../lib/removeCourse';
+
+const userScheduleHandler: MessageHandler<UserScheduleMessages> = {
+    addCourse({ data, sendResponse }) {
+        addCourse(data.scheduleId, new Course(data.course)).then(sendResponse);
+    },
+    removeCourse({ data, sendResponse }) {
+        removeCourse(data.scheduleId, new Course(data.course)).then(sendResponse);
+    },
+};
+
+export default userScheduleHandler;

src/background/lib/addCourse.ts

@@ -0,0 +1,18 @@
+import { userScheduleStore } from 'src/shared/storage/userScheduleStore';
+import { Course } from 'src/shared/types/Course';
+
+/**
+ *
+ */
+export default async function addCourse(scheduleId: string, course: Course): Promise<void> {
+    const schedules = await userScheduleStore.get('schedules');
+    const activeSchedule = schedules.find(s => s.id === scheduleId);
+    if (!activeSchedule) {
+        throw new Error('Schedule not found');
+    }
+
+    activeSchedule.creditHours += course.creditHours;
+    activeSchedule.courses.push(course);
+
+    await userScheduleStore.set('schedules', schedules);
+}

src/background/lib/removeCourse.ts

@@ -0,0 +1,18 @@
+import { userScheduleStore } from 'src/shared/storage/userScheduleStore';
+import { Course } from 'src/shared/types/Course';
+
+/**
+ *
+ */
+export default async function removeCourse(scheduleId: string, course: Course): Promise<void> {
+    const schedules = await userScheduleStore.get('schedules');
+    const activeSchedule = schedules.find(s => s.id === scheduleId);
+    if (!activeSchedule) {
+        throw new Error('Schedule not found');
+    }
+
+    activeSchedule.creditHours -= course.creditHours;
+    activeSchedule.courses = activeSchedule.courses.filter(c => c.uniqueId !== course.uniqueId);
+
+    await userScheduleStore.set('schedules', schedules);
+}