Feature/adhi0331/calendar (#110)

* basic implementation

* completed UI

* basic backend connection

* view calendar

* preliminary calendar complete

* fixed issue with program enrollment

* remove unnecssary logs

Author: adhi0331

backend/src/controllers/calendar.ts

@@ -0,0 +1,116 @@
+/* eslint-disable @typescript-eslint/no-misused-promises */
+/**
+ * Function handlers for calendar route requests
+ */
+import { RequestHandler } from "express";
+import createHttpError from "http-errors";
+
+import EnrollmentModel from "../models/enrollment";
+import SessionModel from "../models/session";
+
+import { Calendar, CalendarSlot } from "./types/calendarTypes";
+
+/**
+ * Calendar Body: {
+ *
+ * studentId: string;
+ * programId: string;
+ * calendar: {
+ *      date: Date;
+ *      hours: number;
+ *      session: string;
+ * }[]
+ *
+ * }
+ */
+
+/**
+ * Request handler for getting calendar for student in program
+ * @param req
+ * @param res
+ * @param next
+ */
+export const getCalendar: RequestHandler = async (req, res, next) => {
+  try {
+    const studentId = req.params.studentId;
+    const programId = req.params.programId;
+
+    const enrollment = EnrollmentModel.find({ studentId, programId });
+    if (!enrollment) {
+      throw createHttpError(404, "Enrollment not found");
+    }
+
+    // get all sessions with studentId and programId
+    const sessions = await SessionModel.find({ programId });
+
+    const calendar: Calendar = { studentId, programId, calendar: [] };
+    for (const session of sessions) {
+      for (const student of session.students) {
+        if (student.studentId.toString() === studentId) {
+          let hours = 0;
+          if (session.marked) {
+            hours = student.hoursAttended;
+          }
+          const date = session.date;
+          const sessionId = session._id.toString();
+          calendar.calendar.push({ date, hours, session: sessionId });
+        }
+      }
+    }
+    console.log(calendar);
+
+    return res.status(200).send(calendar);
+  } catch (error) {
+    next(error);
+  }
+};
+
+/**
+ * Handler for editing a day in a calendar
+ * @param req
+ * @param res
+ * @param next
+ */
+export const editCalendar: RequestHandler = async (req, res, next) => {
+  try {
+    const studentId = req.params.studentId;
+    const programId = req.params.programId;
+
+    const enrollment = await EnrollmentModel.findOne({ studentId, programId });
+    if (!enrollment) {
+      throw createHttpError(404, "Enrollment not found");
+    }
+
+    const { hours, session } = req.body as CalendarSlot;
+
+    const sessionObject = await SessionModel.findById(session);
+
+    if (!sessionObject) {
+      throw createHttpError(404, "Session not found");
+    }
+
+    if (sessionObject.programId.toString() !== programId) {
+      throw createHttpError(404, "Incorrect program for session");
+    }
+
+    const student = sessionObject.students.find((s) => s.studentId.toString() === studentId);
+
+    if (!student) {
+      throw createHttpError(404, "Student not in session");
+    }
+
+    const prevHoursAttended = student.hoursAttended;
+    let hoursLeft = enrollment.hoursLeft + prevHoursAttended;
+
+    student.hoursAttended = hours;
+    hoursLeft -= student.hoursAttended;
+    enrollment.hoursLeft = hoursLeft > 0 ? hoursLeft : 0;
+
+    await sessionObject.save();
+    await enrollment.save();
+
+    res.status(200).send("Updated");
+  } catch (error) {
+    next(error);
+  }
+};

backend/src/controllers/program.ts

@@ -68,6 +68,12 @@ export const updateProgram: RequestHandler = async (req, res, next) => {
       { $set: { status: "Waitlisted", dateUpdated: Date.now() } },
     );
 
+    // update days of week when changed
+    await EnrollmentModel.updateMany(
+      { programId: { $eq: programId }, status: { $eq: "Joined" } },
+      { $set: { schedule: programData.daysOfWeek } },
+    );
+
     res.status(200).json(editedProgram);
   } catch (error) {
     next(error);

backend/src/controllers/session.ts

@@ -266,6 +266,18 @@ export const updateSession: RequestHandler = async (req, res, next) => {
       return res.status(404).json({ message: "No object in database with provided ID" });
     }
 
+    programData.students.forEach(async (student: StudentInfo) => {
+      const enrollment = await EnrollmentModel.findOne({
+        studentId: student.studentId,
+        programId: programData.programId,
+      });
+      if (enrollment) {
+        const hours = enrollment.hoursLeft - student.hoursAttended;
+        enrollment.hoursLeft = hours > 0 ? hours : 0;
+        await enrollment.save();
+      }
+    });
+
     const absentStudents = programData.students.filter((student: StudentInfo) => !student.attended);
 
     const absenceSessions = absentStudents.map((absentStudent) => ({

backend/src/controllers/student.ts

@@ -1,6 +1,6 @@
 /* eslint-disable @typescript-eslint/no-misused-promises */
 /**
- * Functions that process task route requests.
+ * Functions that process student route requests.
  */
 
 import { RequestHandler } from "express";
@@ -9,6 +9,7 @@ import mongoose, { HydratedDocument } from "mongoose";
 
 import EnrollmentModel from "../models/enrollment";
 import { Image } from "../models/image";
+import ProgramModel from "../models/program";
 import ProgressNoteModel from "../models/progressNote";
 import StudentModel from "../models/student";
 import { Enrollment } from "../types/enrollment";
@@ -81,6 +82,10 @@ export const editStudent: RequestHandler = async (req, res, next) => {
       enrollments.map(async (enrollment: Enrollment) => {
         const enrollmentExists = await EnrollmentModel.findById(enrollment._id);
         const enrollmentBody = { ...enrollment, studentId: new mongoose.Types.ObjectId(studentId) };
+        const program = await ProgramModel.findById({ _id: enrollment.programId });
+        if (program?.type === "regular") {
+          enrollmentBody.schedule = program.daysOfWeek;
+        }
         if (!enrollmentExists) {
           return await createEnrollment(enrollmentBody);
         } else {

backend/src/controllers/types/calendarTypes.ts

@@ -0,0 +1,11 @@
+export type CalendarSlot = {
+  date: Date;
+  hours: number;
+  session: string;
+};
+
+export type Calendar = {
+  studentId: string;
+  programId: string;
+  calendar: CalendarSlot[];
+};

backend/src/routes/api.ts

@@ -1,5 +1,6 @@
 import express from "express";
 
+import calendarRouter from "./calendar";
 import imageRouter from "./image";
 import programRoutes from "./program";
 import progressNoteRoutes from "./progressNote";
@@ -15,6 +16,7 @@ router.use("/student", studentRoutes);
 router.use("/program", programRoutes);
 router.use("/session", sessionRoutes);
 router.use("/progressNote", progressNoteRoutes);
+router.use("/calendar", calendarRouter);
 router.use("/image", imageRouter);
 
 export default router;

backend/src/routes/calendar.ts

@@ -0,0 +1,20 @@
+/**
+ * Calendar route requests
+ */
+import express from "express";
+
+import * as CalendarController from "../controllers/calendar";
+import { verifyAuthToken } from "../validators/auth";
+import * as CalendarValidator from "../validators/calendar";
+
+const router = express.Router();
+
+router.get("/:studentId/:programId", [verifyAuthToken], CalendarController.getCalendar);
+router.patch(
+  "/:studentId/:programId",
+  [verifyAuthToken],
+  CalendarValidator.editCalendar,
+  CalendarController.editCalendar,
+);
+
+export default router;

backend/src/routes/student.ts

@@ -1,5 +1,5 @@
 /**
- * Task route requests.
+ * Student route requests.
  */
 
 import express from "express";

backend/src/validators/calendar.ts

@@ -0,0 +1,33 @@
+import { body } from "express-validator";
+
+// const makeDateValidator = () =>
+//     body("date")
+//         .exists()
+//         .withMessage("date needed")
+//         .bail()
+//         .isDate()
+//         .bail();
+
+const makeHoursValidator = () =>
+  body("hours")
+    .exists()
+    .withMessage("hours needed")
+    .bail()
+    .isNumeric()
+    .withMessage("needs to be number")
+    .bail();
+
+const makeSessionValidator = () =>
+  body("session")
+    .exists()
+    .withMessage("sessionId needed")
+    .bail()
+    .isString()
+    .withMessage("needs to be string")
+    .bail();
+
+export const editCalendar = [
+  // makeDateValidator(),
+  makeHoursValidator(),
+  makeSessionValidator(),
+];

frontend/src/api/calendar.ts

@@ -0,0 +1,46 @@
+import { GET, PATCH, createAuthHeader, handleAPIError } from "../api/requests";
+
+import type { APIResult } from "../api/requests";
+
+export type CalendarResponse = {
+  studentId: string;
+  programId: string;
+  calendar: {
+    date: Date;
+    hours: number;
+    session: string;
+  }[];
+};
+
+export async function getCalendar(
+  studentId: string,
+  programId: string,
+  firebaseToken: string,
+): Promise<APIResult<CalendarResponse>> {
+  try {
+    const headers = createAuthHeader(firebaseToken);
+    const response = await GET(`/calendar/${studentId}/${programId}`, headers);
+    const json = (await response.json()) as CalendarResponse;
+    return { success: true, data: json };
+  } catch (error) {
+    return handleAPIError(error);
+  }
+}
+
+export async function editCalendar(
+  studentId: string,
+  programId: string,
+  firebaseToken: string,
+  hours: number,
+  session: string,
+): Promise<APIResult<string>> {
+  try {
+    const headers = createAuthHeader(firebaseToken);
+    const body = { hours, session };
+    const res = await PATCH(`/calendar/${studentId}/${programId}`, body, headers);
+    const json = (await res.json()) as string;
+    return { success: true, data: json };
+  } catch (error) {
+    return handleAPIError(error);
+  }
+}